@powersync/service-module-postgres 0.0.0-dev-20250116115804 → 0.0.0-dev-20250122110924

package/src/replication/WalStream.ts

@@ -1,5 +1,11 @@
 import * as lib_postgres from '@powersync/lib-service-postgres';
-import { container, errors, logger } from '@powersync/lib-services-framework';
+import {
+  container,
+  errors,
+  logger,
+  ReplicationAbortedError,
+  ReplicationAssertionError
+} from '@powersync/lib-services-framework';
 import { getUuidReplicaIdentityBson, Metrics, SourceEntityDescriptor, storage } from '@powersync/service-core';
 import * as pgwire from '@powersync/service-jpgwire';
 import { DatabaseInputRow, SqliteRow, SqlSyncRules, TablePattern, toSyncRulesRow } from '@powersync/service-sync-rules';
@@ -133,7 +139,7 @@ export class WalStream {
     for (let row of tableRows) {
       const name = row.table_name as string;
       if (typeof row.relid != 'bigint') {
-        throw new Error(`missing relid for ${name}`);
+        throw new ReplicationAssertionError(`Missing relid for ${name}`);
       }
       const relid = Number(row.relid as bigint);
 
@@ -294,7 +300,7 @@ export class WalStream {
       }
     }
 
-    throw new Error('Unreachable');
+    throw new ReplicationAssertionError('Unreachable');
   }
 
   async estimatedCount(db: pgwire.PgConnection, table: storage.SourceTable): Promise<string> {
@@ -415,7 +421,7 @@ WHERE  oid = $1::regclass`,
         lastLogIndex = at;
       }
       if (this.abort_signal.aborted) {
-        throw new Error(`Aborted initial replication of ${this.slot_name}`);
+        throw new ReplicationAbortedError(`Aborted initial replication of ${this.slot_name}`);
       }
 
       for (const record of WalStream.getQueryData(rows)) {
@@ -441,7 +447,7 @@ WHERE  oid = $1::regclass`,
 
   async handleRelation(batch: storage.BucketStorageBatch, descriptor: SourceEntityDescriptor, snapshot: boolean) {
     if (!descriptor.objectId && typeof descriptor.objectId != 'number') {
-      throw new Error('objectId expected');
+      throw new ReplicationAssertionError(`objectId expected, got ${typeof descriptor.objectId}`);
     }
     const result = await this.storage.resolveTable({
       group_id: this.group_id,
@@ -484,6 +490,7 @@ WHERE  oid = $1::regclass`,
           await db.query('COMMIT');
         } catch (e) {
           await db.query('ROLLBACK');
+          // TODO: Wrap with custom error type
           throw e;
         }
       } finally {
@@ -501,7 +508,7 @@ WHERE  oid = $1::regclass`,
     if (table == null) {
       // We should always receive a replication message before the relation is used.
       // If we can't find it, it's a bug.
-      throw new Error(`Missing relation cache for ${relationId}`);
+      throw new ReplicationAssertionError(`Missing relation cache for ${relationId}`);
     }
     return table;
   }

package/src/replication/WalStreamReplicator.ts

@@ -2,6 +2,7 @@ import { replication, storage } from '@powersync/service-core';
 import { ConnectionManagerFactory } from './ConnectionManagerFactory.js';
 import { cleanUpReplicationSlot } from './replication-utils.js';
 import { WalStreamReplicationJob } from './WalStreamReplicationJob.js';
+import { PostgresModule } from '../module/PostgresModule.js';
 
 export interface WalStreamReplicatorOptions extends replication.AbstractReplicatorOptions {
   connectionFactory: ConnectionManagerFactory;
@@ -42,4 +43,8 @@ export class WalStreamReplicator extends replication.AbstractReplicator<WalStrea
     await super.stop();
     await this.connectionFactory.shutdown();
   }
+
+  async testConnection() {
+    return await PostgresModule.testConnection(this.connectionFactory.dbConnectionConfig);
+  }
 }

package/src/replication/replication-utils.ts

@@ -1,7 +1,7 @@
 import * as pgwire from '@powersync/service-jpgwire';
 
 import * as lib_postgres from '@powersync/lib-service-postgres';
-import { logger } from '@powersync/lib-services-framework';
+import { ErrorCode, logger, ServiceError } from '@powersync/lib-services-framework';
 import { PatternResult, storage } from '@powersync/service-core';
 import * as sync_rules from '@powersync/service-sync-rules';
 import * as service_types from '@powersync/service-types';
@@ -117,17 +117,22 @@ $$ LANGUAGE plpgsql;`
   });
   const row = pgwire.pgwireRows(rs)[0];
   if (row == null) {
-    throw new Error(
+    throw new ServiceError(
+      ErrorCode.PSYNC_S1141,
       `Publication '${publicationName}' does not exist. Run: \`CREATE PUBLICATION ${publicationName} FOR ALL TABLES\`, or read the documentation for details.`
     );
   }
   if (row.pubinsert == false || row.pubupdate == false || row.pubdelete == false || row.pubtruncate == false) {
-    throw new Error(
+    throw new ServiceError(
+      ErrorCode.PSYNC_S1142,
       `Publication '${publicationName}' does not publish all changes. Create a publication using \`WITH (publish = "insert, update, delete, truncate")\` (the default).`
     );
   }
   if (row.pubviaroot) {
-    throw new Error(`'${publicationName}' uses publish_via_partition_root, which is not supported.`);
+    throw new ServiceError(
+      ErrorCode.PSYNC_S1143,
+      `'${publicationName}' uses publish_via_partition_root, which is not supported.`
+    );
   }
 }
 

package/src/utils/migration_lib.ts

@@ -1,3 +1,4 @@
+import { ServiceAssertionError } from '@powersync/lib-services-framework';
 import * as pgwire from '@powersync/service-jpgwire';
 
 export type MigrationFunction = (db: pgwire.PgConnection) => Promise<void>;
@@ -14,7 +15,7 @@ export class Migrations {
 
   add(id: number, name: string, up: MigrationFunction) {
     if (this.migrations.length > 0 && this.migrations[this.migrations.length - 1].id >= id) {
-      throw new Error('Migration ids must be strictly incrementing');
+      throw new ServiceAssertionError('Migration ids must be strictly incrementing');
     }
     this.migrations.push({ id, up, name });
   }