Skip to content

Commit

Permalink
feat: support chained triggers in sqs ReceiveMessage (#234)
Browse files Browse the repository at this point in the history
* feat: support chained triggers in sqs ReceiveMessage
  • Loading branch information
saartochner-lumigo committed Jan 24, 2023
1 parent 6cf052d commit d29e47a
Show file tree
Hide file tree
Showing 2 changed files with 15 additions and 0 deletions.
1 change: 1 addition & 0 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -17,6 +17,7 @@
"autowrapt>=1.0",
"protobuf>=3.13.0, <4.0.0",
"wrapt>=1.11.0",
"lumigo_tracer==1.1.213",
"opentelemetry-api==1.9.1",
"opentelemetry-sdk==1.9.1",
"opentelemetry-sdk-extension-aws==2.0.1",
Expand Down
Original file line number Diff line number Diff line change
@@ -1,8 +1,10 @@
from __future__ import annotations

import json
from typing import Optional, Type, Dict, Any

from opentelemetry.trace import Span
from lumigo_tracer.event.event_trigger import parse_triggers

from lumigo_opentelemetry.libs.general_utils import lumigo_safe_execute
from lumigo_opentelemetry.libs.json_utils import dump
Expand Down Expand Up @@ -127,6 +129,18 @@ def parse_request(
}
span.set_attributes(attributes)

@staticmethod
def parse_response(
span: Span, service_name: str, operation_name: str, result: Dict[Any, Any]
) -> None:
trigger_details = parse_triggers(
{"service_name": service_name, "operation_name": operation_name, **result}
)
if trigger_details:
span.set_attributes(
{"lumigoData": json.dumps({"trigger": trigger_details})}
)


class LambdaParser(AwsParser):
@staticmethod
Expand Down

0 comments on commit d29e47a

Please sign in to comment.