From 1ef5cb1cd1799b65116ebc236313429431b79c0c Mon Sep 17 00:00:00 2001
From: CeDo <cedoogle@gmail.com>
Date: 星期四, 13 五月 2021 13:16:12 +0800
Subject: [PATCH] add:小程序添加视频上传

---
 springcloud_k8s_panzhihuazhihuishequ/applets/src/main/java/com/panzhihua/applets/api/CommonApi.java |   35 +++++++++++++++++++++++++++++------
 1 files changed, 29 insertions(+), 6 deletions(-)

diff --git a/springcloud_k8s_panzhihuazhihuishequ/applets/src/main/java/com/panzhihua/applets/api/CommonApi.java b/springcloud_k8s_panzhihuazhihuishequ/applets/src/main/java/com/panzhihua/applets/api/CommonApi.java
index e1cb5bb..158d294 100644
--- a/springcloud_k8s_panzhihuazhihuishequ/applets/src/main/java/com/panzhihua/applets/api/CommonApi.java
+++ b/springcloud_k8s_panzhihuazhihuishequ/applets/src/main/java/com/panzhihua/applets/api/CommonApi.java
@@ -1,15 +1,12 @@
 package com.panzhihua.applets.api;
 
 import cn.binarywang.wx.miniapp.api.WxMaSecCheckService;
-import com.netflix.discovery.converters.Auto;
 import com.panzhihua.applets.config.WxMaConfiguration;
 import com.panzhihua.common.constants.FtpConstants;
 import com.panzhihua.common.constants.UserConstants;
 import com.panzhihua.common.model.vos.R;
-import com.panzhihua.common.model.vos.area.AreaInfoVo;
 import com.panzhihua.common.model.vos.user.UserPhoneVO;
 import com.panzhihua.common.service.community.CommunityService;
-import com.panzhihua.common.service.user.UserService;
 import com.panzhihua.common.utlis.SFTPUtil;
 import io.swagger.annotations.Api;
 import io.swagger.annotations.ApiOperation;
@@ -28,10 +25,12 @@
 import java.io.File;
 import java.io.IOException;
 import java.io.InputStream;
+import java.util.Arrays;
 import java.util.Base64;
 import java.util.List;
 import java.util.UUID;
 import java.util.concurrent.TimeUnit;
+import java.util.concurrent.atomic.AtomicBoolean;
 
 /**
  * @program: springcloud_k8s_panzhihuazhihuishequ
@@ -61,22 +60,46 @@
     @Value("${ftp.url}")
     private String url;
 
+    /**
+     * 允许的图片文件后缀
+     */
+    private static  List<String> fileExtensionAllow = Arrays.asList("gif", "jpg", "png", "jpeg");
+    /**
+     * 允许的视频文件后缀
+     */
+    private static  List<String> videoExtensionAllow = Arrays.asList("mp4", "mov");
+
     @Resource
     private WxMaConfiguration wxMaConfiguration;
     @Resource
     private CommunityService communityService;
 
-    @ApiOperation(value = "上传照片")
+    @ApiOperation(value = "上传照片/视频 (jpg/jpeg/png/mp4/mov)")
     @PostMapping(value = "uploadimage", consumes = "multipart/*", headers = "content-type=multipart/form-date")
     public R uploadImage(@RequestParam MultipartFile file, HttpServletRequest request) throws IOException {
 //        微信图片内容校验
         WxMaSecCheckService wxMaSecCheckService = wxMaConfiguration.getMaService().getSecCheckService();
         String property = System.getProperty("user.dir");
-        String fileName = property + File.separator + UUID.randomUUID().toString().replace("-", "") + ".jpg";
+
+        String fileExtension = ".jpg";
+
+        String originName = file.getOriginalFilename();
+        AtomicBoolean isVideo = new AtomicBoolean(false);
+        videoExtensionAllow.forEach(ext ->{
+            if(ext.equalsIgnoreCase(originName)){
+                isVideo.set(true);
+            }
+        });
+        if(isVideo.get()){
+            fileExtension = ".mp4";
+        }
+
+
+        String fileName = property + File.separator + UUID.randomUUID().toString().replace("-", "") + fileExtension;
         File file1 = new File(fileName);
 
         String name = file.getOriginalFilename();
-        name = UUID.randomUUID().toString().replaceAll("-", "") + ".jpg";
+        name = UUID.randomUUID().toString().replaceAll("-", "") + fileExtension;
         try {
             SFTPUtil sftp = new SFTPUtil(userName, password, host, port);
             sftp.login();

--
Gitblit v1.7.1