fr.cph.stock.web.servlet.portfolio.AddEquityServlet.java Source code

Introduction

Here is the source code for fr.cph.stock.web.servlet.portfolio.AddEquityServlet.java

Source

/**
 * Copyright 2013 Carl-Philipp Harmant
 *
 * Licensed 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 fr.cph.stock.web.servlet.portfolio;

import java.util.Arrays;

import javax.servlet.ServletException;
import javax.servlet.annotation.WebServlet;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpSession;

import org.apache.commons.lang.math.NumberUtils;
import org.apache.log4j.Logger;

import fr.cph.stock.business.Business;
import fr.cph.stock.business.IBusiness;
import fr.cph.stock.entities.Company;
import fr.cph.stock.entities.Equity;
import fr.cph.stock.entities.User;
import fr.cph.stock.enumtype.Currency;
import fr.cph.stock.exception.EquityException;
import fr.cph.stock.exception.YahooException;
import fr.cph.stock.language.LanguageFactory;
import fr.cph.stock.web.servlet.CookieManagement;

/**
 * This servlet is called when the user want to add an equity
 * 
 * @author Carl-Philipp Harmant
 * 
 */
@WebServlet(name = "AddEquityServlet", urlPatterns = { "/add" })
public class AddEquityServlet extends HttpServlet {

    /** Serialization **/
    private static final long serialVersionUID = -4917456731220463031L;
    /** Logger **/
    private static final Logger LOG = Logger.getLogger(AddEquityServlet.class);
    /** Business **/
    private IBusiness business;

    @Override
    public final void init() {
        business = Business.getInstance();
    }

    @Override
    protected final void doGet(final HttpServletRequest request, final HttpServletResponse response)
            throws ServletException {
        try {
            String lang = CookieManagement.getCookieLanguage(Arrays.asList(request.getCookies()));
            LanguageFactory language = LanguageFactory.getInstance();
            HttpSession session = request.getSession();
            User user = (User) session.getAttribute("user");
            String manual = request.getParameter("manual");
            if (manual != null && manual.equals("true")) {
                String manualName = request.getParameter("manualName");
                String manualUnitCostPrice = request.getParameter("manualUnitCostPrice");
                String manualQuantity = request.getParameter("manualQuantity");
                String manualParityPersonal = request.getParameter("manualParityPersonal");
                String manualCurrency = request.getParameter("manualCurrency");
                String manualIndustry = request.getParameter("manualIndustry");
                String manualSector = request.getParameter("manualSector");
                String manualQuote = request.getParameter("manualQuote");

                Double quantity = NumberUtils.createDouble(manualQuantity);
                Double unitCostPrice = NumberUtils.createDouble(manualUnitCostPrice);
                Double parityPersonal = null;
                if (!manualParityPersonal.equals("")) {
                    parityPersonal = NumberUtils.createDouble(manualParityPersonal);
                }
                Double quote = NumberUtils.createDouble(manualQuote);
                Company company = business.createManualCompany(manualName, manualIndustry, manualSector,
                        Currency.getEnum(manualCurrency), quote);

                Equity equity = new Equity();
                equity.setQuantity(quantity);
                equity.setUnitCostPrice(unitCostPrice);
                equity.setParityPersonal(parityPersonal);
                try {
                    business.createManualEquity(user.getId(), company, equity);
                    request.setAttribute("added", language.getLanguage(lang).get("CONSTANT_ADDED") + " !");
                } catch (EquityException e) {
                    request.setAttribute("addError", e.getMessage());
                }
            } else {
                String ticker = request.getParameter("ticker").toUpperCase();
                String unitCostP = request.getParameter("unitCostPrice");
                String quant = request.getParameter("quantity");
                String parityPerso = request.getParameter("parityPersonal");

                Double quantity = NumberUtils.createDouble(quant);
                Double unitCostPrice = NumberUtils.createDouble(unitCostP);
                Double parityPersonal = null;
                if (!parityPerso.equals("")) {
                    parityPersonal = NumberUtils.createDouble(parityPerso);
                }

                Equity equity = new Equity();
                equity.setQuantity(quantity);
                equity.setUnitCostPrice(unitCostPrice);
                equity.setParityPersonal(parityPersonal);
                try {
                    business.createEquity(user.getId(), ticker, equity);
                    request.setAttribute("added", language.getLanguage(lang).get("CONSTANT_ADDED") + " !");
                } catch (YahooException | EquityException e) {
                    request.setAttribute("addError", e.getMessage());
                }
            }

            request.getRequestDispatcher("home").forward(request, response);
        } catch (Throwable t) {
            LOG.error(t.getMessage(), t);
            throw new ServletException("Error: " + t.getMessage(), t);
        }
    }

    @Override
    protected final void doPost(final HttpServletRequest request, final HttpServletResponse response)
            throws ServletException {
        doGet(request, response);
    }

}