Add IsLocalActivity to ActivityInfo

activity/activity.go

@@ -101,3 +101,8 @@ func GetHeartbeatDetails(ctx context.Context, d ...interface{}) error {
 func GetWorkerStopChannel(ctx context.Context) <-chan struct{} {
 	return internal.GetWorkerStopChannel(ctx)
 }
+
+// IsActivity check if the context is an activity context from a normal or local activity.
+func IsActivity(ctx context.Context) bool {
+	return internal.IsActivity(ctx)
+}

internal/activity.go

@@ -58,6 +58,7 @@ type (
 		StartedTime       time.Time     // Time of activity start
 		Deadline          time.Time     // Time of activity timeout
 		Attempt           int32         // Attempt starts from 1, and increased by 1 for every retry if retry policy is specified.
+		IsLocalActivity   bool          // true if it is a local activity
 	}
 
 	// RegisterActivityOptions consists of options for registering an activity
@@ -179,6 +180,12 @@ func HasHeartbeatDetails(ctx context.Context) bool {
 	return getActivityOutboundInterceptor(ctx).HasHeartbeatDetails(ctx)
 }
 
+// IsActivity check if the context is an activity context from a normal or local activity.
+func IsActivity(ctx context.Context) bool {
+	a := ctx.Value(activityInterceptorContextKey)
+	return a != nil
+}
+
 // GetHeartbeatDetails extract heartbeat details from last failed attempt. This is used in combination with retry policy.
 // An activity could be scheduled with an optional retry policy on ActivityOptions. If the activity failed then server
 // would attempt to dispatch another activity task to retry according to the retry policy. If there was heartbeat

internal/activity_test.go

@@ -241,3 +241,11 @@ func (s *activityTestSuite) TestGetWorkerStopChannel() {
 	channel := GetWorkerStopChannel(ctx)
 	s.NotNil(channel)
 }
+
+func (s *activityTestSuite) TestIsActivity() {
+	ctx := context.Background()
+	s.False(IsActivity(ctx))
+	ch := make(chan struct{}, 1)
+	ctx, _ = newActivityContext(context.Background(), nil, &activityEnvironment{workerStopChannel: ch})
+	s.True(IsActivity(ctx))
+}

internal/internal_activity.go

@@ -378,6 +378,7 @@ func (a *activityEnvironmentInterceptor) GetInfo(ctx context.Context) ActivityIn
 		Attempt:           a.env.attempt,
 		WorkflowType:      a.env.workflowType,
 		WorkflowNamespace: a.env.workflowNamespace,
+		IsLocalActivity:   a.env.isLocalActivity,
 	}
 }
 

test/activity_test.go

@@ -153,8 +153,12 @@ func (a *Activities) failNTimes(_ context.Context, times int, id int) error {
 	return errFailOnPurpose
 }
 
-func (a *Activities) InspectActivityInfo(ctx context.Context, namespace, taskQueue, wfType string) error {
+func (a *Activities) InspectActivityInfo(ctx context.Context, namespace, taskQueue, wfType string, isLocalActivity bool) error {
 	a.append("inspectActivityInfo")
+	if !activity.IsActivity(ctx) {
+		return fmt.Errorf("expected InActivity to return %v but got %v", true, activity.IsActivity(ctx))
+	}
+
 	info := activity.GetInfo(ctx)
 	if info.WorkflowNamespace != namespace {
 		return fmt.Errorf("expected namespace %v but got %v", namespace, info.WorkflowNamespace)
@@ -165,6 +169,9 @@ func (a *Activities) InspectActivityInfo(ctx context.Context, namespace, taskQue
 	if info.TaskQueue != taskQueue {
 		return fmt.Errorf("expected taskQueue %v but got %v", taskQueue, info.TaskQueue)
 	}
+	if info.IsLocalActivity != isLocalActivity {
+		return fmt.Errorf("expected IsLocalActivity %v but got %v", isLocalActivity, info.IsLocalActivity)
+	}
 	return nil
 }
 

test/workflow_test.go

@@ -1058,7 +1058,7 @@ func (w *Workflows) InspectActivityInfo(ctx workflow.Context) error {
 	wfType := info.WorkflowType.Name
 	taskQueue := info.TaskQueueName
 	ctx = workflow.WithActivityOptions(ctx, w.defaultActivityOptions())
-	return workflow.ExecuteActivity(ctx, "inspectActivityInfo", namespace, taskQueue, wfType).Get(ctx, nil)
+	return workflow.ExecuteActivity(ctx, "inspectActivityInfo", namespace, taskQueue, wfType, false).Get(ctx, nil)
 }
 
 func (w *Workflows) InspectLocalActivityInfo(ctx workflow.Context) error {
@@ -1069,7 +1069,7 @@ func (w *Workflows) InspectLocalActivityInfo(ctx workflow.Context) error {
 	ctx = workflow.WithLocalActivityOptions(ctx, w.defaultLocalActivityOptions())
 	var activities *Activities
 	return workflow.ExecuteLocalActivity(
-		ctx, activities.InspectActivityInfo, namespace, taskQueue, wfType).Get(ctx, nil)
+		ctx, activities.InspectActivityInfo, namespace, taskQueue, wfType, true).Get(ctx, nil)
 }
 
 func (w *Workflows) WorkflowWithLocalActivityCtxPropagation(ctx workflow.Context) (string, error) {