feat(EventBus): 新增 clear, destroy

fjc0k · Oct 18, 2021 · 38558f6 · 38558f6
1 parent 5ece08c
commit 38558f6
Show file tree

Hide file tree

Showing 2 changed files with 32 additions and 2 deletions.
diff --git a/src/utils/EventBus.test.ts b/src/utils/EventBus.test.ts
@@ -168,4 +168,19 @@ describe('EventBus', () => {
     expect(errorCallback).toBeCalled().toBeCalledTimes(2)
     expect(errorCallback2).toBeCalled().toBeCalledTimes(2)
   })
+
+  test('支持 clear', () => {
+    const bus = new EventBus<{
+      test: () => any
+    }>()
+    const fn = jest.fn()
+    bus.on('test', fn)
+    bus.emit('test')
+    expect(fn).toBeCalled().toBeCalledTimes(1)
+    bus.emit('test')
+    expect(fn).toBeCalled().toBeCalledTimes(2)
+    bus.clear()
+    bus.emit('test')
+    expect(fn).toBeCalled().toBeCalledTimes(2)
+  })
 })
diff --git a/src/utils/EventBus.ts b/src/utils/EventBus.ts
@@ -1,7 +1,7 @@
 export type EventBusListenerTag = string | number
 
 export type EventBusListener<
-  TCallback extends (...args: any[]) => any = (...args: any[]) => any
+  TCallback extends (...args: any[]) => any = (...args: any[]) => any,
 > = TCallback & {
   __EVENT_BUS_TAG__?: EventBusListenerTag
 }
@@ -11,7 +11,7 @@ export type EventBusOffListener = () => any
 export type EventBusListeners = Record<string, EventBusListener>
 
 export interface EventBusListenerDescriptor<
-  TListenerName extends keyof EventBusListeners
+  TListenerName extends keyof EventBusListeners,
 > {
   name: TListenerName
   context?: any
@@ -176,4 +176,19 @@ export class EventBus<TListeners extends EventBusListeners> {
       return callback.call(context, ...args)
     })
   }
+
+  /**
+   * 清空事件订阅。
+   */
+  clear() {
+    this.callbacks = Object.create(null)
+  }
+
+  /**
+   * 销毁。
+   */
+  destroy() {
+    // @ts-ignore
+    this.callbacks = null
+  }
 }