@rebasepro/server-postgres 0.10.1-canary.6f89f77 → 0.10.1-canary.7609565

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -23,19 +23,14 @@ export declare function parseCdcPayload(payload: string): CdcChangeEvent | null;
23
23
  /**
24
24
  * Dedicated Postgres LISTEN client for database-level CDC.
25
25
  *
26
- * Mirrors the resilience of the RealtimeService cross-instance LISTEN client: a
27
- * standalone `pg.Client` (outside the Drizzle pool) that stays connected, and
28
- * transparently reconnects on error/disconnect. Each backend instance runs one,
29
- * so every instance observes every committed change regardless of which
30
- * instance (or external process) made the write.
26
+ * A {@link PgNotifyListener} — a connection outside the Drizzle pool that stays
27
+ * open and repairs itself — plus the parsing that turns a `rebase_cdc` payload
28
+ * into a change event. Each backend instance runs one, so every instance
29
+ * observes every committed change regardless of which instance (or external
30
+ * process) made the write.
31
31
  */
32
32
  export declare class CdcListener {
33
- private readonly connectionString;
34
- private readonly onEvent;
35
- private client?;
36
- private running;
37
- private reconnectTimer?;
38
- private static readonly RECONNECT_DELAY_MS;
33
+ private readonly listener;
39
34
  constructor(connectionString: string, onEvent: (event: CdcChangeEvent) => void | Promise<void>);
40
35
  /**
41
36
  * Connect and begin listening. Idempotent.
@@ -44,11 +39,9 @@ export declare class CdcListener {
44
39
  * established (or `LISTEN` is refused), this rejects so callers — notably
45
40
  * `REALTIME_CDC=auto` — can detect an unusable connection and fall back to
46
41
  * app-level realtime. Once the initial connection succeeds, later drops
47
- * self-heal via {@link scheduleReconnect}.
42
+ * self-heal in the background.
48
43
  */
49
44
  start(): Promise<void>;
50
45
  /** Stop listening and release the connection. */
51
46
  stop(): Promise<void>;
52
- private connect;
53
- private scheduleReconnect;
54
47
  }
@@ -0,0 +1,29 @@
1
+ /**
2
+ * Runtime pieces of the channel bus that are not the contract itself.
3
+ *
4
+ * The interface, the frame shape and the implementer's contract live in
5
+ * `@rebasepro/types` (`types/channel_bus.ts`), so a transport shipped as its own
6
+ * package — a Redis one, say — depends on the contract and not on this database
7
+ * adapter. They are re-exported here for convenience: code already importing
8
+ * from the adapter should not have to know where the types are declared.
9
+ */
10
+ import type { ChannelBusFrame } from "@rebasepro/types";
11
+ export type { ChannelBus, ChannelBusFrame, ChannelBusHandler, ChannelBusConfig, ChannelBusSetting } from "@rebasepro/types";
12
+ export { isChannelBusInstance } from "@rebasepro/types";
13
+ /**
14
+ * The default: no cross-instance delivery at all.
15
+ *
16
+ * This is what every deployment ran before the bus existed, and what a
17
+ * single-instance deployment should keep running — `publish` resolves without
18
+ * touching the network, so the broadcast path is the same handful of `ws.send`
19
+ * calls it always was.
20
+ */
21
+ export declare class MemoryChannelBus {
22
+ readonly kind: "memory";
23
+ readonly maxFrameBytes: number;
24
+ start(): Promise<void>;
25
+ publish(): Promise<void>;
26
+ stop(): Promise<void>;
27
+ }
28
+ /** Encoded size of a frame, for a transport's size check. */
29
+ export declare function frameByteLength(frame: ChannelBusFrame): number;
@@ -0,0 +1,111 @@
1
+ /**
2
+ * Channel bus over Postgres LISTEN/NOTIFY.
3
+ *
4
+ * Chosen because it needs nothing that a Rebase deployment does not already
5
+ * have — the same database, the same direct URL the CDC listener uses. Three
6
+ * properties of `NOTIFY` shape everything below:
7
+ *
8
+ * - **8000 bytes per payload.** Presence and cursors fit with room to spare; a
9
+ * scene snapshot does not. Rather than truncate or drop, an oversized frame
10
+ * on a *retained* channel is published as a pointer — the body is already in
11
+ * `rebase.channel_messages` with a sequence number, so the receiver reads it
12
+ * back. That is the same trick the entity path uses (notify an address,
13
+ * refetch the row), applied to a different table. On an ephemeral channel
14
+ * there is nothing to point at, so the publish is refused loudly instead of
15
+ * reaching some instances and not others.
16
+ *
17
+ * - **A notify is a query on the primary database.** Not a slow one, but it
18
+ * competes with the application's real queries, and that — not throughput —
19
+ * is what actually limits this transport. Measured, it carried ~10k
20
+ * cross-instance messages/second and stayed flat out to eight instances; what
21
+ * it should not do is spend 10k queries/second of the database's budget on
22
+ * cursor movement. Hence the batching below.
23
+ *
24
+ * - **Delivery is best-effort.** Retained channels repair themselves through
25
+ * the client's history replay, so a lost frame costs a live update rather
26
+ * than correctness. That is what makes coalescing safe.
27
+ */
28
+ import { NodePgDatabase } from "drizzle-orm/node-postgres";
29
+ import { ChannelBus, ChannelBusFrame, ChannelBusHandler } from "./ChannelBus";
30
+ /** NOTIFY channel carrying channel-bus frames. */
31
+ export declare const CHANNEL_BUS_NOTIFY_CHANNEL = "rebase_channel_bus";
32
+ /**
33
+ * Postgres refuses a NOTIFY payload of 8000 bytes or more. The margin below it
34
+ * is for nothing in particular — it is there so that a payload which passes this
35
+ * check cannot fail at the server for being a few bytes over.
36
+ */
37
+ export declare const PG_NOTIFY_MAX_PAYLOAD_BYTES = 7500;
38
+ /**
39
+ * How long a batching window stays open.
40
+ *
41
+ * Ten milliseconds is below the threshold where a human notices a cursor lag,
42
+ * and it is the difference between one query per message and one query per
43
+ * window under load. Set to 0 to disable coalescing entirely.
44
+ */
45
+ export declare const DEFAULT_BATCH_WINDOW_MS = 10;
46
+ export declare class PostgresChannelBus implements ChannelBus {
47
+ private readonly db;
48
+ private readonly connectionString;
49
+ readonly kind: "postgres";
50
+ readonly maxFrameBytes = 7500;
51
+ private listener?;
52
+ private readonly batchWindowMs;
53
+ /**
54
+ * Frames waiting for the current window to close.
55
+ *
56
+ * The window is opened by a publish that found none open, and that publish
57
+ * is sent *immediately* rather than joining a batch — see {@link publish}.
58
+ */
59
+ private pending;
60
+ private pendingBytes;
61
+ private windowTimer?;
62
+ private stopped;
63
+ constructor(db: NodePgDatabase<Record<string, unknown>>, connectionString: string, options?: {
64
+ batchWindowMs?: number;
65
+ });
66
+ start(handler: ChannelBusHandler): Promise<void>;
67
+ /**
68
+ * Publish, coalescing under load.
69
+ *
70
+ * The window is *leading edge*: a publish arriving when no window is open is
71
+ * sent straight away and opens one, so an idle channel pays no added latency
72
+ * at all. Frames arriving while it is open are collected and leave together
73
+ * when it closes. The effect is that cost tracks elapsed time rather than
74
+ * message count — one query per window instead of one per message — which is
75
+ * the same shape as the retention pruning throttle, for the same reason.
76
+ *
77
+ * The returned promise settles when the frame has actually left, not when it
78
+ * was queued, so the contract ("reaches the other instances, or rejects")
79
+ * still holds.
80
+ */
81
+ publish(frame: ChannelBusFrame): Promise<void>;
82
+ stop(): Promise<void>;
83
+ private openWindow;
84
+ /** Send everything queued and settle the promises waiting on it. */
85
+ private flush;
86
+ /**
87
+ * One NOTIFY.
88
+ *
89
+ * A single frame goes out in the plain, unwrapped shape. That is not just
90
+ * economy: during a rolling deploy an instance running the previous build
91
+ * understands only that shape, and low-rate traffic — presence, the tail of
92
+ * a session — is exactly what is flowing while pods restart. Batching only
93
+ * appears under load, which shrinks the mixed-version window to almost
94
+ * nothing.
95
+ */
96
+ private send;
97
+ }
98
+ /**
99
+ * Parse a bus payload into the frames it carries.
100
+ *
101
+ * Accepts both wire shapes — a bare frame and a `{ batch: [...] }` envelope —
102
+ * so an instance on the new build understands one on the old. Returns an empty
103
+ * array for anything unrecognisable: a malformed or future-versioned message
104
+ * must never take the listener down.
105
+ */
106
+ export declare function parseChannelBusPayload(payload: string): ChannelBusFrame[];
107
+ /**
108
+ * Parse a single bus frame, returning null for anything that is not a frame we
109
+ * understand.
110
+ */
111
+ export declare function parseChannelBusFrame(payload: string): ChannelBusFrame | null;
@@ -0,0 +1,55 @@
1
+ /**
2
+ * Resolution of the channel bus from config, environment, or a supplied instance.
3
+ *
4
+ * Opt-in, like every other cross-cutting realtime switch here: with nothing
5
+ * configured a deployment gets the memory bus and behaves exactly as it did
6
+ * before this existed. Unlike `REALTIME_CDC=auto`, there is no "try it and see"
7
+ * default — a bus changes where messages go, and quietly turning on a Postgres
8
+ * NOTIFY per broadcast because a direct URL happened to be set is not a
9
+ * decision to make on the user's behalf.
10
+ *
11
+ * Two transports ship, and neither adds a service to a deployment. A third is
12
+ * not a code change here: `realtime.bus` also accepts an already-constructed
13
+ * {@link ChannelBus}, so a transport published as its own package plugs in
14
+ * without this file learning about it. See `@rebasepro/types` →
15
+ * `types/channel_bus.ts` for the contract such a package implements.
16
+ */
17
+ import { NodePgDatabase } from "drizzle-orm/node-postgres";
18
+ import { type ChannelBus, type ChannelBusConfig, type ChannelBusSetting } from "@rebasepro/types";
19
+ export * from "./ChannelBus";
20
+ export { PostgresChannelBus, CHANNEL_BUS_NOTIFY_CHANNEL, PG_NOTIFY_MAX_PAYLOAD_BYTES, DEFAULT_BATCH_WINDOW_MS, parseChannelBusFrame, parseChannelBusPayload } from "./PostgresChannelBus";
21
+ export interface ChannelBusDeps {
22
+ db: NodePgDatabase<Record<string, unknown>>;
23
+ /**
24
+ * Direct (non-pooled) Postgres URL for the LISTEN client. `LISTEN` is
25
+ * session state, so behind PgBouncer in transaction mode this must be the
26
+ * database itself and not the pooler.
27
+ */
28
+ directUrl?: string;
29
+ }
30
+ /**
31
+ * Merge `REALTIME_CHANNEL_BUS` into the configured bus.
32
+ *
33
+ * The environment wins over a *named* built-in, so the transport can be changed
34
+ * per deployment without a rebuild — the same reason `REALTIME_CDC` is an env
35
+ * var. It does **not** win over a supplied instance: the env var can only name
36
+ * transports this package knows how to construct, so honouring it there would
37
+ * mean silently discarding the object the application handed us.
38
+ */
39
+ export declare function resolveChannelBusSetting(configured?: ChannelBusSetting): ChannelBusSetting;
40
+ /**
41
+ * @deprecated Use {@link resolveChannelBusSetting}, which also accepts a
42
+ * supplied {@link ChannelBus} instance. Kept as a narrow alias so existing
43
+ * config-only callers keep their exact types.
44
+ */
45
+ export declare function resolveChannelBusConfig(configured?: ChannelBusConfig): ChannelBusConfig;
46
+ /**
47
+ * Produce the bus a setting asks for.
48
+ *
49
+ * An instance is handed straight back — constructing it was the application's
50
+ * job, and this function has nothing to add. A named built-in that turns out to
51
+ * be unusable degrades to the memory bus, with the reason logged, rather than
52
+ * throwing: a misconfigured bus should cost a deployment its cross-instance
53
+ * fan-out, not its ability to boot.
54
+ */
55
+ export declare function createChannelBus(setting: ChannelBusSetting, deps: ChannelBusDeps): ChannelBus;
@@ -105,6 +105,17 @@ export declare class ChannelHistoryStore {
105
105
  messages: ChannelHistoryEntry[];
106
106
  latestSeq: number;
107
107
  }>;
108
+ /**
109
+ * One retained message by its address.
110
+ *
111
+ * This is what makes the cross-instance pointer path work: a broadcast too
112
+ * large to travel inside a `pg_notify` payload is already stored here, so
113
+ * the notification carries `(channel, seq)` and each receiving instance
114
+ * reads the body back. Returns null when the message has since been pruned
115
+ * — a receiver that is that far behind has nothing useful to deliver, and
116
+ * the client's own `channel_history` replay is the repair path.
117
+ */
118
+ getBySeq(channel: string, seq: number): Promise<ChannelHistoryEntry | null>;
108
119
  /**
109
120
  * Enforce a channel's retention bounds.
110
121
  *
@@ -0,0 +1,66 @@
1
+ /**
2
+ * The shared presence roster.
3
+ *
4
+ * Broadcast only ever needed *fan-out* to work across instances — a frame goes
5
+ * out, whoever is connected receives it. Presence needs more than that, because
6
+ * `presence_state` is a question ("who is in this document?") and a per-process
7
+ * `Map` can only answer for the clients that happen to share a replica with the
8
+ * asker. Two people editing the same scene through different pods would each
9
+ * see an empty room while broadcasting cursors at each other perfectly.
10
+ *
11
+ * So presence gets one row per tracked client, in Postgres, readable by every
12
+ * instance. Three consequences worth stating:
13
+ *
14
+ * - **The table is the roster; the in-process map is a cache of our own
15
+ * clients.** Reads answer from the table when this store is active, so the
16
+ * answer is the same whichever instance is asked.
17
+ *
18
+ * - **`last_seen` is the liveness signal, and it is already there.** The client
19
+ * heartbeats presence every ~20 s against a 30 s window; the sweep that has
20
+ * always reaped local stale entries now also reaps rows belonging to
21
+ * instances that stopped writing — which is exactly what a crashed pod looks
22
+ * like. Crash recovery is a property of the TTL, not a separate mechanism.
23
+ *
24
+ * - **The sweep deletes with `RETURNING`.** Whichever instance wins the delete
25
+ * is the one that announces the departures, so a stale client produces one
26
+ * `presence_diff` for the cluster rather than one per replica.
27
+ */
28
+ import { NodePgDatabase } from "drizzle-orm/node-postgres";
29
+ /** A tracked client, as any instance sees it. */
30
+ export interface PresenceRow {
31
+ channel: string;
32
+ clientId: string;
33
+ state: Record<string, unknown>;
34
+ }
35
+ export declare class ChannelPresenceStore {
36
+ private readonly db;
37
+ private readonly instanceId;
38
+ private tablesReady;
39
+ constructor(db: NodePgDatabase<Record<string, unknown>>, instanceId: string);
40
+ /** Create the roster table. Idempotent. */
41
+ ensureTables(): Promise<void>;
42
+ /** Record (or refresh) a client's presence. */
43
+ track(channel: string, clientId: string, state: Record<string, unknown>): Promise<void>;
44
+ /** Drop one client's presence in one channel. */
45
+ remove(channel: string, clientId: string): Promise<void>;
46
+ /** Drop a client from every channel — used when its socket closes. */
47
+ removeClient(clientId: string): Promise<void>;
48
+ /** The global roster for a channel. */
49
+ roster(channel: string): Promise<Record<string, Record<string, unknown>>>;
50
+ /**
51
+ * Reap rows this instance is not responsible for and that have gone quiet.
52
+ *
53
+ * Own rows are excluded because the in-process sweep already handles them —
54
+ * and handles them better, since it can tell "the socket is gone" from "the
55
+ * heartbeat is late". What is left is precisely the interesting case: rows
56
+ * written by an instance that is no longer writing.
57
+ *
58
+ * Returns what was removed, so the caller can announce it.
59
+ */
60
+ sweepStale(ttlMs: number): Promise<PresenceRow[]>;
61
+ /**
62
+ * Remove every row this instance owns. Called on graceful shutdown so a
63
+ * rolling deploy does not leave a TTL window of ghosts in every roster.
64
+ */
65
+ removeInstance(): Promise<void>;
66
+ }
@@ -0,0 +1,47 @@
1
+ /**
2
+ * A dedicated, self-healing Postgres `LISTEN` connection.
3
+ *
4
+ * Every cross-instance feature in the backend needs the same thing: one
5
+ * connection *outside* the Drizzle pool that stays open, holds a `LISTEN`, and
6
+ * comes back on its own after the database or the network drops it. CDC needed
7
+ * it first; the channel bus needs it too. This is that connection, with the one
8
+ * behaviour that matters to callers preserved: the **first** connect is
9
+ * validated and rethrown, so a caller can fall back to a different strategy,
10
+ * while every later drop is repaired quietly in the background.
11
+ *
12
+ * `LISTEN` is session state, so this connection must not go through a
13
+ * transaction-mode pooler (PgBouncer): give it the direct database URL.
14
+ */
15
+ export interface PgNotifyListenerOptions {
16
+ /** Direct Postgres connection string (must bypass a transaction-mode pooler). */
17
+ connectionString: string;
18
+ /** NOTIFY channel to LISTEN on. Must be a plain identifier — it is interpolated. */
19
+ channel: string;
20
+ /** Called for every notification payload received. */
21
+ onPayload: (payload: string) => void | Promise<void>;
22
+ /** Prefix for log lines, e.g. `"[CDC]"`. */
23
+ logLabel: string;
24
+ /** Delay before a reconnect attempt. */
25
+ reconnectDelayMs?: number;
26
+ }
27
+ export declare class PgNotifyListener {
28
+ private readonly options;
29
+ private client?;
30
+ private running;
31
+ private reconnectTimer?;
32
+ constructor(options: PgNotifyListenerOptions);
33
+ /** Whether the listener is meant to be connected right now. */
34
+ get active(): boolean;
35
+ /**
36
+ * Connect and begin listening. Idempotent.
37
+ *
38
+ * Rejects if the *initial* connection or `LISTEN` fails, leaving the
39
+ * listener stopped — callers use that to degrade deliberately instead of
40
+ * running blind against a channel nothing is delivering.
41
+ */
42
+ start(): Promise<void>;
43
+ /** Stop listening and release the connection. Idempotent. */
44
+ stop(): Promise<void>;
45
+ private connect;
46
+ private scheduleReconnect;
47
+ }
@@ -4,6 +4,7 @@ import { DataDriver, WebSocketMessage } from "@rebasepro/types";
4
4
  import { NodePgDatabase } from "drizzle-orm/node-postgres";
5
5
  import { RealtimeProvider, CollectionSubscriptionConfig, SingleSubscriptionConfig } from "../interfaces";
6
6
  import { PostgresCollectionRegistry } from "../collections/PostgresCollectionRegistry";
7
+ import { ChannelBus } from "./channel-bus";
7
8
  import type { ChannelRetentionRule } from "@rebasepro/types";
8
9
  /**
9
10
  * Auth context stored per-subscription so real-time refetches respect RLS.
@@ -45,8 +46,33 @@ export declare class RealtimeService extends EventEmitter implements RealtimePro
45
46
  * wait on each other.
46
47
  */
47
48
  private channelSendQueues;
49
+ /**
50
+ * Cross-instance transport for channel frames and presence.
51
+ *
52
+ * Defaults to the memory bus, which publishes nowhere — so a single-instance
53
+ * deployment runs the same fan-out it always did, with one resolved promise
54
+ * per broadcast for company. See `channel-bus/ChannelBus.ts`.
55
+ */
56
+ private bus;
57
+ /**
58
+ * The shared presence roster, present only when a real bus is active.
59
+ *
60
+ * Fan-out alone is not enough for presence: `presence_state` has to answer
61
+ * with everyone in the channel, and per-process maps can only answer for
62
+ * this replica's clients. See `channel-presence.ts`.
63
+ */
64
+ private presenceStore?;
65
+ /** Sweeps roster rows left behind by instances that stopped heartbeating. */
66
+ private presenceSweepInterval?;
67
+ /**
68
+ * Channels whose oversized ephemeral broadcasts have already been reported,
69
+ * so a hot channel logs the problem once rather than once per message.
70
+ */
71
+ private oversizedBroadcastWarned;
48
72
  private presenceInterval?;
49
73
  private static readonly PRESENCE_TIMEOUT_MS;
74
+ /** How often stale roster rows from other instances are reaped. */
75
+ private static readonly PRESENCE_SWEEP_INTERVAL_MS;
50
76
  private dataService;
51
77
  private _subscriptions;
52
78
  private subscriptionCallbacks;
@@ -245,6 +271,44 @@ export declare class RealtimeService extends EventEmitter implements RealtimePro
245
271
  private persistAndFanOut;
246
272
  /** Deliver a broadcast frame to every member of a channel but the sender. */
247
273
  private fanOutBroadcast;
274
+ /**
275
+ * Install the transport that carries channel frames between instances.
276
+ *
277
+ * Called once at boot. A bus that cannot start is reported and replaced with
278
+ * the memory bus: losing cross-instance fan-out degrades collaboration to
279
+ * what it was before this existed, whereas refusing to boot takes the whole
280
+ * backend down for it.
281
+ */
282
+ configureChannelBus(bus: ChannelBus): Promise<void>;
283
+ /** Which transport is in use — `"memory"` means per-instance only. */
284
+ getChannelBusKind(): ChannelBus["kind"];
285
+ /**
286
+ * Send a broadcast to the other instances.
287
+ *
288
+ * Fire-and-forget by design: the clients on this instance have already been
289
+ * served, and a bus that is briefly unreachable must not turn a broadcast
290
+ * into an error for the sender.
291
+ */
292
+ private publishBroadcast;
293
+ private publishFrame;
294
+ /**
295
+ * Tell the sender that a message was delivered locally but nowhere else.
296
+ *
297
+ * Staying quiet here would be the worst option available: on one instance
298
+ * the app works, on two it works for half the users, and nothing in the
299
+ * logs connects the two. The fix is a one-liner in config — give the
300
+ * channel a retention rule and the message travels as a pointer instead —
301
+ * so the message says exactly that.
302
+ */
303
+ private reportOversizedBroadcast;
304
+ /**
305
+ * Deliver a frame published by another instance to this one's clients.
306
+ *
307
+ * Frames we published ourselves are dropped on arrival — the local fan-out
308
+ * happened before the publish — exactly as the entity-change handler skips
309
+ * its own `sid`.
310
+ */
311
+ private handleBusFrame;
248
312
  /**
249
313
  * Install retention rules and create the tables they need.
250
314
  *
@@ -266,16 +330,60 @@ export declare class RealtimeService extends EventEmitter implements RealtimePro
266
330
  */
267
331
  private handleChannelHistoryRequest;
268
332
  private sendChannelHistory;
269
- /** Track presence in a channel */
333
+ /**
334
+ * Track presence in a channel.
335
+ *
336
+ * The client re-sends this every ~20s as a heartbeat against the 30s
337
+ * timeout, so most calls carry the state that is already recorded. Those
338
+ * refresh `last_seen` and stop there: re-announcing an unchanged state to
339
+ * every instance would put a bus message per client per heartbeat on the
340
+ * wire to tell everyone nothing happened.
341
+ */
270
342
  trackPresence(clientId: string, channel: string, state: Record<string, unknown>): void;
271
- /** Remove presence from a channel */
272
- removePresence(clientId: string, channel: string): void;
273
- /** Send full presence state to a specific client */
343
+ /**
344
+ * Remove presence from a channel.
345
+ *
346
+ * `skipStore` is for the socket-close path, which clears every channel at
347
+ * once and then deletes the client's rows in a single statement instead of
348
+ * one per channel.
349
+ */
350
+ removePresence(clientId: string, channel: string, options?: {
351
+ skipStore?: boolean;
352
+ }): void;
353
+ /**
354
+ * Send the full roster for a channel to one client.
355
+ *
356
+ * Answered from the shared table when there is one, because "who is in this
357
+ * document?" has a single answer that must not depend on which replica the
358
+ * asker happens to be connected to. Without a bus there is nothing to share
359
+ * and the local map *is* the roster — that path stays synchronous, which is
360
+ * what it always was.
361
+ */
274
362
  sendPresenceState(clientId: string, channel: string): void;
275
- /** Broadcast presence diff (joins/leaves) to channel */
276
- private broadcastPresenceDiff;
363
+ /** Presence of the clients connected to this instance. */
364
+ private localPresences;
365
+ private sendPresenceStateMessage;
366
+ /** Deliver a presence diff to this instance's members of the channel. */
367
+ private deliverPresenceDiff;
368
+ /** Tell the other instances about a presence change. */
369
+ private publishPresenceDiff;
370
+ /** Run a roster write when there is a roster, and never let it throw. */
371
+ private presenceStoreOp;
277
372
  /** Periodic cleanup for stale presences */
278
373
  private ensurePresenceCleanup;
374
+ /**
375
+ * Reap roster rows whose owning instance stopped heartbeating.
376
+ *
377
+ * This is the cross-instance half of the sweep above, and it doubles as
378
+ * crash recovery: a pod that dies takes its clients with it but leaves
379
+ * their rows behind, and after one TTL window they look exactly like any
380
+ * other client that went quiet. The delete returns what it removed, so
381
+ * whichever instance wins the race is the one that announces the
382
+ * departures — once for the cluster, not once per replica.
383
+ */
384
+ private ensurePresenceSweep;
385
+ /** One pass of the stale-roster sweep. See {@link ensurePresenceSweep}. */
386
+ private sweepStalePresence;
279
387
  /**
280
388
  * Gracefully tear down all realtime resources.
281
389
  *
package/package.json CHANGED
@@ -1,7 +1,7 @@
1
1
  {
2
2
  "name": "@rebasepro/server-postgres",
3
3
  "type": "module",
4
- "version": "0.10.1-canary.6f89f77",
4
+ "version": "0.10.1-canary.7609565",
5
5
  "description": "PostgreSQL data source backend implementation for Rebase with Drizzle ORM",
6
6
  "funding": {
7
7
  "url": "https://github.com/sponsors/rebaseco"
@@ -47,11 +47,11 @@
47
47
  "execa": "^9.6.1",
48
48
  "pg": "^8.21.0",
49
49
  "ws": "^8.21.0",
50
- "@rebasepro/codegen": "0.10.1-canary.6f89f77",
51
- "@rebasepro/server": "0.10.1-canary.6f89f77",
52
- "@rebasepro/common": "0.10.1-canary.6f89f77",
53
- "@rebasepro/utils": "0.10.1-canary.6f89f77",
54
- "@rebasepro/types": "0.10.1-canary.6f89f77"
50
+ "@rebasepro/codegen": "0.10.1-canary.7609565",
51
+ "@rebasepro/common": "0.10.1-canary.7609565",
52
+ "@rebasepro/server": "0.10.1-canary.7609565",
53
+ "@rebasepro/utils": "0.10.1-canary.7609565",
54
+ "@rebasepro/types": "0.10.1-canary.7609565"
55
55
  },
56
56
  "devDependencies": {
57
57
  "@hono/node-server": "^2.0.9",
@@ -37,6 +37,8 @@ import { buildCollectionsFromSchema, introspectSchema, readRlsStatus } from "./s
37
37
  import { buildDrizzleTablesFromSchema, buildDrizzleRelationsFromSchema } from "./schema/dynamic-tables";
38
38
  import { detectConnectionPosture, ensureAppRole, validatePolicyPgRoles, warnOnAnonymousGrants, REBASE_USER_ROLE, type RawSqlRunner } from "./security/rls-enforcement";
39
39
  import { provisionTriggerCdc, type CdcTableRef } from "./services/cdc/trigger-cdc";
40
+ import { createChannelBus, resolveChannelBusSetting } from "./services/channel-bus";
41
+ import { isChannelBusInstance } from "@rebasepro/types";
40
42
 
41
43
  export interface PostgresDriverConfig {
42
44
  connectionString?: string;
@@ -54,9 +56,13 @@ export interface PostgresDriverConfig {
54
56
  */
55
57
  introspectionSchema?: string;
56
58
  /**
57
- * Realtime options. Currently only channel retention, which is opt-in:
58
- * without rules here no channel keeps any history and broadcast stays
59
- * fire-and-forget. See {@link ChannelRetentionRule}.
59
+ * Realtime options, both opt-in:
60
+ *
61
+ * - `channels` — retention. Without rules no channel keeps any history and
62
+ * broadcast stays fire-and-forget. See {@link ChannelRetentionRule}.
63
+ * - `bus` — the cross-instance transport for channel broadcast and
64
+ * presence. Defaults to in-process only, which is correct for a single
65
+ * instance and wrong for two. See {@link ChannelBusConfig}.
60
66
  */
61
67
  realtime?: RealtimeChannelsConfig;
62
68
  }
@@ -337,6 +343,29 @@ export function createPostgresBootstrapper(pgConfig: PostgresDriverConfig): Back
337
343
  // Prefer DATABASE_DIRECT_URL to bypass PgBouncer for LISTEN/NOTIFY.
338
344
  const directUrl = process.env.DATABASE_DIRECT_URL || pgConfig.connectionString;
339
345
 
346
+ // ── Cross-instance channel bus ───────────────────────────────────
347
+ // Entity changes already span instances (CDC / LISTEN below).
348
+ // Channel broadcast and presence did not — they lived in per-process
349
+ // maps, so behind two replicas the clients of one were invisible to
350
+ // the other. Opt-in, and a no-op when left at "memory".
351
+ try {
352
+ const busSetting = resolveChannelBusSetting(pgConfig.realtime?.bus);
353
+ // A supplied instance is always installed; a named built-in only
354
+ // when it is not the in-process default, so the common case
355
+ // touches none of this machinery.
356
+ const wantsBus = isChannelBusInstance(busSetting) || busSetting.type !== "memory";
357
+ if (wantsBus) {
358
+ await realtimeService.configureChannelBus(
359
+ createChannelBus(busSetting, {
360
+ db: schemaAwareDb as unknown as NodePgDatabase<Record<string, unknown>>,
361
+ directUrl
362
+ })
363
+ );
364
+ }
365
+ } catch (err) {
366
+ logger.warn("⚠️ [ChannelBus] Could not configure the channel bus — channel broadcast and presence stay per-instance", { error: err });
367
+ }
368
+
340
369
  // Database-level Change Data Capture. When active, realtime events are
341
370
  // emitted for EVERY committed write — including ones that bypass the
342
371
  // Rebase API (psql, another service's cron, raw SQL, the Studio SQL
package/src/index.ts CHANGED
@@ -7,6 +7,10 @@ export * from "./schema/generate-drizzle-schema-logic";
7
7
  export * from "./schema/generate-drizzle-schema";
8
8
  export * from "./utils/drizzle-conditions";
9
9
  export * from "./services/realtimeService";
10
+ // The channel bus is a public extension point: a transport published as its own
11
+ // package implements `ChannelBus` (declared in @rebasepro/types and re-exported
12
+ // here) and is passed to `realtime.bus`. Without this export it would be one.
13
+ export * from "./services/channel-bus";
10
14
  export * from "./websocket";
11
15
  export * from "./collections/PostgresCollectionRegistry";
12
16
  export * from "./services/BranchService";