feat: we can produce with context now

w-h-a · Jul 14, 2024 · 753352a · 753352a
1 parent 063c657
commit 753352a
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/streams/options.go b/streams/options.go
@@ -94,6 +94,7 @@ type ProduceOptions struct {
 	Timestamp time.Time
 	Metadata  map[string]string
 	Processed map[string]bool
+	Context   context.Context
 }
 
 func ProduceWithTimestamp(t time.Time) ProduceOption {
@@ -108,11 +109,24 @@ func ProduceWithMetadata(md map[string]string) ProduceOption {
 	}
 }
 
+func ProduceWithProcessed(p map[string]bool) ProduceOption {
+	return func(o *ProduceOptions) {
+		o.Processed = p
+	}
+}
+
+func ProduceWithContext(ctx context.Context) ProduceOption {
+	return func(o *ProduceOptions) {
+		o.Context = ctx
+	}
+}
+
 func NewProduceOptions(opts ...ProduceOption) ProduceOptions {
 	options := ProduceOptions{
 		Timestamp: time.Now(),
 		Metadata:  map[string]string{},
 		Processed: map[string]bool{},
+		Context:   context.Background(),
 	}
 
 	for _, fn := range opts {