LiEventBus增加自定义注册事件、卸载事件

leaderli · Oct 30, 2024 · b0c5e94 · b0c5e94
1 parent 23534d5
commit b0c5e94
Show file tree

Hide file tree

Showing 3 changed files with 41 additions and 0 deletions.
diff --git a/litool-core/src/main/java/io/leaderli/litool/core/event/ILiEventListener.java b/litool-core/src/main/java/io/leaderli/litool/core/event/ILiEventListener.java
@@ -23,6 +23,23 @@
 @FunctionalInterface
 public interface ILiEventListener<E extends LiEventObject<S>, S> extends ComponentType<E>, OnError {
 
+    /**
+     * 当注册监听器时触发
+     * {@link LiEventBus#registerListener(ILiEventListener)}
+     */
+    default void onRegister() {
+
+    }
+
+    /**
+     * 当卸载监听器时触发
+     * {@link LiEventBus#unRegisterListener(ILiEventListener)}
+     */
+    default void onUnRegister() {
+
+    }
+
+
     /**
      * 监听事件。当 {@link LiEventBus#push(LiEventObject)} 推送事件时，可以监听该事件的监听器会接收到该事件。
      *

diff --git a/litool-core/src/main/java/io/leaderli/litool/core/event/LiEventBus.java b/litool-core/src/main/java/io/leaderli/litool/core/event/LiEventBus.java
@@ -24,6 +24,7 @@ public class LiEventBus implements LiEventBusBehavior {
      */
     public <E extends LiEventObject<S>, S> void registerListener(ILiEventListener<E, S> listener) {
         liEventMap.put(listener.componentType(), listener);
+        listener.onRegister();
     }
 
     /**
@@ -34,6 +35,7 @@ public <E extends LiEventObject<S>, S> void registerListener(ILiEventListener<E,
     @Override
     public <E extends LiEventObject<S>, S> void unRegisterListener(ILiEventListener<E, S> listener) {
         liEventMap.remove(listener);
+        listener.onUnRegister();
     }
 
     /**

diff --git a/litool-core/src/test/java/io/leaderli/litool/core/event/LiEventBusTest.java b/litool-core/src/test/java/io/leaderli/litool/core/event/LiEventBusTest.java
@@ -38,6 +38,28 @@ void getPublisher() {
         Assertions.assertDoesNotThrow(() -> {
             eventBus.push(new TestStringLiEventObject("456"));
         });
+        LiBox<Integer> count = LiBox.of(0);
+        ILiEventListener<TestStringLiEventObject, String> listener = new ILiEventListener<TestStringLiEventObject, String>() {
+            @Override
+            public void onRegister() {
+                count.value(1);
+            }
+
+            @Override
+            public void onUnRegister() {
+                count.value(2);
+            }
+
+            @Override
+            public void listen(String source) {
+
+            }
+        };
+        eventBus.registerListener(listener);
+        Assertions.assertEquals(1, count.value());
+        eventBus.unRegisterListener(listener);
+        Assertions.assertEquals(2, count.value());
+
     }
 
     @Test