o iu@snddlmZddlmZmZddlmZddlZddlZ Gddde Z da e Z ddZ d d Zd d ZdS) ) ThreadStats)LockThread)apiNc@sLeZdZdZdZeZeZdZddZ e ddZ e dd Z d d Z d S) _LambdaDecoratorzY DEPRECATED Decorator to automatically init & flush metrics, created for Lambda functionsrFcCs ||_dS)N)func)selfrr R/home/ubuntu/.local/lib/python3.10/site-packages/datadog/threadstats/aws_lambda.py__init__s z_LambdaDecorator.__init__cCs|j5|js*d|_tjdtjdt_tjddt_tt d}| t |j d|_ WddS1s;wYdS)NTDATADOG_API_KEY DD_API_KEY DATADOG_HOSTzhttps://api.datadoghq.com)target) _counter_lock_was_initializedosenvirongetr_api_key _api_hostr_init_api_clientstart_get_lambda_stats_counter)clstr r r _enter"s "z_LambdaDecorator._enterc Csd}|j|jd|_|jdkrd}Wdn1swY|rf|j5|j|jdkr3d}Wdn1s=wY|rTttdWddSWddS1s_wYdSdS)NFrrTinf)rr _flush_lockrflushfloat)r should_flushr r r _close4s(   "z_LambdaDecorator._closecOs:tdttz |j|i|WtStw)Nz=datadog_lambda_wrapper() is relocated to https://git.io/fjy8o)warningswarnDeprecationWarningrrrr$)rargskwr r r __call__Gs z_LambdaDecorator.__call__N)__name__ __module__ __qualname____doc__rrrr rr classmethodrr$r*r r r r rs   rcCs tdurtatjdddtS)NF)flush_in_greenletflush_in_thread) _lambda_statsrrr r r r rTsrcOstj|i|dS)z8 Alias to expose only distributions for lambda functionsN)r distribution)r(r)r r r lambda_metric]sr4cCs,z tjjddWdStyYdSw)aNo-op GET to initialize the requests connection with DD's endpoints The goal here is to make the final flush faster: we keep alive the Requests session, this means that we can re-use the connection The consequence is that the HTTP Handshake, which can take hundreds of ms, is now made at the beginning of a lambda instead of at the end. By making the initial request async, we spare a lot of execution time in the lambdas. GETvalidateN)r api_client APIClientsubmit Exceptionr r r r rbs   r)datadog.threadstatsr threadingrrdatadogrrr%objectrr2datadog_lambda_wrapperrr4rr r r r s   :