o i[@sddlmZddlZddlmZddlmZddlmZddlmZddl m Z ddl m Z dd l m Z dd lmZeeZd ZGd d d Zdededeje ddfddZGddde ZGdddeZdS))datetimeN)Callable)Optional)Flare) get_logger)Payload) RCCallback)PublisherSubscriberConnector)RemoteConfigSubscriberc@seZdZdZdddZdS)TracerFlareStatez:Shared state between the callback and stale check service.returnNcCs d|_dSN)current_request_startselfrW/home/ubuntu/.local/lib/python3.10/site-packages/ddtrace/internal/flare/_subscribers.py__init__s zTracerFlareState.__init__r N)__name__ __module__ __qualname____doc__rrrrrr sr flarestatepayloadsr c Cst|D]}|jdurtdq|jj}t|jtr|jn|jg}|D]}t|ts4tdtt |q"| ||}| rj|j durMt dt|j q"td|j}|dur_t dq"||rit|_ q"|r|j durtd|drt|_ nt dq"td |||d|_ q"|rtd ||d|_ q"t d |q"qdS) Nz$No metadata for tracer flare payloadzCConfig item is not type dict, received type %s instead. Skipping...zHThere is already a tracer flare job started at %s. Skipping new request.zPreparing tracer flarez+Received set flare action without log levelzCStarting tracer flare job for AGENT_TASK without prior AGENT_CONFIGDEBUGz5Failed to prepare tracer flare. Skipping new request.z#Generating and sending tracer flarezIReverting tracer flare configurations and cleaning up any generated filesz5Received unexpected product type for tracer flare: %s)metadatalogdebug product_name isinstancecontentlistdictstrtypehandle_remote_config_datais_setrwarninginfolevelpreparernowis_sendrevert_configssendis_unsetclean_up_files) rrrpayload product_typeitemsitem flare_action log_levelrrr_process_payloadssZ                 r:c @speZdZdZefdeeeegdfdede de ddf dd Z defd d Z dd d Z dejeddfddZdS)TracerFlareCallbackzRemote config callback for tracer flare requests. The periodic() method performs stale flare checks at every polling operation. The __call__() method processes tracer flare payloads when present. callbackNrrstale_duration_minsr cCs ||_||_||_|d|_dS)aAInitialize the tracer flare callback. Args: callback: Handler function for flare operations flare: Flare instance for generating flares state: Shared state for tracking flare requests stale_duration_mins: Minutes before a flare request is considered stale <N) _callback_flare_state_stale_duration_secs)rr<rrr=rrrrTszTracerFlareCallback.__init__cCs.|jjrt}||jj}||jkSdS)z,Check if the current flare request is stale.F)rArrr. total_secondsrB)rcurr flare_agerrr_has_stale_flarehs  z$TracerFlareCallback._has_stale_flarecCs8|rtd|jjd|j_||jiddSdS)zPeriodic method called at every polling operation. Checks for stale flare requests and cleans them up if necessary. jTracer flare request started at %s is stale, reverting logger configurations and cleaning up resources nowNT)rFrr+rArr?r@rrrrperiodicpszTracerFlareCallback.periodicrcCs|sdSt|j|j|dS)zProcess tracer flare configuration payloads. Args: payloads: Sequence of configuration payloads to process N)r:r@rA)rrrrr__call__~szTracerFlareCallback.__call__r)rrrr!DEFAULT_STALE_FLARE_DURATION_MINSrrr%boolr intrrFrHtSequencerrIrrrrr;Ms"   r;csJeZdZdZefdededeffdd Zde fdd Z d d d Z Z S)TracerFlareSubscriberz%Subscriber for tracer flare requests.data_connectorrstale_flare_agecs*t|dddd|_||_||_dS)NcSsdSrr)_datarrrsz0TracerFlareSubscriber.__init__..TracerFlareConfig)superrrstale_tracer_flare_num_minsr)rrPrrQ __class__rrrs zTracerFlareSubscriber.__init__r cCs2|jrt}||j}|jd}||kSdS)Nr>F)rrr.rCrV)rrDrE stale_agerrrhas_stale_flares  z%TracerFlareSubscriber.has_stale_flareNcCsx|rtd|jd|_|j|jdS|j}|s(t ddSt }|j|_t |j|||j|_dS)NrGz$No data received from data connector) rZrr+rrr0r3_data_connectorreadr r r:)r_datarrrr!_get_data_from_connector_and_execs"     z7TracerFlareSubscriber._get_data_from_connector_and_execr) rrrrrJr rrLrrKrZr_ __classcell__rrrWrrOs rO)rtypingrMrrddtrace.internal.flare.flarerddtrace.internal.loggerrddtrace.internal.remoteconfigrr)ddtrace.internal.remoteconfig._connectorsr *ddtrace.internal.remoteconfig._subscribersr rrrJr rNr:r;rOrrrrs          3=