Fix AutoConfigHint population in the mapper (#1415)

What changed? add missing mapper for autoconfighint
cadence-workflow · Jan 6, 2025 · 6c105e0 · 6c105e0
1 parent 1fd8ba0
commit 6c105e0
Show file tree

Hide file tree

Showing 4 changed files with 18 additions and 0 deletions.
diff --git a/internal/compatibility/proto/response.go b/internal/compatibility/proto/response.go
@@ -210,6 +210,7 @@ func PollForActivityTaskResponse(t *shared.PollForActivityTaskResponse) *apiv1.P
 		WorkflowType:               WorkflowType(t.WorkflowType),
 		WorkflowDomain:             t.GetWorkflowDomain(),
 		Header:                     Header(t.Header),
+		AutoConfigHint:             AutoConfigHint(t.AutoConfigHint),
 	}
 }
 
@@ -234,6 +235,7 @@ func PollForDecisionTaskResponse(t *shared.PollForDecisionTaskResponse) *apiv1.P
 		Queries:                   WorkflowQueryMap(t.Queries),
 		NextEventId:               t.GetNextEventId(),
 		TotalHistoryBytes:         t.GetTotalHistoryBytes(),
+		AutoConfigHint:            AutoConfigHint(t.AutoConfigHint),
 	}
 }
 

diff --git a/internal/compatibility/proto/types.go b/internal/compatibility/proto/types.go
@@ -77,6 +77,16 @@ func ActivityType(t *shared.ActivityType) *apiv1.ActivityType {
 	}
 }
 
+func AutoConfigHint(t *shared.AutoConfigHint) *apiv1.AutoConfigHint {
+	if t == nil {
+		return nil
+	}
+	return &apiv1.AutoConfigHint{
+		EnableAutoConfig:   t.GetEnableAutoConfig(),
+		PollerWaitTimeInMs: t.GetPollerWaitTimeInMs(),
+	}
+}
+
 func WorkflowType(t *shared.WorkflowType) *apiv1.WorkflowType {
 	if t == nil {
 		return nil

diff --git a/internal/compatibility/testdata/common.go b/internal/compatibility/testdata/common.go
@@ -208,6 +208,10 @@ var (
 	ActivityLocalDispatchInfoMap = map[string]*apiv1.ActivityLocalDispatchInfo{
 		ActivityID: &ActivityLocalDispatchInfo,
 	}
+	AutoConfigHint = apiv1.AutoConfigHint{
+		EnableAutoConfig:   true,
+		PollerWaitTimeInMs: 100,
+	}
 	TaskListMetadata = apiv1.TaskListMetadata{
 		MaxTasksPerSecond: &gogo.DoubleValue{Value: RatePerSecond},
 	}

diff --git a/internal/compatibility/testdata/service.go b/internal/compatibility/testdata/service.go
@@ -199,6 +199,7 @@ var (
 		StartedTime:               Timestamp2,
 		Queries:                   WorkflowQueryMap,
 		NextEventId:               EventID3,
+		AutoConfigHint:            &AutoConfigHint,
 	}
 	RespondDecisionTaskCompletedRequest = apiv1.RespondDecisionTaskCompletedRequest{
 		TaskToken:                  TaskToken,
@@ -245,6 +246,7 @@ var (
 		WorkflowType:               &WorkflowType,
 		WorkflowDomain:             DomainName,
 		Header:                     &Header,
+		AutoConfigHint:             &AutoConfigHint,
 	}
 	RespondActivityTaskCompletedRequest = apiv1.RespondActivityTaskCompletedRequest{
 		TaskToken: TaskToken,