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());