feat(EventBus): 支持 filter 过滤回调

fjc0k · Jul 26, 2024 · 8290e32 · 8290e32
1 parent 0a231a4
commit 8290e32
Show file tree

Hide file tree

Showing 2 changed files with 58 additions and 0 deletions.
diff --git a/src/utils/EventBus.test.ts b/src/utils/EventBus.test.ts
@@ -199,4 +199,48 @@ describe('EventBus', () => {
     bus.emit('test')
     expect(fn).toBeCalled().toBeCalledTimes(2)
   })
+
+  test('支持 filter', () => {
+    const bus = new EventBus<{
+      test: () => any
+    }>()
+    const fn = jest.fn()
+    const fn1 = jest.fn()
+    const fn2 = jest.fn()
+
+    bus.on('test', fn)
+    bus.on('test', fn1)
+    bus.on('test', fn2)
+
+    bus.emit({
+      name: 'test',
+    })
+    expect(fn).toBeCalled().toBeCalledTimes(1)
+    expect(fn1).toBeCalled().toBeCalledTimes(1)
+    expect(fn2).toBeCalled().toBeCalledTimes(1)
+
+    bus.emit({
+      name: 'test',
+      filter: (_, index) => index === 0,
+    })
+    expect(fn).toBeCalled().toBeCalledTimes(2)
+    expect(fn1).toBeCalled().toBeCalledTimes(1)
+    expect(fn2).toBeCalled().toBeCalledTimes(1)
+
+    bus.emit({
+      name: 'test',
+      filter: (_, index, { length }) => index === length - 1,
+    })
+    expect(fn).toBeCalled().toBeCalledTimes(2)
+    expect(fn1).toBeCalled().toBeCalledTimes(1)
+    expect(fn2).toBeCalled().toBeCalledTimes(2)
+
+    bus.emit({
+      name: 'test',
+      filter: (_, index) => index === 0 || index === 1,
+    })
+    expect(fn).toBeCalled().toBeCalledTimes(3)
+    expect(fn1).toBeCalled().toBeCalledTimes(2)
+    expect(fn2).toBeCalled().toBeCalledTimes(2)
+  })
 })
diff --git a/src/utils/EventBus.ts b/src/utils/EventBus.ts
@@ -16,6 +16,11 @@ export interface EventBusListenerDescriptor<
   name: TListenerName
   context?: any
   tag?: EventBusListenerTag
+  filter?: (
+    callback: EventBusListener,
+    index: number,
+    callbacks: EventBusListener[],
+  ) => boolean
 }
 
 export type EventBusBeforeOn<TListeners extends EventBusListeners> = {
@@ -153,10 +158,16 @@ export class EventBus<TListeners extends EventBusListeners> {
       name,
       context,
       tag,
+      filter,
     }: {
       name: TListenerName
       context?: any
       tag?: string | number
+      filter?: (
+        callback: EventBusListener,
+        index: number,
+        callbacks: EventBusListener[],
+      ) => boolean
     } =
       typeof eventName === 'object'
         ? eventName
@@ -171,6 +182,9 @@ export class EventBus<TListeners extends EventBusListeners> {
           tag === callback.__EVENT_BUS_TAG__,
       )
     }
+    if (typeof filter === 'function') {
+      callbacks = callbacks.filter(filter)
+    }
     this.options?.beforeEmit?.[name]?.call(this, context)
     return callbacks.map(callback => {
       return callback.call(context, ...args)