p96170835
/
diboot
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Merge remote-tracking branch 'origin/develop' into develop 

# Conflicts:
#	diboot-file-starter/src/main/java/com/diboot/file/controller/BaseFileController.java
This commit is contained in:
mazhicheng 2020-04-26 11:23:46 +08:00
parent c0e65df23a f803390056
commit 6965b0e9e7
1 changed files with 11 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
diboot-file-starter/src/main/java/com/diboot/file/controller/BaseFileController.java
View File

@ -63,11 +63,12 @@ public abstract class BaseFileController extends BaseController {
/*** /***
* 直接上传文件 * 直接上传文件
* @param request
* @param * @param
* @return * @return
* @throws Exception * @throws Exception
*/ */
public <T> JsonResult uploadFile(MultipartFile file, Class<T> entityClass) throws Exception { public <T> JsonResult uploadFile(MultipartFile file, Class<T> entityClass, HttpServletRequest request) throws Exception {
if(file == null) { if(file == null) {
throw new BusinessException(Status.FAIL_INVALID_PARAM, "未获取待处理的文件！"); throw new BusinessException(Status.FAIL_INVALID_PARAM, "未获取待处理的文件！");
} }
@ -77,9 +78,9 @@ public abstract class BaseFileController extends BaseController {
throw new BusinessException(Status.FAIL_VALIDATION, "请上传合法的文件格式！"); throw new BusinessException(Status.FAIL_VALIDATION, "请上传合法的文件格式！");
} }
// 保存文件 // 保存文件
UploadFile uploadFile = saveFile(file, entityClass); UploadFile uploadFile = saveFile(file, entityClass, request);
// 保存上传记录 // 保存上传记录
createUploadFile(uploadFile); createUploadFile(uploadFile, request);
// 返回结果 // 返回结果
Map<String, String> dataMap = new HashMap<>(); Map<String, String> dataMap = new HashMap<>();
dataMap.put("uuid", uploadFile.getUuid()); dataMap.put("uuid", uploadFile.getUuid());
@ -91,11 +92,12 @@ public abstract class BaseFileController extends BaseController {
* 保存文件 * 保存文件
* @param file * @param file
* @param entityClass * @param entityClass
* @param request
* @param <T> * @param <T>
* @return * @return
* @throws Exception * @throws Exception
*/ */
protected <T> UploadFile saveFile(MultipartFile file, Class<T> entityClass) throws Exception{ protected <T> UploadFile saveFile(MultipartFile file, Class<T> entityClass, HttpServletRequest request) throws Exception{
// 文件后缀 // 文件后缀
String originFileName = file.getOriginalFilename(); String originFileName = file.getOriginalFilename();
String ext = FileHelper.getFileExtByName(file.getOriginalFilename()); String ext = FileHelper.getFileExtByName(file.getOriginalFilename());
@ -117,11 +119,12 @@ public abstract class BaseFileController extends BaseController {
/** /**
* 保存上传文件信息 * 保存上传文件信息
* @param uploadFile * @param uploadFile
* @param request
* @throws Exception * @throws Exception
*/ */
protected void createUploadFile(UploadFile uploadFile) throws Exception{ protected void createUploadFile(UploadFile uploadFile, HttpServletRequest request) throws Exception{
// 保存文件之后的处理逻辑 // 保存文件之后的处理逻辑
int dataCount = extractDataCount(uploadFile.getUuid(), uploadFile.getStoragePath()); int dataCount = extractDataCount(uploadFile.getUuid(), uploadFile.getStoragePath(), request);
uploadFile.setDataCount(dataCount); uploadFile.setDataCount(dataCount);
// 保存文件上传记录 // 保存文件上传记录
uploadFileService.createEntity(uploadFile); uploadFileService.createEntity(uploadFile);
@ -130,7 +133,7 @@ public abstract class BaseFileController extends BaseController {
/** /**
* 保存文件之后的处理逻辑，如解析excel * 保存文件之后的处理逻辑，如解析excel
*/ */
protected int extractDataCount(String fileUuid, String fullPath) throws Exception{ protected int extractDataCount(String fileUuid, String fullPath, HttpServletRequest request) throws Exception{
return 0; return 0;
} }