代码规范优化

Richardlsr · Aug 27, 2024 · 2071873 · 2071873
1 parent d3fcf3f
commit 2071873
Show file tree

Hide file tree

Showing 3 changed files with 58 additions and 8 deletions.
diff --git a/yami-shop-admin/src/main/java/com/yami/shop/admin/controller/FileController.java b/yami-shop-admin/src/main/java/com/yami/shop/admin/controller/FileController.java
@@ -10,6 +10,7 @@
 
 package com.yami.shop.admin.controller;
 
+import com.yami.shop.bean.enums.UploadType;
 import com.yami.shop.common.bean.Qiniu;
 import com.yami.shop.common.response.ServerResponseEntity;
 import com.yami.shop.common.util.ImgUploadUtil;
@@ -32,14 +33,14 @@
 @RestController
 @RequestMapping("/admin/file")
 public class FileController {
-	
+
 	@Autowired
 	private AttachFileService attachFileService;
 	@Autowired
 	private Qiniu qiniu;
 	@Autowired
 	private ImgUploadUtil imgUploadUtil;
-	
+
 	@PostMapping("/upload/element")
 	public ServerResponseEntity<String> uploadElementFile(@RequestParam("file") MultipartFile file) throws IOException{
 		if(file.isEmpty()){
@@ -48,17 +49,17 @@ public ServerResponseEntity<String> uploadElementFile(@RequestParam("file") Mult
 		String fileName = attachFileService.uploadFile(file);
         return ServerResponseEntity.success(fileName);
 	}
-	
+
 	@PostMapping("/upload/tinymceEditor")
 	public ServerResponseEntity<String> uploadTinymceEditorImages(@RequestParam("editorFile") MultipartFile editorFile) throws IOException{
 		String fileName =  attachFileService.uploadFile(editorFile);
 		String data = "";
-		if (Objects.equals(imgUploadUtil.getUploadType(), 1)) {
+		if (Objects.equals(imgUploadUtil.getUploadType(), UploadType.LOCAL.value())) {
 			data = imgUploadUtil.getUploadPath() + fileName;
-		} else if (Objects.equals(imgUploadUtil.getUploadType(), 2)) {
+		} else if (Objects.equals(imgUploadUtil.getUploadType(), UploadType.QINIU.value())) {
 			data = qiniu.getResourcesUrl() + fileName;
 		}
         return ServerResponseEntity.success(data);
 	}
-	
+
 }
diff --git a/yami-shop-bean/src/main/java/com/yami/shop/bean/enums/UploadType.java b/yami-shop-bean/src/main/java/com/yami/shop/bean/enums/UploadType.java
@@ -0,0 +1,48 @@
+/*
+ * Copyright (c) 2018-2999 广州市蓝海创新科技有限公司 All rights reserved.
+ *
+ * https://www.mall4j.com/
+ *
+ * 未经允许，不可做商业用途！
+ *
+ * 版权所有，侵权必究！
+ */
+
+package com.yami.shop.bean.enums;
+
+/**
+ * @author lanhai
+ * 文件上传方式 1.本地文件上传 2.七牛云
+ */
+public enum UploadType {
+
+    /**
+     * 本地文件上传
+     */
+    LOCAL(1),
+
+    /**
+     * 七牛云
+     */
+    QINIU(2);
+
+    private Integer num;
+
+    public Integer value() {
+        return num;
+    }
+
+    UploadType(Integer num) {
+        this.num = num;
+    }
+
+    public static UploadType instance(Integer value) {
+        UploadType[] enums = values();
+        for (UploadType statusEnum : enums) {
+            if (statusEnum.value().equals(value)) {
+                return statusEnum;
+            }
+        }
+        return null;
+    }
+}
diff --git a/yami-shop-service/src/main/java/com/yami/shop/service/impl/AttachFileServiceImpl.java b/yami-shop-service/src/main/java/com/yami/shop/service/impl/AttachFileServiceImpl.java
@@ -21,6 +21,7 @@
 import com.qiniu.storage.UploadManager;
 import com.qiniu.storage.model.DefaultPutRet;
 import com.qiniu.util.Auth;
+import com.yami.shop.bean.enums.UploadType;
 import com.yami.shop.bean.model.AttachFile;
 import com.yami.shop.common.bean.Qiniu;
 import com.yami.shop.common.util.ImgUploadUtil;
@@ -83,9 +84,9 @@ public String uploadFile(MultipartFile file) throws IOException {
 	public void deleteFile(String fileName){
 		attachFileMapper.delete(new LambdaQueryWrapper<AttachFile>().eq(AttachFile::getFilePath,fileName));
 		try {
-			if (Objects.equals(imgUploadUtil.getUploadType(), 1)) {
+			if (Objects.equals(imgUploadUtil.getUploadType(), UploadType.LOCAL.value())) {
 				imgUploadUtil.delete(fileName);
-			} else if (Objects.equals(imgUploadUtil.getUploadType(), 2)) {
+			} else if (Objects.equals(imgUploadUtil.getUploadType(), UploadType.QINIU.value())) {
 				bucketManager.delete(qiniu.getBucket(), fileName);
 			}
 		} catch (QiniuException e) {