add a task interface

internal/internal_task_handlers.go

@@ -59,6 +59,11 @@ const (
 	defaultMediumLivedWorkflowTimeoutUpperLimitInSec = 8 * 3600
 )
 
+var (
+	_ autoConfigHintAwareTask = (*workflowTask)(nil)
+	_ autoConfigHintAwareTask = (*activityTask)(nil)
+)
+
 type (
 	// workflowExecutionEventHandler process a single event.
 	workflowExecutionEventHandler interface {
@@ -72,6 +77,11 @@ type (
 		Close()
 	}
 
+	// autoConfigHintAwareTask is a task that can provide auto config hint
+	autoConfigHintAwareTask interface {
+		getAutoConfigHint() *s.AutoConfigHint
+	}
+
 	// workflowTask wraps a decision task.
 	workflowTask struct {
 		task            *s.PollForDecisionTaskResponse
@@ -153,6 +163,20 @@ type (
 	}
 )
 
+func (t *workflowTask) getAutoConfigHint() *s.AutoConfigHint {
+	if t.task != nil {
+		return t.task.AutoConfigHint
+	}
+	return nil
+}
+
+func (t *activityTask) getAutoConfigHint() *s.AutoConfigHint {
+	if t.task != nil {
+		return t.task.AutoConfigHint
+	}
+	return nil
+}
+
 func newHistory(task *workflowTask, eventsHandler *workflowExecutionEventHandlerImpl) *history {
 	result := &history{
 		workflowTask:  task,

internal/internal_worker_base.go

@@ -332,8 +332,8 @@ func (bw *baseWorker) pollTask() {
 			bw.retrier.Failed()
 		} else {
 			bw.retrier.Succeeded()
-			if bw.concurrencyAutoScaler != nil {
-				bw.concurrencyAutoScaler.ProcessPollerHint(getAutoConfigHint(task))
+			if t, ok :=task.(autoConfigHintAwareTask); bw.concurrencyAutoScaler != nil && ok {
+				bw.concurrencyAutoScaler.ProcessPollerHint(t.getAutoConfigHint())
 			}
 		}
 	}