58 lines
2.3 KiB
Java
58 lines
2.3 KiB
Java
package com.baeldung.servlets;
|
|
|
|
import org.apache.commons.fileupload.FileItem;
|
|
import org.apache.commons.fileupload.disk.DiskFileItemFactory;
|
|
import org.apache.commons.fileupload.servlet.ServletFileUpload;
|
|
|
|
import javax.servlet.ServletException;
|
|
import javax.servlet.http.HttpServlet;
|
|
import javax.servlet.http.HttpServletRequest;
|
|
import javax.servlet.http.HttpServletResponse;
|
|
import java.io.File;
|
|
import java.io.IOException;
|
|
import java.util.List;
|
|
|
|
import static com.baeldung.Constants.*;
|
|
|
|
public class UploadServlet extends HttpServlet {
|
|
|
|
private static final long serialVersionUID = 1L;
|
|
|
|
protected void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
|
|
|
|
if (ServletFileUpload.isMultipartContent(request)) {
|
|
|
|
DiskFileItemFactory factory = new DiskFileItemFactory();
|
|
factory.setSizeThreshold(MEMORY_THRESHOLD);
|
|
factory.setRepository(new File(System.getProperty("java.io.tmpdir")));
|
|
|
|
ServletFileUpload upload = new ServletFileUpload(factory);
|
|
upload.setFileSizeMax(MAX_FILE_SIZE);
|
|
upload.setSizeMax(MAX_REQUEST_SIZE);
|
|
String uploadPath = getServletContext().getRealPath("") + File.separator + UPLOAD_DIRECTORY;
|
|
File uploadDir = new File(uploadPath);
|
|
if (!uploadDir.exists()) {
|
|
uploadDir.mkdir();
|
|
}
|
|
|
|
try {
|
|
List<FileItem> formItems = upload.parseRequest(request);
|
|
|
|
if (formItems != null && formItems.size() > 0) {
|
|
for (FileItem item : formItems) {
|
|
if (!item.isFormField()) {
|
|
String fileName = new File(item.getName()).getName();
|
|
String filePath = uploadPath + File.separator + fileName;
|
|
File storeFile = new File(filePath);
|
|
item.write(storeFile);
|
|
request.setAttribute("message", "File " + fileName + " has uploaded successfully!");
|
|
}
|
|
}
|
|
}
|
|
} catch (Exception ex) {
|
|
request.setAttribute("message", "There was an error: " + ex.getMessage());
|
|
}
|
|
getServletContext().getRequestDispatcher("/result.jsp").forward(request, response);
|
|
}
|
|
}
|
|
} |