chore: Detect Azure Function and AWS Lambda and report as custom inst…

…all type
newrelic · Sep 18, 2024 · 0b37308 · 0b37308
1 parent 80cf9ef
commit 0b37308
Showing 1 changed file with 28 additions and 0 deletions.
diff --git a/src/Agent/NewRelic/Agent/Core/AgentInstallConfiguration.cs b/src/Agent/NewRelic/Agent/Core/AgentInstallConfiguration.cs
@@ -202,9 +202,37 @@ public static AgentInfo GetAgentInfo()
                 }
             }
             else
+            {
+                if (TryGetAzureFunctionInstallType(out var agentInfo))
+                    return agentInfo;
+
+                if (TryGetLambdaServerlessInstallType(out agentInfo))
+                    return agentInfo;
+
                 Log.Debug($"Could not get agent info from {agentInfoPath}. File does not exist.");
+            }
 
             return null;
         }
+
+        private static bool TryGetLambdaServerlessInstallType(out AgentInfo agentInfo)
+        {
+            agentInfo = null;
+            if (System.Environment.GetEnvironmentVariable("AWS_LAMBDA_FUNCTION_NAME") != null)
+            {
+                agentInfo = new AgentInfo { InstallType = "Lambda" };
+            }
+            return agentInfo != null;
+        }
+
+        private static bool TryGetAzureFunctionInstallType(out AgentInfo agentInfo)
+        {
+            agentInfo = null;
+            if (System.Environment.GetEnvironmentVariable("FUNCTIONS_WORKER_RUNTIME") != null)
+            {
+                agentInfo = new AgentInfo { InstallType = "AzureFunction" };
+            }
+            return agentInfo != null;
+        }
     }
 }