111

src/main/java/com/fdkankan/modeling/receiver/RabbitMqListener.java

@@ -12,10 +12,7 @@ import com.fdkankan.modeling.constants.SysConstants;
 import com.fdkankan.modeling.entity.BuildLog;
 import com.fdkankan.modeling.exception.BuildException;
 import com.fdkankan.modeling.handler.LaserSceneObjGenerateHandler;
-import com.fdkankan.modeling.service.IBuildLogService;
-import com.fdkankan.modeling.service.IBuildService;
-import com.fdkankan.modeling.service.IMqEcsService;
-import com.fdkankan.modeling.service.ISceneBuildProcessLogService;
+import com.fdkankan.modeling.service.*;
 import com.fdkankan.modeling.service.impl.MqEcsServiceImpl;
 import com.fdkankan.rabbitmq.bean.BuildSceneCallMessage;
 import com.fdkankan.rabbitmq.bean.BuildSceneProcessLogMessage;
@@ -86,6 +83,8 @@ public class RabbitMqListener {
 
     @Autowired
     private IBuildService buildService;
+    @Autowired
+    private IReverseE57Service reverseE57Service;
 
     /**
      * 场景计算
@@ -131,7 +130,11 @@ public class RabbitMqListener {
         buildLog.setMessageId(messageId);
         buildLog.setContent(msg);
 
-        this.process(buildSceneMessage,buildLog);
+        if(StrUtil.isNotEmpty(buildSceneMessage.getBizType()) && buildSceneMessage.getBizType().equals("reverseE57")){
+            reverseE57Service.build(buildSceneMessage,buildLog);
+        }else{
+            this.process(buildSceneMessage,buildLog);
+        }
         log.info("场景计算结束,队列名:{},id:{}", queueModelingCall, messageId);
 
         //计算完毕，将当前系统构建状态改为false

src/main/java/com/fdkankan/modeling/service/IReverseE57Service.java

@@ -0,0 +1,10 @@
+package com.fdkankan.modeling.service;
+
+import com.fdkankan.modeling.entity.BuildLog;
+import com.fdkankan.rabbitmq.bean.BuildSceneCallMessage;
+
+public interface IReverseE57Service {
+
+    void build(BuildSceneCallMessage message, BuildLog buildLog);
+
+}

src/main/java/com/fdkankan/modeling/service/impl/ReverseE57ServiceImpl.java

@@ -0,0 +1,164 @@
+package com.fdkankan.modeling.service.impl;
+
+import cn.hutool.core.util.StrUtil;
+import com.alibaba.fastjson.JSONObject;
+import com.fdkankan.common.constant.ModelingBuildStatus;
+import com.fdkankan.common.util.FileUtils;
+import com.fdkankan.model.utils.ComputerUtil;
+import com.fdkankan.modeling.bean.BuildSceneResultBean;
+import com.fdkankan.modeling.constants.SysConstants;
+import com.fdkankan.modeling.entity.BuildLog;
+import com.fdkankan.modeling.exception.BuildException;
+import com.fdkankan.modeling.service.IBuildLogService;
+import com.fdkankan.modeling.service.IReverseE57Service;
+import com.fdkankan.rabbitmq.bean.BuildSceneCallMessage;
+import com.fdkankan.rabbitmq.bean.BuildSceneProcessLogMessage;
+import com.fdkankan.rabbitmq.util.RabbitMqProducer;
+import lombok.extern.slf4j.Slf4j;
+import org.apache.commons.lang3.StringUtils;
+import org.apache.commons.lang3.time.StopWatch;
+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 javax.annotation.Resource;
+import java.io.File;
+import java.util.Date;
+import java.util.HashMap;
+import java.util.Map;
+import java.util.concurrent.Future;
+import java.util.concurrent.TimeUnit;
+import java.util.concurrent.TimeoutException;
+
+@Slf4j
+@Service
+public class ReverseE57ServiceImpl implements IReverseE57Service {
+
+    @Value("${queue.modeling.modeling-call}")
+    private String queueModelingCall;
+    @Value("${queue.modeling.modeling-process-log}")
+    private String queueModelingProcessLog;
+    @Value("${queue.modeling.modeling-post}")
+    private String queueModelingPost;
+    @Autowired
+    private IBuildLogService buildLogService;
+    @Resource
+    private RabbitMqProducer rabbitMqProducer;
+
+    @Override
+    public void build(BuildSceneCallMessage message, BuildLog buildLog) {
+
+        //开始计时
+        StopWatch watch = new StopWatch();
+        watch.start();
+
+        String num = message.getSceneNum();
+
+        //发送记录计算状态为计算中
+        rabbitMqProducer.sendByWorkQueue(queueModelingProcessLog,
+            BuildSceneProcessLogMessage.builder().num(message.getSceneNum()).buildStatus(ModelingBuildStatus.CALCULATING.code()).queueName(queueModelingCall).build());
+
+        final BuildSceneResultBean buildSceneResult = new BuildSceneResultBean();
+        String resultQueueName = StrUtil.isNotBlank(message.getResultReceiverMqName()) ?
+                message.getResultReceiverMqName() : queueModelingPost;
+        buildSceneResult.setResultQueueName(resultQueueName);
+        ModelingBuildStatus status = null;
+        Future<ModelingBuildStatus> future = null;
+        try {
+            Map<String,Object> buildContext = new HashMap<>();
+
+            buildLog.setHostName(SysConstants.hostName);
+            buildLog.setQueueName(queueModelingCall);
+            buildLog.setSceneNum(message.getSceneNum());
+            buildLog.setDataSource(message.getPath());
+            buildLog.setCreateTime(new Date());
+            buildLog.setUpdateTime(new Date());
+            buildLog.setBuildType(0);
+            buildLog.setStatus(0);
+            buildLogService.save(buildLog);
+
+            future = SysConstants.executorService.submit(() -> {
+                try {
+                    return buildScene(buildContext,message, buildSceneResult,buildLog);
+                } catch (Exception e) {
+                    log.error("服务实例：{} 构建异常：", SysConstants.hostName, e);
+                    e.printStackTrace();
+                    return ModelingBuildStatus.FAILED;
+                }
+            });
+            status = future.get(SysConstants.modelTimeOut, TimeUnit.HOURS);
+
+            //结束计时
+            watch.stop();
+            buildLog.setDuration(watch.getTime(TimeUnit.SECONDS));
+        } catch (TimeoutException ex) {
+            log.error("服务实例：{} 构建异常：",SysConstants.hostName,ex);
+            status = ModelingBuildStatus.OVERTIME;
+            log.info("释放线程");
+            //场景计算超时，需要中断当前执行任务，释放线程池中的任务线程，否则下个场景计算获取不到线程进行计算
+            future.cancel(true);
+        } catch (BuildException e){
+            status = e.getBuildStatus();
+        } catch(Exception e) {
+            log.error("服务实例：{} 构建异常：", SysConstants.hostName, e);
+            if (e.getCause() instanceof BuildException) {
+                status = ((BuildException) e.getCause()).getBuildStatus();
+                buildLog.setStatus(((BuildException) e.getCause()).getBuildStatus().code());
+            } else {
+                status = ModelingBuildStatus.FAILED;
+            }
+        }
+        buildLog.setStatus(status.code());
+        buildSceneResult.setCameraType(message.getCameraType());
+        buildSceneResult.setBuildStatus(status);
+        //计算后处理
+        ModelingBuildStatus buildStatus = null;
+        try {
+            buildStatus = buildSceneResult.getBuildStatus();
+            log.info("服务{} 计算结束:{},计算状态：{}", SysConstants.hostName, buildLog.getSceneNum(),buildStatus.message());
+            buildLog.setUpdateTime(new Date());
+            buildLog.setResultQuequeName(buildSceneResult.getResultQueueName());
+            buildLogService.updateById(buildLog);
+        } catch (Exception e) {
+            log.error("计算后业务处理出错！", e);
+        }finally {
+            buildSceneResult.setDuration(buildLog.getDuration());
+            rabbitMqProducer.sendByWorkQueue(queueModelingProcessLog,
+                    BuildSceneProcessLogMessage.builder().num(message.getSceneNum()).buildStatus(buildStatus.code()).queueName(queueModelingCall).build());
+        }
+
+    }
+
+    private ModelingBuildStatus buildScene(Map<String,Object> buildContext,BuildSceneCallMessage message, BuildSceneResultBean buildSceneResult,BuildLog logRecord) throws Exception{
+        //如果mq生产者在消息体中设置了结算结果mq队列名，就发到这个队列，否则就发送到默认队列
+        String path = message.getPath();
+        String num = message.getSceneNum();
+        //不同的相机不同的方法
+        String cameraType = message.getCameraType();
+        //0表示有4k图，1表示没有
+        //判断调用V2还是V3版本的算法
+        String buildType = ObjectUtils.isEmpty(message.getBuildType()) ? "V2" : message.getBuildType();
+
+        buildSceneResult.setCameraType(cameraType);
+        buildSceneResult.setNum(num);
+        buildSceneResult.setPath(path);
+
+
+        String splitType = null;
+        String skyboxType = null;
+
+        ComputerUtil.createProjectAndDataFile(path,num, splitType, skyboxType,null,null);
+        //计算模型并返回需要上传oss的文件集合
+        ComputerUtil.computer(num, path, buildType);
+
+        // 检测计算结果文件是否有生成
+        String resultsPath = path + File.separator + "results" + File.separator;
+        if (!new File(resultsPath + "upload.json").exists()) {
+            log.error("未检测到计算结果文件：upload.json");
+            return ModelingBuildStatus.FAILED;
+        }
+        log.info("八目上完oss结束修改数据："+num);
+        return ModelingBuildStatus.SUCCESS;
+    }
+}