se.vgregion.webbisar.tool.WebbisGenerator.java Source code

Introduction

Here is the source code for se.vgregion.webbisar.tool.WebbisGenerator.java

Source

/**
 * Copyright 2010 Vstra Gtalandsregionen
 *
 *   This library is free software; you can redistribute it and/or modify
 *   it under the terms of version 2.1 of the GNU Lesser General Public
 *   License as published by the Free Software Foundation.
 *
 *   This library is distributed in the hope that it will be useful,
 *   but WITHOUT ANY WARRANTY; without even the implied warranty of
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *   GNU Lesser General Public License for more details.
 *
 *   You should have received a copy of the GNU Lesser General Public
 *   License along with this library; if not, write to the
 *   Free Software Foundation, Inc., 59 Temple Place, Suite 330,
 *   Boston, MA 02111-1307  USA
 *
 */

package se.vgregion.webbisar.tool;

import java.util.ArrayList;
import java.util.Comparator;
import java.util.Date;
import java.util.List;
import java.util.Random;
import java.util.SortedSet;
import java.util.TreeSet;

import org.joda.time.DateTime;
import org.springframework.context.support.AbstractRefreshableApplicationContext;
import org.springframework.context.support.ClassPathXmlApplicationContext;

import se.vgregion.webbisar.svc.WebbisService;
import se.vgregion.webbisar.types.BirthTime;
import se.vgregion.webbisar.types.Hospital;
import se.vgregion.webbisar.types.MultimediaFile;
import se.vgregion.webbisar.types.Name;
import se.vgregion.webbisar.types.Sex;
import se.vgregion.webbisar.types.Webbis;
import se.vgregion.webbisar.types.MultimediaFile.MediaType;

public class WebbisGenerator {

    private static Random rand = new Random(new Date().getTime());

    /**
     * @param args
     */
    public static void main(String[] args) {
        new WebbisGenerator().run();
    }

    public void run() {
        AbstractRefreshableApplicationContext ctx = new ClassPathXmlApplicationContext(
                new String[] { "services-config.xml", "applicationContext-hibernate.xml" });

        WebbisService svc = (WebbisService) ctx.getBean("webbisService");

        SortedSet<Webbis> webbisar = new TreeSet<Webbis>(new Comparator<Webbis>() {
            public int compare(Webbis o1, Webbis o2) {
                if (o1.getBirthTime().compareTo(o2.getBirthTime()) == 0) {
                    return -1;
                }
                return o1.getBirthTime().compareTo(o2.getBirthTime());
            }
        });
        for (int i = 0; i < 20000; i++) {
            webbisar.add(generateWebbis());
        }
        svc.saveAll(webbisar);

        ctx.close();
    }

    private Webbis generateWebbis() {
        String n = NameGenerator.generateWebbisName();
        String name = (rand.nextInt(5) != 0) ? n.substring(1) : null;
        Sex sex = n.charAt(0) == 'f' ? Sex.Female : Sex.Male;

        BirthTime birthTime = getBirthTime();

        int weight = 2000 + rand.nextInt(3000);
        int length = 40 + rand.nextInt(15);

        Hospital hospital = generateHospital();
        String home = "Gteborg";

        List<Name> parents = new ArrayList<Name>();
        parents.add(NameGenerator.generateFemaleName());
        parents.add(NameGenerator.generateMaleName());

        List<MultimediaFile> images = new ArrayList<MultimediaFile>();
        images.add(generateImage());
        for (int i = 0; i < rand.nextInt(3); i++) {
            images.add(generateImage());
        }

        String siblings = generateSiblings();
        String authorId = generateAuthorId();

        String message = "Vlkommen till vrlden, h vad vi har lngtat efter dig!";
        String email = "mamma@pappa.se";
        String homePage = "http://www.blogg.se./mamma";

        return new Webbis(name, authorId, sex, birthTime, weight, length, hospital, home, parents, images, siblings,
                message, email, homePage);
    }

    private String generateSiblings() {
        int numSiblings = rand.nextInt(4);
        String siblings = "";
        if (numSiblings == 1) {
            siblings = generateWebbisNameNoGender();
        }
        if (numSiblings == 2) {
            siblings = generateWebbisNameNoGender() + " och " + generateWebbisNameNoGender();
        }
        if (numSiblings == 3) {
            siblings = generateWebbisNameNoGender() + ", " + generateWebbisNameNoGender() + " och "
                    + generateWebbisNameNoGender();
        }

        return siblings;
    }

    private String generateWebbisNameNoGender() {
        return NameGenerator.generateWebbisName().substring(1);
    }

    private Hospital generateHospital() {
        int l = Hospital.values().length;
        return Hospital.values()[rand.nextInt(l - 1)]; // remove unknown
    }

    private MultimediaFile generateImage() {
        int n = rand.nextInt(11);
        return new MultimediaFile(n + ".jpg", "lilla gosan", MediaType.IMAGE, "image/jpeg");
    }

    private String generateAuthorId() {
        StringBuffer sb = new StringBuffer();
        for (int i = 0; i < 10; i++) {
            sb.append(rand.nextInt(10));
        }
        return sb.toString();
    }

    private BirthTime getBirthTime() {

        DateTime now = new DateTime();
        DateTime oneYearBack = now.minusYears(1);

        int secondsInYear = (int) ((now.getMillis() - oneYearBack.getMillis()) / 1000L);

        int randomSecond = rand.nextInt(secondsInYear);

        DateTime bd = new DateTime(oneYearBack.getMillis() + randomSecond * 1000L);
        return new BirthTime(bd.toDate());
    }
}