JavaWeb中的文件上传的方法和注意事项

桃扇骨 2022-12-15 14:09 132阅读 0赞

文件上传的调优考虑：

1.  为保证服务器安全，上传文件应该放在外界无法直接访问的目录下，比如放在WEB-INF目录下
2.  为方式文件覆盖的现象发生，要为上传文件产生一个唯一的文件名
3.  要限制上传文件的最大值
4.  可以限制上传文件的类型

package com.gong.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 FileUpdateServlet extends HttpServlet {
        @Override
        protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
            doPost(req,resp);
        }
    
        @Override
        protected void doPost(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
            System.out.println("进入dopost");
            //判断上传的文件是普通的表单还是带文件的表单
            if (!ServletFileUpload.isMultipartContent(req)){
                    return;//终止方法运行，说明这是一个普通的表单，直接返回
                }
            //创建上传文件的保存路径，建议在WEB-INF路径下，安全，用户无法直接访问上传的文件
            String uploadPath = this.getServletContext().getRealPath("WEB-INF/file");
            File uploadFile = new File(uploadPath);
            if (!uploadFile.exists()){
                uploadFile.mkdir();//创建这个目录
            }
            //缓存，临时文件
            //临时路径，假如文件超过了预期的大小，我们就把它放到一个临时文件中，过几天自动删除，或者提醒用户转存为永久
            String temPath = this.getServletContext().getRealPath("WEB-INF/file");
            File temFile = new File(temPath);
            if (!uploadFile.exists()){
                uploadFile.mkdir();//创建这个临时目录
            }
            /*处理上传的文件，一般都需要通过流来获取，我们可以使用request.getInputStream,原生态的文件上传流获取，十分麻烦
            所以我们都建议使用Apache的文件上传组件来实现，common-fileupload，它需要依赖于commons-io组件*/
    
            /*ServletFileUpload负责处理上传的文件数据，并将表单中的每个输入项封装成一个FileItem对象，
            在使用SerbletFileUpload对象解析请求时需要DiskFileItemFactory对象，
            所以我们需要在进行解析工作前构造好DiskFileItemFactory对象，通过ServletFileUpload对象的构造方法
            或setFileItemFactory()方法设置ServletFileUpload对象的fileItemFactory属性*/
    
            //1，创建DiskFileItemFactory对象，处理文件上传路径或者大小限制的；
            DiskFileItemFactory factory = new DiskFileItemFactory();
            //通过这个工厂设置一个缓冲区，当上传的文件大于这个缓冲区的时候，将它放到临时文件中
            factory.setSizeThreshold(1024*1024);//缓存区大小为1M
            factory.setRepository(temFile);//临时目录的保存目录，需要一个File
    
            //1，创建DiskFileItemFactory对象，处理文件上传路径或者大小限制的；
            //2.获取ServletUpload
            ServletFileUpload upload = new ServletFileUpload(factory);
            //监听文件上传进度
            upload.setProgressListener(new ProgressListener() {
                //l已经读到的文件大小
                //l1文件大小
                //i上传时间
                @Override
                public void update(long l, long l1, int i) {
                    System.out.println("总大小："+l1+"已上传："+l);
                }
            });
            //处理乱码问题
            upload.setHeaderEncoding("UTF-8");
            //设置单个文件的最大值
            upload.setFileSizeMax(1024*1024*10);
            //设置总共能够上传文件的大小
            //1024 = 1kb * 1024 = 1M * 10 = 10M
            upload.setSizeMax(1024*1024*10);
            //3.处理上传文件
            //把前端请求解析，封装成一个FileItem对象，需要从ServletFileUpload对象中获取
            try {
                List<FileItem> fileItems = upload.parseRequest(req);
                for (FileItem fileItem : fileItems) {
                    //判断是普通的表单域还是文件
                    if (fileItem.isFormField()){
                        //getFileName指的是前端表单控件的name
                        String name = fileItem.getFieldName();
                        String value = fileItem.getString("UTF-8");//处理乱码
                        System.out.println(name+"-"+value);
                    }else{//如果是文件的情况下
                        //处理文件
                        String fieldName = fileItem.getName();
                        //可能存在文件不合法的情况
                        if (fieldName.trim().equals("") || fieldName == null){
                            continue;
                        }
                        //获取上传的文件名，/images/girl/01.png
                        String name = fieldName.substring(fieldName.lastIndexOf("/") + 1);
                        //获取上传文件的后缀名
                        String lastName = fieldName.substring(fieldName.lastIndexOf(".")+1);
                        /*
                          这里可以判断文件名的后缀是否合法，如果不合法直接return,通知用户文件类型不对
                        }*/
                        /*
                        *可以使用UUID（唯一识别的通用码），保证文件名唯一
                        * UUID.randomUUID();随机生成一个位置识别的通用吗；
                        * 网络传输中的东西，都需要序列化
                        * POJO，实体类，如果想要在多个电脑上运行，传输→需要把对象都序列化了
                        * JNI = Java Native Interface
                        * implements Serializable ：标记接口，JVM→Java栈 本地方法栈 native→C++
                        */
                        String uid = UUID.randomUUID().toString();
                        //存放地址
                        //存到哪？uploadPath
                        //文件真实存在的路径 realPath
                        String realPath = uploadPath+"/"+uid;
                        //给每个文件创建一个对应的文件夹
                        File realPathFile = new File(realPath);
                        if (!realPathFile.exists()){
                            realPathFile.mkdir();
                        }
                        //文件传输
                        //获得文件上传的流
                        InputStream inputStream = fileItem.getInputStream();
                        //创建一个文件输出流
                        //realPath = 真实的文件夹
                        //差了一个文件；加上输出文件的名字+"/"+uid
                        FileOutputStream fos = new FileOutputStream(realPath+"/"+name);
                        //创建一个缓冲区
                        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();
                    }
                }
            } catch (FileUploadException e) {
                e.printStackTrace();
            }
    
    
            System.out.println("finish");
        }
    }