1 package com.rzk.servlet;
2
3
4 import org.apache.commons.fileupload.FileItem;
5 import org.apache.commons.fileupload.FileUploadException;
6 import org.apache.commons.fileupload.ProgressListener;
7 import org.apache.commons.fileupload.disk.DiskFileItemFactory;
8 import org.apache.commons.fileupload.servlet.ServletFileUpload;
9
10 import javax.servlet.ServletException;
11 import javax.servlet.http.HttpServlet;
12 import javax.servlet.http.HttpServletRequest;
13 import javax.servlet.http.HttpServletResponse;
14 import java.io.File;
15 import java.io.FileOutputStream;
16 import java.io.IOException;
17 import java.io.InputStream;
18 import java.util.List;
19 import java.util.UUID;
20
21
22 public class FileServlet extends HttpServlet {
23 protected void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
24 System.out.println("进来了");
25 //判断上传的文件是普通表单还是带文件的表单
26 if (!ServletFileUpload.isMultipartContent(request)) {
27 return;//终止方法运行，说明这是一个普通表单，直接返回
28 }
29
30 //创建上传文件的保存路径，建议在WEB-INF路径下，安全，用户无法直接访问上传的文件
31 String uploadPath = this.getServletContext().getRealPath("/WEB-INF/upload");
32 File uploadFile = new File(uploadPath);
33 if (!uploadFile.exists()) {
34 uploadFile.mkdir();//创建这个目录
35 }
36
37 //缓存:临时文件
38 //临时路径：加入文件超过了预期的大小，我们就把他放到一个临时文件，过几天，或者提醒用户转存为永久
39 String tmpPath = this.getServletContext().getRealPath("/WEB-INF/tmp");
40 File file = new File(tmpPath);
41 if (!file.exists()) {//如果临时文件不存在
42 file.mkdir();//创建这个目录
43 }
44
45 // 处理上传的文件，一般都需要通过流来获取，我们可以使用request.getInputStream(),原生态的文件上传流获取，十分麻烦
46 //但是我们都建议使用 Apache的文件上传组件来实现 common-fileupload ,他需要依赖于commons-io 组件
47
48
49 /*
50 * ServletFileUpload 负责处理上传的文件数据，并将表单中每个输入项封装成一个Fileitem对象
51 * 在使用servletFileUpload 对象解析请求时需要 DiskFileItemFactory对象
52 * 所以，我们需要在进行解析工作前构造好 DiskFileItemFactory 对象
53 * 通过ServletFileUpload对象的构造方法 或 setFileItemFoctory()方法
54 * 设置ServletFileUpload对象的fileItemFactory属性
55 *
56 * */
57
58 try {
59 //1.创建DiskFileItemFactory对象 处理文件上传路径或者大小限制
60 DiskFileItemFactory factory = getDiskFileItemFactory(file);
61 //2.获取ServletFileUpload
62 ServletFileUpload upload = getServletFileUpload(factory);
63 //3.处理上传的文件
64 String msg = uploadParseRequest(upload,request,uploadPath);
65
66 //servlet请求转发消息
67 request.setAttribute("msg",msg);
68 request.getRequestDispatcher("info.jsp").forward(request,response);
69 } catch (FileUploadException ex) {
70 ex.printStackTrace();
71 }
72
73
74 }
75
76 public static DiskFileItemFactory getDiskFileItemFactory(File file) {
77 DiskFileItemFactory factory = new DiskFileItemFactory();
78 //通过这个工厂设置一个缓冲区，当上传的文件大于这个缓冲区的时候，将他放进临时文件
79 factory.setSizeThreshold(1024 * 1024);//设置缓冲区大小为1M
80 factory.setRepository(file);//临时目录的保存目录，需要一个File
81 return factory;
82 }
83 public static ServletFileUpload getServletFileUpload(DiskFileItemFactory factory){
84 //2.获取ServletFileUpload
85 ServletFileUpload upload = new ServletFileUpload(factory);
86 //监听文件上传进度
87 upload.setProgressListener(new ProgressListener() {
88 //pBytesRead:已经读取到的文件大小
89 //pContentLength : 文件大小
90 public void update(long pBytesRead, long pContentLength, int pItems) {
91 System.out.println("总大小" + pContentLength + "已上传" + pBytesRead);
92 }
93 });
94 //处理乱码问题
95 upload.setHeaderEncoding("UTF-8");
96 //设置单个文件的最大值
97 upload.setFileSizeMax(1024 * 1024 * 10);
98 //设置总共能够上传文件的大小
99 //1024 = 1kb * 1024 = 1M * 10 = 10M
100 upload.setSizeMax(1024 * 1024 * 10);
101 return upload;
102 }
103
104 public static String uploadParseRequest(ServletFileUpload upload, HttpServletRequest request, String uploadPath) throws FileUploadException, IOException {
105 //3.处理上传文件
106 System.out.println("进来");
107 String msg = "";
108 //把前端请求解析，封装成FileItem对象，需要从ServletFileUpload对象中获取
109 List<FileItem> fileItems = upload.parseRequest(request);
110 //fileItem 每一个表单对象
111 for (FileItem fileItem : fileItems) {//fileItem
112 //判断上传的文件是普通表单还是带文件的表单
113 if (fileItem.isFormField()) {
114 //getFieldName指的是前端表单控件的name
115 String name = fileItem.getFieldName();
116 String value = fileItem.getString("utf-8");
117 System.out.println(name + " : " + value);
118 } else {//如果是文件
119 // ---------------------------处理文件---------------------
120 String uploadFileName = fileItem.getName();
121 //可能存在文件名不合法
122 if (uploadFileName.trim().equals("") || uploadFileName == null) {
123 //如果名字为空
124 continue;
125 }
126 //获得上传的文件名 /images/popj.png
127 String fileName = uploadFileName.substring(uploadFileName.lastIndexOf("/") + 1);
128 //获得文件后缀名
129 String fileExtName = uploadFileName.substring(uploadFileName.lastIndexOf(".") + 1);
130 /*
131 如果文件后缀名 fileExtNAME 不是我们锁需要的
132 就直接return 不处理告诉用户名类型不对
133 */
134 //可以使用UUID（唯一识别的通用码 ）保证文件名唯一
135 //UUID.randomUUID() 随机生一个唯一识别的通用码
136
137 //网络传输中的东西 都需要序列化
138 // pojo 实体类 如果想要在多个电脑上运行 传输--》需要把对象都系列化了
139 //implements Serializable ： 标记接口
140
141
142 String uuidPath = UUID.randomUUID().toString();
143 // ---------------------------存放地址---------------------
144 //存到哪? 路径uploadPath
145 //文件真实存在的路径realPath
146 String realPath = uploadPath + "/" + uuidPath;
147 //给每个文件创建一个对应的文件夹
148 File realPathFile = new File(realPath);
149 if (!realPathFile.exists()) {
150 realPathFile.mkdir();
151 }
152
153 // ---------------------------文件传输---------------------
154 //获得文件上传的流
155 InputStream inputStream = fileItem.getInputStream();
156
157 //创建一个文件输出流
158 //realPath = 真实的文件夹
159 // 差一个文件 ； 加上输出文件的名字+ "/" + uuidFileName
160 FileOutputStream fos = new FileOutputStream(realPath + "/" + fileName);
161
162 //创建一个缓冲区
163 byte[] buffer = new byte[1024 * 1024];
164
165 //判断是否读取完毕
166 int len = 0;
167 //如果大于0说明还存在数据
168 while ((len = inputStream.read(buffer)) > 0) {
169 fos.write(buffer, 0, len);
170 }
171
172 //关闭流
173 fos.close();
174 inputStream.close();
175
176 msg = "文件上传成功";
177 fileItem.delete();//上传成功，清除临时文件
178 }
179 }
180 return msg;
181
182 }
183
184 }
185
