j ZddlZddlmZddlmZddlmZGddeZGddZy) N)Enum)Optional)connection_probeceZdZdZdZdZdZy)ConnectionStatusaO Represents the connection status for a system. Provides an enumeration of various states a connection can be in, such as successful connection, connection failure, and restored connection. Used primarily to indicate the status of a network, server, or other connectivity-dependent services. Attributes: connection_ok: Indicates the connection is currently active and functioning properly. connection_failed: Indicates the connection has failed. connection_restored: Indicates the connection, which was previously failed, has been restored. N)__name__ __module__ __qualname____doc__ connection_okconnection_failedconnection_restored|/Users/manta/Documents/Projects/TheRoad-I1/backend/.venv/lib/python3.12/site-packages/opik/healthcheck/connection_monitor.pyrrs MrrceZdZdZdededej fdZede fdZ dd Z d e e ddfd Zdefd Zd e d e e defdZy)OpikConnectionMonitora Monitors the connection to the Opik server and tracks its status. This class is responsible for checking the server connectivity using a probe object at regular intervals. It maintains the status of the server connection, including disconnection details and reasons. Attributes: ping_interval: Interval in seconds between server connection pings. disconnect_time: Time in seconds when the server connection was lost. Defaults to 0 if the connection has not been lost yet. disconnect_reason: Reason for the disconnection, if applicable. Defaults to None if there is no disconnection. ping_interval check_timeoutprobecx||_d|_||_td|_||_d|_d|_y)a Initializes an instance of the class to handle connection probing and monitoring. Args: ping_interval: The interval, in seconds, at which pings are sent to probe the connection. check_timeout: The threshold duration, in seconds, before a connection is considered lost if no response is received. probe: An instance of the ``connection_probe.ConnectionProbe`` class used for sending probe requests and handling connectivity checks. Tz-infN)_check_timeout_has_server_connection_probefloat last_beatrdisconnect_timedisconnect_reason)selfrrrs r__init__zOpikConnectionMonitor.__init__+s>",&*# v*"04rreturnc|jS)N)rr#s rhas_server_connectionz+OpikConnectionMonitor.has_server_connectionEs***rNc.d|_d|_d|_y)NrT)r!r"rr's rresetzOpikConnectionMonitor.resetIs"!%&*#rfailure_reasoncj|jr tj|_||_d|_y)a Handles the procedure when the connection to the server is determined to have failed. This method processes the failure by capturing the disconnection time and the reason for the failure (if provided). It also updates the state to reflect that the connection to the server is no longer active. Args: failure_reason: Optional reason for the connection failure. Defaults to None if no reason is provided. FN)rtimer!r")r#r+s rrz'OpikConnectionMonitor.connection_failedNs+  & &#'99;D %3D "&+#rc|j|jz}tj}||krf|jj |j }tj|_|j |j|jS|jrtjStjS)a Checks and updates the connection status during a timed heartbeat event. This method ensures that the connection is actively monitored, sends probe requests to check the connection status if necessary, and updates the last heartbeat timestamp. Returns: ConnectionStatus: Indicates the current status of the connection. It may return one of the following statuses: - `ConnectionStatus.connection_ok`: The connection is active. - `ConnectionStatus.connection_failed`: The connection is not active. - Result based on the probe check indicating whether the connection is healthy or not. )timeout)r+) r rr-rcheck_connectionr_on_ping_result is_healthy error_messager(rrr)r# next_beatnowresults rtickzOpikConnectionMonitor.tick`s NNT%7%77 iik  [[11$:M:M1NF!YY[DN''!!&2F2F(  ' '#11 1#55 5rsuccessc|r6|jstj}ntj}d|_|Stj }|j ||S)z)Invoked to process a received ping resultT)r(rrrrr)r#r8r+statuss rr1z%OpikConnectionMonitor._on_ping_result}sU --)==)77*.D '  &77F  " "> 2 r)r%N)r r r rrrConnectionProber$propertyboolr(r*rstrrrr7r1rrrrrs 555 // 54+t+++ , ,$,$6&6:-5c] rr) r-enumrtypingropik.healthcheckrrrrrrrBs) -t&qqr