1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54
|
/**
* Servlet implementation class for Servlet: UploadFileServlet
*/
public class UploadFileServlet extends HttpServlet implements Servlet {
private static final long serialVersionUID = 8305367618713715640L;
protected void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
response.setContentType("text/html");
response.setCharacterEncoding("UTF-8");
FileItem uploadItem = getFileItem(request);
if (uploadItem == null) {
response.getWriter().write("NO-SCRIPT-DATA");
return;
}
int i = 0;
String fileName = new String();
do
{ i++;
fileName = "Logos/logo" + i + ".jpg";
} while ((new File(fileName)).exists());
try {
File newlogo = new File(fileName);
uploadItem.write(newlogo);
} catch (Exception e) {
e.printStackTrace();
response.getWriter().write("NO-SCRIPT-DATA");
return;
}
response.getWriter().write(fileName);
}
private FileItem getFileItem(HttpServletRequest request) {
FileItemFactory factory = new DiskFileItemFactory();
ServletFileUpload upload = new ServletFileUpload(factory);
try {
List items = upload.parseRequest(request);
Iterator it = items.iterator();
while (it.hasNext()) {
FileItem item = (FileItem) it.next();
if (!item.isFormField() && "uploadFormElement".equals(item.getFieldName())) {
return item;
}
}
} catch (FileUploadException e) {
return null;
}
return null;
}
} |
Partager