金蝶push

zhangmeiyang
1 parent b287db95
Showing 16 changed files with 296 additions and 36 deletions
ereport-start/build.gradle
ereport-start/src/main/java/com/diligrp/etrade/boss/EtradeBossBootstrap.java
etrade-core/src/main/java/com/diligrp/etrade/core/service/ApplicationContextHolder.java
etrade-thirdparty/build.gradle
etrade-thirdparty/src/main/java/com/diligrp/etrade/thirdparty/ThirdPartyConfiguration.java
etrade-thirdparty/src/main/java/com/diligrp/etrade/thirdparty/co/BaseCo.java
etrade-thirdparty/src/main/java/com/diligrp/etrade/thirdparty/co/ErrorMessageCo.java
etrade-thirdparty/src/main/java/com/diligrp/etrade/thirdparty/co/TestCo.java
etrade-rpc/src/main/java/com/diligrp/etrade/rpc/executor/ZrAutoPush.java → etrade-thirdparty/src/main/java/com/diligrp/etrade/thirdparty/exec/ZrAutoPush.java
etrade-thirdparty/src/main/java/com/diligrp/etrade/thirdparty/exec/ZrHolder.java
etrade-rpc/src/main/java/com/diligrp/etrade/rpc/executor/ZrReceiver.java → etrade-thirdparty/src/main/java/com/diligrp/etrade/thirdparty/exec/ZrReceiver.java
etrade-thirdparty/src/main/java/com/diligrp/etrade/thirdparty/handler/AbstractBusinessHandler.java
etrade-thirdparty/src/main/java/com/diligrp/etrade/thirdparty/handler/ErrorBusinessHandler.java
etrade-thirdparty/src/main/java/com/diligrp/etrade/thirdparty/handler/TestBusinessHandler.java
etrade-thirdparty/src/main/java/com/diligrp/etrade/thirdparty/type/BusinessEnum.java
settings.gradle
@@ -9,6 +9,7 @@ bootJar {
 dependencies {
     implementation project(':ereport-core')
+    implementation project(':etrade-thirdparty')
     implementation 'com.alibaba.cloud:spring-cloud-starter-alibaba-nacos-discovery:2022.0.0.0-RC2'
     implementation 'com.alibaba.cloud:spring-cloud-starter-alibaba-nacos-config:2022.0.0.0-RC2'
 }
@@ -4,6 +4,7 @@ import com.diligrp.etrade.core.CoreConfiguration;
 import com.diligrp.etrade.report.ReportConfiguration;
 import com.diligrp.etrade.rpc.RpcConfiguration;
 import com.diligrp.etrade.shared.SharedConfiguration;
+import com.diligrp.etrade.thirdparty.ThirdPartyConfiguration;
 import org.springframework.boot.SpringApplication;
 import org.springframework.boot.SpringBootConfiguration;
 import org.springframework.boot.autoconfigure.EnableAutoConfiguration;
@@ -13,7 +14,7 @@ import org.springframework.context.annotation.Import;
 @SpringBootConfiguration
 @EnableAutoConfiguration(exclude = { DataSourceAutoConfiguration.class })
-@Import({BossConfiguration.class, SharedConfiguration.class, RpcConfiguration.class, CoreConfiguration.class, ReportConfiguration.class})
+@Import({BossConfiguration.class, SharedConfiguration.class, RpcConfiguration.class, CoreConfiguration.class, ReportConfiguration.class, ThirdPartyConfiguration.class})
 @EnableDiscoveryClient
 public class EtradeBossBootstrap {
     public static void main(String[] args) {
+package com.diligrp.etrade.core.service;
+
+import org.springframework.context.ApplicationContext;
+
+/**
+ * ApplicationContext 持有者
+ */
+public enum ApplicationContextHolder {
+
+    INSTANCE;
+
+    private ApplicationContext applicationContext;
+
+    /**
+     * 赋值
+     *
+     * @param applicationContext
+     */
+    public void setApplicationContext(ApplicationContext applicationContext) {
+        this.applicationContext = applicationContext;
+    }
+
+    /**
+     * 获取bean
+     *
+     * @param cls
+     * @return
+     */
+    public <T> T getBean(Class<T> cls) {
+        return this.applicationContext.getBean(cls);
+    }
+}
+group = 'com.diligrp'
+archivesBaseName = 'etrade-thirdparty'
+
+dependencies {
+    api project(':etrade-shared')
+}
+package com.diligrp.etrade.thirdparty;
+
+import org.springframework.context.annotation.ComponentScan;
+import org.springframework.context.annotation.Configuration;
+
+/**
+ * @Author: zhangmeiyang
+ * @CreateTime: 2024-09-10 11:09
+ * @Version: todo
+ */
+@Configuration
+@ComponentScan("com.diligrp.etrade.thirdparty")
+public class ThirdPartyConfiguration {
+}
+package com.diligrp.etrade.thirdparty.co;
+
+import com.diligrp.etrade.thirdparty.type.BusinessEnum;
+
+/**
+ * @Author: zhangmeiyang
+ * @CreateTime: 2024-09-10 11:30
+ * @Version: todo
+ */
+public class BaseCo {
+    private String errorMessage;
+    private Integer retryCount;
+    private BusinessEnum businessEnum;
+
+    public BaseCo() {
+        this.retryCount = 0;
+    }
+
+    public BusinessEnum getBusinessEnum() {
+        return businessEnum;
+    }
+
+    public void setBusinessEnum(BusinessEnum businessEnum) {
+        this.businessEnum = businessEnum;
+    }
+
+    public Integer getRetryCount() {
+        return retryCount;
+    }
+
+    public void setRetryCount(Integer retryCount) {
+        this.retryCount = retryCount;
+    }
+
+    public String getErrorMessage() {
+        return errorMessage;
+    }
+
+    public void setErrorMessage(String errorMessage) {
+        this.errorMessage = errorMessage;
+    }
+}
+package com.diligrp.etrade.thirdparty.co;
+
+/**
+ * @Author: zhangmeiyang
+ * @CreateTime: 2024-09-10 11:32
+ * @Version: todo
+ */
+public class ErrorMessageCo extends BaseCo{
+
+}
+package com.diligrp.etrade.thirdparty.co;
+
+/**
+ * @Author: zhangmeiyang
+ * @CreateTime: 2024-09-10 15:06
+ * @Version: todo
+ */
+public class TestCo extends BaseCo{
+    private String name;
+
+    public String getName() {
+        return name;
+    }
+
+    public void setName(String name) {
+        this.name = name;
+    }
+}
-package com.diligrp.etrade.rpc.executor;
+package com.diligrp.etrade.thirdparty.exec;
-import jakarta.annotation.Resource;
 import org.springframework.amqp.core.*;
-import org.springframework.amqp.rabbit.annotation.QueueBinding;
-import org.springframework.amqp.rabbit.annotation.RabbitListener;
-import org.springframework.amqp.rabbit.core.RabbitTemplate;
 import org.springframework.beans.factory.annotation.Qualifier;
 import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.Configuration;
-import org.springframework.stereotype.Component;
-import java.nio.charset.StandardCharsets;
 import java.util.HashMap;
 import java.util.Map;
+package com.diligrp.etrade.thirdparty.exec;
+
+import com.diligrp.etrade.thirdparty.handler.AbstractBusinessHandler;
+import com.diligrp.etrade.thirdparty.type.BusinessEnum;
+import jakarta.annotation.Resource;
+import org.springframework.beans.factory.DisposableBean;
+import org.springframework.beans.factory.InitializingBean;
+import org.springframework.stereotype.Component;
+
+import java.util.List;
+import java.util.Map;
+import java.util.concurrent.ConcurrentHashMap;
+
+/**
+ * @Author: zhangmeiyang
+ * @CreateTime: 2024-09-10 10:49
+ * @Version: todo
+ */
+@Component
+public class ZrHolder implements InitializingBean, DisposableBean {
+
+    public final static Map<BusinessEnum, AbstractBusinessHandler> CONTEXT = new ConcurrentHashMap<>();
+
+    @Resource
+    private List<AbstractBusinessHandler> abstractBusinessHandlerList;
+
+    @Override
+    public void afterPropertiesSet() throws Exception {
+        abstractBusinessHandlerList.forEach(e->{
+            CONTEXT.put(e.getType(), e);
+        });
+    }
+
+    @Override
+    public void destroy() throws Exception {
+
+    }
+}
-package com.diligrp.etrade.rpc.executor;
+package com.diligrp.etrade.thirdparty.exec;
 import com.diligrp.etrade.core.util.JsonUtils;
+import com.diligrp.etrade.thirdparty.co.BaseCo;
+import com.diligrp.etrade.thirdparty.type.BusinessEnum;
 import com.rabbitmq.client.Channel;
 import jakarta.annotation.Resource;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
 import org.springframework.amqp.core.Message;
 import org.springframework.amqp.core.MessageBuilder;
 import org.springframework.amqp.rabbit.annotation.Exchange;
@@ -14,9 +18,8 @@ import org.springframework.stereotype.Component;
 import java.io.IOException;
 import java.nio.charset.StandardCharsets;
-import java.util.Map;
-import static com.diligrp.etrade.rpc.executor.ZrAutoPush.*;
+import static com.diligrp.etrade.thirdparty.exec.ZrAutoPush.*;
 /**
  * @Author: zhangmeiyang
@@ -25,47 +28,49 @@ import static com.diligrp.etrade.rpc.executor.ZrAutoPush.*;
  */
 @Component
 public class ZrReceiver {
-
+    private final static Logger LOG = LoggerFactory.getLogger(ZrReceiver.class);
+    private static final Long DELAY = 5000L;
+    private static final String DELAY_PREFIX = "x-delay";
     @Resource
     private RabbitTemplate rabbitTemplate;
     @RabbitListener(bindings = @QueueBinding(value = @Queue(value = ZR_NORMAL_QUEUE, autoDelete = "false"), exchange = @Exchange(value = ZR_NORMAL_EXCHANGE), key = ZR_NORMAL_ROUTING), ackMode = "MANUAL")
     public void receiveMessage(Channel channel, Message message) throws IOException {
+        var content = new String(message.getBody(), StandardCharsets.UTF_8);
+        BaseCo baseCo = JsonUtils.fromJsonString(content, BaseCo.class);
         try {
-            var temp = 1/0;
+            ZrHolder.CONTEXT.get(baseCo.getBusinessEnum()).handle(content);
             channel.basicAck(message.getMessageProperties().getDeliveryTag(), false);
-        }catch (Exception e){
-            StackTraceElement[] stackTrace = e.getStackTrace();
-            String jsonString = JsonUtils.toJsonString(stackTrace);
-            System.out.println(jsonString);
-            sendDelayMsg(3000L,message.getBody());
+        } catch (Exception e) {
             channel.basicAck(message.getMessageProperties().getDeliveryTag(), false);
+            var t = baseCo.getBusinessEnum().getBusinessParam(content);
+            t.setRetryCount(baseCo.getRetryCount() + 1);
+            sendDelayMsg(DELAY, JsonUtils.toJsonString(t));
         }
     }
-    @RabbitListener(queues = ZR_DEAD_QUEUE)
+    @RabbitListener(queues = ZR_DEAD_QUEUE, ackMode = "MANUAL")
     public void handleCustomDelayMsg(Channel channel, Message message) throws IOException {
-        Map<String, Object> headersMap = message.getMessageProperties().getHeaders();
-        int retryCount = (int) headersMap.get("x-retry-count");
+        var content = new String(message.getBody(), StandardCharsets.UTF_8);
+        BaseCo baseCo = JsonUtils.fromJsonString(content, BaseCo.class);
         try {
-            if (retryCount > 3){
-                //db
-            }else {
-
-            }
+            ZrHolder.CONTEXT.get(baseCo.getBusinessEnum()).handle(content);
             channel.basicAck(message.getMessageProperties().getDeliveryTag(), false);
-        }catch (Exception e){
-            retryCount++;
-            headersMap.put("x-retry-count", retryCount);
-            channel.basicNack(message.getMessageProperties().getDeliveryTag(), false, true);
+        } catch (Exception e) {
+            channel.basicAck(message.getMessageProperties().getDeliveryTag(), false);
+            var t = baseCo.getBusinessEnum().getBusinessParam(content);
+            t.setRetryCount(baseCo.getRetryCount() + 1);
+            if (t.getRetryCount() > 3) {
+                t.setErrorMessage(e.fillInStackTrace().getMessage());
+                t.setBusinessEnum(BusinessEnum.ERROR);
+            }
+            sendDelayMsg(DELAY, JsonUtils.toJsonString(t));
         }
     }
-    public void sendDelayMsg(Long currentMills,byte[] bytes) {
-        Message msg = MessageBuilder.withBody(bytes)
-                .setHeader("x-delay", currentMills)
-                .setHeader("x-retry-count", 0)
-                .build();
+    public void sendDelayMsg(Long currentMills, String message) {
+        byte[] bytes = message.getBytes(StandardCharsets.UTF_8);
+        Message msg = MessageBuilder.withBody(bytes).setHeader(DELAY_PREFIX, currentMills).build();
         rabbitTemplate.convertAndSend(ZR_DEAD_EXCHANGE, ZR_DEAD_QUEUE, msg);
     }
 }
+package com.diligrp.etrade.thirdparty.handler;
+
+import com.diligrp.etrade.thirdparty.type.BusinessEnum;
+
+public abstract class AbstractBusinessHandler {
+
+   public abstract void handle(String json);
+
+   public abstract BusinessEnum getType();
+}
+package com.diligrp.etrade.thirdparty.handler;
+
+import com.diligrp.etrade.core.util.JsonUtils;
+import com.diligrp.etrade.thirdparty.co.ErrorMessageCo;
+import com.diligrp.etrade.thirdparty.type.BusinessEnum;
+import org.springframework.stereotype.Component;
+
+/**
+ * @Author: zhangmeiyang
+ * @CreateTime: 2024-09-10 10:56
+ * @Version: todo
+ */
+@Component
+public class ErrorBusinessHandler extends AbstractBusinessHandler {
+    @Override
+    public void handle(String json) {
+        ErrorMessageCo co = JsonUtils.fromJsonString(json, ErrorMessageCo.class);
+        System.out.println(co.getErrorMessage());
+    }
+
+    @Override
+    public BusinessEnum getType() {
+        return BusinessEnum.ERROR;
+    }
+}
+package com.diligrp.etrade.thirdparty.handler;
+
+import com.diligrp.etrade.core.util.JsonUtils;
+import com.diligrp.etrade.thirdparty.co.TestCo;
+import com.diligrp.etrade.thirdparty.type.BusinessEnum;
+import org.springframework.stereotype.Component;
+
+/**
+ * @Author: zhangmeiyang
+ * @CreateTime: 2024-09-10 15:03
+ * @Version: todo
+ */
+@Component
+public class TestBusinessHandler extends AbstractBusinessHandler{
+    @Override
+    public void handle(String json) {
+        TestCo testCo = JsonUtils.fromJsonString(json, TestCo.class);
+        var s =  1/0;
+    }
+
+    @Override
+    public BusinessEnum getType() {
+        return BusinessEnum.TEST;
+    }
+}
+package com.diligrp.etrade.thirdparty.type;
+
+import com.diligrp.etrade.core.util.JsonUtils;
+import com.diligrp.etrade.thirdparty.co.BaseCo;
+import com.diligrp.etrade.thirdparty.co.ErrorMessageCo;
+import com.diligrp.etrade.thirdparty.co.TestCo;
+
+/**
+ * @Author: zhangmeiyang
+ * @CreateTime: 2024-09-10 10:53
+ * @Version: todo
+ */
+public enum BusinessEnum implements BusinessKlass{
+
+    ERROR("ERROR"){
+        @Override
+        public ErrorMessageCo getBusinessParam(String content) {
+            return JsonUtils.fromJsonString(content,ErrorMessageCo.class);
+        }
+    },
+
+    TEST("TEST"){
+        @Override
+        public TestCo getBusinessParam(String content) {
+            return JsonUtils.fromJsonString(content,TestCo.class);
+        }
+    };
+
+    public final String code;
+
+    BusinessEnum(String code) {
+        this.code = code;
+    }
+}
+
+interface BusinessKlass<T extends BaseCo> {
+    T getBusinessParam(String content);
+
+}
 rootProject.name = 'dili-ereport'
 include 'etrade-core'
+include 'etrade-thirdparty'
 include 'etrade-rpc'
 include 'etrade-shared'
 include 'ereport-start'
 include 'ereport-core'
-