修改event bus

eventbus/eventbus.go

 package eventbus
 
 import (
+	"context"
 	"fmt"
 	"reflect"
 	"sync"
@@ -15,7 +16,7 @@ type BusSubscriber interface {
 }
 
 type BusPublisher interface {
-	Publish(topic string, args ...interface{})
+	Publish(ctx context.Context, topic string, args ...interface{})
 }
 
 type BusController interface {
@@ -106,7 +107,7 @@ func (bus *EventBus) Unsubscribe(topic string, handler interface{}) error {
 	return fmt.Errorf("topic %s doesn't exist", topic)
 }
 
-func (bus *EventBus) Publish(topic string, args ...interface{}) {
+func (bus *EventBus) Publish(ctx context.Context, topic string, args ...interface{}) {
 	bus.lock.Lock()
 	defer bus.lock.Unlock()
 	if handlers, ok := bus.handlers[topic]; ok && 0 < len(handlers) {
@@ -117,7 +118,7 @@ func (bus *EventBus) Publish(topic string, args ...interface{}) {
 				bus.removeHandler(topic, i)
 			}
 			if !handler.async {
-				bus.doPublish(handler, args...)
+				bus.doPublish(ctx, handler, args...)
 			} else {
 				bus.wg.Add(1)
 				if handler.transactional {
@@ -126,24 +127,27 @@ func (bus *EventBus) Publish(topic string, args ...interface{}) {
 					bus.lock.Lock()
 				}
 				go func() {
-					bus.doPublishAsync(handler, args...)
+					bus.doPublishAsync(ctx, handler, args...)
 				}()
 			}
 		}
 	}
 }
 
-func (bus *EventBus) doPublish(handler *eventHandler, args ...interface{}) {
-	passedArguments := bus.setUpPublish(handler, args...)
+func (bus *EventBus) doPublish(ctx context.Context, handler *eventHandler, args ...interface{}) {
+	allArgs := make([]any, 0, len(args)+1)
+	allArgs = append(allArgs, ctx)
+	allArgs = append(allArgs, args...)
+	passedArguments := bus.setUpPublish(handler, allArgs...)
 	handler.callBack.Call(passedArguments)
 }
 
-func (bus *EventBus) doPublishAsync(handler *eventHandler, args ...interface{}) {
+func (bus *EventBus) doPublishAsync(ctx context.Context, handler *eventHandler, args ...interface{}) {
 	defer bus.wg.Done()
 	if handler.transactional {
 		defer handler.Unlock()
 	}
-	bus.doPublish(handler, args...)
+	bus.doPublish(ctx, handler, args...)
 }
 
 func (bus *EventBus) removeHandler(topic string, idx int) {