o i@sddlZddlmZddlZddlZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z dd l m Z dd l mZdd lmZeeZGd d d Zeddde edefdeefddZdeddfddZdS)N) lru_cache)Any)Literal)Optional)Union)CONTAINER_ID_HEADER_NAME)ENTITY_ID_HEADER_NAME)!EXTERNAL_ENV_ENVIRONMENT_VARIABLE)EXTERNAL_ENV_HEADER_NAME) get_loggerc@seZdZUeed<eed<eed<eed<eed<eed<eed<dZd Zd Zd Ze d Z e d eZ e d eeeZ       dddZddZedededfddZdS) CGroupInfoidgroupspath container_id controllerspod_id node_inoder rrrrrrzh[0-9a-f]{8}[-_][0-9a-f]{4}[-_][0-9a-f]{4}[-_][0-9a-f]{4}[-_][0-9a-f]{12}|([0-9a-f]{8}(-[0-9a-f]{4}){4}$)z [0-9a-f]{64}z[0-9a-f]{32}-\d+z^(\d+):([^:]*):(.+)$zpod({0})(?:\.slice)?$z!(?:.+)?({0}|{1}|{2})(?:\.scope)?$NcCs.||_||_||_||_||_||_||_dSNr)selfr rrrrrrrV/home/ubuntu/.local/lib/python3.10/site-packages/ddtrace/internal/runtime/container.py__init__,s  zCGroupInfo.__init__cCs^t|to.|j|jko.|j|jko.|j|jko.|j|jko.|j|jko.|j|jko.|j|jkSr) isinstancer r rrrrrr)rotherrrr__eq__>s        zCGroupInfo.__eq__linereturnc Cs|}|j|}|sdS|\}}}dd|dD}dd|dD}d}t|r>|j|}|r>|d}d} t|rS|j |}|rS|d} zt d|dj } Wnt yrtd |d} Ynw||||||| | d S) a[ Parse a new :class:`CGroupInfo` from the provided line :param line: A line from a cgroup file (e.g. /proc/self/cgroup) to parse information from :type line: str :returns: A :class:`CGroupInfo` object with all parsed data, if the line is valid, otherwise `None` :rtype: :class:`CGroupInfo` | None NcSsg|] }|r|qSr)strip).0crrr `sz(CGroupInfo.from_line..,cSsg|]}|qSrr)r prrrr"es/z/sys/fs/cgroup/z+Failed to stat cgroup node file for path %rr)rLINE_REmatchrsplitlen CONTAINER_REpopgroupPOD_REosstatst_ino Exceptionlogdebug) clsrr(id_rrrpartsrrrrrr from_lineJs@     zCGroupInfo.from_line)NNNNNNN)__name__ __module__ __qualname__r__annotations__ __slots__UUID_SOURCE_PATTERNCONTAINER_SOURCE_PATTERN TASK_PATTERNrecompiler'formatr.r+rr classmethodstrrr8rrrrr s:     r @rpidrc Csz=td|ddd(}|D]}t|}|r)|js|jr)|WdWSqWdWdS1s6wYWdStye}z|jtjkrZtj d|ddWYd}~dSWYd}~dSd}~wt yvtj d |ddYdSw) z Helper to fetch the current container id, if we are running in a container. We will parse `/proc/{pid}/cgroup` to determine our container id. The results of calling this function are cached. z/proc/z/cgroupr)modeNz%Failed to open cgroup file for pid %rT)exc_infoz&Failed to parse cgroup file for pid %r) openr r8rrIOErrorerrnoENOENTr3r4r2)rGfprinfoerrrget_container_infos0        rRheaderscCstt}|dur'|jr|t|jtd|jin|jr'|td|jitjt }|r8|t |idSdS)z_Get the container info (either the container ID or the cgroup inode) and add it to the headers.Nzci-zin-) rRrupdaterrrr/environgetr r )rScontainer_info external_inforrrupdate_headerss(   rY)r)rM functoolsrr/rAtypingrrrrddtrace.internal.constantsrrr r ddtrace.internal.loggerr r9r3r intrRdictrYrrrrs$          r&