Merge pull request #9 from Talexs/develop

Develop

Author: TalexDreamSoul

src/main/java/com/talexframe/frame/core/function/command/frame/StopCmd.java

@@ -1,5 +1,6 @@
 package com.talexframe.frame.core.function.command.frame;
 
+import com.talexframe.frame.TalexFrameApplication;
 import com.talexframe.frame.core.function.command.BaseCommand;
 import com.talexframe.frame.core.function.command.ISender;
 import com.talexframe.frame.core.talex.TFrame;
@@ -20,7 +21,11 @@ public StopCmd() {
     @Override
     public boolean executeCommand(ISender sender, String wholeCommand, String matchedLabel, String[] args) {
 
-        tframe.shutdown();
+        if ( TalexFrameApplication.context != null ) {
+
+            TalexFrameApplication.context.stop();
+
+        }
 
         return false;
 

src/main/java/com/talexframe/frame/core/function/listener/FrameMajorListener.java

@@ -84,6 +84,8 @@ public void onStopped(ContextStoppedEvent event) {
     @EventListener
     public void onStopped(ContextClosedEvent event) {
 
+        TFrame.tframe.shutdown();
+
         log.warn("框架已停止.");
 
         TFrame.tframe.callEvent(new FramePostUnInstallEvent(System.nanoTime() - TalexFrameApplication.startedTimeStamp));

src/main/java/com/talexframe/frame/core/function/listener/FrameSelfListener.java

@@ -1,7 +1,7 @@
 package com.talexframe.frame.core.function.listener;
 
 import com.talexframe.frame.core.modules.event.FrameListener;
-import com.talexframe.frame.core.modules.event.THandler;
+import com.talexframe.frame.core.modules.event.service.THandler;
 import com.talexframe.frame.core.modules.event.events.dao.DAOProcessorPreShutdownEvent;
 import com.talexframe.frame.core.modules.repository.TRepo;
 import com.talexframe.frame.core.modules.repository.TRepoPlus;

src/main/java/com/talexframe/frame/core/modules/application/TAppManager.java

@@ -3,7 +3,6 @@
 import com.talexframe.frame.core.modules.event.events.app.AppPostRegisterEvent;
 import com.talexframe.frame.core.modules.event.events.app.AppPreRegisterEvent;
 import com.talexframe.frame.core.modules.event.events.app.AppUnRegisteredEvent;
-import com.talexframe.frame.core.modules.network.connection.RequestAnalyser;
 import com.talexframe.frame.core.modules.network.interfaces.IUnRegisterHandler;
 import com.talexframe.frame.core.modules.plugins.core.WebPlugin;
 import com.talexframe.frame.core.modules.repository.TRepo;
@@ -64,7 +63,7 @@ public boolean registerController(WebPlugin plugin, TApp controller) {
 
         AppPreRegisterEvent event = new AppPreRegisterEvent(plugin, controller);
 
-        TFrame.tframe.callEvent(new AppPostRegisterEvent(plugin, controller));
+        TFrame.tframe.callEvent(event);
 
         if( event.isCancelled() ) return false;
 

src/main/java/com/talexframe/frame/core/modules/event/MethodManager.java

@@ -1,5 +1,6 @@
 package com.talexframe.frame.core.modules.event;
 
+import com.talexframe.frame.core.modules.event.service.THandler;
 import lombok.Getter;
 import lombok.Setter;
 
@@ -16,22 +17,21 @@ public class MethodManager {
     private final FrameListener owner;
     @Getter
     private final Map<String, Long> listenedEvents = new HashMap<>();
-    private Class<?> paramType;
+
     private Method method;
     @Setter
     @Getter
     private THandler tHandler;
 
-    public MethodManager(FrameListener listener, Class<?> paramType, Method method, THandler tHandler) {
+    public MethodManager(FrameListener listener, Method method, THandler tHandler) {
 
         this.owner = listener;
-        this.paramType = paramType;
         this.method = method;
         this.tHandler = tHandler;
 
     }
 
-    public void listen(String key) {
+    public void listened(String key) {
 
         if ( listenedEvents.containsKey(key) ) {
 
@@ -43,10 +43,6 @@ public void listen(String key) {
 
     }
 
-    public Class<?> getParamType() {return paramType;}
-
-    public void setParamType(Class<?> paramType) {this.paramType = paramType;}
-
     public Method getMethod() {return method;}
 
     public void setMethod(Method method) {this.method = method;}

src/main/java/com/talexframe/frame/core/modules/event/TalexEventBus.java

@@ -0,0 +1,254 @@
+package com.talexframe.frame.core.modules.event;
+
+import cn.hutool.core.thread.ThreadUtil;
+import cn.hutool.json.JSONUtil;
+import com.google.common.collect.ArrayListMultimap;
+import com.google.common.collect.Multimap;
+import com.talexframe.frame.core.modules.event.service.IContinue;
+import com.talexframe.frame.core.modules.event.service.IEventBus;
+import com.talexframe.frame.core.modules.event.service.THandler;
+import com.talexframe.frame.core.modules.event.service.TalexEvent;
+import com.talexframe.frame.core.pojo.enums.ThreadMode;
+import lombok.Getter;
+import lombok.extern.slf4j.Slf4j;
+
+import java.lang.reflect.Method;
+import java.lang.reflect.Parameter;
+import java.lang.reflect.Type;
+import java.util.*;
+
+/**
+ * @author TalexDreamSoul
+ */
+@Slf4j
+public class TalexEventBus implements IEventBus {
+
+    private static TalexEventBus INSTANCE;
+
+    @Getter
+    private final Multimap<FrameListener, MethodManager> listenerManager = ArrayListMultimap.create();
+
+    @Getter
+    private final Map<String, TalexEvent> continueEvents = new HashMap<>(16);
+
+    private TalexEventBus() {
+
+    }
+
+    public static TalexEventBus getDefault() {
+
+        if ( INSTANCE == null ) {
+
+            INSTANCE = new TalexEventBus();
+
+            return INSTANCE;
+
+        }
+
+        return null;
+
+    }
+
+    @Override
+    public TalexEventBus registerListener(FrameListener listener) {
+
+        if( listenerManager.containsKey(listener) ) {
+
+            return this;
+
+        }
+
+        List<MethodManager> methodsList = new ArrayList<>(listenerManager.get(listener));
+
+        Class<?> clazz = listener.getClass();
+
+        Arrays.asList( clazz.getMethods() ).forEach((method) -> {
+
+            THandler annotation = method.getAnnotation(THandler.class);
+
+            if ( annotation == null ) return;
+
+            Type type = method.getGenericReturnType();
+
+            if ( !"void".equals(type.toString()) ) {
+
+                throw new RuntimeException(method.getName() + " 返回值类型必须为void类型");
+
+            }
+
+            Class<?>[] parameterTypes = method.getParameterTypes();
+
+            if ( parameterTypes.length > 2 ) {
+
+                throw new RuntimeException(method.getName() + " 参数个数必须 <= 2");
+
+            }
+
+            MethodManager methodManager = new MethodManager(listener, method, annotation);
+
+            methodsList.add(methodManager);
+
+            for ( TalexEvent continueEvent : continueEvents.values() ) {
+
+                this._callEvent(continueEvent, methodManager);
+
+            }
+
+        });
+
+        listenerManager.putAll(listener, methodsList);
+
+        return this;
+
+    }
+
+    private void _callEvent(TalexEvent event, MethodManager methodManager) {
+
+        // 判断 event 参数是否和 continue event 相同
+        if( !methodManager.getMethod().getParameterTypes()[0].isAssignableFrom(event.getClass()) ) return;
+
+        log.debug("[Event] For _event: {} | {}", event, JSONUtil.toJsonStr(methodManager));
+
+        // execute
+        if ( methodManager.getTHandler().threadMode() == ThreadMode.ASYNC ) {
+
+            ThreadUtil.execAsync(() -> event(event, methodManager));
+
+        } else {
+
+            event(event, methodManager);
+
+        }
+
+    }
+
+    @Override
+    public TalexEventBus callEvent(TalexEvent event) {
+
+        // 如果是 ContinuouslyEvent 放到 map 中
+        if ( event instanceof IContinue ) {
+
+            String matchKey = ( (IContinue) event ).getMatchKey();
+            continueEvents.put(matchKey, event);
+
+            log.debug("[Event] For continuously event: {} | {}", event, JSONUtil.toJsonStr(continueEvents));
+
+        }
+
+        // 筛选出匹配的 methodManager
+        Map<Integer, MethodManager> preList = new HashMap<>();
+
+        listenerManager.entries().forEach((entry) -> {
+
+            if( entry.getValue().getMethod().getParameterTypes()[0].isAssignableFrom(event.getClass()) ) {
+
+                preList.put(entry.getKey().hashCode(), entry.getValue());
+
+            }
+
+        });
+
+        log.debug("[Event] For event preList: {}", preList);
+
+        List<MethodManager> pl = new ArrayList<>(preList.values());
+
+        pl.sort((o1, o2) -> {
+
+            THandler ts1 = o1.getTHandler();
+            THandler ts2 = o2.getTHandler();
+
+            return Integer.compare(ts1.priority().getLevel(), ts2.priority().getLevel());
+
+        });
+
+        for ( MethodManager methodManager : pl ) {
+
+            if ( methodManager.getTHandler().threadMode() == ThreadMode.ASYNC ) {
+
+                log.debug("[Event] Execute async event: {}", methodManager.getMethod().getName());
+
+                ThreadUtil.execAsync(() -> event(event, methodManager));
+
+            } else {
+
+                log.debug("[Event] Execute event: {}", methodManager.getMethod().getName());
+
+                event(event, methodManager);
+
+            }
+
+        }
+
+        return this;
+
+    }
+
+    private void event(TalexEvent event, MethodManager methodManager) {
+
+        Method method = methodManager.getMethod();
+
+        log.debug("[Event] --> Invoke: " + event.getClass().getName() + " | " + method.getName() + " <> " + methodManager.getOwner().getClass().getName());
+
+        try {
+
+            List<Object> objs = new ArrayList<>();
+
+            objs.add(event);
+
+            for ( Parameter parameter : method.getParameters() ) {
+
+                if ( parameter.getType().isInstance(MethodManager.class) ) {
+
+                    objs.add(methodManager);
+
+                }
+
+            }
+
+            method.invoke(methodManager.getOwner(), objs.toArray(new Object[0]));
+
+            if ( event instanceof IContinue ) {
+
+                String matchKey = ( (IContinue) event ).getMatchKey();
+                continueEvents.put(matchKey, event);
+
+                methodManager.listened(matchKey);
+
+            }
+
+            if ( methodManager.getTHandler().once() ) {
+
+                FrameListener listener = methodManager.getOwner();
+
+                listenerManager.remove(listener, methodManager);
+
+            }
+
+        } catch ( Exception e ) {
+
+            if ( e.getCause() == null ) {
+
+                log.error("[事件] 在调用 " + method.getName() + " 时, 目标抛出异常" + "   @" + method.getDeclaringClass().getName() + "." + method.getName());
+
+            } else {
+                log.error("[事件] 在调用 " + method.getName() + " 时, 目标抛出异常: " + e.getCause().getMessage() + "   @" + method.getDeclaringClass().getName() + "." + method.getName());
+            }
+
+            e.printStackTrace();
+
+        }
+
+    }
+
+    @Override
+    public TalexEventBus unRegisterListener(FrameListener listener) {
+
+        listenerManager.removeAll(listener);
+
+        return this;
+
+    }
+
+}
+
+

src/main/java/com/talexframe/frame/core/modules/event/events/app/AppPostRegisterEvent.java

@@ -1,7 +1,7 @@
 package com.talexframe.frame.core.modules.event.events.app;
 
 import com.talexframe.frame.core.modules.application.TApp;
-import com.talexframe.frame.core.modules.event.TalexEvent;
+import com.talexframe.frame.core.modules.event.service.TalexEvent;
 import com.talexframe.frame.core.modules.plugins.core.WebPlugin;
 import lombok.Getter;
 

src/main/java/com/talexframe/frame/core/modules/event/events/app/AppPreRegisterEvent.java

@@ -1,8 +1,8 @@
 package com.talexframe.frame.core.modules.event.events.app;
 
 import com.talexframe.frame.core.modules.application.TApp;
-import com.talexframe.frame.core.modules.event.Cancellable;
-import com.talexframe.frame.core.modules.event.TalexEvent;
+import com.talexframe.frame.core.modules.event.service.Cancellable;
+import com.talexframe.frame.core.modules.event.service.TalexEvent;
 import com.talexframe.frame.core.modules.plugins.core.WebPlugin;
 import lombok.Getter;
 

src/main/java/com/talexframe/frame/core/modules/event/events/app/AppUnRegisteredEvent.java

@@ -1,7 +1,7 @@
 package com.talexframe.frame.core.modules.event.events.app;
 
 import com.talexframe.frame.core.modules.application.TApp;
-import com.talexframe.frame.core.modules.event.TalexEvent;
+import com.talexframe.frame.core.modules.event.service.TalexEvent;
 import com.talexframe.frame.core.modules.plugins.core.WebPlugin;
 import lombok.Getter;
 

src/main/java/com/talexframe/frame/core/modules/event/events/dao/BaseDAOEvent.java

@@ -1,6 +1,6 @@
 package com.talexframe.frame.core.modules.event.events.dao;
 
-import com.talexframe.frame.core.modules.event.TalexEvent;
+import com.talexframe.frame.core.modules.event.service.TalexEvent;
 import lombok.AllArgsConstructor;
 import lombok.Getter;