Struct ConnectionRegistry 

pub struct ConnectionRegistry { /* private fields */ }

Sharded, lock-free-on-the-hot-path connection registry.

One instance is created at launch, leaked to &'static, and shared by every gateway route. All mutating operations are O(1) amortised and touch only a single shard; broadcast is O(connections) with no global lock.

Implementations

impl ConnectionRegistry

pub fn new() -> Self

pub fn register( &self, tx: Sender<WsMessage>, claims: Option<Arc<Claims>>, ) -> ConnId

Register a freshly-upgraded socket. Returns its assigned ConnId.

tx must be the sending half of a bounded channel — the queue depth is the slow-client memory ceiling (LaunchConfig::ws_outbound_buffer).

pub fn touch(&self, id: ConnId)

Record inbound activity for the idle sweeper. Called by the reader on every frame (including pongs); one relaxed store.

pub fn close_all(&self) -> usize

Send a Close frame to every live socket. Used by graceful shutdown: WS connections otherwise keep axum’s drain phase open indefinitely (until clients disconnect or the supervisor SIGKILLs), which would mean plugin on_shutdown hooks never run in production. Each socket’s writer task sends the frame and exits; readers follow.

pub fn sweep_idle(&self, max_idle_secs: u64) -> Vec<ConnId>

Close connections with no inbound activity for max_idle_secs. Dead TCP links (NAT timeouts, vanished mobile clients) never send a Close frame — without reaping they linger in the registry forever, eating queue memory on every broadcast. Returns the ids reaped.

Pair with a server ping (ws_ping_interval): pings provoke pongs, pongs touch last_seen, so only truly dead links exceed the window.

pub fn unregister(&self, id: ConnId)

Remove a socket and evict it from every room it had joined.

pub fn connection_count(&self) -> usize

Number of live connections. O(1).

pub fn join_room(&self, id: ConnId, room: String)

Add a connection to a room (Socket.IO-style logical channel).

TOCTOU safety

Both member_rooms and rooms are updated while holding the DashMap shard read-lock on conns (the Ref keeps the shard locked until it is dropped). This prevents a concurrent unregister() — which needs the shard write-lock — from removing the entry between the two writes, which would otherwise leave a stale ConnId in rooms forever.

pub fn leave_room(&self, id: ConnId, room: &str)

Remove a connection from a room.

Mirrors join_room’s lock order: hold the conns shard read-lock first so a concurrent unregister() (which needs the write-lock) cannot remove the entry between the two writes and leave a stale id in rooms.

pub fn send_text(&self, id: ConnId, text: Arc<str>)

Enqueue a text frame to one connection. Non-blocking; a full queue evicts the slow client, a dead receiver is dropped silently.

pub fn broadcast_text(&self, text: &str)

Fan a text frame out to every live connection. Creates one Arc<str> allocation then enqueues N pointer copies — no per-connection memcpy. Clients whose queues are full are evicted after the sweep (removal inside iteration would deadlock the DashMap shard).

pub fn broadcast_room_text(&self, room: &str, text: &str)

Fan a text frame out to the members of one room.

Trait Implementations

impl Default for ConnectionRegistry

fn default() -> Self

Returns the “default value” for a type.