PrintBill.java 2.66 KB
package fi.insomnia.bortal.servlet;

import java.io.ByteArrayOutputStream;
import java.io.IOException;

import javax.ejb.EJB;
import javax.servlet.ServletException;
import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import fi.insomnia.bortal.beans.BillBeanLocal;
import fi.insomnia.bortal.beans.EventBeanLocal;
import fi.insomnia.bortal.model.Bill;

/**
 * Servlet implementation class PrintBill
 */
public class PrintBill extends HttpServlet {
    private static final long serialVersionUID = 1L;

    private static final String BILL_ID = "billid";

    @EJB
    private BillBeanLocal billentity;
    @EJB
    private EventBeanLocal eventbean;

    /**
     * @see HttpServlet#HttpServlet()
     */
    public PrintBill() {
        super();
        // TODO Auto-generated constructor stub
    }

    /**
     * @see HttpServlet#doGet(HttpServletRequest request, HttpServletResponse
     *      response)
     */
    protected void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
        ouput(request, response);
    }

    private void ouput(HttpServletRequest request, HttpServletResponse response) throws IOException {

        Integer billid = getIntegerParameter(request, BILL_ID);
        Bill bill = billentity.findById( billid);
        if (bill == null) {
            return;
        }
        ByteArrayOutputStream billstream = billentity.getPdfBillStream(bill);
        if (billstream == null) {
            return;
        }
        response.setContentLength(billstream.size());

        response.setContentType("application/pdf");
        response.setHeader("Content-Disposition", "filename=Bill-" + bill.getBillNumber() + ".pdf");

        ServletOutputStream ostream = response.getOutputStream();
        billstream.writeTo(ostream);
        ostream.close();
    }

    /**
     * @see HttpServlet#doPost(HttpServletRequest request, HttpServletResponse
     *      response)
     */
    protected void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
        ouput(request, response);
    }

    /***
     * Convert request parameter into integer
     * 
     * @param request
     * @param parameter
     * @return
     */
    private static Integer getIntegerParameter(HttpServletRequest request, String parameter) {
        try {
            String valueString = request.getParameter(parameter);
            Integer value = Integer.parseInt(valueString);
            return value;
        } catch (NumberFormatException nfe) {
        }
        return null;
    }
}