geronimo-scm mailing list archives

Site index · List index
Message view « Date » · « Thread »
Top « Date » · « Thread »
From viol...@apache.org
Subject svn commit: r1155997 [4/4] - in /geronimo/daytrader/trunk/javaee6/modules/web/src/main/java/org/apache/geronimo/daytrader/javaee6/core: ./ api/ beans/ direct/
Date Wed, 10 Aug 2011 01:45:07 GMT
Added: geronimo/daytrader/trunk/javaee6/modules/web/src/main/java/org/apache/geronimo/daytrader/javaee6/core/direct/TradeJPADirect.java
URL: http://svn.apache.org/viewvc/geronimo/daytrader/trunk/javaee6/modules/web/src/main/java/org/apache/geronimo/daytrader/javaee6/core/direct/TradeJPADirect.java?rev=1155997&view=auto
==============================================================================
--- geronimo/daytrader/trunk/javaee6/modules/web/src/main/java/org/apache/geronimo/daytrader/javaee6/core/direct/TradeJPADirect.java
(added)
+++ geronimo/daytrader/trunk/javaee6/modules/web/src/main/java/org/apache/geronimo/daytrader/javaee6/core/direct/TradeJPADirect.java
Wed Aug 10 01:45:07 2011
@@ -0,0 +1,1041 @@
+/**
+ *  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.geronimo.daytrader.javaee6.core.direct;
+
+import java.math.BigDecimal;
+import java.sql.Timestamp;
+import java.util.ArrayList;
+import java.util.Collection;
+import java.util.Iterator;
+
+import javax.persistence.EntityManager;
+import javax.persistence.EntityManagerFactory;
+import javax.persistence.Persistence;
+import javax.persistence.PersistenceUnit;
+import javax.persistence.Query;
+
+import org.apache.geronimo.daytrader.javaee6.entities.*;
+import org.apache.geronimo.daytrader.javaee6.core.api.*;
+import org.apache.geronimo.daytrader.javaee6.core.beans.*;
+
+import org.apache.geronimo.daytrader.javaee6.utils.*;
+
+/**
+ * TradeJPADirect uses JPA to implement the business methods of the Trade online
+ * broker application. These business methods represent the features and
+ * operations that can be performed by customers of the brokerage such as login,
+ * logout, get a stock quote, buy or sell a stock, etc. and are specified in the
+ * {@link org.apache.geronimo.samples.daytrader.TradeServices} interface
+ * 
+ * Note: In order for this class to be thread-safe, a new TradeJPA must be
+ * created for each call to a method from the TradeInterface interface.
+ * Otherwise, pooled connections may not be released.
+ * 
+ * @see org.apache.geronimo.samples.daytrader.TradeServices
+ * 
+ */
+
+public class TradeJPADirect implements TradeServices, TradeDBServices {
+
+    @PersistenceUnit(unitName="daytrader")
+    private static EntityManagerFactory emf;
+
+    private static BigDecimal ZERO = new BigDecimal(0.0);
+
+    private static boolean initialized = false;
+    
+    private Integer soldholdingID = null;
+    private Object soldholdingIDlock = new Object();
+
+    /**
+     * Zero arg constructor for TradeJPADirect
+     */
+    public TradeJPADirect() {
+
+
+        // FIXME - Why is this here???
+        TradeConfig.setPublishQuotePriceChange(false);
+        if (emf == null) {
+            Log.error("TradeJPADirect.ctor:  Calling createEntityManagerFactory()");
+            // creating entity manager factory. the persistence xml must be
+            // place under src/META-INF/
+            emf = Persistence.createEntityManagerFactory("daytrader");
+        }
+
+        if (initialized == false)
+            init();
+    }
+
+    public void setEmf (EntityManagerFactory em) { 
+        emf = em;
+    }
+
+    public static synchronized void init() {
+        if (initialized)
+            return;
+        if (Log.doTrace())
+            Log.trace("TradeJPADirect:init -- *** initializing");  
+
+        TradeConfig.setPublishQuotePriceChange(false);
+
+        if (Log.doTrace())
+            Log.trace("TradeJPADirect:init -- +++ initialized");
+
+        initialized = true;
+    }
+
+    public static void destroy() {
+        try {
+            if (!initialized)
+                return;
+            Log.trace("TradeJPADirect:destroy");
+        }
+        catch (Exception e) {
+            Log.error("TradeJPADirect:destroy", e);
+        }
+
+    }
+
+    public MarketSummaryDataBean getMarketSummary() {
+        MarketSummaryDataBean marketSummaryData;
+
+        /*
+         * Creating entiManager
+         */
+        EntityManager entityManager = emf.createEntityManager();
+
+        try {
+            if (Log.doTrace())
+                Log.trace("TradeJPADirect:getMarketSummary -- getting market summary");
+
+            // Find Trade Stock Index Quotes (Top 100 quotes)
+            // ordered by their change in value
+            Collection<QuoteDataBean> quotes;
+
+            Query query = entityManager
+                          .createNamedQuery("quoteejb.quotesByChange");
+            quotes = query.getResultList();
+
+            QuoteDataBean[] quoteArray = (QuoteDataBean[]) quotes.toArray(new QuoteDataBean[quotes.size()]);
+            ArrayList<QuoteDataBean> topGainers = new ArrayList<QuoteDataBean>(
+                                                                              5);
+            ArrayList<QuoteDataBean> topLosers = new ArrayList<QuoteDataBean>(5);
+            BigDecimal TSIA = FinancialUtils.ZERO;
+            BigDecimal openTSIA = FinancialUtils.ZERO;
+            double totalVolume = 0.0;
+
+            if (quoteArray.length > 5) {
+                for (int i = 0; i < 5; i++)
+                    topGainers.add(quoteArray[i]);
+                for (int i = quoteArray.length - 1; i >= quoteArray.length - 5; i--)
+                    topLosers.add(quoteArray[i]);
+
+                for (QuoteDataBean quote : quoteArray) {
+                    BigDecimal price = quote.getPrice();
+                    BigDecimal open = quote.getOpen();
+                    double volume = quote.getVolume();
+                    TSIA = TSIA.add(price);
+                    openTSIA = openTSIA.add(open);
+                    totalVolume += volume;
+                }
+                TSIA = TSIA.divide(new BigDecimal(quoteArray.length),
+                                   FinancialUtils.ROUND);
+                openTSIA = openTSIA.divide(new BigDecimal(quoteArray.length),
+                                           FinancialUtils.ROUND);
+            }
+
+            marketSummaryData = new MarketSummaryDataBean(TSIA, openTSIA,
+                                                          totalVolume, topGainers, topLosers);
+        }
+        catch (Exception e) {
+            Log.error("TradeJPADirect:getMarketSummary", e);
+            throw new RuntimeException("TradeJPADirect:getMarketSummary -- error ", e);
+        } finally {
+            entityManager.close();
+        }
+
+        return marketSummaryData;
+    }
+
+    public OrderDataBean buy(String userID, String symbol, double quantity, int orderProcessingMode)
{
+        OrderDataBean order = null;
+        BigDecimal total;
+        /*
+         * creating entitymanager
+         */
+        EntityManager entityManager = emf.createEntityManager();
+
+        try {
+            if (Log.doTrace())
+                Log.trace("TradeJPADirect:buy", userID, symbol, quantity, orderProcessingMode);
+
+            entityManager.getTransaction().begin();
+
+            AccountProfileDataBean profile = entityManager.find(
+                                                               AccountProfileDataBean.class,
userID);
+            AccountDataBean account = profile.getAccount();
+
+            QuoteDataBean quote = entityManager.find(QuoteDataBean.class,
+                                                     symbol);
+
+            HoldingDataBean holding = null; // The holding will be created by this buy order
+
+            order = createOrder(account, quote, holding, "buy", quantity, entityManager);
+
+            // order = createOrder(account, quote, holding, "buy", quantity);
+            // UPDATE - account should be credited during completeOrder
+
+            BigDecimal price = quote.getPrice();
+            BigDecimal orderFee = order.getOrderFee();
+            BigDecimal balance = account.getBalance();
+            total = (new BigDecimal(quantity).multiply(price)).add(orderFee);
+            account.setBalance(balance.subtract(total));
+
+            // commit the transaction before calling completeOrder
+            entityManager.getTransaction().commit();
+
+            if (orderProcessingMode == TradeConfig.SYNCH)
+                completeOrder(order.getOrderID(), false);
+            else if (orderProcessingMode == TradeConfig.ASYNCH_2PHASE)
+                queueOrder(order.getOrderID(), true);
+        }
+        catch (Exception e) {
+            Log.error("TradeJPADirect:buy(" + userID + "," + symbol + "," + quantity + ")
--> failed", e);
+            /* On exception - cancel the order */
+            // TODO figure out how to do this with JPA
+            if (order != null)
+                order.cancel();
+
+            entityManager.getTransaction().rollback();
+
+            // throw new EJBException(e);
+            throw new RuntimeException(e);
+        } finally {
+            if (entityManager != null) {
+                entityManager.close();
+                entityManager = null;
+            }
+
+        }
+
+        // after the purchase or sell of a stock, update the stocks volume and
+        // price
+        updateQuotePriceVolume(symbol, TradeConfig.getRandomPriceChangeFactor(), quantity);
+
+        return order;
+    }
+
+    public OrderDataBean sell(String userID, Integer holdingID,
+                              int orderProcessingMode) {
+        EntityManager entityManager = emf.createEntityManager();
+
+        OrderDataBean order = null;
+        BigDecimal total;
+        try {
+            entityManager.getTransaction().begin();
+            if (Log.doTrace())
+                Log.trace("TradeJPADirect:sell", userID, holdingID, orderProcessingMode);
+
+            AccountProfileDataBean profile = entityManager.find(
+                                                               AccountProfileDataBean.class,
userID);
+
+            AccountDataBean account = profile.getAccount();
+            HoldingDataBean holding = entityManager.find(HoldingDataBean.class,
+                                                         holdingID);
+
+            if (holding == null) {
+                Log.error("TradeJPADirect:sell User " + userID
+                          + " attempted to sell holding " + holdingID
+                          + " which has already been sold");
+
+                OrderDataBean orderData = new OrderDataBean();
+                orderData.setOrderStatus("cancelled");
+
+                entityManager.persist(orderData);
+                entityManager.getTransaction().commit();
+                return orderData;
+            }
+
+            QuoteDataBean quote = holding.getQuote();
+            double quantity = holding.getQuantity();
+
+            order = createOrder(account, quote, holding, "sell", quantity,
+                                entityManager);
+            // UPDATE the holding purchase data to signify this holding is
+            // "inflight" to be sold
+            // -- could add a new holdingStatus attribute to holdingEJB
+            holding.setPurchaseDate(new java.sql.Timestamp(0));
+
+            // UPDATE - account should be credited during completeOrder
+            BigDecimal price = quote.getPrice();
+            BigDecimal orderFee = order.getOrderFee();
+            BigDecimal balance = account.getBalance();
+            total = (new BigDecimal(quantity).multiply(price)).subtract(orderFee);
+
+            account.setBalance(balance.add(total));
+
+            // commit the transaction before calling completeOrder
+            entityManager.getTransaction().commit();
+
+            if (orderProcessingMode == TradeConfig.SYNCH) {                
+                synchronized(soldholdingIDlock) {
+                    this.soldholdingID = holding.getHoldingID();
+                    completeOrder(order.getOrderID(), false);
+                }                
+            } else if (orderProcessingMode == TradeConfig.ASYNCH_2PHASE)
+                queueOrder(order.getOrderID(), true);
+
+        }
+        catch (Exception e) {
+            Log.error("TradeJPADirect:sell(" + userID + "," + holdingID + ") --> failed",
e);
+            // TODO figure out JPA cancel
+            if (order != null)
+                order.cancel();
+
+            entityManager.getTransaction().rollback();
+
+            throw new RuntimeException("TradeJPADirect:sell(" + userID + "," + holdingID
+ ")", e);
+        } finally {
+            if (entityManager != null) {
+                entityManager.close();
+                entityManager = null;
+            }
+        }
+
+        if (!(order.getOrderStatus().equalsIgnoreCase("cancelled")))
+            //after the purchase or sell of a stock, update the stocks volume and price
+            updateQuotePriceVolume(order.getSymbol(), TradeConfig.getRandomPriceChangeFactor(),
order.getQuantity());
+
+        return order;
+    }
+
+    public void queueOrder(Integer orderID, boolean twoPhase) {
+        Log
+        .error("TradeJPADirect:queueOrder() not implemented for this runtime mode");
+        throw new UnsupportedOperationException(
+                                               "TradeJPADirect:queueOrder() not implemented
for this runtime mode");
+    }
+
+    public OrderDataBean completeOrder(Integer orderID, boolean twoPhase)
+    throws Exception {
+        EntityManager entityManager = emf.createEntityManager();
+        OrderDataBean order = null;
+
+        if (Log.doTrace())
+            Log.trace("TradeJPADirect:completeOrder", orderID + " twoPhase=" + twoPhase);
+
+        order = entityManager.find(OrderDataBean.class, orderID);
+        order.getQuote();
+
+        if (order == null) {
+            Log.error("TradeJPADirect:completeOrder -- Unable to find Order " + orderID +
" FBPK returned " + order);
+            return null;
+        }
+
+        if (order.isCompleted()) {
+            throw new RuntimeException("Error: attempt to complete Order that is already
completed\n" + order);
+        }
+
+        AccountDataBean account = order.getAccount();
+        QuoteDataBean quote = order.getQuote();
+        HoldingDataBean holding = null;
+        if(order.isSell() && this.soldholdingID != null){
+            holding = entityManager.find(HoldingDataBean.class, this.soldholdingID);
+        }        
+        BigDecimal price = order.getPrice();
+        double quantity = order.getQuantity();
+
+        //String userID = account.getProfile().getUserID();
+
+        if (Log.doTrace())
+            Log.trace("TradeJPADirect:completeOrder--> Completing Order "
+                      + order.getOrderID() + "\n\t Order info: " + order
+                      + "\n\t Account info: " + account + "\n\t Quote info: "
+                      + quote + "\n\t Holding info: " + holding);
+
+        HoldingDataBean newHolding = null;        
+        
+        if (order.isBuy()) {
+            /*
+             * Complete a Buy operation - create a new Holding for the Account -
+             * deduct the Order cost from the Account balance
+             */
+            //newHolding = createHolding(account, quote, quantity, price, entityManager);
+            entityManager.getTransaction().begin();
+            newHolding = new HoldingDataBean(quantity, price, new Timestamp(System.currentTimeMillis()),
account, quote);
+            entityManager.persist(newHolding);            
+            
+            if (newHolding != null) {
+                order.setHolding(newHolding);
+            }
+            entityManager.getTransaction().commit();
+        }
+
+        try {
+            if (order.isSell()) {
+                /*
+                 * Complete a Sell operation - remove the Holding from the Account -
+                 * deposit the Order proceeds to the Account balance
+                 */
+                if (holding == null) {
+                    Log.error("TradeJPADirect:completeOrder -- Unable to sell order " + order.getOrderID()
+ " holding already sold");
+                    order.cancel();                    
+                    return order;
+                }
+                else {
+                    entityManager.getTransaction().begin();
+                    entityManager.remove(holding);                    
+                    order.setHolding(null);
+                    this.soldholdingID = null;
+                    entityManager.getTransaction().commit();
+                }
+            }
+
+            order.setOrderStatus("closed");
+
+            order.setCompletionDate(new java.sql.Timestamp(System.currentTimeMillis()));
+
+            if (Log.doTrace())
+                Log.trace("TradeJPADirect:completeOrder--> Completed Order "
+                          + order.getOrderID() + "\n\t Order info: " + order
+                          + "\n\t Account info: " + account + "\n\t Quote info: "
+                          + quote + "\n\t Holding info: " + holding);
+
+        }
+        catch (Exception e) {
+            e.printStackTrace();
+            entityManager.getTransaction().rollback();
+        } finally {
+            if (entityManager != null) {
+                entityManager.close();
+                entityManager = null;
+            }
+        }
+
+        return order;
+    }
+
+    public void cancelOrder(Integer orderID, boolean twoPhase) {
+        EntityManager entityManager = emf.createEntityManager();
+
+        if (Log.doTrace())
+            Log.trace("TradeJPADirect:cancelOrder", orderID + " twoPhase=" + twoPhase);
+
+        OrderDataBean order = entityManager.find(OrderDataBean.class, orderID);
+        /*
+         * managed transaction
+         */
+        try {
+            entityManager.getTransaction().begin();
+            order.cancel();
+            entityManager.getTransaction().commit();
+        }
+        catch (Exception e) {
+            entityManager.getTransaction().rollback();
+        } finally {
+            entityManager.close();
+        }
+    }
+
+    public void orderCompleted(String userID, Integer orderID) {
+        if (Log.doActionTrace())
+            Log.trace("TradeAction:orderCompleted", userID, orderID);
+        if (Log.doTrace())
+            Log.trace("OrderCompleted", userID, orderID);
+    }
+
+    public Collection<OrderDataBean> getOrders(String userID) {
+        if (Log.doTrace())
+            Log.trace("TradeJPADirect:getOrders", userID);
+        EntityManager entityManager = emf.createEntityManager();
+        AccountProfileDataBean profile = entityManager.find(
+                                                           AccountProfileDataBean.class,
userID);
+        AccountDataBean account = profile.getAccount();
+        entityManager.close();
+        return account.getOrders();
+    }
+
+    public Collection<OrderDataBean> getClosedOrders(String userID) {
+
+        if (Log.doTrace())
+            Log.trace("TradeJPADirect:getClosedOrders", userID);
+        EntityManager entityManager = emf.createEntityManager();
+
+        try {
+
+            // Get the primary keys for all the closed Orders for this
+            // account.
+            /*
+             * managed transaction
+             */
+            //entityManager.getTransaction().begin();
+            Query query = entityManager
+                          .createNamedQuery("orderejb.closedOrders");
+            query.setParameter("userID", userID);
+
+            //entityManager.getTransaction().commit();
+            Collection results = query.getResultList();
+            Iterator itr = results.iterator();
+            // entityManager.joinTransaction();
+            // Spin through the orders to populate the lazy quote fields
+            while (itr.hasNext()) {
+                OrderDataBean thisOrder = (OrderDataBean) itr.next();
+                thisOrder.getQuote();
+            }
+
+            if (TradeConfig.jpaLayer == TradeConfig.OPENJPA) {
+                Query updateStatus = entityManager
+                                     .createNamedQuery("orderejb.completeClosedOrders");
+                /*
+                 * managed transaction
+                 */
+                try {
+                    entityManager.getTransaction().begin();
+                    updateStatus.setParameter("userID", userID);
+
+                    updateStatus.executeUpdate();
+                    entityManager.getTransaction().commit();
+                }
+                catch (Exception e) {
+                    entityManager.getTransaction().rollback();
+                    entityManager.close();
+                    entityManager = null;
+                }
+            }
+            else if (TradeConfig.jpaLayer == TradeConfig.HIBERNATE) {
+                try {
+                /*
+                 * Add logic to do update orders operation, because JBoss5'
+                 * Hibernate 3.3.1GA DB2Dialect and MySQL5Dialect do not work
+                 * with annotated query "orderejb.completeClosedOrders" defined
+                 * in OrderDatabean
+                 */
+                Query findaccountid = entityManager
+                                      .createNativeQuery(
+                                                        "select "
+                                                        + "a.ACCOUNTID, "
+                                                        + "a.LOGINCOUNT, "
+                                                        + "a.LOGOUTCOUNT, "
+                                                        + "a.LASTLOGIN, "
+                                                        + "a.CREATIONDATE, "
+                                                        + "a.BALANCE, "
+                                                        + "a.OPENBALANCE, "
+                                                        + "a.PROFILE_USERID "
+                                                        + "from accountejb a where a.profile_userid
= ?",
+                                                        AccountDataBean.class);
+                findaccountid.setParameter(1, userID);
+                AccountDataBean account = (AccountDataBean) findaccountid.getSingleResult();
+                Integer accountid = account.getAccountID();
+                Query updateStatus = entityManager.createNativeQuery("UPDATE orderejb o SET
o.orderStatus = 'completed' WHERE "
+                                                                     + "o.orderStatus = 'closed'
AND o.ACCOUNT_ACCOUNTID  = ?");
+                updateStatus.setParameter(1, accountid.intValue());
+                entityManager.getTransaction().begin();
+                updateStatus.executeUpdate();
+                entityManager.getTransaction().commit();
+                } catch (Exception e){
+                    entityManager.getTransaction().rollback();
+                    entityManager.close();
+                    entityManager = null;
+                }
+            }
+            
+            if (entityManager != null) {
+                entityManager.close();
+                entityManager = null;
+            }
+            return results;
+        }
+        catch (Exception e) {
+            Log.error("TradeJPADirect.getClosedOrders", e);
+            entityManager.close();
+            entityManager = null;
+            throw new RuntimeException(
+                                      "TradeJPADirect.getClosedOrders - error", e);
+
+        } finally {
+            if (entityManager != null) {
+                entityManager.close();
+                entityManager = null;
+            }
+        }
+
+    }
+
+    public QuoteDataBean createQuote(String symbol, String companyName,
+                                     BigDecimal price) {
+        EntityManager entityManager = emf.createEntityManager();
+        try {
+            QuoteDataBean quote = new QuoteDataBean(symbol, companyName, 0, price, price,
price, price, 0);
+            /*
+             * managed transaction
+             */
+            try {
+                entityManager.getTransaction().begin();
+                entityManager.persist(quote);
+                entityManager.getTransaction().commit();
+            }
+            catch (Exception e) {
+                entityManager.getTransaction().rollback();
+            }
+
+            if (Log.doTrace())
+                Log.trace("TradeJPADirect:createQuote-->" + quote);
+
+            if (entityManager != null) {
+                entityManager.close();
+                entityManager = null;
+            }
+            return quote;
+        }
+        catch (Exception e) {
+            Log.error("TradeJPADirect:createQuote -- exception creating Quote", e);
+            entityManager.close();
+            entityManager = null;
+            throw new RuntimeException(e);
+        } finally {
+            if (entityManager != null) {
+                entityManager.close();
+                entityManager = null;
+            }
+        }
+
+    }
+
+    public QuoteDataBean getQuote(String symbol) {
+        if (Log.doTrace())
+            Log.trace("TradeJPADirect:getQuote", symbol);
+        EntityManager entityManager = emf.createEntityManager();
+
+        QuoteDataBean qdb = entityManager.find(QuoteDataBean.class, symbol);
+
+        if (entityManager != null) {
+            entityManager.close();
+            entityManager = null;
+        }
+        return qdb;
+    }
+
+    public Collection<QuoteDataBean> getAllQuotes() {
+        if (Log.doTrace())
+            Log.trace("TradeJPADirect:getAllQuotes");
+        EntityManager entityManager = emf.createEntityManager();
+
+        Query query = entityManager.createNamedQuery("quoteejb.allQuotes");
+
+        if (entityManager != null) {
+            entityManager.close();
+            entityManager = null;
+
+        }
+        return query.getResultList();
+    }
+
+    public QuoteDataBean updateQuotePriceVolume(String symbol,
+                                                BigDecimal changeFactor, double sharesTraded)
{
+        if (!TradeConfig.getUpdateQuotePrices())
+            return new QuoteDataBean();
+
+        if (Log.doTrace())
+            Log.trace("TradeJPADirect:updateQuote", symbol, changeFactor);
+
+        /*
+         * Add logic to determine JPA layer, because JBoss5' Hibernate 3.3.1GA
+         * DB2Dialect and MySQL5Dialect do not work with annotated query
+         * "quoteejb.quoteForUpdate" defined in QuoteDatabean
+         */
+        EntityManager entityManager = emf.createEntityManager();
+        QuoteDataBean quote = null;
+        if (TradeConfig.jpaLayer == TradeConfig.HIBERNATE) {
+            quote = entityManager.find(QuoteDataBean.class, symbol);
+        } else if (TradeConfig.jpaLayer == TradeConfig.OPENJPA) {
+  
+            Query q = entityManager.createNamedQuery("quoteejb.quoteForUpdate");
+            q.setParameter(1, symbol);
+  
+            quote = (QuoteDataBean) q.getSingleResult();
+        }
+
+        BigDecimal oldPrice = quote.getPrice();
+
+        if (quote.getPrice().equals(TradeConfig.PENNY_STOCK_PRICE)) {
+            changeFactor = TradeConfig.PENNY_STOCK_RECOVERY_MIRACLE_MULTIPLIER;
+        }
+
+        BigDecimal newPrice = changeFactor.multiply(oldPrice).setScale(2, BigDecimal.ROUND_HALF_UP);
+
+        /*
+         * managed transaction
+         */
+
+        try {
+
+            quote.setPrice(newPrice);
+            quote.setVolume(quote.getVolume() + sharesTraded);
+            quote.setChange((newPrice.subtract(quote.getOpen()).doubleValue()));
+
+            entityManager.getTransaction().begin();
+            entityManager.merge(quote);
+            entityManager.getTransaction().commit();
+        }
+        catch (Exception e) {
+            entityManager.getTransaction().rollback();
+        } finally {
+            if (entityManager != null) {
+                entityManager.close();
+                entityManager = null;
+            }
+        }
+
+        this.publishQuotePriceChange(quote, oldPrice, changeFactor, sharesTraded);
+
+        return quote;
+    }
+
+    public Collection<HoldingDataBean> getHoldings(String userID) {
+        if (Log.doTrace())
+            Log.trace("TradeJPADirect:getHoldings", userID);
+        EntityManager entityManager = emf.createEntityManager();
+        /*
+         * managed transaction
+         */
+        entityManager.getTransaction().begin();
+
+        Query query = entityManager.createNamedQuery("holdingejb.holdingsByUserID");
+        query.setParameter("userID", userID);
+
+        entityManager.getTransaction().commit();
+        Collection<HoldingDataBean> holdings = query.getResultList();
+        /*
+         * Inflate the lazy data memebers
+         */
+        Iterator itr = holdings.iterator();
+        while (itr.hasNext()) {
+            ((HoldingDataBean) itr.next()).getQuote();
+        }
+
+        entityManager.close();
+        entityManager = null;
+        return holdings;
+    }
+
+    public HoldingDataBean getHolding(Integer holdingID) {
+        if (Log.doTrace())
+            Log.trace("TradeJPADirect:getHolding", holdingID);
+        HoldingDataBean holding;
+        EntityManager entityManager = emf.createEntityManager();
+        holding = entityManager.find(HoldingDataBean.class, holdingID);
+        entityManager.close();
+        return holding;
+    }
+
+    public AccountDataBean getAccountData(String userID) {
+        if (Log.doTrace())
+            Log.trace("TradeJPADirect:getAccountData", userID);
+
+        EntityManager entityManager = emf.createEntityManager();
+
+        AccountProfileDataBean profile = entityManager.find(AccountProfileDataBean.class,
userID);
+        /*
+         * Inflate the lazy data memebers
+         */
+        AccountDataBean account = profile.getAccount();
+        account.getProfile();
+
+        // Added to populate transient field for account
+        account.setProfileID(profile.getUserID());
+        entityManager.close();
+        entityManager = null;
+
+        return account;
+    }
+
+    public AccountProfileDataBean getAccountProfileData(String userID) {
+        if (Log.doTrace())
+            Log.trace("TradeJPADirect:getProfileData", userID);
+        EntityManager entityManager = emf.createEntityManager();
+
+        AccountProfileDataBean apb = entityManager.find(AccountProfileDataBean.class, userID);
+        entityManager.close();
+        entityManager = null;
+        return apb;
+    }
+
+    public AccountProfileDataBean updateAccountProfile(AccountProfileDataBean profileData)
{
+
+        EntityManager entityManager = emf.createEntityManager();
+
+        if (Log.doTrace())
+            Log.trace("TradeJPADirect:updateAccountProfileData", profileData);
+        /*
+         * // Retrieve the previous account profile in order to get account
+         * data... hook it into new object AccountProfileDataBean temp =
+         * entityManager.find(AccountProfileDataBean.class,
+         * profileData.getUserID()); // In order for the object to merge
+         * correctly, the account has to be hooked into the temp object... // -
+         * may need to reverse this and obtain the full object first
+         * 
+         * profileData.setAccount(temp.getAccount());
+         * 
+         * //TODO this might not be correct temp =
+         * entityManager.merge(profileData); //System.out.println(temp);
+         */
+
+        AccountProfileDataBean temp = entityManager.find(AccountProfileDataBean.class, profileData.getUserID());
+        temp.setAddress(profileData.getAddress());
+        temp.setPassword(profileData.getPassword());
+        temp.setFullName(profileData.getFullName());
+        temp.setCreditCard(profileData.getCreditCard());
+        temp.setEmail(profileData.getEmail());
+        /*
+         * Managed Transaction
+         */
+        try {
+
+            entityManager.getTransaction().begin();
+            entityManager.merge(temp);
+            entityManager.getTransaction().commit();
+        }
+        catch (Exception e) {
+            entityManager.getTransaction().rollback();
+        } finally {
+            entityManager.close();
+        }
+
+        return temp;
+    }
+
+    public AccountDataBean login(String userID, String password)
+    throws Exception {
+
+        EntityManager entityManager = emf.createEntityManager();
+
+        AccountProfileDataBean profile = entityManager.find(AccountProfileDataBean.class,
userID);
+
+        if (profile == null) {
+            throw new RuntimeException("No such user: " + userID);
+        }
+        /*
+         * Managed Transaction
+         */
+        entityManager.getTransaction().begin();
+        entityManager.merge(profile);
+
+        AccountDataBean account = profile.getAccount();
+
+        if (Log.doTrace())
+            Log.trace("TradeJPADirect:login", userID, password);
+
+        account.login(password);
+        entityManager.getTransaction().commit();
+        if (Log.doTrace())
+            Log.trace("TradeJPADirect:login(" + userID + "," + password + ") success" + account);
+        entityManager.close();
+        return account;
+    }
+
+    public void logout(String userID) {
+        if (Log.doTrace())
+            Log.trace("TradeJPADirect:logout", userID);
+        EntityManager entityManager = emf.createEntityManager();
+
+        AccountProfileDataBean profile = entityManager.find(AccountProfileDataBean.class,
userID);
+        AccountDataBean account = profile.getAccount();
+
+        /*
+         * Managed Transaction
+         */
+        try {
+            entityManager.getTransaction().begin();
+            account.logout();
+            entityManager.getTransaction().commit();
+        }
+        catch (Exception e) {
+            entityManager.getTransaction().rollback();
+        } finally {
+            entityManager.close();
+        }
+
+        if (Log.doTrace())
+            Log.trace("TradeJPADirect:logout(" + userID + ") success");
+    }
+
+    public AccountDataBean register(String userID, String password, String fullname, 
+                                    String address, String email, String creditcard,
+                                    BigDecimal openBalance) {
+        AccountDataBean account = null;
+        AccountProfileDataBean profile = null;
+        EntityManager entityManager = emf.createEntityManager();
+
+        if (Log.doTrace())
+            Log.trace("TradeJPADirect:register", userID, password, fullname, address, email,
creditcard, openBalance);
+
+        // Check to see if a profile with the desired userID already exists
+
+        profile = entityManager.find(AccountProfileDataBean.class, userID);
+
+        if (profile != null) {
+            Log.error("Failed to register new Account - AccountProfile with userID(" + userID
+ ") already exists");
+            return null;
+        }
+        else {
+            profile = new AccountProfileDataBean(userID, password, fullname,
+                                                 address, email, creditcard);
+            account = new AccountDataBean(0, 0, null, new Timestamp(System.currentTimeMillis()),
openBalance, openBalance, userID);
+            profile.setAccount(account);
+            account.setProfile(profile);
+            /*
+             * managed Transaction
+             */
+            try {
+                entityManager.getTransaction().begin();
+                entityManager.persist(profile);
+                entityManager.persist(account);
+                entityManager.getTransaction().commit();
+            }
+            catch (Exception e) {
+                entityManager.getTransaction().rollback();
+            } finally {
+                entityManager.close();
+            }
+
+        }
+
+        return account;
+    }
+
+    // @TransactionAttribute(TransactionAttributeType.NOT_SUPPORTED)
+    public RunStatsDataBean resetTrade(boolean deleteAll) throws Exception {
+        if (Log.doTrace())
+            Log.trace("TradeJPADirect:resetTrade", deleteAll);
+
+        return(new TradeJDBCDirect(false)).resetTrade(deleteAll);
+    }
+
+    /*
+     * NO LONGER USE
+     */
+
+    private void publishQuotePriceChange(QuoteDataBean quote,
+                                         BigDecimal oldPrice, BigDecimal changeFactor, double
sharesTraded) {
+        if (!TradeConfig.getPublishQuotePriceChange())
+            return;
+        Log.error("TradeJPADirect:publishQuotePriceChange - is not implemented for this runtime
mode");
+        throw new UnsupportedOperationException("TradeJPADirect:publishQuotePriceChange -
is not implemented for this runtime mode");
+    }
+
+    /*
+     * new Method() that takes EntityManager as a parameter
+     */
+    private OrderDataBean createOrder(AccountDataBean account,
+                                      QuoteDataBean quote, HoldingDataBean holding, String
orderType,
+                                      double quantity, EntityManager entityManager) {
+        OrderDataBean order;
+        if (Log.doTrace())
+            Log.trace("TradeJPADirect:createOrder(orderID=" + " account="
+                      + ((account == null) ? null : account.getAccountID())
+                      + " quote=" + ((quote == null) ? null : quote.getSymbol())
+                      + " orderType=" + orderType + " quantity=" + quantity);
+        try {
+            order = new OrderDataBean(orderType, 
+                                      "open", 
+                                      new Timestamp(System.currentTimeMillis()), 
+                                      null, 
+                                      quantity, 
+                                      quote.getPrice().setScale(FinancialUtils.SCALE, FinancialUtils.ROUND),
+                                      TradeConfig.getOrderFee(orderType), 
+                                      account, 
+                                      quote, 
+                                      holding);
+                entityManager.persist(order);
+        }
+        catch (Exception e) {
+            entityManager.getTransaction().rollback();
+            Log.error("TradeJPADirect:createOrder -- failed to create Order", e);
+            throw new RuntimeException("TradeJPADirect:createOrder -- failed to create Order",
e);
+        }
+        return order;
+    }
+
+    /*private HoldingDataBean createHolding(AccountDataBean account,
+                                          QuoteDataBean quote, double quantity, BigDecimal
purchasePrice,
+                                          EntityManager entityManager) throws Exception {
+        HoldingDataBean newHolding = new HoldingDataBean(quantity,
+                                                         purchasePrice, new Timestamp(System.currentTimeMillis()),
+                                                         account, quote);
+        try {            
+            entityManager.persist(newHolding);            
+        }
+        catch (Exception e) {
+            entityManager.getTransaction().rollback();
+        } 
+        
+        return newHolding;
+    }*/
+
+    public double investmentReturn(double investment, double NetValue)
+    throws Exception {
+        if (Log.doTrace())
+            Log.trace("TradeJPADirect:investmentReturn");
+
+        double diff = NetValue - investment;
+        double ir = diff / investment;
+        return ir;
+    }
+
+    public QuoteDataBean pingTwoPhase(String symbol) throws Exception {
+        Log
+        .error("TradeJPADirect:pingTwoPhase - is not implemented for this runtime mode");
+        throw new UnsupportedOperationException("TradeJPADirect:pingTwoPhase - is not implemented
for this runtime mode");
+    }
+
+    class quotePriceComparator implements java.util.Comparator {
+        public int compare(Object quote1, Object quote2) {
+            double change1 = ((QuoteDataBean) quote1).getChange();
+            double change2 = ((QuoteDataBean) quote2).getChange();
+            return new Double(change2).compareTo(change1);
+        }
+    }
+
+    /**
+     * TradeBuildDB needs this abstracted method
+     */
+    public String checkDBProductName() throws Exception {
+        if (Log.doTrace())
+            Log.trace("TradeJPADirect:checkDBProductName");
+        return(new TradeJDBCDirect(false)).checkDBProductName();
+    }
+
+    /**
+     * TradeBuildDB needs this abstracted method
+     */
+    public boolean recreateDBTables(Object[] sqlBuffer, java.io.PrintWriter out)
+    throws Exception {
+        if (Log.doTrace())
+            Log.trace("TradeJPADirect:checkDBProductName");
+        return(new TradeJDBCDirect(false)).recreateDBTables(sqlBuffer, out);
+    }
+
+    /**
+     * Get mode - returns the persistence mode (TradeConfig.JPA)
+     * 
+     * @return int mode
+     */
+    public int getMode() {
+        return TradeConfig.JPA;
+    }
+
+}



Mime
View raw message