4dkankan-center-modeling-control/src/main/java/com/fdkankan/contro/service/impl/JmgaServiceImpl.java

package com.fdkankan.contro.service.impl;

import cn.hutool.core.codec.Base64;
import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.io.FileUtil;
import cn.hutool.core.lang.UUID;
import cn.hutool.core.util.ZipUtil;
import cn.hutool.http.HttpUtil;
import com.alibaba.excel.EasyExcel;
import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
import com.fdkankan.common.constant.ErrorCode;
import com.fdkankan.common.constant.SceneStatus;
import com.fdkankan.common.exception.BusinessException;
import com.fdkankan.common.util.CmdUtils;
import com.fdkankan.common.util.DateExtUtil;
import com.fdkankan.contro.bean.SendCallAlgorithmDetail;
import com.fdkankan.contro.constant.ZipConstant;
import com.fdkankan.contro.entity.*;
import com.fdkankan.contro.mq.service.IBuildLogService;
import com.fdkankan.contro.service.*;
import com.fdkankan.contro.vo.SendCallAlgorithmParam;
import com.fdkankan.contro.vo.UploadSceneOrigParamVo;
import com.fdkankan.fyun.face.FYunFileServiceInterface;
import com.fdkankan.model.constants.ConstantFilePath;
import com.fdkankan.rabbitmq.util.RabbitMqProducer;
import com.fdkankan.web.util.RSAEncrypt;
import lombok.extern.slf4j.Slf4j;
import net.lingala.zip4j.core.ZipFile;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import javax.annotation.Resource;
import java.io.File;
import java.nio.charset.StandardCharsets;
import java.util.*;
import java.util.stream.Collectors;

@Slf4j
@Service
public class JmgaServiceImpl implements IJmgaService {

    @Autowired
    private IOrigFileUploadBatchService origFileUploadBatchService;

    @Autowired
    private IOrigFileUploadService origFileUploadService;
    @Resource
    private FYunFileServiceInterface fileServiceInterface;
    @Resource
    private RabbitMqProducer mqProducer;
    @Resource
    private FYunFileServiceInterface fYunFileService;
    @Autowired
    private ISceneFileBuildService sceneFileBuildService;
    @Autowired
    private IUserService userService;
    @Autowired
    private IScenePlusService scenePlusService;
    @Autowired
    private IScenePlusExtService scenePlusExtService;
    @Autowired
    private ISceneBuildProcessLogService sceneBuildProcessLogService;
    @Autowired
    private IBuildLogService buildLogService;

    @Override
    public void checkFileWhole(String uuid, String dataSource, JSONObject fdageData) {

        int camType = fdageData.getJSONObject("cam").getIntValue("type");

        //从data.fdage中获取需要上传的文件列表
        JSONArray points = fdageData.getJSONArray("points");
        if(CollUtil.isEmpty(points)){
            return;
        }

        List<String> fileList = new ArrayList<>();
        for (Object point : points) {
            JSONObject pointJson = (JSONObject) point;
            if(camType == 5){//圆周率相机
                String name = pointJson.getString("name");
                fileList.add(name);
            }else{//四维相机
                JSONArray imgs = pointJson.getJSONArray("imgs");
                if(CollUtil.isEmpty(imgs)){
                    continue;
                }
                for (Object img : imgs) {
                    JSONObject imgJson = (JSONObject) img;
                    String name = imgJson.getString("name");
                    fileList.add(name);
                }
            }
        }

        //查询当前待通知计算批次文件是否存在
        List<OrigFileUploadBatch> batchList = origFileUploadBatchService.list(new LambdaQueryWrapper<OrigFileUploadBatch>().eq(OrigFileUploadBatch::getUuid, uuid));
        if(CollUtil.isNotEmpty(batchList)){
            Set<String> batchIds = batchList.stream().map(v -> v.getBatchId()).collect(Collectors.toSet());
            List<OrigFileUpload> origFileList = origFileUploadService.list(new LambdaQueryWrapper<OrigFileUpload>().in(OrigFileUpload::getBatchId, batchIds));
            //过滤出不存在的
            fileList = fileList.stream().filter(v -> origFileList.stream().noneMatch(k -> k.getFileName().equals(v))).collect(Collectors.toList());
        }

        if(CollUtil.isNotEmpty(fileList)){
            String args = JSON.toJSONString(fileList);
            if(fileList.size() > 5){
                args =  fileList.size() +  "个文件";
            }
            throw new BusinessException(ErrorCode.FAILURE_CODE_4001, args);
        }
    }

    @Override
    public void sendStatus(Map<String, Object> param) {
        mqProducer.sendByWorkQueue("jmga-event-notice", param);
    }

    @Override
    public void checkLackFile(String dataSource) {

        String dataFdagePath = dataSource.concat(File.separator).concat("capture").concat(File.separator).concat("data.fdage");
        if(!FileUtil.exist(dataFdagePath)){
            throw new BusinessException(ErrorCode.FAILURE_CODE_4001.code(), ErrorCode.FAILURE_CODE_4001.formatMessage("data.fdage"));
        }
        JSONObject fdageData = null;
        try {
            fdageData = JSON.parseObject(FileUtil.readUtf8String(dataFdagePath));
        }catch (Exception e){
            throw new BusinessException(ErrorCode.FAILURE_CODE_4001.code(), ErrorCode.FAILURE_CODE_4001.formatMessage("data.fdage"));
        }

        int camType = fdageData.getJSONObject("cam").getIntValue("type");
        if(camType == 1 || camType == 2 || camType == 9 || camType == 10 || camType == 11){
            Integer shootCount = fdageData.getJSONArray("points").size();
            if(shootCount > 0){//有点位代表架站式
                String parametersPath = dataSource.concat(File.separator).concat("capture").concat(File.separator).concat("parameters.json");
                if(!FileUtil.exist(parametersPath)){
                    throw new BusinessException(ErrorCode.FAILURE_CODE_4001.code(), ErrorCode.FAILURE_CODE_4001.formatMessage("parameters.json"));
                }

                String sfmDataBinPath = dataSource.concat(File.separator).concat("capture").concat(File.separator).concat("sfm_data.bin");
                if(!FileUtil.exist(sfmDataBinPath)){
                    throw new BusinessException(ErrorCode.FAILURE_CODE_4001.code(), ErrorCode.FAILURE_CODE_4001.formatMessage("sfm_data.bin"));
                }

                String upName = camType > 2 ? "Up.txt" : "Up.xml";
                String upTxtPath = dataSource.concat(File.separator).concat("capture").concat(File.separator).concat(upName);
                if(!FileUtil.exist(upTxtPath)){
                    throw new BusinessException(ErrorCode.FAILURE_CODE_4001.code(), ErrorCode.FAILURE_CODE_4001.formatMessage(upName));
                }
            }
        }
        if(camType == 5){
            String parametersPath = dataSource.concat(File.separator).concat("capture").concat(File.separator).concat("images/parameters.json");
            if(!FileUtil.exist(parametersPath)){
                throw new BusinessException(ErrorCode.FAILURE_CODE_4001.code(), ErrorCode.FAILURE_CODE_4001.formatMessage("parameters.json"));
            }

            String sfmDataBinPath = dataSource.concat(File.separator).concat("capture").concat(File.separator).concat("result/reconstruction/sfm_data.bin");
            if(!FileUtil.exist(sfmDataBinPath)){
                throw new BusinessException(ErrorCode.FAILURE_CODE_4001.code(), ErrorCode.FAILURE_CODE_4001.formatMessage("sfm_data.bin"));
            }
        }

        //从data.fdage中获取需要上传的文件列表
        JSONArray points = fdageData.getJSONArray("points");
        if(CollUtil.isEmpty(points)){
            return;
        }

        for (Object point : points) {
            JSONObject pointJson = (JSONObject) point;
            if(camType == 5){//圆周率相机
                String name = pointJson.getString("name");
                this.checkFileAvailable(dataSource, name);
            }else{//四维相机
                JSONArray imgs = pointJson.getJSONArray("imgs");
                for (Object img : imgs) {
                    JSONObject imgJson = (JSONObject) img;
                    String name = imgJson.getString("name");
                    this.checkFileAvailable(dataSource, name);
                }
            }
        }
    }

    private void checkFileAvailable(String dataSource, String fileName){
        List<File> fileList = FileUtil.loopFiles(dataSource.concat(File.separator).concat("capture"));
        if(CollUtil.isEmpty(fileList)){
            throw new BusinessException(ErrorCode.FAILURE_CODE_4001.code(), ErrorCode.FAILURE_CODE_4001.formatMessage(fileName));
        }
        for (File file : fileList) {
            if(file.getName().equals(fileName) && file.length() > 0){
                return;
            }
        }
        throw new BusinessException(ErrorCode.FAILURE_CODE_4001.code(), ErrorCode.FAILURE_CODE_4001.formatMessage(fileName));
    }

    @Override
    public void uploadSceneOrig(UploadSceneOrigParamVo param) throws Exception {

        User user = userService.getById(param.getUserId());
        if(Objects.isNull(user)){
            throw new BusinessException(ErrorCode.USER_NOT_EXIST);
        }

        String uuid = UUID.randomUUID().toString();
        String zipName = uuid + ".zip";
        String zipDir = "/oss/4dkankan/bd_server/";
        fYunFileService.downloadFileByCommand(zipDir + zipName, param.getFilePath());
//        fYunFileService.deleteFile(param.getFilePath());

        //解压缩
//        String unzipCmd = "unzip -P " + ZipConstant.zipPassword + " " + zipDir +  zipName + " -d " + zipDir + uuid;
//        CmdUtils.callLineSh(unzipCmd);
        ZipFile zipFile = new ZipFile(new File(zipDir +  zipName));
        if (zipFile.isEncrypted()) {
            zipFile.setPassword(ZipConstant.zipPassword);
            zipFile.extractAll(zipDir + uuid);
        } else {
            ZipUtil.unzip(zipDir +  zipName, zipDir + uuid);
        }
//        FileUtil.del(zipDir +  zipName);

        //去读data.fdage
        File dataFdageFile = FileUtil.loopFiles(zipDir + uuid).stream().filter(v -> v.getName().equals("data.fdage")).findFirst().get();
        if(dataFdageFile == null){
            throw new BusinessException(ErrorCode.FAILURE_CODE_4002.code(), ErrorCode.FAILURE_CODE_4002.formatMessage("data.fdage"));
        }
        JSONObject dataFdageObj = JSONObject.parseObject(FileUtil.readUtf8String(dataFdageFile));
        int camType = dataFdageObj.getJSONObject("cam").getIntValue("type");
        String snCode = dataFdageObj.getJSONObject("cam").getString("uuid");
        String uuidTime = dataFdageObj.getString("uuidtime");
        String uniCode = snCode + "_" + uuidTime;
        String fileId = sceneFileBuildService.getFileId(snCode, uniCode);
        String homePath = "/oss/4dkankan/" + ConstantFilePath.OSS_PREFIX + snCode + "/" + fileId + "/" + uniCode;
        FileUtil.mkdir(homePath);
        String cpCmd = "cp -p -r " + zipDir + uuid + "/*/* " + homePath;
        CmdUtils.callLineSh(cpCmd);
//        FileUtil.del(zipDir + uuid);

        String params = snCode + "#" + fileId + "#" + uniCode;
        String encode = Base64.encode(RSAEncrypt.encrypt(RSAEncrypt.loadPublicKeyByStr(RSAEncrypt.loadPublicKeyByFile()), params.getBytes(StandardCharsets.UTF_8)));
        if(camType == 9 || camType == 10 || camType == 11){
            sceneFileBuildService.turntableUploadSuccess(encode, user);
        }else{
            sceneFileBuildService.uploadSuccessBuild(encode, user);
        }
    }


    @Override
    public void noticeBuildBd(String ossHost, String uuid, String dir) {
        String prefix = "http://" + ossHost + "/";
        String path = "/oss/4dkankan/bd_server/" + dir;
        List<File> files = FileUtil.loopFiles(path).stream().filter(file -> !file.getAbsolutePath().contains("backup")).collect(Collectors.toList());
        files.parallelStream().forEach(file ->{
            String url = prefix.concat(file.getAbsolutePath().substring(1).replace("/4dkankan",""));
            SendCallAlgorithmDetail detail = new SendCallAlgorithmDetail();
            detail.setFileName(FileUtil.getName(file.getAbsolutePath()));
            detail.setUuid(uuid);
            SendCallAlgorithmParam param = new SendCallAlgorithmParam();
            param.setFilepath(url);
            param.setDetails(detail);
            HttpUtil.post("http://127.0.0.1:8085/api/scene/file/sendCallAlgorithm", JSON.toJSONString(param));
        });
    }

    @Override
    public void updateSceneFailAndZipLog(String num, String failReason) {
        scenePlusService.update(new LambdaUpdateWrapper<ScenePlus>()
                .set(ScenePlus::getSceneStatus, SceneStatus.FAILD.code())
                .set(ScenePlus::getBuildFailReason, failReason)
                .eq(ScenePlus::getNum, num));

        String logBasePath = File.separator + "mnt" + File.separator + "logs" + File.separator;
        String tmpPath = logBasePath + "tmp" + File.separator;
        String logPath = tmpPath + num + "-" + Calendar.getInstance().getTimeInMillis() + File.separator;
        String zipPath = tmpPath + num + "-" + Calendar.getInstance().getTimeInMillis() + ".zip";
        FileUtil.mkdir(logPath);
        //打包服务端日志
        File[] ls = FileUtil.ls(logBasePath);
        for (File l : ls) {
            if(l.isFile() || !l.getName().startsWith("modeling-control")){
                continue;
            }
            this.copyLog(l.getAbsolutePath(), logPath, "info","sql","error");
        }
        //打包算法日志
        ScenePlus scenePlus = scenePlusService.getScenePlusByNum(num);
        ScenePlusExt scenePlusExt = scenePlusExtService.getScenePlusExtByPlusId(scenePlus.getId());
        String consoleLogPath = scenePlusExt.getDataSource() + "console.log";
        String consoleOldLogPath = scenePlusExt.getDataSource() + "console.old.log";
        if(FileUtil.exist(consoleLogPath)){
            FileUtil.copy(consoleLogPath, logPath,true);
        }
        if(FileUtil.exist(consoleOldLogPath)){
            FileUtil.copy(consoleOldLogPath, logPath,true);
        }
        //打包数据库记录
        String[] split = scenePlusExt.getDataSource().split("/");
        String uuid = split[split.length -1];
        List<OrigFileUploadBatch> batchList = origFileUploadBatchService.list(new LambdaQueryWrapper<OrigFileUploadBatch>().eq(OrigFileUploadBatch::getUuid, uuid));
        if(CollUtil.isNotEmpty(batchList)){
            EasyExcel.write(logPath.concat("origFileUploadBatch.xlsx"), OrigFileUploadBatch.class).sheet("origFileUploadBatch").doWrite(batchList);
            List<String> batchIdList = batchList.stream().map(v -> v.getBatchId()).collect(Collectors.toList());
            List<OrigFileUpload> uploadList = origFileUploadService.list(new LambdaQueryWrapper<OrigFileUpload>().in(OrigFileUpload::getBatchId, batchIdList));
            if(CollUtil.isNotEmpty(uploadList)){
                EasyExcel.write(logPath.concat("origFileUpload.xlsx"), OrigFileUpload.class).sheet("origFileUpload").doWrite(uploadList);
            }
        }
        List<SceneBuildProcessLog> processLogList = sceneBuildProcessLogService.listByNum(num);
        if(CollUtil.isNotEmpty(processLogList)){
            EasyExcel.write(logPath.concat("sceneBuildProcessLog.xlsx"), SceneBuildProcessLog.class).sheet("sceneBuildProcessLog").doWrite(processLogList);
        }
        List<BuildLog> buildLogList = buildLogService.list(new LambdaQueryWrapper<BuildLog>().eq(BuildLog::getSceneNum, num));
        if(CollUtil.isNotEmpty(buildLogList)){
            EasyExcel.write(logPath.concat("buildLog.xlsx"), BuildLog.class).sheet("buildLog").doWrite(buildLogList);
        }

        //上传日志，删除本地压缩包
        ZipUtil.zip(logPath, zipPath, false);
        String logName = "build_log/" + num + "/" + num + "-log.zip";
        if(fYunFileService.fileExist(logName)){
            fYunFileService.copyFileInBucket(logName, logName + "-bak-" + DateExtUtil.format(Calendar.getInstance().getTime(), DateExtUtil.dateStyle11));
        }
        fYunFileService.uploadFile(zipPath, "build_log/" + num + "/" + num + "-log");

    }

    private void copyLog(String serverPath, String tmpPath, String... levels){
        level: for (String level : levels) {
            String infoPath = serverPath + File.separator + level + File.separator;
            List<File> fileList = FileUtil.loopFiles(infoPath);
            if(CollUtil.isEmpty(fileList)){
                continue;
            }
            CollUtil.reverse(fileList);
            for (File file : fileList) {
                int i = fileList.indexOf(file);
                if(i > 7){
                    continue level;
                }
                FileUtil.copy(file.getAbsolutePath(), file.getAbsolutePath().replace(FileUtil.getParent(serverPath, 1), tmpPath), true);
            }
        }
    }
}