java文件上傳

文件上傳

ServeltFileUpload負責處理上傳的文件數據，並將表單中每個輸入項封裝成一個FileItem對象，在是用ServletFileUpload對象解析請求時需要DiskFileItemFactory對象。所以，我們需要在進行解析工作前構造好DiskFileItemFactory對象，通過ServletFileUpload對象的構造方法或setFileItemFactory()方法設置ServletFileUpload對象的fileItemFactory屬性。

表單如果包含一個文件傳輸項的話，這個表單的enctype屬性必須設置為multipart/form-data

注意事項:

1. 為保證服務器安全，上傳文件應該放在外界無法直接訪問的目錄下，比如放於WEB-INFO目錄下
2. 為防止文件覆蓋的現象產生，要為上傳文件產生一個唯一的文件名
3. 限制上傳文件的最大值
4. 可以限制上傳文件的類型，在收到上傳文件名時，判斷后綴名是否合法

FileServlet

package com.yu.servlet;


import org.apache.commons.fileupload.FileItem;
import org.apache.commons.fileupload.FileUploadException;
import org.apache.commons.fileupload.ProgressListener;
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.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.List;
import java.util.UUID;

public class FileServlet extends HttpServlet {
    protected void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
        //判斷上傳的文件是普通表單還是文件表單
        if (!ServletFileUpload.isMultipartContent(request)) {
            return;//終止方法運行，說明這是一個普通的表單
        }

        //創建上傳文件的保存路徑，建議保存在WEB-INF下，用戶無法直接訪問上傳的文件
        String uploadPath = this.getServletContext().getRealPath("/WEB-INF/upload");
        File uploadfile = new File(uploadPath);
        if (!uploadfile.exists()) {
            uploadfile.mkdir();//創建這個目錄
        }
        //緩存，臨時文件
        //臨時路徑，假如文件超過預期大小，放到臨時文件中，過幾天自動刪除，或者提醒用戶轉存為永久
        String tmpPath = this.getServletContext().getRealPath("/WEB-INF/tmp");
        File file = new File(tmpPath);
        if (!file.exists()) {
            file.mkdir();//創建這個臨時目錄
        }
        //處理上傳的文件一般需要流來獲取，我們可以使用request.getInputStream()，原生態的文件上傳流獲取，十分麻煩
        //建議使用Apache的文件上傳組件來實現，common-fileupload,他需要依賴於commons-io
        /*
        ServletFileUpload負責處理上傳的文件數據，並將表單中每個輸入項封裝成一個FileItem對象
        在使用ServletFileUpload對象解析請求時需要DiskFileItemFactory對象。所以我們需要在解析工作前構造好DiskFileItemFactory對象
        通過ServletFileUpload對象的構造方法或setFileItemFactory()方法設置ServletFileUpload對象的fileItemFactory屬性
        */
        try {
            //1.創建DiskFileItemFactory對象，處理文件上傳路徑或者大小限制
            DiskFileItemFactory factory = getDiskFileItemFactory(file);
            //2.獲取ServletFileUpload
            ServletFileUpload upload = getServletFileUpload(factory);
            //3.處理上傳的文件
            String msg = uploadParseRequest(upload, request, uploadPath);
            request.setAttribute("msg", msg);
            request.getRequestDispatcher("info.jsp").forward(request, response);
        } catch (FileUploadException e) {
            e.printStackTrace();
        }


    }

    public static DiskFileItemFactory getDiskFileItemFactory(File file) {
        //1.創建DiskFileItemFactory對象，處理文件上傳路徑或者大小限制的
        DiskFileItemFactory factory = new DiskFileItemFactory();
        //通過這個工廠設置一個緩沖區，當上傳的文件大於這個緩沖區的時候，將他放到臨時文件中；
        factory.setSizeThreshold(1024 * 1024);//緩沖區大小為1m
        factory.setRepository(file);//臨時目錄的保存目錄，需要一個file
        return factory;
    }

    public static ServletFileUpload getServletFileUpload(DiskFileItemFactory factory) {
        //2.獲取ServletFileUpload
        ServletFileUpload upload = new ServletFileUpload(factory);
        //監聽文件上傳進度
        upload.setProgressListener(new ProgressListener() {
            @Override
            public void update(long pBytesRead, long pContentLength, int pItems) {
                System.out.println("總大小：" + pContentLength + "已上傳：" + pBytesRead);
            }
        });
        //處理亂碼問題
        upload.setHeaderEncoding("utf-8");
        //設置單個文件的最大值
        upload.setFileSizeMax(1024 * 1024 * 10);
        //設置總共能夠上傳文件的大小
        upload.setSizeMax(1024 * 1024 * 10);
        return upload;
    }

    public static String uploadParseRequest(ServletFileUpload upload, HttpServletRequest request, String uploadPath)
            throws FileUploadException, IOException {
        String msg = "";
        //3.把前端請求解析，封裝成一個FileItem對象
        List<FileItem> fileItems = upload.parseRequest(request);
        for (FileItem fileItem : fileItems) {
            if (fileItem.isFormField()) { //判斷上傳的文件是普通的表單還是帶文件的表單
                //getFieldName指的是前端表單控件的name;
                String name = fileItem.getFieldName();
                String value = fileItem.getString("UTF-8"); //處理亂碼
                System.out.println(name + ":" + value);
            } else { //判斷它是上傳的文件
                //處理文件
                //拿到文件名字
                String uploadFileName = fileItem.getName();
                System.out.println("上傳的文件名: " + uploadFileName);
                if (uploadFileName.trim().equals("") || uploadFileName == null) {
                    continue;
                }
                //獲得上傳的文件名/images/girl/paojie. png
                String fileName = uploadFileName.substring(uploadFileName.lastIndexOf("/") + 1);
                //獲得文件的后綴名
                String fileExtName = uploadFileName.substring(uploadFileName.lastIndexOf(".") + 1);
                /*
                如果文件后綴名fileExtName 不是我們所需要的
                就直接return,不處理，告訴用戶文件類型不對。
                */
                System.out.println("文件信息[件名: " + fileName + "---文件類型" + fileExtName + "]");
                //可以使用UUID (唯一識別的通用碼)，保證文件名唯一;
                //UUID. randomUUID()，隨機生一個唯一識別的通 用碼;
                String uuidPath = UUID.randomUUID().toString();
                //存到uploadPath
                //文件真實存在的路徑realPath
                String realPath = uploadPath + "/" + uuidPath;
                //給每個文件創建一個對應的文件夾
                File realPathFile = new File(realPath);
                if (!realPathFile.exists()) {
                    realPathFile.mkdir();
                }
                //獲得文件上傳的流
                InputStream inputStream = fileItem.getInputStream();
                //創建一個文件輸出流
                //realPath =真實的文件夾;
                //差了一個文件;加上輸出文件的名字+"/"+uuidFileName
                FileOutputStream fos = new FileOutputStream(realPath + "/" + fileName);
                //創建一個緩沖區
                byte[] buffer = new byte[1024 * 1024];
                //判斷是否讀取完畢
                int len = 0;
                //如果大於0說明還存在數據;
                while ((len = inputStream.read(buffer)) > 0) {
                    fos.write(buffer, 0, len);
                }
                //關閉流
                fos.close();
                inputStream.close();
                msg = "文件上傳成功!";
                fileItem.delete(); //上傳成功，清除臨時文件
            }
        }
        return msg;
    }
}

index.jsp

<%@ page contentType="text/html;charset=UTF-8" language="java" %>
<html>
  <head>
    <title>$Title$</title>
  </head>
  <body>
  <%--
    通過表單上傳文件
    get:上傳文件大小有限制
    post:上傳文件大小沒有限制
   --%>
  <form action="${pageContext.request.contextPath}/upload.do" method="post" enctype="multipart/form-data">
    上傳用戶：<input type="text" name="username"><br>
    <p><input type="file" name="file1"></p>
    <p><input type="file" name="file2"></p>
    <p><input type="submit"> <input type="reset"></p>
  </form>
  </body>
</html>

info.jsp

<%@ page contentType="text/html;charset=UTF-8" language="java" %>
<html>
<head>
    <title>Title</title>
</head>
<body>
${msg}
</body>
</html>

web.xml

<?xml version="1.0" encoding="UTF-8"?>
<web-app xmlns="http://xmlns.jcp.org/xml/ns/javaee"
         xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
         xsi:schemaLocation="http://xmlns.jcp.org/xml/ns/javaee http://xmlns.jcp.org/xml/ns/javaee/web-app_4_0.xsd"
         version="4.0">
    <servlet>
        <servlet-name>FileServlet</servlet-name>
        <servlet-class>com.yu.servlet.FileServlet</servlet-class>
    </servlet>
    <servlet-mapping>
        <servlet-name>FileServlet</servlet-name>
        <url-pattern>/upload.do</url-pattern>
    </servlet-mapping>
</web-app>