015-Java Web文件上传

public class FileServlet extends javax.servlet.http.HttpServlet {
protected void doPost(javax.servlet.http.HttpServletRequest request, javax.servlet.http.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组件
try {
//1.创建DiskFileItemFactory对象,处理文件上传路径或者大小限制
DiskFileItemFactory factory = getDiskFileItemFactory(file);
//2.获取ServletFileUpload
ServletFileUpload upload = getServletFileUpload(factory);
//3.处理上传的文件
String msg = uploadParseRequest(upload, request, uploadPath);
//servlet请求转发消息
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
        //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);
    //设置总共能够上传文件的大小
    upload.setSizeMax(1024 * 1024 * 10);
    return upload;
}

public static String uploadParseRequest(ServletFileUpload upload, HttpServletRequest request, String uploadPath) throws FileUploadException,IOException {
    String msg = "";
    //3.处理上传的文件
    //把前端请求解析,封装成一个FileItem对象,需要从ServletFileUpload对象中获取

    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(); //获得文件名
            //可能存在文件名不合法的情况
            if (uploadFileName.trim().equals("") || uploadFileName == null) {
                continue;
            }
            //获得上传文件的文件名和后缀名
            String fileName = uploadFileName.substring(uploadFileName.lastIndexOf("/") + 1);
            String fileExtName = uploadFileName.substring(uploadFileName.lastIndexOf(".") + 1);
            //如果后缀名不是我们需要的,就直接return,不处理,告诉用户文件类型不对
            //可以使用UUID(唯一识别的通用码),保证文件名唯一
            //UUID.randomUUID(),随机生成一个唯一识别的通用码
            //网络传输中的东西,都需要序列化
            //POJO,实体类,如果想要在多个电脑上运行,传输===>需要把对象都序列化了
            //JNI = Java Native Interface
            //implements Serializable
            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 + "." + fileExtName);
            //创建一个缓冲区
            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;
}

}

猜你喜欢

转载自blog.51cto.com/12859164/2590407