com.sshdemo.common.report.manage.web.ParameterSetAction.java Source code

Introduction

Here is the source code for com.sshdemo.common.report.manage.web.ParameterSetAction.java

Source

/**
 * Copyright (c)2010-2011 Enterprise Website Content Management System(EWCMS), All rights reserved.
 * EWCMS PROPRIETARY/CONFIDENTIAL. Use is subject to license terms.
 * http://www.ewcms.com
 */
package com.sshdemo.common.report.manage.web;

import java.io.ByteArrayInputStream;
import java.io.InputStream;
import java.io.PrintWriter;
import java.util.Arrays;
import java.util.List;
import java.util.Map;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.apache.struts2.ServletActionContext;
import org.springframework.beans.factory.annotation.Autowired;

import com.opensymphony.xwork2.ActionSupport;
import com.sshdemo.common.report.generate.factory.ChartFactoryable;
import com.sshdemo.common.report.generate.factory.TextFactoryable;
import com.sshdemo.common.report.generate.util.ParamConversionPage;
import com.sshdemo.common.report.generate.vo.PageShowParam;
import com.sshdemo.common.report.manage.ReportFacable;
import com.sshdemo.common.report.model.ChartReport;
import com.sshdemo.common.report.model.TextReport;
import com.sshdemo.common.report.model.TextReport.Type;

public class ParameterSetAction extends ActionSupport {

    private static final long serialVersionUID = 3622306004223474063L;

    @Autowired
    private ReportFacable reportFac;
    @Autowired
    private TextFactoryable textFactory;
    @Autowired
    private ChartFactoryable chartFactory;

    private Type fileFormat;
    private Long reportId;
    private String reportType;
    private Map<String, String> paraMap;
    private List<PageShowParam> parameters;

    public void setFileFormat(Type fileFormat) {
        this.fileFormat = fileFormat;
    }

    public List<Type> getFileFormatList() {
        return Arrays.asList(Type.values());
    }

    public Long getReportId() {
        return reportId;
    }

    public void setReportId(Long reportId) {
        this.reportId = reportId;
    }

    public String getReportType() {
        return reportType;
    }

    public void setReportType(String eventOP) {
        this.reportType = eventOP;
    }

    public Map<String, String> getParaMap() {
        return paraMap;
    }

    public void setParaMap(Map<String, String> paraMap) {
        this.paraMap = paraMap;
    }

    public List<PageShowParam> getParameters() {
        return parameters;
    }

    public void setParameters(List<PageShowParam> parameters) {
        this.parameters = parameters;
    }

    public Boolean isTextable() {
        if (reportType.indexOf("text") != -1)
            return true;
        return false;
    }

    @Override
    public String execute() throws Exception {
        if (reportType.equals("text")) {
            parameters = ParamConversionPage.conversion(reportFac.findTextReportById(reportId).getParameters());
            return SUCCESS;
        }

        if (reportType.equals("chart")) {
            parameters = ParamConversionPage.conversion(reportFac.findChartReportById(reportId).getParameters());
            return SUCCESS;
        }
        return null;
    }

    public void build() {
        if (reportType.equals("text")) {
            buildText();
        } else if (reportType.equals("chart")) {
            buildChart();
        }
    }

    private void buildText() {
        PrintWriter pw = null;
        InputStream in = null;
        try {
            TextReport report = reportFac.findTextReportById(reportId);
            HttpServletResponse response = ServletActionContext.getResponse();
            HttpServletRequest request = ServletActionContext.getRequest();

            pw = response.getWriter();
            response.reset();// 
            response.setContentLength(0);
            response.setCharacterEncoding("utf-8");
            response.setContentType("text/html; charset=utf-8");
            byte[] bytes = textFactory.export(paraMap, report, fileFormat, response, request);
            in = new ByteArrayInputStream(bytes);
            int len = 0;
            while ((len = in.read()) > -1) {
                pw.write(len);
            }
            pw.flush();
        } catch (Exception e) {
            // log.error(e.toString());
        } finally {
            if (pw != null) {
                try {
                    pw.close();
                    pw = null;
                } catch (Exception e) {
                }
            }
            if (in != null) {
                try {
                    in.close();
                    in = null;
                } catch (Exception e) {
                }
            }
        }
    }

    private void buildChart() {
        PrintWriter pw = null;
        InputStream in = null;
        try {
            ChartReport chart = reportFac.findChartReportById(reportId);

            HttpServletResponse response = ServletActionContext.getResponse();
            pw = response.getWriter();

            response.reset();// 
            response.setContentLength(0);
            byte[] bytes = chartFactory.export(chart, paraMap);
            response.setContentLength(bytes.length);
            response.setHeader("Content-Type", "image/png");
            in = new ByteArrayInputStream(bytes);
            int len = 0;
            while ((len = in.read()) > -1) {
                pw.write(len);
            }
            pw.flush();
        } catch (Exception e) {
            // log.error(e.toString());
        } finally {
            if (pw != null) {
                try {
                    pw.close();
                    pw = null;
                } catch (Exception e) {
                }
            }
            if (in != null) {
                try {
                    in.close();
                    in = null;
                } catch (Exception e) {
                }
            }
        }
    }

}