refactor aws lambda to use new semantics

Author: weslayer

instrumentation/opentelemetry-instrumentation-aws-lambda/src/opentelemetry/instrumentation/aws_lambda/__init__.py

@@ -86,8 +86,25 @@ def custom_event_context_extractor(lambda_event):
 from opentelemetry.instrumentation.utils import unwrap
 from opentelemetry.metrics import MeterProvider, get_meter_provider
 from opentelemetry.propagate import get_global_textmap
+from opentelemetry.semconv._incubating.attributes.cloud_attributes import (
+    CLOUD_RESOURCE_ID,
+)
+from opentelemetry.semconv._incubating.attributes.faas_attributes import (
+    FAAS_INVOCATION_ID,
+    FAAS_TRIGGER,
+)
+from opentelemetry.semconv._incubating.attributes.http_attributes import (
+    HTTP_METHOD,
+    HTTP_ROUTE,
+    HTTP_SCHEME,
+    HTTP_STATUS_CODE,
+    HTTP_TARGET,
+    HTTP_USER_AGENT,
+)
+from opentelemetry.semconv._incubating.attributes.net_attributes import (
+    NET_HOST_NAME,
+)
 from opentelemetry.semconv.resource import ResourceAttributes
-from opentelemetry.semconv.trace import SpanAttributes
 from opentelemetry.trace import (
     Span,
     SpanKind,
@@ -171,38 +188,34 @@ def _set_api_gateway_v1_proxy_attributes(
     More info:
     https://docs.aws.amazon.com/apigateway/latest/developerguide/set-up-lambda-proxy-integrations.html#api-gateway-simple-proxy-for-lambda-input-format
     """
-    span.set_attribute(
-        SpanAttributes.HTTP_METHOD, lambda_event.get("httpMethod")
-    )
+    span.set_attribute(HTTP_METHOD, lambda_event.get("httpMethod"))
 
     if lambda_event.get("headers"):
         if "User-Agent" in lambda_event["headers"]:
             span.set_attribute(
-                SpanAttributes.HTTP_USER_AGENT,
+                HTTP_USER_AGENT,
                 lambda_event["headers"]["User-Agent"],
             )
         if "X-Forwarded-Proto" in lambda_event["headers"]:
             span.set_attribute(
-                SpanAttributes.HTTP_SCHEME,
+                HTTP_SCHEME,
                 lambda_event["headers"]["X-Forwarded-Proto"],
             )
         if "Host" in lambda_event["headers"]:
             span.set_attribute(
-                SpanAttributes.NET_HOST_NAME,
+                NET_HOST_NAME,
                 lambda_event["headers"]["Host"],
             )
     if "resource" in lambda_event:
-        span.set_attribute(SpanAttributes.HTTP_ROUTE, lambda_event["resource"])
+        span.set_attribute(HTTP_ROUTE, lambda_event["resource"])
 
         if lambda_event.get("queryStringParameters"):
             span.set_attribute(
-                SpanAttributes.HTTP_TARGET,
+                HTTP_TARGET,
                 f"{lambda_event['resource']}?{urlencode(lambda_event['queryStringParameters'])}",
             )
         else:
-            span.set_attribute(
-                SpanAttributes.HTTP_TARGET, lambda_event["resource"]
-            )
+            span.set_attribute(HTTP_TARGET, lambda_event["resource"])
 
     return span
 
@@ -217,34 +230,34 @@ def _set_api_gateway_v2_proxy_attributes(
     """
     if "domainName" in lambda_event["requestContext"]:
         span.set_attribute(
-            SpanAttributes.NET_HOST_NAME,
+            NET_HOST_NAME,
             lambda_event["requestContext"]["domainName"],
         )
 
     if lambda_event["requestContext"].get("http"):
         if "method" in lambda_event["requestContext"]["http"]:
             span.set_attribute(
-                SpanAttributes.HTTP_METHOD,
+                HTTP_METHOD,
                 lambda_event["requestContext"]["http"]["method"],
             )
         if "userAgent" in lambda_event["requestContext"]["http"]:
             span.set_attribute(
-                SpanAttributes.HTTP_USER_AGENT,
+                HTTP_USER_AGENT,
                 lambda_event["requestContext"]["http"]["userAgent"],
             )
         if "path" in lambda_event["requestContext"]["http"]:
             span.set_attribute(
-                SpanAttributes.HTTP_ROUTE,
+                HTTP_ROUTE,
                 lambda_event["requestContext"]["http"]["path"],
             )
             if lambda_event.get("rawQueryString"):
                 span.set_attribute(
-                    SpanAttributes.HTTP_TARGET,
+                    HTTP_TARGET,
                     f"{lambda_event['requestContext']['http']['path']}?{lambda_event['rawQueryString']}",
                 )
             else:
                 span.set_attribute(
-                    SpanAttributes.HTTP_TARGET,
+                    HTTP_TARGET,
                     lambda_event["requestContext"]["http"]["path"],
                 )
 
@@ -319,11 +332,11 @@ def _instrumented_lambda_handler_call(  # noqa pylint: disable=too-many-branches
                     # See more:
                     # https://github.com/open-telemetry/semantic-conventions/blob/main/docs/faas/aws-lambda.md#resource-detector
                     span.set_attribute(
-                        SpanAttributes.CLOUD_RESOURCE_ID,
+                        CLOUD_RESOURCE_ID,
                         lambda_context.invoked_function_arn,
                     )
                     span.set_attribute(
-                        SpanAttributes.FAAS_INVOCATION_ID,
+                        FAAS_INVOCATION_ID,
                         lambda_context.aws_request_id,
                     )
 
@@ -354,7 +367,7 @@ def _instrumented_lambda_handler_call(  # noqa pylint: disable=too-many-branches
                 if isinstance(lambda_event, dict) and lambda_event.get(
                     "requestContext"
                 ):
-                    span.set_attribute(SpanAttributes.FAAS_TRIGGER, "http")
+                    span.set_attribute(FAAS_TRIGGER, "http")
 
                     if lambda_event.get("version") == "2.0":
                         _set_api_gateway_v2_proxy_attributes(
@@ -367,7 +380,7 @@ def _instrumented_lambda_handler_call(  # noqa pylint: disable=too-many-branches
 
                     if isinstance(result, dict) and result.get("statusCode"):
                         span.set_attribute(
-                            SpanAttributes.HTTP_STATUS_CODE,
+                            HTTP_STATUS_CODE,
                             result.get("statusCode"),
                         )
         finally: