App Insights Tags for Orchestrators and Entities

src/WebJobs.Extensions.DurableTask/Listener/TaskEntityShim.cs

@@ -148,6 +148,7 @@ internal void Rehydrate(string serializedInput)
 
         public override async Task<string> Execute(OrchestrationContext innerContext, string serializedInput)
         {
+            string status;
             if (this.operationBatch.Count == 0 && this.lockRequest == null)
             {
                 // we are idle after a ContinueAsNew - the batch is empty.
@@ -167,6 +168,7 @@ public override async Task<string> Execute(OrchestrationContext innerContext, st
                 this.Config.GetIntputOutputTrace(serializedInput),
                 FunctionType.Entity,
                 isReplay: false);
+            status = "Running";
 
             if (this.NumberEventsToReceive > 0)
             {
@@ -211,6 +213,8 @@ public override async Task<string> Execute(OrchestrationContext innerContext, st
                     description,
                     functionType: FunctionType.Entity,
                     isReplay: false);
+                status = "Failed";
+
             }
             else
             {
@@ -222,7 +226,21 @@ public override async Task<string> Execute(OrchestrationContext innerContext, st
                     continuedAsNew: true,
                     functionType: FunctionType.Entity,
                     isReplay: false);
+                status = "Completed";
+            }
+
+#if !FUNCTIONS_V1
+            // Adding "Tags" to activity allows using App Insights to query current state of entities 
+            var activity = Activity.Current;
+
+            // The activity may be null when running unit tests, but should be non-null otherwise
+            if (activity != null)
+            {
+                activity.AddTag("DurableFunctionsType", "Entity");
+                activity.AddTag("DurableFunctionsInstanceId", this.context.InstanceId);
+                activity.AddTag("DurableFunctionsRuntimeStatus", status);
             }
+#endif
 
             // The return value is not used.
             return string.Empty;

src/WebJobs.Extensions.DurableTask/Listener/TaskOrchestrationShim.cs

@@ -49,6 +49,7 @@ public override void RaiseEvent(OrchestrationContext unused, string eventName, s
 
         public override async Task<string> Execute(OrchestrationContext innerContext, string serializedInput)
         {
+            string status;
             if (this.FunctionInvocationCallback == null)
             {
                 throw new InvalidOperationException($"The {nameof(this.FunctionInvocationCallback)} has not been assigned!");
@@ -69,6 +70,7 @@ public override async Task<string> Execute(OrchestrationContext innerContext, st
                 this.Config.GetIntputOutputTrace(serializedInput),
                 FunctionType.Orchestrator,
                 this.context.IsReplaying);
+            status = "Running";
 
             var orchestratorInfo = this.Config.GetOrchestratorInfo(new FunctionName(this.context.Name));
 
@@ -106,6 +108,8 @@ public override async Task<string> Execute(OrchestrationContext innerContext, st
                     exceptionDetails,
                     FunctionType.Orchestrator,
                     this.context.IsReplaying);
+                status = "Failed";
+
 
                 if (!this.context.IsReplaying)
                 {
@@ -169,6 +173,8 @@ public override async Task<string> Execute(OrchestrationContext innerContext, st
                 this.context.ContinuedAsNew,
                 FunctionType.Orchestrator,
                 this.context.IsReplaying);
+            status = "Completed";
+
             if (!this.context.IsReplaying)
             {
                 this.context.AddDeferredTask(() => this.Config.LifeCycleNotificationHelper.OrchestratorCompletedAsync(
@@ -179,6 +185,19 @@ public override async Task<string> Execute(OrchestrationContext innerContext, st
                     this.context.IsReplaying));
             }
 
+#if !FUNCTIONS_V1
+            // Adding "Tags" to activity allows using App Insights to query current state of orchestrations 
+            var activity = Activity.Current;
+
+            // The activity may be null when running unit tests, but should be non-null otherwise
+            if (activity != null)
+            {
+                activity.AddTag("DurableFunctionsType", "Orchestrator");
+                activity.AddTag("DurableFunctionsInstanceId", context.InstanceId);
+                activity.AddTag("DurableFunctionsRuntimeStatus", status);
+            }
+#endif
+
             return serializedOutput;
         }
     }