javaweb--文件上传

JavaWeb文件上传

依赖的jar包

java 复制代码

https://mvnrepository.com/artifact/commons-io/commons-io
https://mvnrepository.com/artifact/commons-fileupload/commons-fileupload

注意事项

为保护服务器的安全，上传的文件不能直接被访问到，比如放在WEB-INF目录下；
限制上传文件的大小；
限制上传文件的格式；
为防止上传文件名字重复导致文件被覆盖，为每个文件设置一个唯一的文件名。

代码解释

ServletFileUpload负责处理上传的文件数据,并将表单中每个输入项封装成一个FileItem对象，在使用ServletFileUpload对象解析请求时需要DiskFileItemFactory对象。所以，我们需要在进行解析工作前构造好DiskFileItemFactory对象，通过ServletFileUpload对象的构造方法或setFileItemFactory()方法设置ServletFileUpload对象的fileItemFactory属性。

缓冲区和临时文件夹的重要性

缓冲区的重要性

防止服务器一次接收到的上传请求过多，导致服务器OOM；设置缓冲区的大小，可以有效解决这一问题。

临时文件夹的重要性

在上传文件时，现将上传的文件上传到临时文件夹中，等待临时文件夹中的文件上传成功后，再移到真正的文件目录中。

好处：

**注意：**不能单纯根据请求头中的内容进行判断文件的大小和格式等信息，请求头的信息可能被篡改！！！

用户中途取消上传 / 网络中断

文件只传了一半，upload 文件夹残留半截损坏文件，占磁盘，还无法区分有效文件和残缺垃圾文件，后期很难清理。
文件超出大小限制（代码限制单文件 10M）

比如用户传 20M 文件，数据已经写了 10M 到 upload，校验时才发现超限，此时正式目录已经产生无效碎片文件。

3.文件后缀黑名单（如禁止 exe、jsp 上传）

文件已经大半写入正式目录，校验发现是禁止类型，文件残片留在 upload，存在安全风险。

前端代码

在HTML页面input 必须有 name

表单如果包含一个文件上传输入项的话，这个表单的enctype属性就必须设置为multipart/form-data

浏览器表单的类型如果为multipart/form-data , 在服务器端想获取数据就要通过流。

jsp 复制代码

<%@ page contentType="text/html;charset=UTF-8" language="java" %>
<html>
<body>

<form action = "${pageContext.request.contextPath}/upFile.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"></p>
</form>
</body>
</html>

后端代码

java 复制代码

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 FileUpServlet extends HttpServlet {
    protected void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
        doGet(request, response);
    }

    protected void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {

        try {

            //判断上传的文件是普通的表单还是带文件的表单
            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 tmpFile = new File(tmpPath);
            if (!tmpFile.exists()) {//如果目录不存在，创建这样一个目录；
                tmpFile.mkdir();
            }

            //处理上传的文件，一般都需要通过流来获取，我们可以使用request.getInputStream(),原生态的文件上传流获取，十分麻烦
            //但是我们都建议使用 Apache的文件上传组件来实现，common-fileupload，它需要依赖于 commons-io组件；
            //1.创建DiskFileItemFactory对象，处理文件上传路径或者大小限制的；
            DiskFileItemFactory factory = getDiskFileItemFactory(tmpFile);
            //2.获取ServletFileUpload
            ServletFileUpload upload = getServletFileUpload(factory);
            //3.处理上传的文件
            String msg = uploadParseRequest(upload, request, uploadPath);

            //servlet请求转发消息
            request.setAttribute("msg",msg);
            request.getRequestDispatcher("upRes.jsp").forward(request,response);

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

    }


    public static DiskFileItemFactory getDiskFileItemFactory(File tmpFile) {
        DiskFileItemFactory factory = new DiskFileItemFactory();
        //通过这个工厂设置一个缓冲区，当上传的文件大于这个缓冲区的时候，将他放到临时文件中；
        factory.setSizeThreshold(1024 * 1024); //缓存区大小为1M
        factory.setRepository(tmpFile);//临时目录的保存目录，需要一个File
        return factory;
    }

    public static ServletFileUpload getServletFileUpload(DiskFileItemFactory factory) {
        ServletFileUpload upload = new ServletFileUpload(factory);
        //监听文件上传进度；
        upload.setProgressListener(new ProgressListener() {
            @Override
            //pBytesRead:已经读取到的文件大小
            //pContentLength ： 文件大小
            public void update(long pBytesRead, long pContentLength, int pItems) {
                System.out.println("总大小：" + pContentLength + "已上传：" + pBytesRead);
            }
        });

        //处理乱码问题
        upload.setHeaderEncoding("UTF-8");
        //设置单个文件的最大值
        upload.setFileSizeMax(1024 * 1024 * 10);
        //设置总共能够上传文件的大小
        //1024 = 1kb * 1024 = 1M * 10 = 10M
        upload.setSizeMax(1024 * 1024 * 10);

        return upload;
    }


    public static String uploadParseRequest(ServletFileUpload upload,HttpServletRequest request,String uploadPath)
            throws FileUploadException, IOException {

        String msg = "";

        //3.把前端请求解析，封装成一个FileItem对象
        try {
            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 == null || uploadFileName.trim().equals("")){
                        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 + File.separator + uuidPath;
                    System.out.println("文件上传的路径："+realPath);
                    //给每个文件创建一个对应的文件夹
                    File realPathFile = new File(realPath);
                    if (!realPathFile.exists()){
                        realPathFile.mkdir();
                    }
                    //=======================存放地址完毕===============================//
                    //获得文件上传的流
                    InputStream inputStream = fileItem.getInputStream();
                    //创建一个文件输出流
                    //realPath = 真实的文件夹；
                    //差了一个文件; 加上输出文件的名字+"/"+uuidFileName
                    FileOutputStream fos = new FileOutputStream(realPath + File.separator + 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(); //上传成功，清除临时文件
                    //=======================文件传输完毕===============================//
                }
            }
        } catch (Exception e) {
            e.printStackTrace();
        }

        return msg;
    }
}