feat: denote end of injected subset snapshot with an additional message

icehaunter · icehaunter · commit 214bdd67c8bd · 2025-12-11T13:58:14.000+03:00
diff --git a/.changeset/swift-brooms-double.md b/.changeset/swift-brooms-double.md
@@ -0,0 +1,5 @@
+---
+'@electric-sql/client': minor
+---
+
+feat: denote end of injected subset snapshot with an additional message
diff --git a/packages/typescript-client/src/client.ts b/packages/typescript-client/src/client.ts
@@ -7,6 +7,7 @@ import {
   GetExtensions,
   ChangeMessage,
   SnapshotMetadata,
+  SubsetParams,
 } from './types'
 import { MessageParser, Parser, TransformFunction } from './parser'
 import { ColumnMapper, encodeWhereClause } from './column-mapper'
@@ -128,14 +129,6 @@ export type ExternalParamsRecord<T extends Row<unknown> = Row> = {
   [K in string]: ParamValue | undefined
 } & Partial<PostgresParams<T>> & { [K in ReservedParamKeys]?: never }
 
-export type SubsetParams = {
-  where?: string
-  params?: Record<string, string>
-  limit?: number
-  offset?: number
-  orderBy?: string
-}
-
 type ReservedParamKeys =
   | typeof LIVE_CACHE_BUSTER_QUERY_PARAM
   | typeof SHAPE_HANDLE_QUERY_PARAM
@@ -1457,6 +1450,7 @@ export class ShapeStream<T extends Row<unknown> = Row>
 
       const dataWithEndBoundary = (data as Array<Message<T>>).concat([
         { headers: { control: `snapshot-end`, ...metadata } },
+        { headers: { control: `subset-end`, ...opts } },
       ])
 
       this.#snapshotTracker.addSnapshot(
diff --git a/packages/typescript-client/src/helpers.ts b/packages/typescript-client/src/helpers.ts
@@ -66,11 +66,9 @@ export function isUpToDateMessage<T extends Row<unknown> = Row>(
  * If we are not in SSE mode this function will return undefined.
  */
 export function getOffset(message: ControlMessage): Offset | undefined {
+  if (message.headers.control != 'up-to-date') return
   const lsn = message.headers.global_last_seen_lsn
-  if (!lsn) {
-    return
-  }
-  return `${lsn}_0` as Offset
+  return lsn ? (`${lsn}_0` as Offset) : undefined
 }
 
 /**
diff --git a/packages/typescript-client/src/types.ts b/packages/typescript-client/src/types.ts
@@ -68,13 +68,22 @@ export type MoveTag = string
  */
 export type MoveOutPattern = { pos: number; value: string }
 
+export type SubsetParams = {
+  where?: string
+  params?: Record<string, string>
+  limit?: number
+  offset?: number
+  orderBy?: string
+}
+
 export type ControlMessage = {
   headers:
     | (Header & {
         control: `up-to-date` | `must-refetch`
         global_last_seen_lsn?: string
       })
     | (Header & { control: `snapshot-end` } & PostgresSnapshot)
+    | ({ control: `subset-end` } & SubsetParams)
 }
 
 export type EventMessage = {

-Original file line number
+Diff line change
@@ @@ -0,0 +1,5 @@ @@
 +---
 +'@electric-sql/client': minor
 +---
++
 +feat: denote end of injected subset snapshot with an additional message