[MongoDB] Short timeout for testing connections

.changeset/moody-cobras-walk.md

@@ -0,0 +1,5 @@
+---
+'@powersync/service-module-mongodb': patch
+---
+
+Use short timeout for testing mongodb connections.

modules/module-mongodb/src/module/MongoModule.ts

@@ -55,9 +55,14 @@ export class MongoModule extends replication.ReplicationModule<types.MongoConnec
   async testConnection(config: types.MongoConnectionConfig): Promise<void> {
     this.decodeConfig(config);
     const normalisedConfig = this.resolveConfig(this.decodedConfig!);
-    const connectionManager = new MongoManager(normalisedConfig);
+    const connectionManager = new MongoManager(normalisedConfig, {
+      // Use short timeouts for testing connections.
+      // Must be < 30s, to ensure we get a proper timeout error.
+      socketTimeoutMS: 5_000,
+      serverSelectionTimeoutMS: 5_000
+    });
     try {
-      return checkSourceConfiguration(connectionManager);
+      return await checkSourceConfiguration(connectionManager);
     } finally {
       await connectionManager.end();
     }

modules/module-mongodb/src/replication/MongoManager.ts

@@ -8,7 +8,10 @@ export class MongoManager {
   public readonly client: mongo.MongoClient;
   public readonly db: mongo.Db;
 
-  constructor(public options: NormalizedMongoConnectionConfig) {
+  constructor(
+    public options: NormalizedMongoConnectionConfig,
+    overrides?: mongo.MongoClientOptions
+  ) {
     // The pool is lazy - no connections are opened until a query is performed.
     this.client = new mongo.MongoClient(options.uri, {
       auth: {
@@ -28,7 +31,8 @@ export class MongoManager {
       maxPoolSize: 8,
 
       maxConnecting: 3,
-      maxIdleTimeMS: 60_000
+      maxIdleTimeMS: 60_000,
+      ...overrides
     });
     this.db = this.client.db(options.database, {});
   }

modules/module-postgres/src/module/PostgresModule.ts

@@ -133,7 +133,7 @@ export class PostgresModule extends replication.ReplicationModule<types.Postgres
     });
     const connection = await connectionManager.snapshotConnection();
     try {
-      return checkSourceConfiguration(connection, PUBLICATION_NAME);
+      return await checkSourceConfiguration(connection, PUBLICATION_NAME);
     } finally {
       await connectionManager.end();
     }