001/*
002 * Stallion Core: A Modern Web Framework
003 *
004 * Copyright (C) 2015 - 2016 Stallion Software LLC.
005 *
006 * This program is free software: you can redistribute it and/or modify it under the terms of the
007 * GNU General Public License as published by the Free Software Foundation, either version 2 of
008 * the License, or (at your option) any later version. This program is distributed in the hope that
009 * it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of
010 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public
011 * License for more details. You should have received a copy of the GNU General Public License
012 * along with this program.  If not, see <http://www.gnu.org/licenses/gpl-2.0.html>.
013 *
014 *
015 *
016 */
017
018package io.stallion.tools;
019
020import io.stallion.exceptions.UsageException;
021
022import java.util.Set;
023
024import static io.stallion.utils.Literals.*;
025import static io.stallion.Context.*;
026
027
028public abstract class SampleDataGenerator {
029    private Set<Long> seenIds = set();
030
031    public abstract Long getBaseId();
032
033
034    public abstract void generate();
035
036    public Long newId(int extra) {
037        Long id = getBaseId() + extra;
038        if (seenIds.contains(id)) {
039            throw new UsageException("The id " + extra + " has already been used.");
040        }
041        seenIds.add(id);
042        return id;
043    }
044
045    public Long getId(int extra) {
046        return getBaseId() + extra;
047    }
048
049
050}