openjpa-integration/daytrader/src/test/java/org/apache/openjpa/integration/daytrader/TradeBuildDB.java

/**
 *  Licensed to the Apache Software Foundation (ASF) under one or more
 *  contributor license agreements.  See the NOTICE file distributed with
 *  this work for additional information regarding copyright ownership.
 *  The ASF licenses this file to You under the Apache License, Version 2.0
 *  (the "License"); you may not use this file except in compliance with
 *  the License.  You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 */
package org.apache.openjpa.integration.daytrader;

import java.math.BigDecimal;

import org.apache.openjpa.lib.log.Log;

// import org.apache.geronimo.samples.daytrader.core.*;
// import org.apache.geronimo.samples.daytrader.core.direct.*;
// import org.apache.geronimo.samples.daytrader.beans.*;
// import org.apache.geronimo.samples.daytrader.util.*;

/**
 * TradeBuildDB uses operations provided by the TradeApplication to
 *   (a) create the Database tables
 *   (b) populate a DayTrader database without creating the tables.
 * Specifically, a new DayTrader User population is created using
 * UserIDs of the form "uid:xxx" where xxx is a sequential number
 * (e.g. uid:0, uid:1, etc.). New stocks are also created of the form "s:xxx",
 * again where xxx represents sequential numbers (e.g. s:1, s:2, etc.)
 */
public class TradeBuildDB {

    private TradeAction trade = null;

    /**
     * Re-create the DayTrader db tables and populate them OR just populate a
     * DayTrader DB, logging to the provided output stream
     */
    public TradeBuildDB(Log log, TradeAction trade) throws Exception {
        this.trade = trade;
        // update config

        // always use TradeJPADirect mode

        // removed - createDBTables

        // removed - Attempt to delete all of the Trade users and Trade Quotes first

    }

    public void setup(int quotes, int users) {
        createQuotes(quotes);
        createAccounts(users);
    }
    private void createQuotes(int quotes) {
        int errorCount = 0;
        String symbol, companyName;
        TradeConfig.log.info("TradeBuildDB.createQuotes(" + quotes + ")");
        for (int i = 0; i < quotes; i++) {
            symbol = "s:" + i;
            companyName = "S" + i + " Incorporated";
            try {
                QuoteDataBean quoteData = trade.createQuote(symbol, companyName,
                    new java.math.BigDecimal(TradeConfig.rndPrice()));
            } catch (Exception e) {
                if (errorCount++ >= 10) {
                    TradeConfig.log.error("createQuotes - aborting after 10 create quote errors", e);
                    throw new RuntimeException(e);
                }
            }
        }

    }

    private void createAccounts(int users) {
        TradeConfig.log.info("TradeBuildDB.createAccounts(" + users + ")");
        for (int i = 0; i < users; i++) {
            String userID = "uid:" + i;
            String fullname = TradeConfig.rndFullName();
            String email = TradeConfig.rndEmail(userID);
            String address = TradeConfig.rndAddress();
            String creditcard = TradeConfig.rndCreditCard();
            double initialBalance = (double) (TradeConfig.rndInt(100000)) + 200000;
            if (i == 0) {
                initialBalance = 1000000; // uid:0 starts with a cool million.
            }

            AccountDataBean accountData = trade.register(userID, "xxx", fullname, address,
                email, creditcard, new BigDecimal(initialBalance));

            String symbol;
            if (accountData != null) {
                // 0-MAX_HOLDING (inclusive), avg holdings per user = (MAX-0)/2
                // int holdings = TradeConfig.rndInt(TradeConfig.getMAX_HOLDINGS() + 1);
                int holdings = TradeConfig.getMAX_HOLDINGS();
                double quantity = 0;
                OrderDataBean orderData = null;
                for (int j = 0; j < holdings; j++) {
                    symbol = TradeConfig.rndSymbol();
                    quantity = TradeConfig.rndQuantity();
                    orderData = trade.buy(userID, symbol, quantity, TradeConfig.orderProcessingMode);
                }
                if (TradeConfig.log.isTraceEnabled()) {
                    TradeConfig.log.trace("createAccounts - created " + holdings + " for userID=" + userID + " order=" + orderData);
                }
            } else {
                TradeConfig.log.error("createAccounts - userID=" + userID + " already registered.");
                throw new RuntimeException("createAccounts - userID=" + userID + " already registered.");
            }
        }
    }

}