com.baldmountain.depot.DbSeed.java Source code

Introduction

Here is the source code for com.baldmountain.depot.DbSeed.java

Source

package com.baldmountain.depot;

import com.baldmountain.depot.models.Product;
import io.vertx.core.Vertx;
import io.vertx.core.json.JsonObject;
import io.vertx.ext.mongo.MongoService;

import java.math.BigDecimal;
import java.math.RoundingMode;
import java.util.LinkedList;
import java.util.List;

/**
 * @author Geoffrey Clements
 *
 * The MIT License (MIT)
 *
 * Copyright (c) 2015 Geoffrey Clements
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in all
 * copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
 *
 */
public class DbSeed {
    static boolean done = false;
    static Object lock = new Object();
    static MongoService mongoService;

    static void setDone(String msg) {
        System.out.println(msg);
        mongoService.stop();
        done = true;
        synchronized (lock) {
            lock.notify();
        }
    }

    static void saveAProduct(List<Product> products) {
        if (!products.isEmpty()) {
            Product product = products.remove(0);
            product.save(mongoService, res -> {
                if (res.succeeded()) {
                    saveAProduct(products);
                } else {
                    setDone(res.cause().getMessage());
                }
            });
        } else {
            setDone("Finished!");
        }
    }

    public static void main(String[] args) {
        Vertx vertx = Vertx.vertx();
        JsonObject config = new JsonObject().put("db_name", "depot_development");

        mongoService = MongoService.create(vertx, config);
        mongoService.start();

        mongoService.dropCollection("line_items", res1 -> {
            if (res1.succeeded()) {
                mongoService.dropCollection("carts", res2 -> {
                    if (res2.succeeded()) {
                        mongoService.dropCollection("products", event1 -> {
                            if (event1.succeeded()) {
                                LinkedList<Product> products = new LinkedList<>();
                                products.add(new Product("CoffeeScript", "<p>\n"
                                        + "        CoffeeScript is JavaScript done right. It provides all of JavaScript's "
                                        + "functionality wrapped in a cleaner, more succinct syntax. In the first "
                                        + "book on this exciting new language, CoffeeScript guru Trevor Burnham "
                                        + "shows you how to hold onto all the power and flexibility of JavaScript "
                                        + "while writing clearer, cleaner, and safer code.</p>", "/images/cs.jpg",
                                        new BigDecimal(36.00).setScale(2, RoundingMode.CEILING)));
                                products.add(new Product("Programming Ruby 1.9 & 2.0",
                                        "<p>Ruby is the fastest growing and most exciting dynamic language "
                                                + "out there. If you need to get working programs delivered fast, "
                                                + "you should add Ruby to your toolbox. </p>",
                                        "/images/ruby.jpg",
                                        new BigDecimal(49.95).setScale(2, RoundingMode.CEILING)));
                                products.add(new Product("Rails Test Prescriptions",
                                        "<p><em>Rails Test Prescriptions</em> is a comprehensive guide to testing "
                                                + "Rails applications, covering Test-Driven Development from both a theoretical perspective (why to test) and from a practical perspective "
                                                + "(how to test effectively). It covers the core Rails testing tools and procedures for Rails 2 and Rails 3, and introduces popular add-ons, "
                                                + "including Cucumber, Shoulda, Machinist, Mocha, and Rcov.</p>",
                                        "/images/rtp.jpg",
                                        new BigDecimal(34.95).setScale(2, RoundingMode.CEILING)));
                                saveAProduct(products);
                            } else {
                                setDone("Trouble dropping products");
                            }
                        });
                    } else {
                        setDone("Trouble dropping carts");
                    }
                });
            } else {
                setDone("Trouble dropping line_items");
            }
        });

        try {
            synchronized (lock) {
                while (!done)
                    lock.wait();
            }
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
        vertx.close();
    }
}