add EventTraceContext

xlorne · xlorne · commit d494eba7c0ec · 2024-11-08T21:39:23.000+08:00
diff --git a/example/example-application/src/main/java/com/codingapi/example/event/AEvent.java b/example/example-application/src/main/java/com/codingapi/example/event/AEvent.java
@@ -0,0 +1,8 @@
+package com.codingapi.example.event;
+
+import com.codingapi.springboot.framework.event.IEvent;
+
+public class AEvent implements IEvent {
+
+
+}
diff --git a/example/example-application/src/main/java/com/codingapi/example/event/BEvent.java b/example/example-application/src/main/java/com/codingapi/example/event/BEvent.java
@@ -0,0 +1,8 @@
+package com.codingapi.example.event;
+
+import com.codingapi.springboot.framework.event.IEvent;
+
+public class BEvent implements IEvent {
+
+
+}
diff --git a/example/example-application/src/main/java/com/codingapi/example/event/CEvent.java b/example/example-application/src/main/java/com/codingapi/example/event/CEvent.java
@@ -0,0 +1,7 @@
+package com.codingapi.example.event;
+
+import com.codingapi.springboot.framework.event.IEvent;
+
+public class CEvent implements IEvent {
+
+}
diff --git a/example/example-application/src/main/java/com/codingapi/example/handler/AHandler.java b/example/example-application/src/main/java/com/codingapi/example/handler/AHandler.java
@@ -0,0 +1,21 @@
+package com.codingapi.example.handler;
+
+import com.codingapi.example.event.AEvent;
+import com.codingapi.example.event.BEvent;
+import com.codingapi.springboot.framework.event.EventPusher;
+import com.codingapi.springboot.framework.event.EventTraceContext;
+import com.codingapi.springboot.framework.event.IHandler;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.stereotype.Service;
+
+@Slf4j
+@Service
+public class AHandler implements IHandler<AEvent> {
+
+    @Override
+    public void handler(AEvent event) {
+        log.info("a event:{},traceId:{}",event, EventTraceContext.getInstance().getListenerKey());
+
+        EventPusher.push(new BEvent());
+    }
+}
diff --git a/example/example-application/src/main/java/com/codingapi/example/handler/BHandler.java b/example/example-application/src/main/java/com/codingapi/example/handler/BHandler.java
@@ -0,0 +1,21 @@
+package com.codingapi.example.handler;
+
+import com.codingapi.example.event.BEvent;
+import com.codingapi.example.event.CEvent;
+import com.codingapi.springboot.framework.event.EventPusher;
+import com.codingapi.springboot.framework.event.EventTraceContext;
+import com.codingapi.springboot.framework.event.IHandler;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.stereotype.Service;
+
+@Slf4j
+@Service
+public class BHandler implements IHandler<BEvent> {
+
+    @Override
+    public void handler(BEvent event) {
+        log.info("b event:{},traceId:{}",event, EventTraceContext.getInstance().getListenerKey());
+
+        EventPusher.push(new CEvent());
+    }
+}
diff --git a/example/example-application/src/main/java/com/codingapi/example/handler/CHandler.java b/example/example-application/src/main/java/com/codingapi/example/handler/CHandler.java
@@ -0,0 +1,21 @@
+package com.codingapi.example.handler;
+
+import com.codingapi.example.event.AEvent;
+import com.codingapi.example.event.CEvent;
+import com.codingapi.springboot.framework.event.EventPusher;
+import com.codingapi.springboot.framework.event.EventTraceContext;
+import com.codingapi.springboot.framework.event.IHandler;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.stereotype.Service;
+
+@Slf4j
+@Service
+public class CHandler implements IHandler<CEvent> {
+
+    @Override
+    public void handler(CEvent event) {
+        log.info("c event:{},traceId:{}",event, EventTraceContext.getInstance().getListenerKey());
+
+        EventPusher.push(new AEvent());
+    }
+}
diff --git a/example/example-application/src/main/java/com/codingapi/example/handler/TestHandler.java b/example/example-application/src/main/java/com/codingapi/example/handler/TestHandler.java
@@ -1,9 +1,11 @@
 package com.codingapi.example.handler;
 
+import com.codingapi.example.event.AEvent;
 import com.codingapi.example.event.TestEvent;
 import com.codingapi.example.infra.entity.TestEntity;
 import com.codingapi.example.infra.jpa.TestEntityRepository;
-import com.codingapi.springboot.framework.handler.IHandler;
+import com.codingapi.springboot.framework.event.EventPusher;
+import com.codingapi.springboot.framework.event.IHandler;
 import lombok.AllArgsConstructor;
 import org.springframework.stereotype.Repository;
 
@@ -18,8 +20,13 @@ public void handler(TestEvent event) {
         TestEntity entity = new TestEntity(event.getName()+"123");
         testEntityRepository.save(entity);
 
-        throw new IllegalArgumentException("123");
+        new Thread(()->{
+            EventPusher.push(new AEvent());
+        }).start();
 
+        new Thread(()->{
+            EventPusher.push(new AEvent());
+        }).start();
     }
 
 
diff --git a/springboot-starter-flow/src/main/java/com/codingapi/springboot/flow/build/FlowWorkBuilder.java b/springboot-starter-flow/src/main/java/com/codingapi/springboot/flow/build/FlowWorkBuilder.java
@@ -10,7 +10,7 @@
 import com.codingapi.springboot.flow.matcher.OperatorMatcher;
 import com.codingapi.springboot.flow.trigger.OutTrigger;
 import com.codingapi.springboot.flow.user.IFlowOperator;
-import com.codingapi.springboot.flow.utils.RandomGenerator;
+import com.codingapi.springboot.framework.utils.RandomGenerator;
 
 /**
  * 流程工作构建器
diff --git a/springboot-starter-flow/src/main/java/com/codingapi/springboot/flow/domain/FlowWork.java b/springboot-starter-flow/src/main/java/com/codingapi/springboot/flow/domain/FlowWork.java
@@ -3,7 +3,7 @@
 import com.codingapi.springboot.flow.build.SchemaReader;
 import com.codingapi.springboot.flow.serializable.FlowWorkSerializable;
 import com.codingapi.springboot.flow.user.IFlowOperator;
-import com.codingapi.springboot.flow.utils.RandomGenerator;
+import com.codingapi.springboot.framework.utils.RandomGenerator;
 import lombok.AllArgsConstructor;
 import lombok.Getter;
 import lombok.Setter;
diff --git a/springboot-starter-flow/src/main/java/com/codingapi/springboot/flow/record/FlowProcess.java b/springboot-starter-flow/src/main/java/com/codingapi/springboot/flow/record/FlowProcess.java
@@ -1,7 +1,7 @@
 package com.codingapi.springboot.flow.record;
 
 import com.codingapi.springboot.flow.user.IFlowOperator;
-import com.codingapi.springboot.flow.utils.RandomGenerator;
+import com.codingapi.springboot.framework.utils.RandomGenerator;
 import lombok.AllArgsConstructor;
 import lombok.Getter;
 
diff --git a/springboot-starter/pom.xml b/springboot-starter/pom.xml
@@ -21,6 +21,11 @@
             <artifactId>spring-data-commons</artifactId>
         </dependency>
 
+        <dependency>
+            <groupId>com.esotericsoftware</groupId>
+            <artifactId>kryo</artifactId>
+        </dependency>
+
         <dependency>
             <groupId>jakarta.persistence</groupId>
             <artifactId>jakarta.persistence-api</artifactId>
diff --git a/springboot-starter/src/main/java/com/codingapi/springboot/framework/event/ApplicationHandlerUtils.java b/springboot-starter/src/main/java/com/codingapi/springboot/framework/event/ApplicationHandlerUtils.java
@@ -1,6 +1,4 @@
-package com.codingapi.springboot.framework.handler;
-
-import com.codingapi.springboot.framework.event.IEvent;
+package com.codingapi.springboot.framework.event;
 
 import java.util.ArrayList;
 import java.util.List;
diff --git a/springboot-starter/src/main/java/com/codingapi/springboot/framework/event/DomainEvent.java b/springboot-starter/src/main/java/com/codingapi/springboot/framework/event/DomainEvent.java
@@ -18,10 +18,14 @@ public class DomainEvent extends ApplicationEvent {
     @Getter
     private final boolean sync;
 
-    public DomainEvent(Object source, boolean sync) {
+    @Getter
+    private final String traceId;
+
+
+    public DomainEvent(Object source, boolean sync, String traceId) {
         super(source);
         this.event = (IEvent) source;
         this.sync = sync;
+        this.traceId = traceId;
     }
-
 }
diff --git a/springboot-starter/src/main/java/com/codingapi/springboot/framework/event/DomainEventContext.java b/springboot-starter/src/main/java/com/codingapi/springboot/framework/event/DomainEventContext.java
@@ -25,7 +25,9 @@ public static DomainEventContext getInstance() {
 
     private void push(IEvent event, boolean sync) {
         if (context != null) {
-            context.publishEvent(new DomainEvent(event, sync));
+            String traceId = EventTraceContext.getInstance().getOrCreateTrace();
+            EventTraceContext.getInstance().addEvent(traceId,event);
+            context.publishEvent(new DomainEvent(event, sync,traceId));
         }
     }
 
diff --git a/springboot-starter/src/main/java/com/codingapi/springboot/framework/event/EventTraceContext.java b/springboot-starter/src/main/java/com/codingapi/springboot/framework/event/EventTraceContext.java
@@ -0,0 +1,81 @@
+package com.codingapi.springboot.framework.event;
+
+import com.codingapi.springboot.framework.exception.EventLoopException;
+import com.codingapi.springboot.framework.utils.RandomGenerator;
+import lombok.Getter;
+
+import java.util.*;
+
+public class EventTraceContext {
+
+    @Getter
+    private final static EventTraceContext instance = new EventTraceContext();
+
+    // trace key
+    private final Set<String> traceKeys = new HashSet<>();
+
+    // thread local
+    private final ThreadLocal<String> threadLocal = new ThreadLocal<>();
+
+    // event listener state
+    private final Map<String, Boolean> eventListenerState = new HashMap<>();
+
+    // event stack
+    private final Map<String, List<Class<?>>> eventStack = new HashMap<>();
+
+    private EventTraceContext() {
+    }
+
+    String getOrCreateTrace() {
+        String listenerKey = threadLocal.get();
+        if (listenerKey != null) {
+            return listenerKey.split("#")[0];
+        }
+        String traceId = UUID.randomUUID().toString();
+        traceKeys.add(traceId);
+        return traceId;
+    }
+
+    public String getListenerKey(){
+        return threadLocal.get();
+    }
+
+    void createEventListener(String traceId) {
+        String listenerKey = traceId + "#" + RandomGenerator.randomString(8);
+        eventListenerState.put(listenerKey, false);
+        threadLocal.set(listenerKey);
+    }
+
+    void checkListener() {
+        String listenerKey = threadLocal.get();
+        if (listenerKey != null) {
+            boolean state = eventListenerState.get(listenerKey);
+            if (!state) {
+                // event execute finish
+                String traceId = listenerKey.split("#")[0];
+                traceKeys.remove(traceId);
+                eventStack.remove(traceId);
+            }
+        }
+        eventListenerState.remove(listenerKey);
+        threadLocal.remove();
+    }
+
+    void addEvent(String traceId, IEvent event) {
+        List<Class<?>> stack = eventStack.get(traceId);
+        if (stack == null) {
+            stack = new ArrayList<>();
+        } else {
+            if (stack.contains(event.getClass())) {
+                //清空trace记录
+                traceKeys.remove(traceId);
+                eventStack.remove(traceId);
+                eventListenerState.remove(traceId);
+                threadLocal.remove();
+                throw new EventLoopException(stack, event);
+            }
+        }
+        stack.add(event.getClass());
+        eventStack.put(traceId, stack);
+    }
+}
diff --git a/springboot-starter/src/main/java/com/codingapi/springboot/framework/event/Handler.java b/springboot-starter/src/main/java/com/codingapi/springboot/framework/event/Handler.java
@@ -1,4 +1,4 @@
-package com.codingapi.springboot.framework.handler;
+package com.codingapi.springboot.framework.event;
 
 import java.lang.annotation.*;
 
diff --git a/springboot-starter/src/main/java/com/codingapi/springboot/framework/event/HandlerBeanDefinitionRegistrar.java b/springboot-starter/src/main/java/com/codingapi/springboot/framework/event/HandlerBeanDefinitionRegistrar.java
@@ -1,4 +1,4 @@
-package com.codingapi.springboot.framework.handler;
+package com.codingapi.springboot.framework.event;
 
 import com.codingapi.springboot.framework.registrar.RegisterBeanScanner;
 import lombok.SneakyThrows;
diff --git a/springboot-starter/src/main/java/com/codingapi/springboot/framework/event/IEvent.java b/springboot-starter/src/main/java/com/codingapi/springboot/framework/event/IEvent.java
@@ -1,14 +1,16 @@
 package com.codingapi.springboot.framework.event;
 
 
+import java.io.Serializable;
+
 /**
  * 默认同步事件
  * <p>
  * 关于事件与事务之间的关系说明:
  * 事件本身不应该同步主业务的事务，即事件对于主业务来说，可成功可失败，成功与失败都不应该强关联主体业务。
  * 若需要让主体业务与分支做事务同步的时候，那不应该采用事件机制，而应该直接采用调用的方式实现业务绑定。
  */
-public interface IEvent {
+public interface IEvent extends Serializable {
 
 
 }
diff --git a/springboot-starter/src/main/java/com/codingapi/springboot/framework/event/IHandler.java b/springboot-starter/src/main/java/com/codingapi/springboot/framework/event/IHandler.java
@@ -1,6 +1,5 @@
-package com.codingapi.springboot.framework.handler;
+package com.codingapi.springboot.framework.event;
 
-import com.codingapi.springboot.framework.event.IEvent;
 import org.springframework.core.ResolvableType;
 
 /**
@@ -22,7 +21,7 @@ public interface IHandler<T extends IEvent> {
      *
      * @param exception 异常信息
      */
-    default void error(Exception exception) throws Exception{
+    default void error(Exception exception) throws Exception {
         throw exception;
     }
 
@@ -32,7 +31,7 @@ default void error(Exception exception) throws Exception{
      */
     default Class<?> getHandlerEventClass() {
         ResolvableType resolvableType = ResolvableType.forClass(getClass()).as(IHandler.class);
-        return (Class<T>) resolvableType.getGeneric(0).resolve();
+        return resolvableType.getGeneric(0).resolve();
     }
 
 
diff --git a/springboot-starter/src/main/java/com/codingapi/springboot/framework/event/SpringEventHandler.java b/springboot-starter/src/main/java/com/codingapi/springboot/framework/event/SpringEventHandler.java
@@ -1,6 +1,5 @@
-package com.codingapi.springboot.framework.handler;
+package com.codingapi.springboot.framework.event;
 
-import com.codingapi.springboot.framework.event.DomainEvent;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.context.ApplicationListener;
 
@@ -28,11 +27,17 @@ public SpringEventHandler(List<IHandler> handlers) {
 
     @Override
     public void onApplicationEvent(DomainEvent domainEvent) {
+        String traceId = domainEvent.getTraceId();
+
         if (domainEvent.isSync()) {
+            EventTraceContext.getInstance().createEventListener(traceId);
             ApplicationHandlerUtils.getInstance().handler(domainEvent.getEvent());
+            EventTraceContext.getInstance().checkListener();
         } else {
             executorService.execute(() -> {
+                EventTraceContext.getInstance().createEventListener(traceId);
                 ApplicationHandlerUtils.getInstance().handler(domainEvent.getEvent());
+                EventTraceContext.getInstance().checkListener();
             });
         }
     }
diff --git a/springboot-starter/src/main/java/com/codingapi/springboot/framework/event/SpringHandlerConfiguration.java b/springboot-starter/src/main/java/com/codingapi/springboot/framework/event/SpringHandlerConfiguration.java
@@ -1,4 +1,4 @@
-package com.codingapi.springboot.framework.handler;
+package com.codingapi.springboot.framework.event;
 
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.context.annotation.Bean;
diff --git a/springboot-starter/src/main/java/com/codingapi/springboot/framework/exception/EventLoopException.java b/springboot-starter/src/main/java/com/codingapi/springboot/framework/exception/EventLoopException.java
@@ -0,0 +1,17 @@
+package com.codingapi.springboot.framework.exception;
+
+import com.codingapi.springboot.framework.event.IEvent;
+import lombok.Getter;
+
+import java.util.List;
+
+@Getter
+public class EventLoopException extends RuntimeException {
+
+    private final List<Class<?>> stack;
+
+    public EventLoopException(List<Class<?>> stack, IEvent event) {
+        super("event loop error current event class:" + event.getClass() + ", history event stack:" + stack);
+        this.stack = stack;
+    }
+}
diff --git a/springboot-starter/src/main/java/com/codingapi/springboot/framework/utils/RandomGenerator.java b/springboot-starter/src/main/java/com/codingapi/springboot/framework/utils/RandomGenerator.java
@@ -1,4 +1,4 @@
-package com.codingapi.springboot.flow.utils;
+package com.codingapi.springboot.framework.utils;
 
 import java.util.UUID;
 
diff --git a/springboot-starter/src/main/resources/META-INF/spring.factories b/springboot-starter/src/main/resources/META-INF/spring.factories
@@ -2,6 +2,6 @@ org.springframework.boot.autoconfigure.EnableAutoConfiguration=\
 com.codingapi.springboot.framework.AutoConfiguration,\
 com.codingapi.springboot.framework.event.SpringEventConfiguration,\
 com.codingapi.springboot.framework.exception.ExceptionConfiguration,\
-com.codingapi.springboot.framework.handler.HandlerBeanDefinitionRegistrar,\
-com.codingapi.springboot.framework.handler.SpringHandlerConfiguration,\
-com.codingapi.springboot.framework.servlet.BasicHandlerExceptionResolverConfiguration
+com.codingapi.springboot.framework.event.HandlerBeanDefinitionRegistrar,\
+com.codingapi.springboot.framework.event.SpringHandlerConfiguration,\
+com.codingapi.springboot.framework.servlet.BasicHandlerExceptionResolverConfiguration
diff --git a/springboot-starter/src/main/resources/META-INF/spring/org.springframework.boot.autoconfigure.AutoConfiguration.imports b/springboot-starter/src/main/resources/META-INF/spring/org.springframework.boot.autoconfigure.AutoConfiguration.imports
diff --git a/springboot-starter/src/test/java/com/codingapi/springboot/framework/handler/DemoChangeLogHandler.java b/springboot-starter/src/test/java/com/codingapi/springboot/framework/handler/DemoChangeLogHandler.java
diff --git a/springboot-starter/src/test/java/com/codingapi/springboot/framework/handler/DemoCreateHandler.java b/springboot-starter/src/test/java/com/codingapi/springboot/framework/handler/DemoCreateHandler.java
diff --git a/springboot-starter/src/test/java/com/codingapi/springboot/framework/handler/DemoDeleteHandler.java b/springboot-starter/src/test/java/com/codingapi/springboot/framework/handler/DemoDeleteHandler.java
diff --git a/springboot-starter/src/test/java/com/codingapi/springboot/framework/handler/DemoPersistEventHandler.java b/springboot-starter/src/test/java/com/codingapi/springboot/framework/handler/DemoPersistEventHandler.java
diff --git a/springboot-starter/src/test/java/com/codingapi/springboot/framework/handler/EntityFiledChangeHandler.java b/springboot-starter/src/test/java/com/codingapi/springboot/framework/handler/EntityFiledChangeHandler.java
