populate autoconfig hint field

Author: shijiesheng

internal/internal_task_handlers.go

@@ -85,15 +85,17 @@ type (
 	// workflowTask wraps a decision task.
 	workflowTask struct {
 		task            *s.PollForDecisionTaskResponse
+		autoConfigHint  *s.AutoConfigHint
 		historyIterator HistoryIterator
 		doneCh          chan struct{}
 		laResultCh      chan *localActivityResult
 	}
 
 	// activityTask wraps a activity task.
 	activityTask struct {
-		task          *s.PollForActivityTaskResponse
-		pollStartTime time.Time
+		task           *s.PollForActivityTaskResponse
+		autoConfigHint *s.AutoConfigHint
+		pollStartTime  time.Time
 	}
 
 	// resetStickinessTask wraps a ResetStickyTaskListRequest.
@@ -167,14 +169,14 @@ func (t *workflowTask) getAutoConfigHint() *s.AutoConfigHint {
 	if t.task != nil {
 		return t.task.AutoConfigHint
 	}
-	return nil
+	return t.autoConfigHint
 }
 
 func (t *activityTask) getAutoConfigHint() *s.AutoConfigHint {
 	if t.task != nil {
 		return t.task.AutoConfigHint
 	}
-	return nil
+	return t.autoConfigHint
 }
 
 func newHistory(task *workflowTask, eventsHandler *workflowExecutionEventHandlerImpl) *history {

internal/internal_task_pollers.go

@@ -848,7 +848,7 @@ func (wtp *workflowTaskPoller) poll(ctx context.Context) (interface{}, error) {
 	if response == nil || len(response.TaskToken) == 0 {
 		wtp.metricsScope.Counter(metrics.DecisionPollNoTaskCounter).Inc(1)
 		wtp.updateBacklog(request.TaskList.GetKind(), 0)
-		return &workflowTask{}, nil
+		return &workflowTask{autoConfigHint: response.GetAutoConfigHint()}, nil
 	}
 
 	wtp.updateBacklog(request.TaskList.GetKind(), response.GetBacklogCountHint())
@@ -1116,7 +1116,7 @@ func (atp *activityTaskPoller) pollWithMetrics(ctx context.Context,
 		return nil, err
 	}
 	if response == nil || len(response.TaskToken) == 0 {
-		return &activityTask{}, nil
+		return &activityTask{autoConfigHint: response.GetAutoConfigHint()}, nil
 	}
 
 	workflowType := response.WorkflowType.GetName()

internal/internal_worker_test.go

@@ -1514,13 +1514,25 @@ func TestGetTaskAutoConfigHint(t *testing.T) {
 		{
 			"decision task",
 			&workflowTask{
-				&shared.PollForDecisionTaskResponse{AutoConfigHint: &hint}, nil, nil, nil},
+				task: &shared.PollForDecisionTaskResponse{AutoConfigHint: &hint}},
+			&hint,
+		},
+		{
+			"empty decision task",
+			&workflowTask{
+				autoConfigHint: &hint},
 			&hint,
 		},
 		{
 			"activity task",
 			&activityTask{
-				&shared.PollForActivityTaskResponse{AutoConfigHint: &hint}, time.Now()},
+				task: &shared.PollForActivityTaskResponse{AutoConfigHint: &hint}},
+			&hint,
+		},
+		{
+			"empty activity task",
+			&activityTask{
+				autoConfigHint: &hint},
 			&hint,
 		},
 		{