o i~@s"UdZddlZddlZddlZddlZddlZddlZddlmZe e Z gZ ej ejgdfed<gZej ejgdfed<gZej ejgdfed<dZdadad d Zd d Zd*ddZd efddZdeejgdfd dfddZeeeZeee ZeeeZddZ ee eZ!ee e Z"ee eZ#e"ee#edejgdfd dfddZ$dejgdfd dfddZ%dejgdfd dfddZ&e'ed rej(eeed!e)d"Z*Gd#d$d$ej+ej,e*Z-e.Z/ej.e-ed%<d*d&d'Z0e"e0d ej1fd(d)Z2dS)+z( An API to provide fork-safe functions. N) _unpatched _registry_registry_before_fork_registry_after_parentTFcCdadSNT_forkedr r M/home/ubuntu/.local/lib/python3.10/site-packages/ddtrace/internal/forksafe.py set_forked&r cCtSNrr r r r has_forked,rreturncCrr _fork_childr r r r set_fork_child0r rcCrrrr r r r is_fork_child6rrregistryc Cs:t|D]}z|Wqtytd|YqwdS)Nz%Exception ignored in forksafe hook %r)list Exceptionlog exceptionrhookr r r run_hooks:s   rcCs|||Sr)appendrr r r register_hookHs r after_in_childcC4zt|WdStytd|jYdSw)NzFafter_in_child hook %s was unregistered without first being registered)rremove ValueErrorrinfo__name__)r!r r r unregisterU  r'after_in_parentcCr")NzGafter_in_parent hook %s was unregistered without first being registered)rr#r$rr%r&)r)r r r unregister_parent\r(r* before_forkcCr")NzGbefore_in_child hook %s was unregistered without first being registered)rr#r$rr%r&)r+r r r unregister_before_forkcr(r,register_at_fork)beforer!r)_Tcs8eZdZdZdeeddffdd Zd ddZZS) ResetObjecta`An object wrapper object that is fork-safe and resets itself after a fork. When a Python process forks, a Lock can be in any state, locked or not, by any thread. Since after fork all threads are gone, Lock objects needs to be reset. CPython does this with an internal `threading._after_fork` function. We use the same mechanism here. wrapped_classrNcs&tt||||_t|dSr)superr0__init___self_wrapped_class_resetable_objectsadd)selfr1 __class__r r r3~szResetObject.__init__cCs||_dSr)r4 __wrapped__)r7r r r _reset_objectszResetObject._reset_objectrN) r& __module__ __qualname____doc__typer/r3r; __classcell__r r r8r r0usr0r5c Cs<ttD]}z|Wqtytd|YqwdS)Nz2Exception ignored in object reset forksafe hook %r)rr5r;rrr)objr r r _reset_objectss   rCcCs ttjSr)r0rthreading_Eventr r r r Events rEr<)3r? functoolsloggingostypingweakrefwraptddtrace.internalr getLoggerr&rrListCallable__annotations__rr_softr rr rrboolrrrpartialddtrace_before_forkddtrace_after_in_childddtrace_after_in_parentr register_before_forkregisterregister_after_parentr'r*r,hasattrr-TypeVarr/ ObjectProxyGenericr0WeakSetr5rCrDrEr r r r sT