/* * Copyright (C) 2010 Eko Kurniawan Khannedy * * This library is free software; you can redistribute it and/or * modify it under the terms of the GNU Lesser General Public * License as published by the Free Software Foundation; either * version 2.1 of the License, or (at your option) any later version. * * This library is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU * Lesser General Public License for more details. * * You should have received a copy of the GNU Lesser General Public * License along with this library; if not, write to the Free Software * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA */package khannedy.filmtube.servlet;import java.io.File;import java.io.FileOutputStream;import java.io.IOException;import java.io.PrintWriter;import java.util.List;import java.util.Random;import javax.servlet.ServletException;import javax.servlet.annotation.WebServlet;import javax.servlet.http.HttpServlet;import javax.servlet.http.HttpServletRequest;import javax.servlet.http.HttpServletResponse;import khannedy.filmtube.entity.Member;import khannedy.filmtube.entity.Video;import khannedy.filmtube.utilities.HibernateUtil;import org.apache.commons.fileupload.FileItem;import org.apache.commons.fileupload.FileItemFactory;import org.apache.commons.fileupload.FileUploadException;import org.apache.commons.fileupload.disk.DiskFileItemFactory;import org.apache.commons.fileupload.servlet.ServletFileUpload;import org.hibernate.Session;/** * * @author echo */@WebServlet(name = "UploadServlet", urlPatterns = {"/UploadServlet"})public class UploadServlet extends HttpServlet { private Random random = new Random(); /** * Processes requests for both HTTP <code>GET</code> and <code>POST</code> methods. * @param request servlet request * @param response servlet response * @throws ServletException if a servlet-specific error occurs * @throws IOException if an I/O error occurs */ protected void processRequest(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException { response.setContentType("text/html;charset=UTF-8"); PrintWriter out = response.getWriter(); try { String title = null; String location = null; FileItemFactory itemFactory = new DiskFileItemFactory(); ServletFileUpload fileUpload = new ServletFileUpload(itemFactory); List<FileItem> fileItems = fileUpload.parseRequest(request); String folder = "/home/echo/phpholic/video"; for (FileItem fileItem : fileItems) { if (!fileItem.isFormField()) { location = random.nextLong() + fileItem.getName(); File file = new File(folder, location); FileOutputStream stream = new FileOutputStream(file); stream.write(fileItem.get()); stream.close(); } else { title = fileItem.getString(); } } Session session = HibernateUtil.getSessionFactory().openSession(); session.beginTransaction(); Member member = (Member) request.getSession().getAttribute("member"); session.refresh(member); Video video = new Video(); video.setName(title); video.setLocation(location); video.setOwner(member); session.save(video); session.getTransaction().commit(); session.close(); response.sendRedirect("album.faces"); } catch (FileUploadException ex) { out.write(ex.getMessage()); } finally { out.close(); } } // <editor-fold defaultstate="collapsed" desc="HttpServlet methods. Click on the + sign on the left to edit the code."> /** * Handles the HTTP <code>POST</code> method. * @param request servlet request * @param response servlet response * @throws ServletException if a servlet-specific error occurs * @throws IOException if an I/O error occurs */ @Override protected void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException { processRequest(request, response); } /** * Returns a short description of the servlet. * @return a String containing servlet description */ @Override public String getServletInfo() { return "Short description"; }// </editor-fold>}