Upload

package cs3220.servlet; import java.io.File; import java.io.IOException; import java.io.PrintWriter; import java.util.List; import javax.servlet.ServletContext; 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 org.apache.commons.fileupload.FileItem; import org.apache.commons.fileupload.disk.DiskFileItemFactory; import org.apache.commons.fileupload.servlet.ServletFileUpload; @WebServlet("/Upload") public class Upload extends HttpServlet { private static final long serialVersionUID = 1L; public Upload() { super(); } protected void doGet( HttpServletRequest request, HttpServletResponse response ) throws ServletException, IOException { // Redirect the user to the upload form response.sendRedirect( "Upload.html" ); } protected void doPost( HttpServletRequest request, HttpServletResponse response ) throws ServletException, IOException { // Create a factory for disk-based file items DiskFileItemFactory factory = new DiskFileItemFactory(); // Configure a repository (to ensure a secure temp location is used) ServletContext servletContext = this.getServletConfig() .getServletContext(); File repository = (File) servletContext .getAttribute( "javax.servlet.context.tempdir" ); factory.setRepository( repository ); // Create a new file upload handler ServletFileUpload upload = new ServletFileUpload( factory ); // Count how many files are uploaded int count = 0; // The directory we want to save the uploaded files to. String fileDir = getServletContext().getRealPath( "/WEB-INF/files" ); // Parse the request try { List items = upload.parseRequest( request ); for( FileItem item : items ) { // If the item is not a form field - meaning it's an uploaded // file, we save it to the target dir if( !item.isFormField() ) { // item.getName() will return the full path of the uploaded // file, e.g. "C:/My Documents/files/test.txt", but we only // want the file name part, which is why we first create a // File object, then use File.getName() to get the file // name. String fileName = (new File( item.getName() )).getName(); File file = new File( fileDir, fileName ); item.write( file ); ++count; } } } catch( Exception e ) { throw new IOException( e ); } response.setContentType( "text/html" ); PrintWriter out = response.getWriter(); out.println( "Upload" ); out.println( "

" + count + " file(s) uploaded to " + fileDir ); out.println( "" ); } }