Update Azure Function context

elastic · Apr 1, 2024 · bde44c9 · bde44c9
1 parent 1c29919
commit bde44c9
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/azure/Elastic.Apm.Azure.Functions/AzureFunctionsContext.cs b/src/azure/Elastic.Apm.Azure.Functions/AzureFunctionsContext.cs
@@ -18,7 +18,7 @@ internal AzureFunctionsContext(string loggerScopeName)
 	{
 		Logger = Agent.Instance.Logger.Scoped(loggerScopeName);
 		MetaData = AzureFunctionsMetadataProvider.GetAzureFunctionsMetaData(Logger);
-		UpdateServiceInformation(Agent.Instance.Service);
+		UpdateServiceInformation(Agent.Instance.Service, Agent.Instance.Configuration.OverwriteDiscoverDefaultServiceName);
 		FaasIdPrefix =
 			$"/subscriptions/{MetaData.SubscriptionId}/resourceGroups/{MetaData.WebsiteResourceGroup}/providers/Microsoft.Web/sites/{MetaData.WebsiteSiteName}/functions/";
 		Logger.Trace()?.Log("FaasIdPrefix: {FaasIdPrefix}", FaasIdPrefix);
@@ -32,15 +32,15 @@ internal AzureFunctionsContext(string loggerScopeName)
 
 	internal static bool IsColdStart() => Interlocked.Exchange(ref ColdStart, 0) == 1;
 
-	private void UpdateServiceInformation(Service? service)
+	private void UpdateServiceInformation(Service? service, bool overwriteDiscoverDefaultServiceName)
 	{
 		if (service == null)
 		{
 			Logger.Warning()?.Log($"{nameof(UpdateServiceInformation)}: service is null");
 			return;
 		}
 
-		if (service.Name == AbstractConfigurationReader.AdaptServiceName(AbstractConfigurationReader.DiscoverDefaultServiceName()))
+		if (overwriteDiscoverDefaultServiceName && service.Name == AbstractConfigurationReader.AdaptServiceName(AbstractConfigurationReader.DiscoverDefaultServiceName()))
 		{
 			// Only override the service name if it was set to default.
 			service.Name = MetaData.WebsiteSiteName;