文化中心

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

@@ -61,6 +61,11 @@ public class BuildSceneDTServiceImpl implements IBuildSceneDTService {
     }
 
     @Override
+    public void handModelFail(String reason, String serverPath, String num, String hostName, String logUrl) {
+
+    }
+
+    @Override
     public void handBaseFail(String reason, String serverPath, String num, String hostName) {
 //        CompletableFuture.runAsync(() -> {
 //            try {

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

@@ -2009,4 +2009,180 @@ public class SceneFileBuildServiceImpl extends ServiceImpl<ISceneFileBuildMapper
         //推送mq到前置计算
         return ResultData.ok(num);
     }
+
+    @Override
+    public ResultData uploadFile(MultipartFile file, String params) throws Exception {
+
+        String uuid = UUID.randomUUID().toString();
+        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.info("upload-params解密结果：" + restr);
+
+        //这里增加判断，如果是日志文件，不做任何处理，直接返回
+        if(restr.contains("normal")){
+            return ResultData.ok("参数带有normal，直接丢弃, params=" + params);
+        }
+        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];
+
+        log.info("mac:{},fileId:{},picNum:{},md5:{},chunk:{}", mac, fileId, picNum, md5, chunk);
+
+        ResultData result = null;
+        if (file.isEmpty()){
+            log.error("文件为空。");
+            throw new BusinessException(ErrorCode.PARAM_REQUIRED.code(),"文件为空。");
+        }
+        if (StringUtils.isEmpty(fileId)){
+            log.error("文件Id为空。");
+            throw new BusinessException(ErrorCode.PARAM_REQUIRED.code(),"文件Id为空。");
+        }
+        if (picNum == null){
+            log.error("照片数目为空。");
+            throw new BusinessException(ErrorCode.PARAM_REQUIRED.code(),"照片数目为空。");
+        }
+        if (StringUtils.isEmpty(mac)){
+            log.error("mac为空。");
+            throw new BusinessException(ErrorCode.PARAM_REQUIRED.code(),"mac为空。");
+        }
+        if (StringUtils.isEmpty(md5)){
+            log.error("md5为空。");
+            throw new BusinessException(ErrorCode.PARAM_REQUIRED.code(),"md5为空。");
+        }
+        long size = file.getSize();
+        log.info("mac:{}, fileId:{}, picNum:{}, md5:{}, chunk:{}", mac, fileId, picNum, md5, chunk);
+        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)){
+                log.error("数据不正常，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);
+        log.info("yunFilePath:{}", yunFilePath);
+
+        Map<String, Object> map = new HashMap<>();
+        map.put("uuid", uuid);
+        map.put("params", params);
+        map.put("yunFilePath", yunFilePath);
+        map.put("mac", mac);
+        map.put("fileId", fileId);
+        map.put("picNum", picNum);
+        map.put("md5", md5);
+        map.put("chunk", chunk);
+
+        boolean needUpload = false;
+        File yunFile = new File(yunFilePath);
+        if (yunFile.exists()){
+            String fileMD5 = FileMd5Util.getFileMD5(yunFile);
+            if (md5.equals(fileMD5) && yunFile.length() == size){
+                log.info("文件已存在，MD5和文件大小一致。");
+                SceneFileUpload uploadEntity = sceneFileUploadService.findByFileIdAndChunk(fileId, Integer.valueOf(chunk));
+                if (uploadEntity != null){
+                    uploadEntity.setSize((int) size);
+                    uploadEntity.setMd5(md5);
+                    uploadEntity.setFilePath(yunFilePath);
+                    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);
+                }
+                map.put("needUpload",false);
+                result = ResultData.ok(map);
+            }else if (!md5.equals(fileMD5)) {
+                log.info("文件已上传，上传MD5："+md5+"，服务器MD5："+fileMD5+"。不一致。需要重新上传");
+                FileUtils.delFile(yunFilePath.toString());
+                needUpload = true;
+            }else if (yunFile.length() != size){
+                log.info("文件已上传，文件大小不一致。需要重新上传");
+                FileUtils.delFile(yunFilePath);
+                needUpload = true;
+            }
+        }else {
+            log.info("文件不存在，需要重新上传");
+            needUpload = true;
+        }
+
+        // 4. 上传成功后，校验MD5和文件大小是否相等
+        // 5. 如果相等，更新数据记录。如果不相等，返回上传失败结果。
+        try {
+            if (needUpload){
+
+                map.put("needUpload", true);
+
+                fYunFileService.uploadFile(file.getInputStream(),ConstantFilePath.OSS_PREFIX.concat(filePath).concat(File.separator).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);
+                sceneFileUploadEntity.setFileSourceName(fileName);
+                sceneFileUploadEntity.setFileId(fileId);
+                sceneFileUploadEntity.setChunk(Integer.valueOf(chunk));
+
+                if (md5.equals(fileMD5) && uploadFile.length() == size){
+                    log.info("文件已上传，MD5和文件大小一致。上传成功");
+                    sceneFileUploadEntity.setUploadStatus(1);
+                    sceneFileUploadService.save(sceneFileUploadEntity);
+                    result = ResultData.ok(map);
+                }else if (!md5.equals(fileMD5)) {
+                    log.error("文件已上传，上传MD5："+md5+"，服务器MD5："+fileMD5+"。不一致。上传失败");
+                    sceneFileUploadEntity.setUploadStatus(-1);
+                    sceneFileUploadService.save(sceneFileUploadEntity);
+                    result = ResultData.error(ErrorCode.SYSTEM_ERROR,"上传失败, 请重新上传。");
+                }else if (uploadFile.length() != size){
+                    log.error("文件已上传，文件大小不一致。上传失败");
+                    sceneFileUploadEntity.setUploadStatus(-1);
+                    sceneFileUploadService.save(sceneFileUploadEntity);
+                    result = ResultData.error(ErrorCode.SYSTEM_ERROR,"上传失败, 请重新上传。");
+                }
+            }
+        }catch (IllegalStateException | IOException e) {
+            log.error("上传失败, 请重新上传,md5:{},chunk:{}", md5, chunk);
+            log.error("上传失败", e);
+            result = ResultData.error(ErrorCode.SYSTEM_ERROR,"上传失败, 请重新上传。");
+        }
+        return result;
+    }
 }