feat(issue-5): persistent MQTT publisher prototype and benchmark harness

Splits publish_event.main into three pieces with no behaviour change:

  prepare_event   load the job, consume the seq, build the signed payload,
                  decide retain. The seq is burned here, before any transport
                  runs, so a failed publish can never leave a number a replayed
                  packet could reuse.
  record_outcome  audit record, registry event log and status sync -- all three
                  best effort, all three regardless of the publish result. That
                  is exactly the contract G-1/G-2/G-4 pin down, now in one place.
  main            parse, prepare, publish with retry, record.

publisher.py holds the two transports behind one interface. OneShotPublisher is
the current behaviour. PersistentPublisher keeps a single connection and lets
paho reconnect underneath, raising on an unacknowledged publish so the caller's
retry policy stays the only thing deciding attempt counts. It refuses an event
for a different broker rather than publishing to the wrong place.

Nothing in production uses the persistent path: the CLI and shell helpers exit
after one publish. The consumer is the phase 2a engine, which pays the CONNACK
cost on every event today -- this job's own started event took ~12.5s because
of one CONNACK timeout plus reconnect.

bench.py measures both arms. The fake transport is deterministic and is what
the tests assert on; the real-broker arm is informational, since the PoC
default is a public broker whose latency is not ours to control.

Verified: N publishes cost N connections one-shot and 1 persistent; the two
transports leave field-for-field identical registry and audit records including
the HMAC signature and retain flag; an unacknowledged publish retries on the
same connection and records published=false; a second broker is refused. No
test opens a socket.

Co-Authored-By: Claude Opus 5 <noreply@anthropic.com>
Claude-Session: https://claude.ai/code/session_01UrDNJkHmm18uu7n7JwV5Qd
This commit is contained in:
GodopuandClaude Opus 5 committed 2026-09-06 15:44:33 +09:00
1 parent 78cb9ecfa9
commit a4b7b45e13
4 files changed
+600 -44

No files matched your search

+178
View File
@@ -0,0 +1,178 @@
"""Benchmark harness for the persistent-connection prototype.
Two transports, deliberately separated:
--transport fake a stub client with a synthetic connect latency. Fully
deterministic, touches no network, and is what CI asserts
on: N publishes cost N connections one-shot and exactly 1
persistent.
--transport broker the real broker named by the job record. Informational
only: the PoC default is a public broker whose CONNACK
latency is not ours to control, so no gate depends on it.
Usage:
python -m lib_py.delegation.bench --n 50 --transport fake [--connect-latency-ms 50]
[--job <id>] [--registry-dir DIR] [--out FILE]
"""
from __future__ import annotations
import argparse
import json
import sys
import time
from typing import List, Optional
from lib_py.delegation import mqtt_common, publish_event
from lib_py.delegation.publisher import OneShotPublisher, PersistentPublisher
class FakeInfo(object):
def __init__(self, published=True):
self._published = published
def wait_for_publish(self, timeout=None):
return None
def is_published(self):
return self._published
class FakeClient(object):
"""Minimal paho stand-in: counts connects and charges a synthetic latency."""
def __init__(self, connect_latency=0.0, counters=None):
self.connect_latency = connect_latency
self.counters = counters if counters is not None else {}
self.on_connect = None
def connect(self, host, port, keepalive=60):
if self.connect_latency:
time.sleep(self.connect_latency)
self.counters["connect"] = self.counters.get("connect", 0) + 1
if self.on_connect:
self.on_connect(self, None, None, 0, None)
def loop_start(self):
return None
def loop_stop(self):
return None
def reconnect_delay_set(self, min_delay=1, max_delay=16):
return None
def publish(self, topic, payload=None, qos=0, retain=False):
self.counters["publish"] = self.counters.get("publish", 0) + 1
return FakeInfo()
def disconnect(self):
return None
def fake_factory(connect_latency, counters):
def factory(_role, _config):
return FakeClient(connect_latency=connect_latency, counters=counters)
return factory
def _percentile(values: List[float], pct: float) -> float:
if not values:
return 0.0
ordered = sorted(values)
idx = int(round((pct / 100.0) * (len(ordered) - 1)))
return ordered[idx]
def _run(publisher, prepared, n: int) -> dict:
latencies = []
failures = 0
for _ in range(n):
start = time.monotonic()
ok, _err = publisher.publish_with_retry(prepared, attempts=1)
latencies.append((time.monotonic() - start) * 1000.0)
if not ok:
failures += 1
return {
"publishes": n,
"connects": getattr(publisher, "connect_count", None),
"failures": failures,
"latency_ms": {
"p50": round(_percentile(latencies, 50), 3),
"p95": round(_percentile(latencies, 95), 3),
"max": round(max(latencies), 3) if latencies else 0.0,
},
}
def run_benchmark(n=50, transport="fake", connect_latency_ms=50.0,
job_id=None, registry_dir=None) -> dict:
registry_dir = registry_dir or mqtt_common.DEFAULT_REGISTRY_DIR
if transport == "fake":
prepared = publish_event.PreparedEvent(
job_id="benchjob", event="progress", detail="bench", seq=0,
topic="python/mqtt/jobs/benchjob/events", retain=False,
payload={"seq": 0}, body=b'{"seq": 0}',
config=mqtt_common.broker_config_from_env(),
)
latency = connect_latency_ms / 1000.0
one_counters, persist_counters = {}, {}
one = OneShotPublisher(client_factory=fake_factory(latency, one_counters))
persist = PersistentPublisher(client_factory=fake_factory(latency, persist_counters))
try:
result = {
"transport": "fake",
"connect_latency_ms": connect_latency_ms,
"one_shot": _run(one, prepared, n),
"persistent": _run(persist, prepared, n),
}
finally:
persist.close()
result["one_shot"]["client_connects"] = one_counters.get("connect", 0)
result["persistent"]["client_connects"] = persist_counters.get("connect", 0)
return result
if not job_id:
raise ValueError("--transport broker requires --job")
prepared = publish_event.prepare_event(
job_id=job_id, event="progress", detail="bench", registry_dir=registry_dir,
)
one = OneShotPublisher()
persist = PersistentPublisher()
try:
return {
"transport": "broker",
"broker": "{}:{}".format(prepared.config.host, prepared.config.port),
"one_shot": _run(one, prepared, n),
"persistent": _run(persist, prepared, n),
}
finally:
persist.close()
def main(argv: Optional[List[str]] = None) -> int:
mqtt_common.ensure_env_loaded()
parser = argparse.ArgumentParser(prog="lib_py.delegation.bench")
parser.add_argument("--n", type=int, default=50)
parser.add_argument("--transport", default="fake", choices=["fake", "broker"])
parser.add_argument("--connect-latency-ms", type=float, default=50.0)
parser.add_argument("--job", default=None)
parser.add_argument("--registry-dir", default=None)
parser.add_argument("--out", default=None)
args = parser.parse_args(argv)
result = run_benchmark(
n=args.n, transport=args.transport,
connect_latency_ms=args.connect_latency_ms,
job_id=args.job, registry_dir=args.registry_dir,
)
text = json.dumps(result, indent=2, sort_keys=True)
if args.out:
with open(args.out, "w", encoding="utf-8") as fh:
fh.write(text + "\n")
print(text)
return 0
if __name__ == "__main__":
sys.exit(main())