添加本地文件上传逻辑

src/main/java/com/fdkankan/contro/controller/SceneFileController.java

@@ -68,17 +68,14 @@ public class SceneFileController{
     }
 
     /**
-     * 上传文件
+     * 单个文件上传
      * @param file
+     * @param params
      * @return
      */
-    @RequestMapping(value = "upload", method = RequestMethod.POST)
-    public ResultData uploadFile(String key, @RequestParam("file") MultipartFile file) throws Exception {
-        if(ObjectUtils.isEmpty(key) || ObjectUtils.isEmpty(file)){
-            log.error("参数有误！");
-            throw new BusinessException(ErrorCode.PARAM_REQUIRED);
-        }
-        fYunFileService.uploadFile(file.getInputStream(),key.concat(File.separator).concat(file.getOriginalFilename()));
-        return ResultData.ok();
+    @PostMapping("upload")
+    public ResultData upload(@RequestParam(value = "file",required = false) MultipartFile file,
+                         String params) throws Exception {
+        return sceneFileBuildService.uploadFile(file, params);
     }
 }

src/main/java/com/fdkankan/contro/service/ISceneFileBuildService.java

@@ -4,6 +4,7 @@ import com.baomidou.mybatisplus.extension.service.IService;
 import com.fdkankan.contro.entity.SceneFileBuild;
 import com.fdkankan.contro.vo.ResponseSceneFile;
 import com.fdkankan.web.response.ResultData;
+import org.springframework.web.multipart.MultipartFile;
 
 import java.io.IOException;
 
@@ -27,4 +28,5 @@ public interface ISceneFileBuildService extends IService<SceneFileBuild> {
 
     ResultData rebuildScene(String num,Boolean force) throws IOException;
 
+    ResultData uploadFile(MultipartFile file, String params) throws Exception;
 }

src/main/java/com/fdkankan/contro/service/impl/SceneFileBuildServiceImpl.java

@@ -12,6 +12,7 @@ import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.fdkankan.common.constant.*;
 import com.fdkankan.common.exception.BusinessException;
 import com.fdkankan.common.util.DateExtUtil;
+import com.fdkankan.common.util.FileMd5Util;
 import com.fdkankan.common.util.FileUtils;
 import com.fdkankan.common.util.SnowflakeIdGenerator;
 import com.fdkankan.contro.constant.RedisConstants;
@@ -22,6 +23,7 @@ import com.fdkankan.contro.vo.ResponseSceneFile;
 import com.fdkankan.contro.vo.ScenePlusVO;
 import com.fdkankan.fyun.config.FYunFileConfig;
 import com.fdkankan.fyun.face.FYunFileServiceInterface;
+import com.fdkankan.fyun.local.constant.LocalConstants;
 import com.fdkankan.model.constants.ConstantFilePath;
 import com.fdkankan.model.constants.UploadFilePath;
 import com.fdkankan.rabbitmq.bean.BuildSceneCallMessage;
@@ -36,6 +38,7 @@ import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.beans.factory.annotation.Value;
 import org.springframework.stereotype.Service;
 import org.springframework.util.ObjectUtils;
+import org.springframework.web.multipart.MultipartFile;
 
 import java.io.File;
 import java.io.IOException;
@@ -107,7 +110,7 @@ public class SceneFileBuildServiceImpl extends ServiceImpl<ISceneFileBuildMapper
     private IUserService userService;
 
     @Autowired
-    private ICompanyService companyService;
+    private ISceneFileUploadService sceneFileUploadService;
 
     @Autowired
     private FYunFileServiceInterface fYunFileService;
@@ -665,4 +668,148 @@ public class SceneFileBuildServiceImpl extends ServiceImpl<ISceneFileBuildMapper
 
         return ResultData.ok();
     }
+
+    @Override
+    public ResultData uploadFile(MultipartFile file, String params) throws Exception {
+        log.info("upload-params: "+params);
+        if (StringUtils.isEmpty(params)){
+            throw new BusinessException(ErrorCode.PARAM_REQUIRED);
+        }
+        params = params.replaceAll("%2B", "+");
+        Base64 base64 = new Base64();
+        String cipher = params;
+        // 私钥解密过程
+        byte[] res = RSAEncrypt.decrypt(RSAEncrypt.loadPrivateKeyByStr(RSAEncrypt.loadPrivateKeyByFile()),
+                base64.decode(cipher));
+        String restr = new String(res, "UTF-8");
+        log.debug("upload-params解密结果：" + restr);
+        String[] strArr = restr.split(SPLICE);
+        if (strArr.length != 6) {
+            throw new BusinessException(ErrorCode.PARAM_REQUIRED);
+        }
+        String mac = strArr[0];
+        String fileId = strArr[1];
+        String picNum = strArr[2];
+        String md5 = strArr[3];
+        String chunk = strArr[4];
+
+        ResultData result = null;
+        if (file.isEmpty()){
+            throw new BusinessException(ErrorCode.PARAM_REQUIRED.code(),"文件为空。");
+        }
+        if (StringUtils.isEmpty(fileId)){
+            throw new BusinessException(ErrorCode.PARAM_REQUIRED.code(),"文件Id为空。");
+        }
+        if (picNum == null){
+            throw new BusinessException(ErrorCode.PARAM_REQUIRED.code(),"照片数目为空。");
+        }
+        if (StringUtils.isEmpty(mac)){
+            throw new BusinessException(ErrorCode.PARAM_REQUIRED.code(),"mac为空。");
+        }
+        if (StringUtils.isEmpty(md5)){
+            throw new BusinessException(ErrorCode.PARAM_REQUIRED.code(),"md5为空。");
+        }
+        long size = file.getSize();
+        log.warn("fileId:"+fileId+"---picNum:"+picNum+"---size:"+size+"---md5:"+md5+"---mac:"+mac);
+        chunk = chunk.split(SPLICE)[0];
+        // 获取文件名
+        String fileName = file.getOriginalFilename();
+        log.info("上传的文件名为：" + fileName);
+        // 获取文件的后缀名
+        String folderName = redisUtil.get(String.format(RedisConstants.FILEID_FOLDER_BUILD, fileId));
+        if(StringUtils.isEmpty(folderName)){
+            SceneFileBuild fileBuild = findByFileId(fileId);
+            if(ObjectUtils.isEmpty(fileBuild)){
+                throw new BusinessException(ErrorCode.FAILURE_CODE_5012);
+            }
+            folderName = fileBuild.getUnicode();
+            redisUtil.set(String.format(RedisConstants.FILEID_FOLDER_BUILD, fileId), folderName,2 * 24 * 60 * 60);
+        }
+        // 1. 判断该文件是否已经上传过
+        // 2. 如果已经上传过，判断MD5值和文件大小是否相等。如果相等，更新数据记录。如果不相等，删除该文件，重新上传。
+        // 3. 如果未上传过，需要上传。
+        String filePath = mac.concat(File.separator).concat(fileId).concat(File.separator).concat(folderName);
+
+        String yunFilePath = LocalConstants.BASE_PATH.concat(fYunFileConfig.getBucket()).concat(File.separator).concat(ConstantFilePath.OSS_PREFIX)
+                .concat(filePath).concat(File.separator).concat(fileName);
+
+        boolean needUpload = false;
+        File yunFile = new File(yunFilePath);
+        if (yunFile.exists()){
+            String fileMD5 = FileMd5Util.getFileMD5(yunFile);
+            if (md5.equals(fileMD5) && yunFile.length() == size){
+                log.warn("文件已存在，MD5和文件大小一致。");
+                SceneFileUpload uploadEntity = sceneFileUploadService.findByFileIdAndChunk(fileId, Integer.valueOf(chunk));
+                if (uploadEntity != null){
+                    uploadEntity.setSize((int) size);
+                    uploadEntity.setMd5(md5);
+                    uploadEntity.setFilePath(yunFilePath.toString());
+                    uploadEntity.setFileSourceName(fileName);
+                    uploadEntity.setUploadStatus(1);
+                    sceneFileUploadService.updateById(uploadEntity);
+                }else{
+                    SceneFileUpload sceneFileUploadEntity = new SceneFileUpload();
+                    sceneFileUploadEntity.setSize((int) size);
+                    sceneFileUploadEntity.setMd5(md5);
+                    sceneFileUploadEntity.setFilePath(yunFilePath.toString());
+                    sceneFileUploadEntity.setFileSourceName(fileName);
+                    sceneFileUploadEntity.setUploadStatus(1);
+                    sceneFileUploadEntity.setFileId(fileId);
+                    sceneFileUploadEntity.setChunk(Integer.valueOf(chunk));
+                    sceneFileUploadService.save(sceneFileUploadEntity);
+                }
+                result = ResultData.ok();
+            }else if (!md5.equals(fileMD5)) {
+                log.warn("文件已上传，上传MD5："+md5+"，服务器MD5："+fileMD5+"。不一致。上传失败");
+                FileUtils.delFile(yunFilePath.toString());
+                needUpload = true;
+            }else if (yunFile.length() != size){
+                log.warn("文件已上传，文件大小不一致。上传失败");
+                FileUtils.delFile(yunFilePath.toString());
+                needUpload = true;
+            }
+        }else {
+            log.warn("文件不存在，需要重新上传");
+            needUpload = true;
+        }
+
+        // 4. 上传成功后，校验MD5和文件大小是否相等
+        // 5. 如果相等，更新数据记录。如果不相等，返回上传失败结果。
+        try {
+            if (needUpload){
+                fYunFileService.uploadFile(file.getInputStream(),ConstantFilePath.OSS_PREFIX.concat(filePath).concat(fileName));
+                File uploadFile = new File(yunFilePath);
+                String fileMD5 = FileMd5Util.getFileMD5(uploadFile);
+
+                SceneFileUpload sceneFileUploadEntity = new SceneFileUpload();
+                sceneFileUploadEntity.setSize((int) size);
+                sceneFileUploadEntity.setMd5(md5);
+                sceneFileUploadEntity.setFilePath(yunFilePath.toString());
+                sceneFileUploadEntity.setFileSourceName(fileName);
+                sceneFileUploadEntity.setFileId(fileId);
+                sceneFileUploadEntity.setChunk(Integer.valueOf(chunk));
+
+                if (md5.equals(fileMD5) && uploadFile.length() == size){
+                    log.warn("文件已上传，MD5和文件大小一致。上传成功");
+
+                    sceneFileUploadEntity.setUploadStatus(1);
+                    sceneFileUploadService.save(sceneFileUploadEntity);
+                    result = ResultData.ok();
+                }else if (!md5.equals(fileMD5)) {
+                    log.warn("文件已上传，上传MD5："+md5+"，服务器MD5："+fileMD5+"。不一致。上传失败");
+                    sceneFileUploadEntity.setUploadStatus(-1);
+                    sceneFileUploadService.save(sceneFileUploadEntity);
+                    result = ResultData.error(ErrorCode.SYSTEM_ERROR,"上传失败, 请重新上传。");
+                }else if (uploadFile.length() != size){
+                    log.warn("文件已上传，文件大小不一致。上传失败");
+                    sceneFileUploadEntity.setUploadStatus(-1);
+                    sceneFileUploadService.save(sceneFileUploadEntity);
+                    result = ResultData.error(ErrorCode.SYSTEM_ERROR,"上传失败, 请重新上传。");
+                }
+            }
+        }catch (IllegalStateException | IOException e) {
+            result = ResultData.error(ErrorCode.SYSTEM_ERROR,"上传失败, 请重新上传。");
+        }
+        return result;
+    }
 }