o ڷi @sdZddlmZddlZddlmZmZmZddlZddl Zddl m Z GdddeZ dd d Z dddZe dd ddZe dd!ddZ dd"ddZd dgZdS)#zJModule holding utility and convenience functions for zmq event monitoring.) annotationsN) Awaitable TypedDictoverload)_check_versionc@s&eZdZUded<ded<ded<dS)_MonitorMessageinteventvaluebytesendpointN)__name__ __module__ __qualname____annotations__rrG/home/ubuntu/vllm_env/lib/python3.10/site-packages/zmq/utils/monitor.pyrs  rmsg list[bytes]returncCs^t|dkst|ddkrtd|td|d\}}t|t||dd}|S)adecode zmq_monitor event messages. Parameters ---------- msg : list(bytes) zmq multipart message that has arrived on a monitor PAIR socket. First frame is:: 16 bit event id 32 bit event value no padding Second frame is the endpoint as a bytestring Returns ------- event : dict event description as dict with the keys `event`, `value`, and `endpoint`. rzInvalid event message format: z=hi)r r r )len RuntimeErrorstructunpackzmqEvent)revent_idr r rrrparse_monitor_messagesr awaitable_msgAwaitable[list[bytes]]cs|IdH}t|S)zyLike parse_monitor_msg, but awaitable Given awaitable message, return awaitable for the parsed monitor message. N)r )r!rrrr_parse_monitor_msg_async6s r#socketzmq.asyncio.SocketflagsrAwaitable[_MonitorMessage]cCdSNrr$r&rrrrecv_monitor_messageCr+zmq.Socket[bytes]cCr(r)rr*rrrr+Jr, zmq.Socket,_MonitorMessage | Awaitable[_MonitorMessage]cCs.tdd||}t|trt|St|S)uReceive and decode the given raw message from the monitoring socket and return a dict. Requires libzmq ≥ 4.0 The returned dict will have the following entries: event : int the event id as described in `libzmq.zmq_socket_monitor` value : int the event value associated with the event, see `libzmq.zmq_socket_monitor` endpoint : str the affected endpoint .. versionchanged:: 23.1 Support for async sockets added. When called with a async socket, returns an awaitable for the monitor message. Parameters ---------- socket : zmq.Socket The PAIR socket (created by other.get_monitor_socket()) on which to recv the message flags : int standard zmq recv flags Returns ------- event : dict event description as dict with the keys `event`, `value`, and `endpoint`. )rzlibzmq event API)rrecv_multipart isinstancerr#r )r$r&rrrrr+Qs "  )rrrr)r!r"rr)r)r$r%r&rrr')r$r-r&rrr)r$r.r&rrr/)__doc__ __future__rrtypingrrrr zmq.asyncio zmq.errorrrr r#r+__all__rrrrs&      /