bonus
/
gzdevicemgt
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

文件白名单

This commit is contained in:
sxu 2024-09-28 19:22:58 +08:00
parent e62e1172d7
commit 03141f7d69
1 changed files with 28 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

56
sgzb-modules/sgzb-system/src/main/java/com/bonus/sgzb/system/controller/SysFileController.java
View File

@ -42,34 +42,34 @@ public class SysFileController {
private ResourceLoader resourceLoader; private ResourceLoader resourceLoader;
@PostMapping("/upload") // @PostMapping("/upload")
@ApiOperation(value = "文件上传") // @ApiOperation(value = "文件上传")
public AjaxResult upload(HttpServletRequest request) { // public AjaxResult upload(HttpServletRequest request) {
String limitWords = request.getParameter("limitWords"); // String limitWords = request.getParameter("limitWords");
FileInfo file = new FileInfo(); // FileInfo file = new FileInfo();
try { // try {
file = service.uploadFile(request); // file = service.uploadFile(request);
// String suffix = file.getFileName().substring(file.getFileName().lastIndexOf(".") + 1); // // String suffix = file.getFileName().substring(file.getFileName().lastIndexOf(".") + 1);
//
/* if (! ATTACHMENT_FILE_SUFFIX.contains(suffix) || ! COST_FILE_SUFFIX.contains(suffix)) { // /* if (! ATTACHMENT_FILE_SUFFIX.contains(suffix) || ! COST_FILE_SUFFIX.contains(suffix)) {
return AjaxResult.error(ErrorCode.COMMISSION_BILL_ATTACHMENT_NOT_SUPPORT.getCode(), // return AjaxResult.error(ErrorCode.COMMISSION_BILL_ATTACHMENT_NOT_SUPPORT.getCode(),
ErrorCode.COMMISSION_BILL_ATTACHMENT_NOT_SUPPORT.getMessage()); // ErrorCode.COMMISSION_BILL_ATTACHMENT_NOT_SUPPORT.getMessage());
}*/ // }*/
if (limitWords != null && file.getWords() > Integer.parseInt(limitWords)){ // if (limitWords != null && file.getWords() > Integer.parseInt(limitWords)){
return AjaxResult.error(ErrorCode.ATTACHMENT_UPLOAD_FAILED.getCode(), // return AjaxResult.error(ErrorCode.ATTACHMENT_UPLOAD_FAILED.getCode(),
ErrorCode.ATTACHMENT_UPLOAD_FAILED.getMessage()); // ErrorCode.ATTACHMENT_UPLOAD_FAILED.getMessage());
} // }
}catch (Exception e){ // }catch (Exception e){
log.error(e.getMessage()); // log.error(e.getMessage());
} // }
if (file != null && file.getId() != 0){ // if (file != null && file.getId() != 0){
return AjaxResult.success(file); // return AjaxResult.success(file);
}else { // }else {
return AjaxResult.error(ErrorCode.FAILURE_TO_UPLOAD_FILE.getCode(), // return AjaxResult.error(ErrorCode.FAILURE_TO_UPLOAD_FILE.getCode(),
ErrorCode.FAILURE_TO_UPLOAD_FILE.getMessage()); // ErrorCode.FAILURE_TO_UPLOAD_FILE.getMessage());
} // }
//
} // }
@ApiOperation(value = "模板", httpMethod = "GET") @ApiOperation(value = "模板", httpMethod = "GET")
@GetMapping("download") @GetMapping("download")