Fix ResetStickyTaskList call (#541)

Author: yiminc

evictiontest/workflow_cache_eviction_test.go

@@ -164,7 +164,7 @@ func (s *CacheEvictionSuite) TestResetStickyOnEviction() {
 	// this is the critical point of the test.
 	// ResetSticky should be called exactly once because our workflow cache evicts when full
 	// so if our worker puts *cacheSize* entries in the cache, it should evict exactly one
-	s.service.EXPECT().ResetStickyTaskList(gomock.Any(), gomock.Any()).DoAndReturn(mockResetStickyTaskList).Times(1)
+	s.service.EXPECT().ResetStickyTaskList(gomock.Any(), gomock.Any(), callOptions...).DoAndReturn(mockResetStickyTaskList).Times(1)
 
 	workflowWorker := internal.NewWorker(s.service, "test-domain", "tasklist", worker.Options{DisableActivityWorker: true})
 

internal/internal_task_pollers.go

@@ -256,8 +256,17 @@ func (wtp *workflowTaskPoller) processWorkflowTask(workflowTask *workflowTask) e
 }
 
 func (wtp *workflowTaskPoller) processResetStickinessTask(rst *resetStickinessTask) error {
-	_, err := wtp.service.ResetStickyTaskList(context.Background(), rst.task)
-	return err
+	tchCtx, cancel, opt := newChannelContext(context.Background())
+	defer cancel()
+	if _, err := wtp.service.ResetStickyTaskList(tchCtx, rst.task, opt...); err != nil {
+		wtp.logger.Warn("ResetStickyTaskList failed",
+			zap.String(tagWorkflowID, rst.task.Execution.GetWorkflowId()),
+			zap.String(tagRunID, rst.task.Execution.GetRunId()),
+			zap.Error(err))
+		return err
+	}
+
+	return nil
 }
 
 func (wtp *workflowTaskPoller) scheduleRespondDecisionTaskCompleted(wc WorkflowExecutionContext, workflowTask *workflowTask, startTime time.Time) {