文件上传

  1 package com.oracle.upload;
  2 
  3 import com.google.gson.Gson;
  4 import com.google.gson.JsonObject;
  5 import com.oracle.domain.ResponseUpload;
  6 import com.oracle.domain.UploadBean;
  7 import com.oracle.domain.UserInfo;
  8 import com.oracle.service.UploadService;
  9 import org.apache.commons.fileupload.FileItem;
 10 import org.apache.commons.fileupload.FileUploadException;
 11 import org.apache.commons.fileupload.disk.DiskFileItemFactory;
 12 import org.apache.commons.fileupload.servlet.ServletFileUpload;
 13 import org.apache.commons.io.FileUtils;
 14 
 15 import javax.servlet.ServletException;
 16 import javax.servlet.annotation.WebServlet;
 17 import javax.servlet.http.HttpServlet;
 18 import javax.servlet.http.HttpServletRequest;
 19 import javax.servlet.http.HttpServletResponse;
 20 import java.io.*;
 21 import java.util.HashMap;
 22 import java.util.List;
 23 import java.util.Map;
 24 import java.util.UUID;
 25 
 26 /**
 27  * @Description TODO
 28  * @ClassName UploadServlet
 29  * @Author YHY
 30  * @Date 2019/11/11 11:31
 31  * @Version V1.0
 32  */
 33 @WebServlet(value = "/UploadServlet")
 34 public class UploadServlet extends HttpServlet {
 35     @Override
 36     protected void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
 37         /* 处理响应乱码问题:字节流需getBytes("UTF-8") */
 38         /*response.setContentType("text/html;charset=UTF-8");*/
 39         /* 处理post请求乱码问题 */
 40         request.setCharacterEncoding("UTF-8");
 41 
 42         String fileName = ""; //文件名*/
 43         String href = "";/*服务器存储路径*/
 44         String loaclhost_file_name = "";
 45         UserInfo userInfo = (UserInfo) request.getSession().getAttribute("userinfo");
 46         String upload_those = userInfo.getUid();
 47         String access_url = "";/*用户访问的相对路径*/
 48 
 49         /*用于返回数据*/
 50         Map map = new HashMap();
 51 
 52         boolean flag = ServletFileUpload.isMultipartContent(request);
 53         if (!flag) {
 54             request.setAttribute("msg", "表单的enctype属性不正确");
 55             request.getRequestDispatcher("jsp/upload.jsp").forward(request, response);
 56             return;
 57         }
 58 
 59         /*缓冲字节输出流*/
 60         BufferedOutputStream bos = null;
 61         /*缓冲字节输入流*/
 62         BufferedInputStream bis = null;
 63 
 64         /*1、创建磁盘文件工程*/
 65         DiskFileItemFactory diskFileItemFactory = new DiskFileItemFactory();
 66         /*设置文件上传过程中缓冲区的大小*/
 67         diskFileItemFactory.setSizeThreshold(3);
 68         /*设置文件上传过程中超出缓冲区所产生的临时文件路径*/
 69         diskFileItemFactory.setRepository(new File("D://temporary"));
 70         /*2、创建一个核心解析类*/
 71         ServletFileUpload servletFileUpload = new ServletFileUpload(diskFileItemFactory);
 72         /*解决中文乱码问题*/
 73         servletFileUpload.setHeaderEncoding("UTF-8");
 74 
 75         /*3、利用核心类解析request，解析后得到多个部分。返回list集合。list集合中装的值每个部分内容（FileItem文件项）。*/
 76         try {
 77             List fileItems = servletFileUpload.parseRequest(request);
 78 
 79             /*4、遍历list集合，会得到代表每个部分的文件上传项的对象。根据文件项判断是否是文件上传项。*/
 80             for (FileItem f : fileItems) {
 81                 /*判断这个文件项是普通项还是文件上传项*/
 82                 if (f.isFormField()) {
 83                     /*普通项：*/
 84                     /*接收普通项的值，不能用request.getParameter()*/
 85                     /*获取普通项的key*/
 86                     String name = f.getFieldName();
 87                     /*接收普通项的值*/
 88                     String string = f.getString();
 89                     System.out.println(name + ":" + string);
 90                     if (name.equals("href")) {
 91                         href = string;
 92                     }
 93                     if (name.equals("access_url")) {
 94                         access_url = string;
 95                     }
 96                 }
 97             }
 98             /*文件上传项*/
 99             for (FileItem file : fileItems) {
100                 if (!file.isFormField()) {
101                     /*文件上传项*/
102                     /*获取文件名*/
103                     loaclhost_file_name = file.getName();
104                     fileName = loaclhost_file_name;
105                     int index = 0;
106                     if ((index = fileName.lastIndexOf("\\")) != -1) {
107                         loaclhost_file_name = loaclhost_file_name.substring(index + 1);
108                     }
109                     loaclhost_file_name = UploadServlet.randomFileNanm(loaclhost_file_name);
110                     /*往request域中存储文件名*/
111                     request.setAttribute("randomFileName", loaclhost_file_name.getBytes());
112                     /*获取数据源*/
113                     bis = new BufferedInputStream(file.getInputStream());
114 
115                     /*服务器存储路径*/
116                     href = href + loaclhost_file_name;
117                     /*用户访问的相对路径*/
118                     access_url = access_url + loaclhost_file_name;
119 
120                     /*明确数据源*/
121                     bos = new BufferedOutputStream(new FileOutputStream(href));
122                     byte[] bytes = new byte[1024];
123                     int line = 0;
124                     while ((line = bis.read(bytes)) != -1) {
125                         bos.write(bytes, 0, line);
126                     }
127                     if (bos != null && bis != null) {
128                         /*释放资源*/
129                         bos.close();
130                         bis.close();
131                     }
132                     /*5、删除临时文件*/
133                     file.delete();
134 
135                     FileUtils.copyFile(new File("D:\\Java\\apache-tomcat-9.0.27\\webapps\\YHY\\images\\xishi\\" + loaclhost_file_name),
136                             new File("D:\\YHY\\web\\images\\xishi\\" + loaclhost_file_name));
137 
138 
139                     /*调用sevice方法*/
140                     /*封装参数*/
141                     UploadBean uploadBean = new UploadBean();
142                     uploadBean.setName(fileName);
143                     uploadBean.setLoaclhost_file_name(loaclhost_file_name);
144                     uploadBean.setUpload_those(upload_those);
145                     uploadBean.setAccess_url(access_url);
146                     uploadBean.setHref(href);
147                     new UploadService().insertUpload(uploadBean);
148 
149                     map.put("code", "0");
150                     map.put("msg", "成功");
151                     map.put("url", "url");
152                 }
153             }
154         } catch (FileUploadException e) {
155             map.put("code", "500");
156             map.put("msg", "失败");
157             map.put("url", "url");
158             e.printStackTrace();
159         } finally {
160             if (bos != null && bis != null) {
161                 /*释放资源*/
162                 bos.close();
163                 bis.close();
164             }
165         }
166         response.getWriter().write(new Gson().toJson(map));
167     }
168 
169     @Override
170     protected void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
171         doGet(request, response);
172     }
173 
174     private static String randomFileNanm(String name) {
175         return UUID.randomUUID().toString().replace("-", "") + name;
176     }
177 
178 }