UploadServlet.java 4.08 KB
package fi.insomnia.bortal.servlet;

import java.io.IOException;
import java.io.PrintWriter;
import java.math.BigInteger;
import java.util.List;

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

import org.apache.commons.fileupload.FileItem;
import org.apache.commons.fileupload.FileItemFactory;
import org.apache.commons.fileupload.FileUploadException;
import org.apache.commons.fileupload.disk.DiskFileItem;
import org.apache.commons.fileupload.disk.DiskFileItemFactory;
import org.apache.commons.fileupload.servlet.ServletFileUpload;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import fi.insomnia.bortal.beans.EventMapBeanLocal;
import fi.insomnia.bortal.beans.UserBeanLocal;
import fi.insomnia.bortal.beans.UtilBeanLocal;
import fi.insomnia.bortal.model.UserImage;

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

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

    private static final Logger logger = LoggerFactory.getLogger(UploadServlet.class);

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

    @EJB
    private UserBeanLocal userbean;
    @EJB
    private EventMapBeanLocal eventmapbean;

    @EJB
    private UtilBeanLocal utilbean;

    /**
     * @see HttpServlet#doPost(HttpServletRequest request, HttpServletResponse
     *      response)
     */
    protected void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
        // boolean isMultipart = ServletFileUpload.isMultipartContent(request);
        FileItemFactory factory = new DiskFileItemFactory();
        ServletFileUpload upload = new ServletFileUpload(factory);

        // Parse the request
        CharSequence retmsg = "Tiedosto tallennettu onnistuneesti.";
        try {

            String contenttype = "";
            byte[] imagedata = null;

            int destId = 0;

            String description = "";
            String filename = "";
            String idstr = request.getParameter("id");
            if (idstr != null) {
                destId = Integer.parseInt(idstr);
            }
            for (Object ti : upload.parseRequest(request)) {
                if (ti instanceof FileItem) {

                    FileItem fi = (FileItem) ti;

                    if (fi.getFieldName().equals("id")) {
                        destId = Integer.parseInt(fi.getString());

                    } else if (fi.getFieldName().equals("description")) {
                        description = fi.getString();

                    } else if (fi.getFieldName().equals("file") || fi.getFieldName().equals("Filedata")) {
                        imagedata = fi.get();
                        filename = fi.getName();
                        contenttype = fi.getContentType();
                    }

                }
            }
            String type = request.getParameter("type");
            if (type.equals("userimage")) {
                logger.debug("Uploading userimage for user {}");
                UserImage userimage = userbean.uploadImage(destId, contenttype, imagedata, filename, description);
                utilbean.convertImage(userimage.getUser());
            } else if (type.equals("mapimage")) {
                eventmapbean.sendImage(destId, imagedata);
            } else {
                retmsg = "FATAL: servlet type not defined!!";
            }

        } catch (FileUploadException e) {
            logger.warn("Error uploading image", e);
            retmsg = "Tiedoston tallennuksessa tapahtui virhe!";
        }

        PrintWriter writer = response.getWriter();
        writer.append(retmsg);

    }
}