add bind option to ctx

brickingsoft · Nov 19, 2024 · cd60bbf · cd60bbf
1 parent dd03fa3
commit cd60bbf
Showing 1 changed file with 28 additions and 3 deletions.
diff --git a/async/make.go b/async/make.go
@@ -66,9 +66,11 @@ func WithWait() Option {
 	}
 }
 
-// Make
-// 构建一个许诺
-func Make[R any](ctx context.Context, options ...Option) (p Promise[R], err error) {
+type optionsCtxKey struct{}
+
+// WithOptions
+// 把 Make 的 Options 绑定到 context.Context。常用于设置上下文中的默认选项。
+func WithOptions(ctx context.Context, options ...Option) context.Context {
 	opt := Options{
 		Stream:      false,
 		Mode:        Normal,
@@ -77,6 +79,29 @@ func Make[R any](ctx context.Context, options ...Option) (p Promise[R], err erro
 	for _, o := range options {
 		o(&opt)
 	}
+	return context.WithValue(ctx, optionsCtxKey{}, opt)
+}
+
+func getOptions(ctx context.Context) Options {
+	value := ctx.Value(optionsCtxKey{})
+	if value == nil {
+		return Options{
+			Stream:      false,
+			Mode:        Normal,
+			WaitTimeout: -1,
+		}
+	}
+	opt := value.(Options)
+	return opt
+}
+
+// Make
+// 构建一个许诺
+func Make[R any](ctx context.Context, options ...Option) (p Promise[R], err error) {
+	opt := getOptions(ctx)
+	for _, o := range options {
+		o(&opt)
+	}
 	var submitter rxp.TaskSubmitter
 	switch opt.Mode {
 	case Unlimited: