package com.xiaojing.servlet;

import org.apache.commons.fileupload.FileItem;
import org.apache.commons.fileupload.FileUploadException;
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.*;
import java.util.List;
import java.util.UUID;

public class FileUploadServlet extends HttpServlet {
? ? @Override
? ? protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
? ? ? ? //判斷表單是普通表單，還是上傳文件表單
? ? ? ? if(!ServletFileUpload.isMultipartContent(req)){
? ? ? ? ? ? return;
? ? ? ? }
? ? ? ? //創(chuàng)建上傳文件的路徑,一般放在/WEB-INF下，比較安全
? ? ? ? String uploadPath = this.getServletContext().getRealPath("/WEB-INF/upload");
? ? ? ? File uploadFile = new File(uploadPath);
? ? ? ? if(!uploadFile.exists()){
? ? ? ? ? ? uploadFile.mkdir();
? ? ? ? }
? ? ? ? //如果上傳的文件大小超過了限制，那么會將這個文件放在臨時文件夾，過段時間提醒用戶刪除或者轉(zhuǎn)為永久
? ? ? ? String tempPath = this.getServletContext().getRealPath("/WEB-INF/temp");
? ? ? ? File tempFile = new File(tempPath);
? ? ? ? if(!tempFile.exists()){
? ? ? ? ? ? tempFile.mkdir();
? ? ? ? }

? ? ? ? //1、創(chuàng)建DiskFileItemFactory對象
? ? ? ? DiskFileItemFactory factory = new DiskFileItemFactory();
? ? ? ? //2、獲取ServletFileUpload對象
? ? ? ? ServletFileUpload upload = new ServletFileUpload();
? ? ? ? upload.setFileItemFactory(factory);
? ? ? ? //3、處理上傳文件
? ? ? ? ? ? //將前端請求封裝成Fileitem對象,每一個表單輸入項都是一個Fileitem對象
? ? ? ? try {
? ? ? ? ? ? List<FileItem> items = upload.parseRequest(req);
? ? ? ? ? ? for (FileItem item: items) {
? ? ? ? ? ? ? ? if(item.isFormField()){ //如果是普通的表單
? ? ? ? ? ? ? ? ? ? //getFieldName()是input的name屬性，getString()是input的value屬性
? ? ? ? ? ? ? ? ? ? String fieldName = item.getFieldName();
? ? ? ? ? ? ? ? ? ? String value = item.getString("UTF-8");
? ? ? ? ? ? ? ? ? ? System.out.println(fieldName + ":" +value);
? ? ? ? ? ? ? ? }else{ ?//上傳文件的表單
? ? ? ? ? ? ? ? ? ? //---------------處理文件------------------//
? ? ? ? ? ? ? ? ? ? String uploadFileName = item.getName();
? ? ? ? ? ? ? ? ? ? System.out.println("上傳的文件名："+uploadFileName);
? ? ? ? ? ? ? ? ? ? if(uploadFileName == null || uploadFileName.trim().equals("")){
? ? ? ? ? ? ? ? ? ? ? ? continue;
? ? ? ? ? ? ? ? ? ? }
? ? ? ? ? ? ? ? ? ? //獲取fileName文件名，uploadFileName一般是url/文件路徑/文件
? ? ? ? ? ? ? ? ? ? String fileName = uploadFileName.substring(uploadFileName.lastIndexOf("\\")+1);
? ? ? ? ? ? ? ? ? ? //獲取文件后綴
? ? ? ? ? ? ? ? ? ? String fileExt = uploadFileName.substring(uploadFileName.lastIndexOf(".")+1);
? ? ? ? ? ? ? ? ? ? System.out.println("文件信息:"+fileName);

? ? ? ? ? ? ? ? ? ? //---------------處理地址-----------------//
? ? ? ? ? ? ? ? ? ? //使用UUID唯一標識存放文件夾
? ? ? ? ? ? ? ? ? ? String uuidPath = UUID.randomUUID().toString();
? ? ? ? ? ? ? ? ? ? //獲取真實的文件夾地址
? ? ? ? ? ? ? ? ? ? String realPath = uploadPath + "/" + uuidPath;
? ? ? ? ? ? ? ? ? ? File realFile = new File(realPath);
? ? ? ? ? ? ? ? ? ? if(!realFile.exists()){
? ? ? ? ? ? ? ? ? ? ? ? realFile.mkdir();
? ? ? ? ? ? ? ? ? ? }

? ? ? ? ? ? ? ? ? ? //---------------處理文件流-----------------//
? ? ? ? ? ? ? ? ? ? //獲取文件上傳的流
? ? ? ? ? ? ? ? ? ? InputStream inputStream = item.getInputStream();
? ? ? ? ? ? ? ? ? ? //創(chuàng)建一個文件輸出流
? ? ? ? ? ? ? ? ? ? FileOutputStream outputStream = new FileOutputStream(realFile + "/" + fileName);
? ? ? ? ? ? ? ? ? ? //創(chuàng)建一個緩沖區(qū)
? ? ? ? ? ? ? ? ? ? byte[] buffer = new byte[1024 * 1024];
? ? ? ? ? ? ? ? ? ? int len = 0;
? ? ? ? ? ? ? ? ? ? while((len = inputStream.read(buffer)) > 0){ //每次讀寫1M
? ? ? ? ? ? ? ? ? ? ? ? outputStream.write(buffer,0,len);
? ? ? ? ? ? ? ? ? ? }
? ? ? ? ? ? ? ? ? ? outputStream.close();
? ? ? ? ? ? ? ? ? ? inputStream.close();

? ? ? ? ? ? ? ? ? ? String msg = "文件上傳成功";
? ? ? ? ? ? ? ? ? ? System.out.println("文件上傳成功");
? ? ? ? ? ? ? ? ? ? //清除臨時文件
? ? ? ? ? ? ? ? ? ? item.delete();
? ? ? ? ? ? ? ? ? ? //轉(zhuǎn)發(fā)
? ? ? ? ? ? ? ? ? ? req.setAttribute("msg",msg);
? ? ? ? ? ? ? ? ? ? req.getRequestDispatcher("index.jsp").forward(req,resp);

? ? ? ? ? ? ? ? }
? ? ? ? ? ? }
? ? ? ? } catch (FileUploadException e) {
? ? ? ? ? ? e.printStackTrace();
? ? ? ? }

? ? }

? ? @Override
? ? protected void doPost(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
? ? ? ? doGet(req, resp);
? ? }
}