https://github.com/sam-martin/aws-xray-lambda-segment-shim

A shim which faciliates the AWS X-Ray tracing of Lambda Functions invoked by triggers which don't have native X-Ray support.
https://github.com/sam-martin/aws-xray-lambda-segment-shim

aws aws-lambda aws-lambda-python aws-xray tracing

Last synced: 10 months ago
JSON representation

A shim which faciliates the AWS X-Ray tracing of Lambda Functions invoked by triggers which don't have native X-Ray support.

• Host: GitHub
• URL: https://github.com/sam-martin/aws-xray-lambda-segment-shim
• Owner: Sam-Martin
• License: mit
• Created: 2021-08-21T18:12:09.000Z (over 4 years ago)
• Default Branch: main
• Last Pushed: 2021-08-28T10:14:10.000Z (over 4 years ago)
• Last Synced: 2025-04-30T07:49:26.268Z (10 months ago)
• Topics: aws, aws-lambda, aws-lambda-python, aws-xray, tracing
• Language: Python
• Homepage:
• Size: 165 KB
• Stars: 7
• Watchers: 1
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.rst
  • License: LICENSE
  • Code of conduct: CODE_OF_CONDUCT.md

Awesome Lists containing this project

README

          
aws-xray-lambda-segment-shim

=====================================

|shield1| |shield2|

.. |shield1| image:: https://img.shields.io/github/workflow/status/sam-martin/aws-xray-lambda-segment-shim/Linting%20&%20Testing?style=flat-square

    :target: https://github.com/Sam-Martin/aws-xray-lambda-segment-shim/actions/workflows/continuous-integration.yml?query=branch%3Amain+

    :alt: GitHub Workflow Status

.. |shield2|  image:: https://img.shields.io/pypi/v/aws-xray-lambda-segment-shim?style=flat-square

    :target: https://pypi.org/project/aws-xray-lambda-segment-shim/

    :alt: PyPI

.. image:: https://github.com/Sam-Martin/aws-xray-lambda-segment-shim/blob/main/images/example.png?raw=true

| **Full Docs:** https://aws-xray-lambda-segment-shim.readthedocs.io

| **GitHub:** https://github.com/Sam-Martin/aws-xray-sqs-lambda-segment-shim

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``!

Installation

----------------

.. code-block::

    pip install aws-xray-lambda-segment-shim

Usage

------

.. doctest::

    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 `__.

Caveats

----------

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.

- `Issue on Python SDK `__

- `Issue on .NET SDK `__

- `Issue on Node SDK `__

Testing

---------

You can run the python tests with:

.. code-block::

    pip install -r requirements.txt

    make local-test

Terraform Tests

""""""""""""""""""

You can run the terraform tests with:

.. code-block::

    $ make install-xray-daemon-mac

    $ make run-xray-daemon-mac

Then in a new terminal run:

.. code-block::

    $ 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.