aws_xray_sdk_quickstart_patch_all_with_subsegment_tracing.py

1from aws_xray_sdk.core import xray_recorder
2from aws_xray_sdk.core import patch_all
3
4# Patch all supported libraries (e.g., requests, boto3, pymysql) 
5# to automatically trace downstream calls
6patch_all()
7
8# Configure the recorder to use a specific service name
9xray_recorder.configure(service='MyApplication')
10
11def main():
12    # Start a segment (the main unit of work)
13    segment = xray_recorder.begin_segment('main_logic')
14    
15    try:
16        # Start a subsegment to trace a specific block of code
17        with xray_recorder.in_subsegment('processing_step') as subsegment:
18            # Your application logic here
19            print("Processing data...")
20            
21    except Exception as e:
22        # The recorder automatically captures exceptions when using the context manager
23        raise e
24    finally:
25        # Always end the segment
26        xray_recorder.end_segment()
27
28if __name__ == '__main__':
29    main()