From 34e50f2c23ff7cc7f0dbf2a40e3d038b30aec12f Mon Sep 17 00:00:00 2001
From: huanghongfa <huanghongfa123456>
Date: 星期六, 24 七月 2021 13:47:30 +0800
Subject: [PATCH] 测试社区动态

---
 springcloud_k8s_panzhihuazhihuishequ/applets/src/main/java/com/panzhihua/applets/api/CommonApi.java |   71 ++++++++++++++++++++++++++---------
 1 files changed, 52 insertions(+), 19 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 15184ca..2b4c66d 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
@@ -6,6 +6,7 @@
 import com.panzhihua.common.constants.UserConstants;
 import com.panzhihua.common.model.vos.R;
 import com.panzhihua.common.model.vos.user.UserPhoneVO;
+import com.panzhihua.common.service.community.CommunityService;
 import com.panzhihua.common.utlis.SFTPUtil;
 import io.swagger.annotations.Api;
 import io.swagger.annotations.ApiOperation;
@@ -24,9 +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
@@ -56,39 +60,67 @@
     @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 = "上传照片")
-    @PostMapping(value = "uploadimage",consumes = "multipart/*",headers = "content-type=multipart/form-date")
+    @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();
+        WxMaSecCheckService wxMaSecCheckService = wxMaConfiguration.getMaService().getSecCheckService();
         String property = System.getProperty("user.dir");
-        String fileName = property + File.separator + UUID.randomUUID().toString().replace("-", "")+".jpg";
-        File file1=new File(fileName);
+
+        String fileExtension = ".jpg";
+
+        String originName = file.getOriginalFilename();
+        AtomicBoolean isVideo = new AtomicBoolean(false);
+        videoExtensionAllow.forEach(ext ->{
+            String originNameLowerCase = originName.toLowerCase();
+            if(originNameLowerCase.endsWith("." +ext)){
+                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);
+            SFTPUtil sftp = new SFTPUtil(userName, password, host, port);
             sftp.login();
             InputStream is = file.getInputStream();
             file.transferTo(file1);
-        try {
-            boolean b = wxMaSecCheckService.checkImage(file1);
-            boolean delete = file1.delete();
-            log.info("临时文件删除【{}】",delete);
-            if (!b) {
-                return R.fail("图片内容违规");
+            try {
+                boolean b = wxMaSecCheckService.checkImage(file1);
+                boolean delete = file1.delete();
+                log.info("临时文件删除【{}】", delete);
+                if (!b) {
+                    return R.fail("图片内容违规");
+                }
+            } catch (WxErrorException e) {
+                log.error("微信审核图片出错【{}】", e.getMessage());
+                return R.fail("图片上传失败");
+//                e.printStackTrace();
             }
-        } catch (WxErrorException e) {
-            log.error("微信审核图片出错【{}】",e.getMessage());
-            e.printStackTrace();
-        }
             sftp.uploadMore(FtpConstants.FTPFILEPATH_IDCARD, name, is);
             sftp.logout();
-            return R.ok(url+"/idcard/" + name);
+            return R.ok(url + "/idcard/" + name);
         } catch (Exception e) {
             log.error("上传照片失败【{}】", e.getMessage());
             return R.fail();
@@ -106,10 +138,11 @@
         //todo 发送验证码接第三方
         int nextInt = RandomUtils.nextInt(99999, 1000000);
         ValueOperations<String, String> valueOperations = stringRedisTemplate.opsForValue();
-        valueOperations.set(UserConstants.PHONE_PUT+newPhone,nextInt+"",3, TimeUnit.MINUTES);
+        valueOperations.set(UserConstants.PHONE_PUT + newPhone, nextInt + "", 3, TimeUnit.MINUTES);
         return R.ok(nextInt);
     }
 
+
     public static void main(String[] args) {
 //        int nextInt = RandomUtils.nextInt(99999, 1000000);
 //        System.out.println(nextInt);

--
Gitblit v1.7.1