from __future__ import annotations from typing import Callable PUB_OP = "PUB" HPUB_OP = "HPUB" SUB_OP = "SUB" UNSUB_OP = "UNSUB" _CRLF_ = "\r\n" Command = Callable[..., bytes] def pub_cmd(subject, reply, payload) -> bytes: return f"{PUB_OP} {subject} {reply} {len(payload)}{_CRLF_}".encode() + payload + _CRLF_.encode() def hpub_cmd(subject, reply, hdr, payload) -> bytes: hdr_len = len(hdr) total_size = len(payload) + hdr_len return f"{HPUB_OP} {subject} {reply} {hdr_len} {total_size}{_CRLF_}".encode() + hdr + payload + _CRLF_.encode() def sub_cmd(subject, queue, sid) -> bytes: return f"{SUB_OP} {subject} {queue} {sid}{_CRLF_}".encode() def unsub_cmd(sid, limit) -> bytes: limit_s = "" if limit == 0 else f"{limit}" return f"{UNSUB_OP} {sid} {limit_s}{_CRLF_}".encode()