candishared: add method to set multiple context on event context

candishared/event_context.go

@@ -55,6 +55,15 @@ func (e *EventContext) SetContext(ctx context.Context) {
 	e.ctx = ctx
 }
 
+// SetContextWithValue set context with allow to multiple value
+// If parent context is null will assign with context.Background() as default value
+func (e *EventContext) SetContextWithValue(key, value any) {
+	if e.ctx == nil {
+		e.ctx = context.Background()
+	}
+	e.ctx = context.WithValue(e.ctx, key, value)
+}
+
 // SetWorkerType setter
 func (e *EventContext) SetWorkerType(w string) {
 	e.workerType = w

candishared/event_context_test.go

@@ -0,0 +1,29 @@
+package candishared
+
+import (
+	"context"
+	"github.com/stretchr/testify/assert"
+	"testing"
+)
+
+func TestNewEventContext(t *testing.T) {
+	event := &EventContext{}
+
+	ctx := context.Background()
+	event.SetContext(context.WithValue(ctx, "key1", "value 1"))
+	event.SetContext(context.WithValue(ctx, "key2", "value 2"))
+
+	assert.Equal(t, nil, event.Context().Value("key1"))
+	assert.Equal(t, "value 2", event.Context().Value("key2"))
+}
+
+func TestMultiContext(t *testing.T) {
+	event := &EventContext{}
+	event.SetContext(context.Background())
+
+	event.SetContextWithValue("key1", "value 1")
+	event.SetContextWithValue("key2", "value 2")
+
+	assert.Equal(t, "value 1", event.Context().Value("key1"))
+	assert.Equal(t, "value 2", event.Context().Value("key2"))
+}