lambda_timeout.py

import time

from thundra.thundra_agent import Thundra
from thundra.plugins.trace.traceable import Traceable
THUNDRA_API_KEY = os.environ.get('THUNDRA_API_KEY', '')
thundra = Thundra(api_key=THUNDRA_API_KEY)


@Traceable(trace_args=True, trace_return_value=True)
def _action_a():
    time.sleep(1)
    print('action A')

@Traceable(trace_args=True, trace_return_value=True)
def _action_b():
    time.sleep(2)
    print('action B')

@Traceable(trace_args=True, trace_return_value=True)
def _action_c():
    time.sleep(30)
    print('action C')

@thundra
def handler(event, context):
    _action_a()
    _action_b()
    _action_c()
        
    return