소스 검색

file upload name

wangyunpeng 5 일 전
부모
커밋
8ab2338097
1개의 변경된 파일8개의 추가작업 그리고 1개의 파일을 삭제
  1. 8 1
      api-module/src/main/java/com/tzld/piaoquan/api/controller/FileController.java

+ 8 - 1
api-module/src/main/java/com/tzld/piaoquan/api/controller/FileController.java

@@ -39,7 +39,14 @@ public class FileController {
             return CommonResponse.create(ExceptionEnum.PARAM_ERROR.getCode(), bindingResult.getAllErrors().get(0).getDefaultMessage());
         }
         if (StringUtils.isBlank(param.getFileUri())) {
-            param.setFileUri("temp/" + System.currentTimeMillis() + "_" + RandomUtil.generate18String());
+            String fileName = param.getFile().getOriginalFilename();
+            // 获取文件名中的文件类型
+            if (StringUtils.isBlank(fileName)) {
+                param.setFileUri("temp/" + System.currentTimeMillis() + "_" + RandomUtil.generate18String());
+            } else {
+                String fileType = fileName.substring(fileName.lastIndexOf("."));
+                param.setFileUri("temp/" + System.currentTimeMillis() + "_" + RandomUtil.generate18String() + fileType);
+            }
         }
         String fileUrl = AliOssFileTool.saveInPublicReturnHost(param.getFile().getInputStream(),
                 EnumPublicBuckets.PUBBUCKET, param.getFileUri(), param.getFileType());