FileUpload.java
2.73 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
package com.xiniunet.web.backend.module.screen;
import com.alibaba.citrus.service.requestcontext.buffered.BufferedRequestContext;
import com.alibaba.citrus.turbine.dataresolver.Param;
import com.alibaba.fastjson.JSON;
import com.xiniunet.foundation.contract.UploadTypeEnum;
import com.xiniunet.foundation.request.FileUploadRequest;
import com.xiniunet.foundation.service.FoundationService;
import com.xiniunet.framework.base.BaseResponse;
import com.xiniunet.framework.exception.ErrorType;
import com.xiniunet.framework.log.LogUtil;
import com.xiniunet.framework.security.Passport;
import com.xiniunet.framework.util.auth.LocalData;
import org.apache.commons.fileupload.FileItem;
import org.springframework.beans.factory.annotation.Autowired;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.PrintWriter;
/**
* 文件上传
*/
public class FileUpload {
@Autowired
private HttpServletResponse response;
@Autowired
private BufferedRequestContext brc;
@Autowired
private HttpServletRequest request;
@Autowired
private FoundationService foundationService;
public void execute(@Param("File") FileItem fileItem, @Param("Type") UploadTypeEnum type) throws Exception {
BaseResponse baseResponse = new BaseResponse();
try {
if (fileItem == null) {
return;
}
// 必须关闭buffering,未完成的页面才会被显示在浏览器上。
brc.setBuffering(false);
// 设置content type,但不需要设置charset,框架会设置正确的charset。
response.setContentType("text/plain");
Passport passport = LocalData.getCurrentPassport();
// Passport passport = (Passport) request.getAttribute("passport");
if (passport == null) {
passport = new Passport();
passport.setUserId(100001L);
} else {
if (passport.getUserId() == null) {
passport.setUserId(100001L);
}
}
FileUploadRequest req = new FileUploadRequest();
req.setFileStream(fileItem.get());
req.setFileName(fileItem.getName());
req.setFileExt(fileItem.getContentType());
req.setType(type);
baseResponse = foundationService.uploadFile(req, passport);
} catch (Exception e) {
baseResponse.addError(ErrorType.SYSTEM_ERROR, "系统错误");
baseResponse.addError(ErrorType.STACK_DUMP, LogUtil.dumpException(e));
} finally {
String json = JSON.toJSONString(baseResponse);
PrintWriter out = response.getWriter();
out.println(json);
}
}
}