org.gedantic.web.servlet.AnalyzerServlet.java Source code

Introduction

Here is the source code for org.gedantic.web.servlet.AnalyzerServlet.java

Source

/*
 * Copyright (c) 2016 Matthew R. Harrah
 *
 * MIT License
 *
 * 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.
 */
package org.gedantic.web.servlet;

import java.io.IOException;
import java.util.Collections;
import java.util.List;

import javax.servlet.ServletException;
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.StringEscapeUtils;
import org.apache.poi.ss.usermodel.Workbook;
import org.gedantic.analyzer.AnalysisResult;
import org.gedantic.analyzer.AnalyzerList;
import org.gedantic.analyzer.IAnalyzer;
import org.gedantic.analyzer.comparator.AnalysisResultComparator;
import org.gedantic.web.Constants;
import org.gedcom4j.model.Gedcom;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * Servlet that does analysis and displays results
 * 
 * @author frizbog
 */
public class AnalyzerServlet extends HttpServlet {

    /** Logger */
    private static final Logger LOG = LoggerFactory.getLogger(AnalyzerServlet.class.getName());

    /**
     * Serial Version UID
     */
    private static final long serialVersionUID = -62757248978505969L;

    @Override
    protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
        process(req, resp);
    }

    @Override
    protected void doPost(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
        process(req, resp);
    }

    /**
     * Servlet for handling all analysis requests
     * 
     * @param req
     *            the http request
     * @param resp
     *            the http response
     * @throws ServletException
     *             if there is a code-related problemDescription
     * @throws IOException
     *             if there is a network or storage-related problemDescription
     */
    private void process(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
        HttpSession session = req.getSession();
        Gedcom g = (Gedcom) session.getAttribute(Constants.GEDCOM);
        IAnalyzer a = null;

        String analyzerId = req.getParameter("analyzerId");
        if (analyzerId != null) {
            a = AnalyzerList.getInstance().getAnalyzers().get(analyzerId);
            req.setAttribute(Constants.ANALYSIS_ID, a.getId());
            req.setAttribute(Constants.ANALYSIS_NAME, a.getName());
            req.setAttribute(Constants.ANALYSIS_DESCRIPTION, a.getDescription());
        }

        if (g == null) {
            LOG.info("Redirecting from " + req.getRequestURI()
                    + " to upload page because there is no gedcom in session");
            req.setAttribute(Constants.ALERT_MESSAGE, "Please upload a GEDCOM to analyze");
            req.setAttribute(Constants.ALERT_MESSAGE_TYPE, "alert alert-warning");
            req.getRequestDispatcher(Constants.URL_UPLOAD_PAGE).forward(req, resp);
            return;
        }

        if (a == null) {
            LOG.info("Redirecting from " + req.getRequestURI()
                    + " to upload page because the analysis ID could not be interpreted");
            req.setAttribute(Constants.ALERT_MESSAGE, "Unknown analysis type");
            req.setAttribute(Constants.ALERT_MESSAGE_TYPE, "alert alert-danger");
            req.getRequestDispatcher(Constants.URL_UPLOAD_PAGE).forward(req, resp);
            return;

        }

        Runtime rt = Runtime.getRuntime();
        long maxMemory = rt.maxMemory();
        long freeMemory = rt.freeMemory();
        LOG.info("before analysis " + a.getId() + " - freeMemory:" + Double.toString(freeMemory / (1024 * 1024))
                + " maxMemory:" + Double.toString(maxMemory / (1024 * 1024)));

        List<AnalysisResult> results = a.analyze(g);
        Collections.sort(results, new AnalysisResultComparator());

        maxMemory = rt.maxMemory();
        freeMemory = rt.freeMemory();
        LOG.info("after analysis " + a.getId() + " - freeMemory:" + Double.toString(freeMemory / (1024 * 1024))
                + " maxMemory:" + Double.toString(maxMemory / (1024 * 1024)));

        if ("true".equals(req.getParameter("excel"))) {
            @SuppressWarnings("resource")
            Workbook wb = new WorkbookCreator(a, results).create();

            resp.setContentType("application/vnd.ms-excel");
            resp.setHeader("Content-disposition",
                    "attachment; filename=" + StringEscapeUtils.escapeHtml(a.getName() + ".xlsx"));

            wb.write(resp.getOutputStream());
            resp.getOutputStream().flush();

        } else {
            req.setAttribute(Constants.RESULTS, results);
            req.getRequestDispatcher("results.tiles").forward(req, resp);
        }
    }

}