Fix of the pglite-sync plugin for latest electric (#535)

* Second fix of the pglite-sync plugin for latest electric * working
electric-sql · Feb 13, 2025 · c489ec0 · c489ec0
1 parent b5368a4
commit c489ec0
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 2 deletions.
diff --git a/.changeset/beige-frogs-film.md b/.changeset/beige-frogs-film.md
@@ -0,0 +1,5 @@
+---
+'@electric-sql/pglite-sync': patch
+---
+
+Correctly persist the offset during initial sync
diff --git a/packages/pglite-sync/src/index.ts b/packages/pglite-sync/src/index.ts
@@ -253,6 +253,7 @@ async function createPlugin(
               shapeKey: options.shapeKey,
               shapeId: shapeHandle,
               lastOffset: getMessageOffset(
+                stream,
                 messageAggregator[messageAggregator.length - 1],
               ),
             })
@@ -673,10 +674,18 @@ function subscriptionMetadataTableName(metadatSchema: string) {
 
 const subscriptionTableName = `shape_subscriptions_metadata`
 
-function getMessageOffset(message: LegacyChangeMessage<any>): Offset {
+function getMessageOffset(
+  stream: ShapeStream,
+  message: LegacyChangeMessage<any>,
+): Offset {
   if (message.offset) {
     return message.offset
-  } else {
+  } else if (
+    message.headers.lsn !== undefined &&
+    message.headers.op_position !== undefined
+  ) {
     return `${message.headers.lsn}_${message.headers.op_position}` as Offset
+  } else {
+    return stream.lastOffset
   }
 }