feat:重构file extension,添加七牛云上传

Author: pedro

pom.xml

@@ -84,6 +84,11 @@
             <version>${hutool.version}</version>
         </dependency>
 
+        <dependency>
+            <groupId>com.qiniu</groupId>
+            <artifactId>qiniu-java-sdk</artifactId>
+            <version>7.2.28</version>
+        </dependency>
 
         <dependency>
             <groupId>org.springframework.boot</groupId>

src/main/java/io/github/talelin/merak/extensions/file/AbstractUploader.java

@@ -0,0 +1,203 @@
+package io.github.talelin.merak.extensions.file;
+
+import io.github.talelin.autoconfigure.exception.*;
+import org.springframework.util.MultiValueMap;
+import org.springframework.web.multipart.MultipartFile;
+
+import java.util.ArrayList;
+import java.util.List;
+import java.util.UUID;
+
+/**
+ * 文件上传类的基类
+ * 模版模式
+ */
+public abstract class AbstractUploader implements Uploader {
+
+    private PreHandler preHandler;
+
+    public List<File> upload(MultiValueMap<String, MultipartFile> fileMap) {
+        checkFileMap(fileMap);
+        // 得到单个文件的大小限制
+        // 本地存储需先初始化存储文件夹
+        return handleMultipartFiles(fileMap);
+    }
+
+    public List<File> upload(MultiValueMap<String, MultipartFile> fileMap, PreHandler preHandler) {
+        this.preHandler = preHandler;
+        return this.upload(fileMap);
+    }
+
+    protected List<File> handleMultipartFiles(MultiValueMap<String, MultipartFile> fileMap) {
+        long singleFileLimit = getSingleFileLimit();
+        List<File> res = new ArrayList<>();
+        fileMap.keySet().forEach(key -> fileMap.get(key).forEach(file -> {
+            if (!file.isEmpty()) {
+                handleOneFile0(res, singleFileLimit, file);
+            }
+        }));
+        return res;
+    }
+
+    private void handleOneFile0(List<File> res, long singleFileLimit, MultipartFile file) {
+        byte[] bytes = getFileBytes(file);
+        String[] include = getFileProperties().getInclude();
+        String[] exclude = getFileProperties().getExclude();
+        String ext = checkOneFile(include, exclude, singleFileLimit, file.getOriginalFilename(), bytes.length);
+        String newFilename = getNewFilename(ext);
+        String storePath = getStorePath(newFilename);
+        // 生成文件的md5值
+        String md5 = FileUtil.getFileMD5(bytes);
+        File fileData = File.builder().
+                name(newFilename).
+                md5(md5).
+                path(storePath).
+                size(bytes.length).
+                type(getFileType()).
+                extension(ext).
+                build();
+        // 如果预处理器不为空，且处理结果为false，直接返回, 否则处理
+        if (preHandler != null && !preHandler.handle(fileData))
+            return;
+        boolean ok = handleOneFile(bytes, newFilename);
+        if (ok) {
+            res.add(fileData);
+        }
+    }
+
+    private long getSingleFileLimit() {
+        String singleLimit = getFileProperties().getSingleLimit();
+        return FileUtil.parseSize(singleLimit);
+    }
+
+    /**
+     * 得到文件配置
+     *
+     * @return 文件配置
+     */
+    protected abstract FileProperties getFileProperties();
+
+    /**
+     * 处理一个文件
+     */
+    protected abstract boolean handleOneFile(byte[] bytes, String newFilename);
+
+    /**
+     * 返回文件路径
+     *
+     * @param newFilename 文件名
+     * @return 文件路径
+     */
+    protected abstract String getStorePath(String newFilename);
+
+    /**
+     * 返回文件存储位置类型
+     *
+     * @return LOCAL ｜ REMOTE
+     */
+    protected abstract String getFileType();
+
+    /**
+     * 获得新文件的名称
+     *
+     * @param ext 文件后缀
+     * @return 新名称
+     */
+    protected String getNewFilename(String ext) {
+        String uuid = UUID.randomUUID().toString().replace("-", "");
+        return uuid + ext;
+    }
+
+    /**
+     * 检查文件
+     */
+    protected void checkFileMap(MultiValueMap<String, MultipartFile> fileMap) {
+        if (fileMap.isEmpty()) {
+            throw new NotFoundException("file not found", 10026);
+        }
+        int nums = getFileProperties().getNums();
+        if (fileMap.size() > nums) {
+            throw new FileTooManyException("too many files, amount of files must less than" + nums, 10180);
+        }
+    }
+
+    /**
+     * 获得文件的字节
+     *
+     * @param file 文件
+     * @return 字节
+     */
+    protected byte[] getFileBytes(MultipartFile file) {
+        byte[] bytes;
+        try {
+            bytes = file.getBytes();
+        } catch (Exception e) {
+            throw new FailedException("read file date failed", 10190);
+        }
+        return bytes;
+    }
+
+    /**
+     * 单个文件检查
+     *
+     * @param singleFileLimit 单个文件大小限制
+     * @param originName      文件原始名称
+     * @param length          文件大小
+     * @return 文件的扩展名，例如： .jpg
+     */
+    protected String checkOneFile(String[] include, String[] exclude, long singleFileLimit, String originName, int length) {
+        // 写到了本地
+        String ext = FileUtil.getFileExt(originName);
+        // 检测扩展
+        if (!this.checkExt(include, exclude, ext)) {
+            throw new FileExtensionException(ext + "文件类型不支持");
+        }
+        // 检测单个大小
+        if (length > singleFileLimit) {
+            throw new FileTooLargeException(originName + "文件不能超过" + singleFileLimit);
+        }
+        return ext;
+    }
+
+    /**
+     * 检查文件后缀
+     *
+     * @param ext 后缀名
+     * @return 是否通过
+     */
+    protected boolean checkExt(String[] include, String[] exclude, String ext) {
+        int inLen = include == null ? 0 : include.length;
+        int exLen = exclude == null ? 0 : exclude.length;
+        // 如果两者都有取 include，有一者则用一者
+        if (inLen > 0 && exLen > 0) {
+            return this.findInInclude(include, ext);
+        } else if (inLen > 0) {
+            // 有include，无exclude
+            return this.findInInclude(include, ext);
+        } else if (exLen > 0) {
+            // 有exclude，无include
+            return this.findInExclude(exclude, ext);
+        } else {
+            // 二者都没有
+            return true;
+        }
+    }
+
+    protected boolean findInInclude(String[] include, String ext) {
+        for (String s : include) {
+            if (s.equals(ext)) {
+                return true;
+            }
+        }
+        return false;
+    }
+
+    protected boolean findInExclude(String[] exclude, String ext) {
+        for (String s : exclude) {
+            if (s.equals(ext)) {
+                return true;
+            }
+        }
+        return false;
+    }
+}

src/main/java/io/github/talelin/merak/extensions/file/FileProperties.java

@@ -11,22 +11,24 @@ public class FileProperties {
 
     private static final String[] DEFAULT_EMPTY_ARRAY = new String[0];
 
-    private String dir = "/assets";
+    private String storeDir = "/assets";
 
     private String singleLimit = "2MB";
 
     private Integer nums = 10;
 
+    private String domain;
+
     private String[] exclude = DEFAULT_EMPTY_ARRAY;
 
     private String[] include = DEFAULT_EMPTY_ARRAY;
 
-    public String getDir() {
-        return dir;
+    public String getStoreDir() {
+        return storeDir;
     }
 
-    public void setDir(String dir) {
-        this.dir = dir;
+    public void setStoreDir(String storeDir) {
+        this.storeDir = storeDir;
     }
 
     public String getSingleLimit() {
@@ -60,4 +62,12 @@ public String[] getInclude() {
     public void setInclude(String[] include) {
         this.include = include;
     }
+
+    public String getDomain() {
+        return domain;
+    }
+
+    public void setDomain(String domain) {
+        this.domain = domain;
+    }
 }