feat:ctx fork

.gitignore

@@ -13,3 +13,5 @@
 
 # Dependency directories (remove the comment below to include it)
 # vendor/
+
+.idea/

event_entity.go

@@ -1,12 +1,16 @@
 package easyfsm
 
+import "context"
+
 type (
 	// EventEntity is the core that wraps the basic Event methods.
 	EventEntity struct {
 		hook      EventHook
 		eventName EventName
 		observers []EventObserver
 		eventFunc EventFunc
+		// issue:https://github.com/wuqinqiang/easyfsm/issues/16
+		forkCtxFunc func(ctx context.Context) context.Context
 	}
 
 	EventEntityOpt func(entity *EventEntity)
@@ -34,6 +38,9 @@ func NewEventEntity(event EventName, handler EventFunc,
 		eventName: event,
 		eventFunc: handler,
 		observers: make([]EventObserver, 0),
+		forkCtxFunc: func(ctx context.Context) context.Context {
+			return context.Background()
+		},
 	}
 	for _, opt := range opts {
 		opt(entity)
@@ -61,6 +68,12 @@ func WithHook(hook EventHook) EventEntityOpt {
 	}
 }
 
+func WithForkCtxFunc(fn func(ctx context.Context) context.Context) EventEntityOpt {
+	return func(entity *EventEntity) {
+		entity.forkCtxFunc = fn
+	}
+}
+
 // Execute executes the event.
 func (e *EventEntity) Execute(param *Param) (State, error) {
 	if e.hook != nil {
@@ -78,6 +91,7 @@ func (e *EventEntity) Execute(param *Param) (State, error) {
 
 	// Asynchronous notify observers
 	GoSafe(func() {
+		param.Ctx = e.forkCtxFunc(param.Ctx)
 		e.notify(param)
 	})
 	return state, nil

helper.go

@@ -1,6 +1,9 @@
 package easyfsm
 
-import "github.com/wuqinqiang/easyfsm/log"
+import (
+	"context"
+	"github.com/wuqinqiang/easyfsm/log"
+)
 
 func GoSafe(fn func()) {
 	go goSafe(fn)
@@ -14,3 +17,7 @@ func goSafe(fn func()) {
 	}()
 	fn()
 }
+
+type ForkCtxInterface interface {
+	ForkCtx(ctx context.Context) context.Context
+}