TASK:模块化；

src/main/java/com/ai/da/common/RabbitMQ/GenerateConsumer.java

@@ -5,6 +5,7 @@ import com.ai.da.common.utils.RedisUtil;
 import com.ai.da.model.dto.GenerateThroughImageTextDTO;
 import com.ai.da.model.vo.GenerateResultVO;
 import com.ai.da.model.vo.PoseTransformationVO;
+import com.ai.da.service.DesignService;
 import com.ai.da.service.GenerateService;
 import com.ai.da.service.UserLikeGroupService;
 import com.alibaba.fastjson.JSONObject;
@@ -34,6 +35,9 @@ public class GenerateConsumer {
     @Resource
     private UserLikeGroupService userLikeGroupService;
 
+    @Resource
+    private DesignService designService;
+
     @Autowired
     private RabbitMQProperties rabbitMQProperties;
 
@@ -305,6 +309,15 @@ public class GenerateConsumer {
 
     }
 
+    private void processDesignBatchResult(Message msg, Channel channel) {
+        log.info("============processDesignBatchResult listening==========");
+        long start = System.currentTimeMillis();
+
+        Map<String, Object> generateResult = JSONObject.parseObject(msg.getBody(), Map.class);
+        log.info("designBatch response : {}", generateResult);
+        designService.processDesignBatch(generateResult);
+    }
+
     @RabbitListener(queues = "#{rabbitMQProperties.queues.generate}")
     @RabbitHandler
     public void generateConsumer1(Message msg, Channel channel) {
@@ -382,4 +395,9 @@ public class GenerateConsumer {
     public void getPoseTransformationResult(Message msg, Channel channel) {
         processPoseTransformResult(msg, channel);
     }
+    @RabbitListener(queues = "#{rabbitMQProperties.queues.designBatch}")
+    @RabbitHandler
+    public void getDesignBatchResult(Message msg, Channel channel) {
+        processDesignBatchResult(msg, channel);
+    }
 }

src/main/java/com/ai/da/common/RabbitMQ/RabbitMQProperties.java

@@ -21,6 +21,7 @@ public class RabbitMQProperties {
         private String toProductImageResult;
         private String relightResult;
         private String poseTransform;
+        private String designBatch;
     }
 
     @Data