上传接口替换

haoban-manage3-web/src/main/java/com/gic/haoban/manage/web/controller/UploadController.java

@@ -2,11 +2,18 @@ package com.gic.haoban.manage.web.controller;
 
 import cn.hutool.crypto.SecureUtil;
 import com.alibaba.fastjson.JSONObject;
+import com.gic.clerk.api.dto.AuthorizedUser;
 import com.gic.commons.util.DateUtil;
 import com.gic.haoban.common.utils.HaobanResponse;
 import com.gic.haoban.manage.web.errCode.HaoBanErrCode;
+import com.gic.haoban.manage.web.utils.statistics.UploadUtils;
 import com.gic.qcloud.BucketNameEnum;
 import com.gic.qcloud.FileUploadUtil;
+import com.gic.thirdparty.cloudfile.CloudFileUtil;
+import com.gic.thirdparty.cloudfile.enums.CloudFileBusinessOptEnum;
+import com.gic.thirdparty.cloudfile.enums.CloudFileTypeEnum;
+import com.gic.thirdparty.cloudfile.pojo.CloudFileInfo;
+import com.gic.web.common.utils.SessionContextUtils;
 import org.apache.commons.io.FileUtils;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
@@ -17,6 +24,7 @@ import org.springframework.web.bind.annotation.RestController;
 import org.springframework.web.multipart.MultipartFile;
 
 import java.io.File;
+import java.io.FileInputStream;
 import java.time.LocalDateTime;
 import java.time.format.DateTimeFormatter;
 import java.util.HashSet;
@@ -39,6 +47,8 @@ public class UploadController extends WebBaseController {
         if (file == null || file.isEmpty()) {
             return resultResponse(HaoBanErrCode.ERR_2);
         }
+        //获取登录人信息
+        final AuthorizedUser au = SessionContextUtils.getLoginUser();
         String name = file.getOriginalFilename();
         // 未加密前文件名称
         String mediaName = file.getOriginalFilename();
@@ -59,7 +69,19 @@ public class UploadController extends WebBaseController {
         String key = "haoban/" + dayFilePath + "/" + nameUpload;
         fileType = FILE_TYPE.contains(fileType) ? fileType : BucketNameEnum.OTHER_90000.getName();
         try {
-            String url = FileUploadUtil.simpleUploadFileFromLocal(templateFile, key, fileType);
+            //替换新的上传接口
+//            String url = FileUploadUtil.simpleUploadFileFromLocal(templateFile, key, fileType);
+            CloudFileTypeEnum cloudFileTypeEnum = CloudFileTypeEnum.OTHER;
+            if (UploadUtils.isPicture(suffix)) {
+                cloudFileTypeEnum = CloudFileTypeEnum.IMAGE;
+            } else if (UploadUtils.isOffice(suffix)) {
+                cloudFileTypeEnum = CloudFileTypeEnum.FILE;
+            } else if (UploadUtils.isMusic(suffix)) {
+                cloudFileTypeEnum = CloudFileTypeEnum.AUDIO;
+            } else if (UploadUtils.isVedio(suffix)) {
+                cloudFileTypeEnum = CloudFileTypeEnum.VIDEO;
+            }
+            CloudFileInfo cloudFileInfo =   CloudFileUtil.uploadFile(new FileInputStream(templateFile), suffix, cloudFileTypeEnum,au.getEnterpriseDTO().getFactoryCode(), CloudFileBusinessOptEnum.HAOBAN_COMMON);
             if (templateFile != null) {
                 templateFile.delete();
             }
@@ -69,7 +91,7 @@ public class UploadController extends WebBaseController {
             res.put("ext", suffix);
             res.put("size", len);
             res.put("key", key);
-            res.put("url", "https://" + url);
+            res.put("url", cloudFileInfo.getOrgFileUrl());
             return resultResponse(HaoBanErrCode.ERR_1, res);
         } catch (Exception e) {
             logger.info("异常:{}", e.getMessage(), e);

haoban-manage3-web/src/main/java/com/gic/haoban/manage/web/utils/statistics/UploadUtils.java

+package com.gic.haoban.manage.web.utils.statistics;
+
+/**
+ * @Author jinxin
+ * @Date 2022/12/27 09:23
+ * 文件上传工具类
+ */
+public class UploadUtils {
+    /**
+     * 是否是图片
+     * @param fileSuffix
+     * @return
+     * @throws Exception
+     */
+    public static boolean isPicture(String fileSuffix) throws Exception{
+        //pdf 隐私文件，小程序端，只能用这个
+        String[] imageArray =
+                {"bmp", "dib","gif", "jfif", "jpe","jpeg", "jpg", "png", "tif", "tiff", "ico", "10", "pdf"};
+        for (String str : imageArray) {
+            if (str.equals(fileSuffix.replace(".", ""))) {
+                return true;
+            }
+        }
+        return false;
+    }
+
+    public static boolean isOffice(String fileSuffix) throws Exception{
+        String[] arr =
+                {"doc", "docx","xls", "xlsx", "pptx", "ppt"};
+        for (String str : arr) {
+            if (str.equals(fileSuffix.replace(".", ""))) {
+                return true;
+            }
+        }
+        return false;
+    }
+
+    public static boolean isMusic(String fileSuffix) throws Exception{
+        String[] arr =
+                {"wma", "mpc","ogg", "wav", "mpeg", "mp3"};
+        for (String str : arr) {
+            if (str.equals(fileSuffix.replace(".", ""))) {
+                return true;
+            }
+        }
+        return false;
+    }
+
+    public static boolean isVedio(String fileSuffix) throws Exception{
+        String[] arr =
+                {"rm", "rmvb","mp4", "wmv", "asf", "asx", "3gp", "mov", "m4v", "avi", "dat", "mkv", "flv", "vob"};
+        for (String str : arr) {
+            if (str.equals(fileSuffix.replace(".", ""))) {
+                return true;
+            }
+        }
+        return false;
+    }
+}