It's not currently possibly follow an AWS Xray trace through a Lambda Function triggered by an SQS Queue.
Unless you use aws-xray-lambda-segment-shim
!
pip install aws-xray-lambda-segment-shim
from aws_xray_lambda_segment_shim import get_sqs_triggered_recorder
def lambda_handler(event, context):
for i, record in enumerate(event["Records"]):
recorder = get_sqs_triggered_recorder(
record=record,
lambda_request_id=context.aws_request_id,
lambda_arn=context.invoked_function_arn,
)
with recorder.in_segment():
with recorder.in_subsegment(f"SQS Record {i}") as subsegment:
print(
"I'm triggered by an SQS Record and using trace id ",
subsegment.trace_id,
)
TriggeredXrayRecorder
is a child class of aws_xray_sdk.AWSXRayRecorder
so you can use all the methods you would expect
from following the aws-xray-sdk documentation.
This approach causes all subsegments created with it to appear only in the trace that was passed in by SQS. There will still be a separate Lambda trace that will not contain these subsegments and will not show as being triggered by SQS.
This approach is useful if you are using SQS as an intermediary for a process you're already tracing as it then makes logical sense to view the traces from that starting point.
If you're more likely to view your traces as starting at the lambda function (i.e. you do not have any tracing prior to the SQS queue) then your mileage may vary with this approach.
We are also here working outside the scope of what is expected by the aws-xray-sdk. We are pretending to be AWS Lambda when we're initiating a trace, we're using undocumented fields to pretend to be AWS Lambda, and to allow the correlation of the SQS message and the Lambda Invocation (edge creation).
If this wasn't the only way to pursue a trace through SQS to lambda I would suggest you avoid it! But given the complexity involved in automating this from AWS's side, it may be a while before we see native support.
You can run the python tests with:
pip install -r requirements.txt make local-test
You can run the terraform tests with:
$ make install-xray-daemon-mac $ make run-xray-daemon-mac
Then in a new terminal run:
$ make terraform-init $ make terraform-apply $ make terraform-send-message
Once that's complete you can click on the xray URL provided at the end of the test to observe the trace is correct.