diff --git a/internal-api/src/main/java/datadog/trace/api/gateway/InferredProxySpan.java b/internal-api/src/main/java/datadog/trace/api/gateway/InferredProxySpan.java index aef0fb81a70..0ef894ec095 100644 --- a/internal-api/src/main/java/datadog/trace/api/gateway/InferredProxySpan.java +++ b/internal-api/src/main/java/datadog/trace/api/gateway/InferredProxySpan.java @@ -46,6 +46,7 @@ public class InferredProxySpan implements ImplicitContextKeyed { SUPPORTED_PROXIES = new HashMap<>(); SUPPORTED_PROXIES.put("aws-apigateway", "aws.apigateway"); SUPPORTED_PROXIES.put("aws-httpapi", "aws.httpapi"); + SUPPORTED_PROXIES.put("azure-apim", "azure.apim"); } private final Map headers; diff --git a/internal-api/src/test/java/datadog/trace/api/gateway/InferredProxySpanTests.java b/internal-api/src/test/java/datadog/trace/api/gateway/InferredProxySpanTests.java index c9adcb83551..0d94ae67478 100644 --- a/internal-api/src/test/java/datadog/trace/api/gateway/InferredProxySpanTests.java +++ b/internal-api/src/test/java/datadog/trace/api/gateway/InferredProxySpanTests.java @@ -236,7 +236,10 @@ void testSupportedProxySystems(String proxySystem, String expectedSpanName) { } static Stream supportedProxySystems() { - return Stream.of(of("aws-apigateway", "aws.apigateway"), of("aws-httpapi", "aws.httpapi")); + return Stream.of( + of("aws-apigateway", "aws.apigateway"), + of("aws-httpapi", "aws.httpapi"), + of("azure-apim", "azure.apim")); } @Test