fix(temporal): fix schedule workflow search attributes

formancehq · Nov 27, 2024 · 0ae82c2 · 0ae82c2
1 parent ba234d8
commit 0ae82c2
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 6 deletions.
diff --git a/internal/connectors/engine/activities/temporal_schedule_create.go b/internal/connectors/engine/activities/temporal_schedule_create.go
@@ -6,6 +6,7 @@ import (
 
 	"go.temporal.io/api/enums/v1"
 	"go.temporal.io/sdk/client"
+	"go.temporal.io/sdk/temporal"
 	"go.temporal.io/sdk/workflow"
 )
 
@@ -20,6 +21,19 @@ type ScheduleCreateOptions struct {
 }
 
 func (a Activities) TemporalScheduleCreate(ctx context.Context, options ScheduleCreateOptions) (string, error) {
+	attributes := make([]temporal.SearchAttributeUpdate, 0, len(options.SearchAttributes))
+	for key, value := range options.SearchAttributes {
+		v, ok := value.(string)
+		if !ok {
+			continue
+		}
+
+		attributes = append(attributes,
+			temporal.NewSearchAttributeKeyKeyword(key).ValueSet(v),
+		)
+	}
+	options.Action.TypedSearchAttributes = temporal.NewSearchAttributes(attributes...)
+
 	handle, err := a.temporalClient.ScheduleClient().Create(ctx, client.ScheduleOptions{
 		ID: options.ScheduleID,
 		Spec: client.ScheduleSpec{

diff --git a/internal/connectors/engine/workflow/plugin_workflow.go b/internal/connectors/engine/workflow/plugin_workflow.go
@@ -8,7 +8,6 @@ import (
 	"github.com/pkg/errors"
 	"go.temporal.io/api/enums/v1"
 	"go.temporal.io/sdk/client"
-	"go.temporal.io/sdk/temporal"
 	"go.temporal.io/sdk/workflow"
 )
 
@@ -123,11 +122,6 @@ func (w Workflow) run(
 							task.NextTasks,
 						},
 						TaskQueue: connectorID.String(),
-						// Search attributes are used to query workflows
-						TypedSearchAttributes: temporal.NewSearchAttributes(
-							temporal.NewSearchAttributeKeyKeyword(SearchAttributeScheduleID).ValueSet(scheduleID),
-							temporal.NewSearchAttributeKeyKeyword(SearchAttributeStack).ValueSet(w.stack),
-						),
 					},
 					Overlap:            enums.SCHEDULE_OVERLAP_POLICY_SKIP,
 					TriggerImmediately: true,