PlaceGroupPdf.java

package fi.insomnia.bortal.servlet;

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

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 org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import fi.insomnia.bortal.beans.PlaceGroupBeanLocal;
import fi.insomnia.bortal.beans.UserBeanLocal;
import fi.insomnia.bortal.model.GroupMembership;

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

    @EJB
    private PlaceGroupBeanLocal pgbean;

    @EJB
    private UserBeanLocal userbean;
    private static final Logger logger = LoggerFactory.getLogger(PlaceGroupPdf.class);

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

    protected void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
        output(request, response);
    }

    private void output(HttpServletRequest request, HttpServletResponse response) {
        try {
            response.setContentType("application/pdf");
            response.setHeader("Content-Disposition", "filename=Reservations.pdf");

            List<GroupMembership> members = pgbean.getMembershipsAndCreations(userbean.getCurrentUser());
            ServletOutputStream ostream = response.getOutputStream();

            pgbean.getGroupMembershipPdf(members, ostream);

            // ByteArrayOutputStream billstream = new ByteArrayOutputStream();
            // billentity.getPdfBillStream(bill, billstream);

            // response.setContentLength(billstream.size());

            ostream.close();
        } catch (IOException e) {
            logger.debug("Error writing pdf", e);
        }

    }

    protected void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
        output(request, response);
    }

}