Revert "[SLS-1674] Additional attributes for different managed services (#200)"

This reverts commit ba58ecd.

Author: hghotra

datadog_lambda/tracing.py

@@ -7,13 +7,7 @@
 import os
 import json
 from datetime import datetime, timezone
-from typing import Optional, Dict
-
-try:
-    from typing import Literal
-except ImportError:
-    # Literal was added to typing in python 3.8
-    from typing_extensions import Literal
+from typing import Optional, Literal, Dict
 
 from datadog_lambda.constants import (
     SamplingPriority,
@@ -448,30 +442,24 @@ def is_api_gateway_invocation_async(event):
 
 
 def create_inferred_span_from_api_gateway_websocket_event(event, context):
-    request_context = event["requestContext"]
-    domain = request_context["domainName"]
-    endpoint = request_context["routeKey"]
+    domain = event["requestContext"]["domainName"]
+    endpoint = event["requestContext"]["routeKey"]
     tags = {
         "operation_name": "aws.apigateway.websocket",
         "http.url": domain + endpoint,
         "endpoint": endpoint,
-        "resource_names": endpoint,
-        "apiid": request_context["apiId"],
-        "apiname": request_context["apiId"],
-        "stage": request_context["stage"],
-        "request_id": request_context["requestId"],
-        "connection_id": request_context["connectionId"],
-        "event_type": request_context["eventType"],
-        "message_direction": request_context["messageDirection"],
+        "resource_names": domain + endpoint,
+        "request_id": context.aws_request_id,
+        "connection_id": event["requestContext"]["connectionId"],
     }
-    request_time_epoch = request_context["requestTimeEpoch"]
     if is_api_gateway_invocation_async(event):
         InferredSpanInfo.set_tags(tags, tag_source="self", synchronicity="async")
     else:
         InferredSpanInfo.set_tags(tags, tag_source="self", synchronicity="sync")
+    request_time_epoch = event["requestContext"]["requestTimeEpoch"]
     args = {
         "service": domain,
-        "resource": endpoint,
+        "resource": domain + endpoint,
         "span_type": "web",
     }
     tracer.set_tags({"_dd.origin": "lambda"})
@@ -483,30 +471,24 @@ def create_inferred_span_from_api_gateway_websocket_event(event, context):
 
 
 def create_inferred_span_from_api_gateway_event(event, context):
-    request_context = event["requestContext"]
-    domain = request_context["domainName"]
-    method = event["httpMethod"]
+    domain = event["requestContext"]["domainName"]
     path = event["path"]
-    resource = "{0} {1}".format(method, path)
     tags = {
         "operation_name": "aws.apigateway.rest",
         "http.url": domain + path,
         "endpoint": path,
-        "http.method": method,
-        "resource_names": resource,
-        "apiid": request_context["apiId"],
-        "apiname": request_context["apiId"],
-        "stage": request_context["stage"],
-        "request_id": request_context["requestId"],
+        "http.method": event["httpMethod"],
+        "resource_names": domain + path,
+        "request_id": context.aws_request_id,
     }
-    request_time_epoch = request_context["requestTimeEpoch"]
     if is_api_gateway_invocation_async(event):
         InferredSpanInfo.set_tags(tags, tag_source="self", synchronicity="async")
     else:
         InferredSpanInfo.set_tags(tags, tag_source="self", synchronicity="sync")
+    request_time_epoch = event["requestContext"]["requestTimeEpoch"]
     args = {
         "service": domain,
-        "resource": resource,
+        "resource": domain + path,
         "span_type": "http",
     }
     tracer.set_tags({"_dd.origin": "lambda"})
@@ -518,33 +500,24 @@ def create_inferred_span_from_api_gateway_event(event, context):
 
 
 def create_inferred_span_from_http_api_event(event, context):
-    request_context = event["requestContext"]
-    domain = request_context["domainName"]
-    method = request_context["http"]["method"]
+    domain = event["requestContext"]["domainName"]
     path = event["rawPath"]
-    resource = "{0} {1}".format(method, path)
     tags = {
         "operation_name": "aws.httpapi",
-        "endpoint": path,
         "http.url": domain + path,
-        "http.method": request_context["http"]["method"],
-        "http.protocol": request_context["http"]["protocol"],
-        "http.source_ip": request_context["http"]["sourceIp"],
-        "http.user_agent": request_context["http"]["userAgent"],
-        "resource_names": resource,
+        "endpoint": path,
+        "http.method": event["requestContext"]["http"]["method"],
+        "resource_names": domain + path,
         "request_id": context.aws_request_id,
-        "apiid": request_context["apiId"],
-        "apiname": request_context["apiId"],
-        "stage": request_context["stage"],
     }
-    request_time_epoch = request_context["timeEpoch"]
     if is_api_gateway_invocation_async(event):
         InferredSpanInfo.set_tags(tags, tag_source="self", synchronicity="async")
     else:
         InferredSpanInfo.set_tags(tags, tag_source="self", synchronicity="sync")
+    request_time_epoch = event["requestContext"]["timeEpoch"]
     args = {
         "service": domain,
-        "resource": resource,
+        "resource": domain + path,
         "span_type": "http",
     }
     tracer.set_tags({"_dd.origin": "lambda"})
@@ -557,15 +530,10 @@ def create_inferred_span_from_http_api_event(event, context):
 
 def create_inferred_span_from_sqs_event(event, context):
     event_record = get_first_record(event)
-    event_source_arn = event_record["eventSourceARN"]
-    queue_name = event_source_arn.split(":")[-1]
+    queue_name = event_record["eventSourceARN"].split(":")[-1]
     tags = {
         "operation_name": "aws.sqs",
         "resource_names": queue_name,
-        "queuename": queue_name,
-        "event_source_arn": event_source_arn,
-        "receipt_handle": event_record["receiptHandle"],
-        "sender_id": event_record["attributes"]["SenderId"],
     }
     InferredSpanInfo.set_tags(tags, tag_source="self", synchronicity="async")
     request_time_epoch = event_record["attributes"]["SentTimestamp"]
@@ -584,18 +552,10 @@ def create_inferred_span_from_sqs_event(event, context):
 
 def create_inferred_span_from_sns_event(event, context):
     event_record = get_first_record(event)
-    sns_message = event_record["Sns"]
-    topic_arn = event_record["Sns"]["TopicArn"]
-    topic_name = topic_arn.split(":")[-1]
+    topic_name = event_record["Sns"]["TopicArn"].split(":")[-1]
     tags = {
         "operation_name": "aws.sns",
         "resource_names": topic_name,
-        "topicname": topic_name,
-        "topic_arn": topic_arn,
-        "message_id": sns_message["MessageId"],
-        "type": sns_message["Type"],
-        "subject": sns_message["Subject"],
-        "event_subscription_arn": event_record["EventSubscriptionArn"],
     }
     InferredSpanInfo.set_tags(tags, tag_source="self", synchronicity="async")
     sns_dt_format = "%Y-%m-%dT%H:%M:%S.%fZ"
@@ -617,20 +577,10 @@ def create_inferred_span_from_sns_event(event, context):
 
 def create_inferred_span_from_kinesis_event(event, context):
     event_record = get_first_record(event)
-    event_source_arn = event_record["eventSourceARN"]
-    event_id = event_record["eventID"]
-    stream_name = event_source_arn.split(":")[-1]
-    shard_id = event_id.split(":")[0]
+    stream_name = event_record["eventSourceARN"].split(":")[-1]
     tags = {
         "operation_name": "aws.kinesis",
         "resource_names": stream_name,
-        "streamname": stream_name,
-        "shardid": shard_id,
-        "event_source_arn": event_source_arn,
-        "event_id": event_id,
-        "event_name": event_record["eventName"],
-        "event_version": event_record["eventVersion"],
-        "partition_key": event_record["kinesis"]["partitionKey"],
     }
     InferredSpanInfo.set_tags(tags, tag_source="self", synchronicity="async")
     request_time_epoch = event_record["kinesis"]["approximateArrivalTimestamp"]
@@ -650,22 +600,14 @@ def create_inferred_span_from_kinesis_event(event, context):
 
 def create_inferred_span_from_dynamodb_event(event, context):
     event_record = get_first_record(event)
-    event_source_arn = event_record["eventSourceARN"]
-    table_name = event_source_arn.split("/")[1]
-    dynamodb_message = event_record["dynamodb"]
+    table_name = event_record["eventSourceARN"].split("/")[1]
     tags = {
         "operation_name": "aws.dynamodb",
         "resource_names": table_name,
-        "tablename": table_name,
-        "event_source_arn": event_source_arn,
-        "event_id": event_record["eventID"],
-        "event_name": event_record["eventName"],
-        "event_version": event_record["eventVersion"],
-        "stream_view_type": dynamodb_message["StreamViewType"],
-        "size_bytes": dynamodb_message["SizeBytes"],
     }
     InferredSpanInfo.set_tags(tags, synchronicity="async", tag_source="self")
     request_time_epoch = event_record["dynamodb"]["ApproximateCreationDateTime"]
+
     args = {
         "service": "dynamodb",
         "resource": table_name,
@@ -675,7 +617,6 @@ def create_inferred_span_from_dynamodb_event(event, context):
     span = tracer.trace("aws.dynamodb", **args)
     if span:
         span.set_tags(tags)
-
     span.start = int(request_time_epoch)
     return span
 
@@ -686,12 +627,6 @@ def create_inferred_span_from_s3_event(event, context):
     tags = {
         "operation_name": "aws.s3",
         "resource_names": bucket_name,
-        "event_name": event_record["eventName"],
-        "bucketname": bucket_name,
-        "bucket_arn": event_record["s3"]["bucket"]["arn"],
-        "object_key": event_record["s3"]["object"]["key"],
-        "object_size": event_record["s3"]["object"]["size"],
-        "object_etag": event_record["s3"]["etag"],
     }
     InferredSpanInfo.set_tags(tags, synchronicity="async", tag_source="self")
     dt_format = "%Y-%m-%dT%H:%M:%S.%fZ"
@@ -716,7 +651,6 @@ def create_inferred_span_from_eventbridge_event(event, context):
     tags = {
         "operation_name": "aws.eventbridge",
         "resource_names": source,
-        "detail_type": event["detail-type"],
     }
     InferredSpanInfo.set_tags(
         tags,

pyproject.toml

@@ -28,7 +28,6 @@ datadog = "^0.41.0"
 wrapt = "^1.11.2"
 ddtrace = "^0.50.0"
 importlib_metadata = {version = "^1.0", python = "<3.8"}
-typing_extensions = {version = "^4.0", python = "<3.8"}
 boto3 = { version = "^1.10.33", optional = true }
 requests = { version ="^2.22.0", optional = true }
 nose2 = { version= "^0.9.1", optional = true }