Skip to content

Commit deaf57b

Browse files
authored
Refactor to extractor classes (#4)
1 parent 796e6f9 commit deaf57b

6 files changed

Lines changed: 552 additions & 201 deletions

File tree

aws_lambda_opentelemetry/trace/helpers.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010
)
1111

1212
from aws_lambda_opentelemetry.typing.context import LambdaContext
13-
from aws_lambda_opentelemetry.utils import AwsAttributesMapper
13+
from aws_lambda_opentelemetry.utils import AwsAttributesExtractor
1414

1515

1616
def instrument_handler(**kwargs):
@@ -54,8 +54,8 @@ def wrapper(event: dict, context: LambdaContext):
5454
span.record_exception(exc)
5555
raise
5656
finally:
57-
mapper = AwsAttributesMapper(event, context)
58-
mapper.add_attributes()
57+
extractor = AwsAttributesExtractor(event, context)
58+
extractor.add_attributes()
5959
finally:
6060
provider.force_flush()
6161

0 commit comments

Comments
 (0)