SeedData.java

package com.lambdaschool.orders;

import com.github.javafaker.Faker;
import com.lambdaschool.orders.models.Agent;
import com.lambdaschool.orders.models.Customer;
import com.lambdaschool.orders.models.Order;
import com.lambdaschool.orders.models.Payment;
import com.lambdaschool.orders.repositories.AgentsRepository;
import com.lambdaschool.orders.repositories.CustomersRepository;
import com.lambdaschool.orders.repositories.OrdersRepository;
import com.lambdaschool.orders.repositories.PaymentRepository;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.CommandLineRunner;
import org.springframework.stereotype.Component;
import org.springframework.transaction.annotation.Transactional;

import java.util.HashSet;
import java.util.Locale;
import java.util.Random;
import java.util.Set;

@Transactional
@Component
public class SeedData
        implements CommandLineRunner
{
    /**
     * Connects the customer table to this SeedData method
     */
    @Autowired
    private CustomersRepository custrepos;

    /**
     * Connects the agents table to this SeedData method
     */
    @Autowired
    private AgentsRepository agentrepos;

    /**
     * Connects the orders table to this SeedData method
     */
    @Autowired
    private OrdersRepository ordersrepos;

    /**
     * Connects the payment table to this SeedData method
     */
    @Autowired
    private PaymentRepository paymentrepos;

    /**
     * A Random generator is needed to randomly generate faker data.
     */
    private Random random = new Random();

    /**
     * Generates test, seed data for our application
     * First a set of known data is seeded into our database.
     * Second a random set of data using Java Faker is seeded into our database.
     * Note this process does not remove data from the database. So if data exists in the database
     * prior to running this process, that data remains in the database.
     *
     * @param args The parameter is required by the parent interface but is not used in this process.
     */
    @Transactional
    @Override
    public void run(String[] args) throws
            Exception
    {
        Payment pay1 = new Payment("Cash");
        Payment pay2 = new Payment("Gift Card");
        Payment pay3 = new Payment("Credit Card");
        Payment pay4 = new Payment("Mobile Pay");

        pay1 = paymentrepos.save(pay1);
        pay2 = paymentrepos.save(pay2);
        pay3 = paymentrepos.save(pay3);
        pay4 = paymentrepos.save(pay4);

        Agent a01 = new Agent("Ramasundar",
                              "Bangalore",
                              0.15,
                              "077-25814763",
                              "");
        Agent a02 = new Agent("Alex",
                              "London",
                              0.13,
                              "075-12458969",
                              "");
        Agent a03 = new Agent("Alford",
                              "New York",
                              0.12,
                              "044-25874365",
                              "");
        Agent a04 = new Agent("Ravi",
                              "Bangalore",
                              0.15,
                              "077-45625874",
                              "");
        Agent a05 = new Agent("Santakumar",
                              "Chennai",
                              0.14,
                              "007-22388644",
                              "");
        Agent a06 = new Agent("Lucida",
                              "San Jose",
                              0.12,
                              "044-52981425",
                              "");
        Agent a07 = new Agent("Anderson",
                              "Brisban",
                              0.13,
                              "045-21447739",
                              "");
        Agent a08 = new Agent("Subbarao",
                              "Bangalore",
                              0.14,
                              "077-12346674",
                              "");
        Agent a09 = new Agent("Mukesh",
                              "Mumbai",
                              0.11,
                              "029-12358964",
                              "");
        Agent a10 = new Agent("McDen",
                              "London",
                              0.15,
                              "078-22255588",
                              "");
        Agent a11 = new Agent("Ivan",
                              "Torento",
                              0.15,
                              "008-22544166",
                              "");
        Agent a12 = new Agent("Benjamin",
                              "Hampshair",
                              0.11,
                              "008-22536178",
                              "");

        Customer c01 = new Customer("Holmes",
                                    "London",
                                    "London",
                                    "UK",
                                    "2",
                                    6000.00,
                                    5000.00,
                                    7000.00,
                                    4000.00,
                                    "BBBBBBB",
                                    a03);
        Customer c02 = new Customer("Micheal",
                                    "New York",
                                    "New York",
                                    "USA",
                                    "2",
                                    3000.00,
                                    5000.00,
                                    2000.00,
                                    6000.00,
                                    "CCCCCCC",
                                    a08);
        Customer c03 = new Customer("Albert",
                                    "New York",
                                    "New York",
                                    "USA",
                                    "3",
                                    5000.00,
                                    7000.00,
                                    6000.00,
                                    6000.00,
                                    "BBBBSBB",
                                    a08);
        Customer c04 = new Customer("Ravindran",
                                    "Bangalore",
                                    "Bangalore",
                                    "India",
                                    "2",
                                    5000.00,
                                    7000.00,
                                    4000.00,
                                    8000.00,
                                    "AVAVAVA",
                                    a11);
        Customer c05 = new Customer("Cook",
                                    "London",
                                    "London",
                                    "UK",
                                    "2",
                                    4000.00,
                                    9000.00,
                                    7000.00,
                                    6000.00,
                                    "FSDDSDF",
                                    a06);
        Customer c06 = new Customer("Stuart",
                                    "London",
                                    "London",
                                    "UK",
                                    "1",
                                    6000.00,
                                    8000.00,
                                    3000.00,
                                    11000.00,
                                    "GFSGERS",
                                    a03);
        Customer c07 = new Customer("Bolt",
                                    "New York",
                                    "New York",
                                    "USA",
                                    "3",
                                    5000.00,
                                    7000.00,
                                    9000.00,
                                    3000.00,
                                    "DDNRDRH",
                                    a08);
        Customer c08 = new Customer("Fleming",
                                    "Brisban",
                                    "Brisban",
                                    "Australia",
                                    "2",
                                    7000.00,
                                    7000.00,
                                    9000.00,
                                    5000.00,
                                    "NHBGVFC",
                                    a05);
        Customer c09 = new Customer("Jacks",
                                    "Brisban",
                                    "Brisban",
                                    "Australia",
                                    "1",
                                    7000.00,
                                    7000.00,
                                    7000.00,
                                    7000.00,
                                    "WERTGDF",
                                    a05);
        Customer c10 = new Customer("Yearannaidu",
                                    "Chennai",
                                    "Chennai",
                                    "India",
                                    "1",
                                    8000.00,
                                    7000.00,
                                    7000.00,
                                    8000.00,
                                    "ZZZZBFV",
                                    a10);
        Customer c11 = new Customer("Sasikant",
                                    "Mumbai",
                                    "Mumbai",
                                    "India",
                                    "1",
                                    7000.00,
                                    11000.00,
                                    7000.00,
                                    11000.00,
                                    "147-25896312",
                                    a02);
        Customer c12 = new Customer("Ramanathan",
                                    "Chennai",
                                    "Chennai",
                                    "India",
                                    "1",
                                    7000.00,
                                    11000.00,
                                    9000.00,
                                    9000.00,
                                    "GHRDWSD",
                                    a10);
        Customer c13 = new Customer("Avinash",
                                    "Mumbai",
                                    "Mumbai",
                                    "India",
                                    "2",
                                    7000.00,
                                    11000.00,
                                    9000.00,
                                    9000.00,
                                    "113-12345678",
                                    a02);
        Customer c14 = new Customer("Winston",
                                    "Brisban",
                                    "Brisban",
                                    "Australia",
                                    "1",
                                    5000.00,
                                    8000.00,
                                    7000.00,
                                    6000.00,
                                    "AAAAAAA",
                                    a05);
        Customer c15 = new Customer("Karl",
                                    "London",
                                    "London",
                                    "UK",
                                    "0",
                                    4000.00,
                                    6000.00,
                                    7000.00,
                                    3000.00,
                                    "AAAABAA",
                                    a06);
        Customer c16 = new Customer("Shilton",
                                    "Torento",
                                    "Torento",
                                    "Canada",
                                    "1",
                                    10000.00,
                                    7000.00,
                                    6000.00,
                                    11000.00,
                                    "DDDDDDD",
                                    a04);
        Customer c17 = new Customer("Charles",
                                    "Hampshair",
                                    "Hampshair",
                                    "UK",
                                    "3",
                                    6000.00,
                                    4000.00,
                                    5000.00,
                                    5000.00,
                                    "MMMMMMM",
                                    a09);
        Customer c18 = new Customer("Srinivas",
                                    "Bangalore",
                                    "Bangalore",
                                    "India",
                                    "2",
                                    8000.00,
                                    4000.00,
                                    3000.00,
                                    9000.00,
                                    "AAAAAAB",
                                    a07);
        Customer c19 = new Customer("Steven",
                                    "San Jose",
                                    "San Jose",
                                    "USA",
                                    "1",
                                    5000.00,
                                    7000.00,
                                    9000.00,
                                    3000.00,
                                    "KRFYGJK",
                                    a10);
        Customer c20 = new Customer("Karolina",
                                    "Torento",
                                    "Torento",
                                    "Canada",
                                    "1",
                                    7000.00,
                                    7000.00,
                                    9000.00,
                                    5000.00,
                                    "HJKORED",
                                    a04);
        Customer c21 = new Customer("Martin",
                                    "Torento",
                                    "Torento",
                                    "Canada",
                                    "2",
                                    8000.00,
                                    7000.00,
                                    7000.00,
                                    8000.00,
                                    "MJYURFD",
                                    a04);
        Customer c22 = new Customer("Ramesh",
                                    "Mumbai",
                                    "Mumbai",
                                    "India",
                                    "3",
                                    8000.00,
                                    7000.00,
                                    3000.00,
                                    12000.00,
                                    "Phone No",
                                    a02);
        Customer c23 = new Customer("Rangarappa",
                                    "Bangalore",
                                    "Bangalore",
                                    "India",
                                    "2",
                                    8000.00,
                                    11000.00,
                                    7000.00,
                                    12000.00,
                                    "AAAATGF",
                                    a01);
        Customer c24 = new Customer("Venkatpati",
                                    "Bangalore",
                                    "Bangalore",
                                    "India",
                                    "2",
                                    8000.00,
                                    11000.00,
                                    7000.00,
                                    12000.00,
                                    "JRTVFDD",
                                    a07);
        Customer c25 = new Customer("Sundariya",
                                    "Chennai",
                                    "Chennai",
                                    "India",
                                    "3",
                                    7000.00,
                                    11000.00,
                                    7000.00,
                                    11000.00,
                                    "PPHGRTS",
                                    a10);

        Order o01 = new Order(1000.00,
                              600.00,
                              c13,
                              "SOD");
        o01.getPayments()
                .add(pay1);

        Order o02 = new Order(3000.00,
                              500.00,
                              c19,
                              "SOD");
        o02.getPayments()
                .add(pay2);

        Order o03 = new Order(4500.00,
                              900.00,
                              c07,
                              "SOD");
        o03.getPayments()
                .add(pay3);
        o03.getPayments()
                .add(pay2);

        Order o04 = new Order(2000.00,
                              0.00,
                              c16,
                              "SOD");
        o04.getPayments()
                .add(pay4);

        Order o05 = new Order(4000.00,
                              600.00,
                              c22,
                              "SOD");
        o05.getPayments()
                .add(pay2);

        Order o06 = new Order(2000.00,
                              0.00,
                              c12,
                              "SOD");
        o06.getPayments()
                .add(pay3);

        Order o07 = new Order(3500.00,
                              2000.00,
                              c02,
                              "SOD");
        o07.getPayments()
                .add(pay4);

        Order o08 = new Order(2500.00,
                              400.00,
                              c03,
                              "SOD");
        o08.getPayments()
                .add(pay1);

        Order o09 = new Order(500.00,
                              0.00,
                              c23,
                              "SOD");
        o09.getPayments()
                .add(pay3);

        Order o10 = new Order(4000.00,
                              700.00,
                              c07,
                              "SOD");
        o10.getPayments()
                .add(pay4);

        Order o11 = new Order(1500.00,
                              600.00,
                              c08,
                              "SOD");
        o11.getPayments()
                .add(pay2);

        Order o12 = new Order(2500.00,
                              0.00,
                              c25,
                              "SOD");
        o12.getPayments()
                .add(pay1);

        agentrepos.save(a01);
        agentrepos.save(a02);
        agentrepos.save(a03);
        agentrepos.save(a04);
        agentrepos.save(a05);
        agentrepos.save(a06);
        agentrepos.save(a07);
        agentrepos.save(a08);
        agentrepos.save(a09);
        agentrepos.save(a10);
        agentrepos.save(a11);
        agentrepos.save(a12);

        custrepos.save(c01);
        custrepos.save(c02);
        custrepos.save(c03);
        custrepos.save(c04);
        custrepos.save(c05);
        custrepos.save(c06);
        custrepos.save(c07);
        custrepos.save(c08);
        custrepos.save(c09);
        custrepos.save(c10);
        custrepos.save(c11);
        custrepos.save(c12);
        custrepos.save(c13);
        custrepos.save(c14);
        custrepos.save(c15);
        custrepos.save(c16);
        custrepos.save(c17);
        custrepos.save(c18);
        custrepos.save(c19);
        custrepos.save(c20);
        custrepos.save(c21);
        custrepos.save(c22);
        custrepos.save(c23);
        custrepos.save(c24);
        custrepos.save(c25);

        ordersrepos.save(o01);
        ordersrepos.save(o02);
        ordersrepos.save(o03);
        ordersrepos.save(o04);
        ordersrepos.save(o05);
        ordersrepos.save(o06);
        ordersrepos.save(o07);
        ordersrepos.save(o08);
        ordersrepos.save(o09);
        ordersrepos.save(o10);
        ordersrepos.save(o11);
        ordersrepos.save(o12);

        //Begins the faker data

        Faker dataFaker = new Faker(new Locale("en-US"));
        Set<String> customerNames = new HashSet<>();
        for (int i = 0; i < 100; i++)
        {
            customerNames.add(dataFaker.name()
                                      .fullName());
        }

        for (String theName : customerNames)
        {
            String custcity = dataFaker.address()
                    .city();
            String tempWorkingarea = dataFaker.address()
                    .cityName();
            String tempCustcountry = dataFaker.address()
                    .country();
            String tempGrade = dataFaker.country()
                    .countryCode2();
            double tempOpeningamt = dataFaker.number()
                    .randomDouble(2,
                                  0,
                                  10000);
            double tempReceiveamt = dataFaker.number()
                    .randomDouble(2,
                                  0,
                                  10000);
            double tempPaymentamt = dataFaker.number()
                    .randomDouble(2,
                                  0,
                                  10000);
            double tempOutstandingamt = dataFaker.number()
                    .randomDouble(2,
                                  0,
                                  10000);
            String tempPhone = dataFaker.phoneNumber()
                    .phoneNumber();

            Customer fakeCustomer = new Customer(theName,
                                                 custcity,
                                                 tempWorkingarea,
                                                 tempCustcountry,
                                                 tempGrade,
                                                 tempOpeningamt,
                                                 tempReceiveamt,
                                                 tempPaymentamt,
                                                 tempOutstandingamt,
                                                 tempPhone,
                                                 a10);

            int randomNumber = random.nextInt(10); // random number 0 through 9
            for (int i = 0; i < randomNumber; i++)
            {
                double tempGetOrdamount = dataFaker.number()
                        .randomDouble(2,
                                      0,
                                      10000);
                double tempGetAdvanceamount = dataFaker.number()
                        .randomDouble(2,
                                      0,
                                      10000);
                String tempGetOrderdescription = dataFaker.lorem()
                        .characters();

                Order newOrder = new Order(tempGetOrdamount,
                                           tempGetAdvanceamount,
                                           fakeCustomer,
                                           tempGetOrderdescription);

                newOrder.getPayments().add(pay1);
                fakeCustomer.getOrders()
                        .add(newOrder);
            }

            // this actually saves the faker data.
            custrepos.save(fakeCustomer);
        }
    }
}