feat(adopt): 增加消息转换时的系统类型设置

lvqi
1 parent 47872e7b
Showing 2 changed files with 9 additions and 0 deletions
tax-adopt/src/main/java/com/diligrp/tax/adopt/model/ReceiveMessageCO.java
tax-adopt/src/main/java/com/diligrp/tax/adopt/service/impl/AdoptMessageServiceImpl.java
 package com.diligrp.tax.adopt.model;
+import com.diligrp.tax.central.type.SystemType;
 import lombok.Data;
 import java.util.Map;
@@ -40,4 +41,6 @@ public class ReceiveMessageCO {
      * 消息信息
      */
     private Map<String, Object> msgBody;
+
+    private SystemType systemType;
 }
@@ -4,7 +4,9 @@ import com.diligrp.tax.adopt.model.ReceiveMessageCO;
 import com.diligrp.tax.adopt.service.AdoptMessageService;
 import com.diligrp.tax.central.exception.TaxAgentServiceException;
 import com.diligrp.tax.central.model.PipelineBusinessKeyword;
+import com.diligrp.tax.central.model.TenantPipeline;
 import com.diligrp.tax.central.service.ITaxPipelineBusinessKeywordService;
+import com.diligrp.tax.central.service.ITaxTenantService;
 import com.diligrp.tax.central.utils.AdoptUtils;
 import com.diligrp.tax.central.utils.JsonUtils;
 import lombok.AllArgsConstructor;
@@ -15,6 +17,7 @@ import org.springframework.web.multipart.MultipartFile;
 import java.util.List;
 import java.util.Map;
+import java.util.Optional;
 import static com.diligrp.tax.central.queue.TaxAutoPush.NORMAL_EXCHANGE;
 import static com.diligrp.tax.central.queue.TaxAutoPush.NORMAL_ROUTING;
@@ -28,6 +31,7 @@ public class AdoptMessageServiceImpl implements AdoptMessageService {
     private final ITaxPipelineBusinessKeywordService taxPipelineBusinessKeywordService;
     private final RabbitTemplate rabbitTemplate;
+    private final ITaxTenantService taxTenantService;
     @Override
     public void convertMessage(ReceiveMessageCO receiveMessageCO) {
@@ -41,6 +45,8 @@ public class AdoptMessageServiceImpl implements AdoptMessageService {
         String json = JsonUtils.toJsonString(msgBody);
         Map<String, Object> map = AdoptUtils.handleAdopt(pipelineBusinessKeywords, json);
         receiveMessageCO.setMsgBody(map);
+        Optional<TenantPipeline> byTenantAndPipelineCode = taxTenantService.findByTenantAndPipelineCode(receiveMessageCO.getGroup(), receiveMessageCO.getEntity(), receiveMessageCO.getPipelineCode());
+        byTenantAndPipelineCode.ifPresent(tenantPipeline -> receiveMessageCO.setSystemType(tenantPipeline.getSystemType()));
         // 发送mq
         String messageJson = JsonUtils.toJsonString(receiveMessageCO);
         rabbitTemplate.convertAndSend(NORMAL_EXCHANGE, NORMAL_ROUTING, messageJson);