o i+ @sddlZddlmZddlmZddlZddlZddlZddlZddlZddlm Z ddlm Z ddlm Z ddl Z ddl mZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZddlmZddlmZddlmZddlmZddlm Z ddlm!Z!ddlm"Z"ddlm#Z#ddlm$Z$ddlm%Z%ddlm&Z&ddlm'Z'ddlm(Z(ddlm)Z)ddlm*Z*e+dZ,e+d Z-d!e d"efd#d$Z.d%ed&e/e0e fd"dfd'd(Z1d%ed"dfd)d*Z2d"e efd+d,Z3d-ed"dfd.d/Z4dFd-ed0e0d1e d"dfd2d3Z5d4e0fd5d6Z6d7e0d"e0fd8d9Z7d:e/d"e/e0e ffd;d<Z8d=e d"ejfd>d?Z9dGd=e dAe:d"e;efdBdCZ9s z,_inject_dd_trace_ctx_kwarg..)key) parameters) inspect signaturerr*r KEYWORD_ONLYlistvaluessortedreplace)rold_sig new_param params_list sorted_paramsr&r&r'_inject_dd_trace_ctx_kwarg2s   r6contextkwargscCs<i}t||d|vs|dduri|d<||dt<dS)Nr8)r_injectr)r7r8headersr&r&r'_inject_context_in_kwargs=s  r;cCs8i}t|||ddtjd<|ddtjd<dS)N traceparent tracestate)rr9getosenviron)r7r:r&r&r'_inject_context_in_envEs rBcCsDtjddur tjddur ttjdtjddSdS)Nr<r>)r<r>)r@rAr?r_extractr&r&r&r'!_extract_tracing_context_from_envLs   rDspancCs|dt|tt|td|td|td|t d|t dt j t}|dur<|t|trt}|t||t||}|dura|t||jjtjjjkrx|}|durx|t||}|dur|t |dSdSdS)N componentr)! _set_tag_strrrsocket gethostname set_metricr r r rrr@rAr?rrrayis_initializedrr get_job_idr get_node_id get_worker_idrworkermode_private WORKER_MODE get_task_idr get_actor_idr)rE submission_idruntime_context worker_idtask_idactor_idr&r&r'!_inject_ray_span_tags_and_metricsWs4          r\tag_name tag_valuecCs.t|tkr||tdS|||dS)zWe want to add args/kwargs values as tag when we execute a task/actor method. However they might be really big. In that case we dont way to serialize them AT ALL and we do not want to rely on _encoding.pyx. N)sys getsizeofrset_tagr)rEr]r^r&r&r'set_tag_or_truncatexsrb entrypointcCst|}|r |dSdS)z/ Get the job name from the entrypoint. rN)ENTRY_POINT_REGEXsearchgroup)rcmatchr&r&r'get_dd_job_name_from_entrypoints  rhscs>dddtffdd td|}dfdd |DS) z Redact path-like substrings from an entry-point string. Uses os.sep (and os.altsep if present) to detect paths; preserves spacing. cSsbtj|vrtjn tjrtj|vrtjnd}|s|S||}|s"tS||d}t||S)z If s contains a path separator, replace the directory part with REDACTION, preserving the final component (basename). Trailing separators are ignored. Detects both os.sep and os.altsep if present. N)r@sepaltseprstriprsplit)riused_sepcorebasenamer&r&r'_redact_pathlikes* z&redact_paths.._redact_pathliker csd|vr@|dd\}}t|dkr7|d|dkr7|ddvr7|d}|dd}|d|||S|d|St|dkrh|d|dkrh|ddvrh|d}|dd}|||S|S)N=rrGrrj>"')rnlen)tokr)valqinner)rrr&r' _redact_tokens( ( z#redact_paths.._redact_tokenz(\s+)r=c3s(|]}|dkr |n|VqdS)r=N)strip).0part)r{r&r' s&zredact_paths..)strrernjoin)ripartsr&)rrr{r' redact_pathss rdatacs.t|tsiSifdd|tS)a Converts a JSON (or Python dictionary) structure into a dict mapping dot-notation paths to leaf values, with keys prefixed once by RAY_METADATA_PREFIX. - Assumes the top-level is a dictionary. If a list is encountered anywhere, it is stringified with json.dumps and treated as a leaf (no recursion into list elements). - Leaf values (str, int, float, bool, None) are returned as-is as the dict values. - Returned dict keys are prefixed once with RAY_METADATA_PREFIX. cst|tr |D]\}}|r|d|n|}||q dSt|tr@z tj|dd}Wn ty9d}Ynw||<dS||<dS)N.F) ensure_asciiz[]) isinstancedictitemsr.jsondumps Exception)nodepathr)valuenew_path list_dump_recurseresultr&r'rs      z'flatten_metadata_dict.._recurse)rrr)rr&rr'flatten_metadata_dicts  rfunccshtr/gd}tfdd|Dr(}dd|D] }t|t||qntdtS)aBGet signature parameters. Support Cython functions by grabbing relevant attributes from the Cython function and attaching to a no-op function. This is somewhat brittle, since inspect may change, but given that inspect is written to a PEP, we hope it is relatively stable. Future versions of Python may allow overloading the inspect 'isfunction' and 'ismethod' functions / create ABC for Python functions. Until then, it appears that Cython won't do anything about compatibility with the inspect module. Args: func: The function whose signature should be checked. Returns: A function signature object, which includes the names of the keyword arguments as well as their default values. Raises: TypeError: A type error if the signature is not supported )__code____annotations__ __defaults____kwdefaults__c3s|]}t|VqdSr")hasattr)r}attrrr&r'rsz get_signature..cSsdSr"r&r&r&r&r'rszget_signature..funcz( is not a Python function we can process) is_cythonallsetattrgetattr TypeErrorr+r,)rattrs original_funcrr&rr' get_signatures rF ignore_firstcCsDtt|j}|r t|dkrtd|jd|dd}|S)a\Extract the function signature from the function. Args: func: The function whose signature should be extracted. ignore_first: True if the first argument should be ignored. This should be used when func is a method of a class. Returns: list of Parameter objects representing the function signature. rz5Methods must take a 'self' argument, but the method 'z' does not have one.rN)r.rr*r/rv ValueError__name__)rrsignature_parametersr&r&r'extract_signature s    rcCs$dd}||pt|do||jS)z1Check if an object is a Cython function or methodcSst|jdkS)Ncython_function_or_method)typer)xr&r&r' check_cython+szis_cython..check_cython__func__)rr)objrr&r&r'r$srr")F)>r+rrrr@rrIr_typingrrrrLray.runtime_contextrddtrace._trace._limitsrddtrace._trace.contextr ddtrace._trace.spanr ddtrace.constantsr r r rrddtrace.propagation.httpr constantsrrrrrrrrrrrrrcompileJOB_NAME_REGEXrdr6rrr;rBrDr\rbrhrrrboolr.rrr&r&r&r'sb                                !  /)(