o i @sxddlmZddlmZzddlmZWney#ddlmZYnwddlmZGddde Z Gd d d e Z d S) )wraps)iscoroutinefunction) monotonic)time)_get_wrapped_coc@sJeZdZdZdddZddZdd Zd d Zd d ZddZ ddZ dS)TimedContextManagerDecoratorzx A context manager and a decorator which will report the elapsed time in the context OR in a function call. NrcCs0||_|j|_||_||_||_||_d|_dSN)statsdtiming timing_funcmetrictags sample_rateuse_mselapsedselfr r rrrrT/home/ubuntu/.local/lib/python3.10/site-packages/ddtrace/vendor/dogstatsd/context.py__init__s z%TimedContextManagerDecorator.__init__csDjs djjf_trtStfdd}|S)z Decorator which returns the elapsed time of the function call. Default to the function name if metric was not provided. z%s.%sc s.t}z |i|W|S|wr )r_send)argskwargsstartfuncrrrwrapped.sz6TimedContextManagerDecorator.__call__..wrapped)r __module____name__rrr)rrrrrr__call__ s z%TimedContextManagerDecorator.__call__cCs|jstdt|_|S)Nz#Cannot used timed without a metric!)r TypeErrorr_startrrrr __enter__8sz&TimedContextManagerDecorator.__enter__cCs||jdSr )rr")rtypevalue tracebackrrr__exit__>sz%TimedContextManagerDecorator.__exit__cCsZt|}|jdur |jn|jj}|rttd|n|}||j||j|j||_ dS)Ni) rrr introundr r rrr)rrrrrrrrBs  z"TimedContextManagerDecorator._sendcCs |dSr )r$r#rrrrIs z"TimedContextManagerDecorator.startcCs|ddddSr )r(r#rrrstopLsz!TimedContextManagerDecorator.stopNNrN) rr __qualname____doc__rr r$r(rrr+rrrrrs   rcs"eZdZdZdfdd ZZS)"DistributedContextManagerDecoratorz A context manager and a decorator which will report the elapsed time in the context OR in a function call using the custom distribution metric. Nrcs$tt|||||||j|_dSr )superr/r distributionr r __class__rrrVs z+DistributedContextManagerDecorator.__init__r,)rrr-r.r __classcell__rrr2rr/Psr/N) functoolsrinspectrrr ImportError context_asyncrobjectrr/rrrrs    ?