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
* @return
* @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) {
throw new BusinessException(Status.FAIL_INVALID_PARAM, "未获取待处理的文件！");
}
@ -77,9 +78,9 @@ public abstract class BaseFileController extends BaseController {
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<>();
dataMap.put("uuid", uploadFile.getUuid());
@ -91,11 +92,12 @@ public abstract class BaseFileController extends BaseController {
* 保存文件
* @param file
* @param entityClass
* @param request
* @param <T>
* @return
* @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 ext = FileHelper.getFileExtByName(file.getOriginalFilename());
@ -117,11 +119,12 @@ public abstract class BaseFileController extends BaseController {
/**
* 保存上传文件信息
* @param uploadFile
* @param request
* @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);
// 保存文件上传记录
uploadFileService.createEntity(uploadFile);
@ -130,7 +133,7 @@ public abstract class BaseFileController extends BaseController {
/**
* 保存文件之后的处理逻辑，如解析excel
*/
protected int extractDataCount(String fileUuid, String fullPath) throws Exception{
protected int extractDataCount(String fileUuid, String fullPath, HttpServletRequest request) throws Exception{
return 0;
}