@axiom-lattice/pg-stores 1.0.5 → 1.0.7

package/src/stores/PostgreSQLThreadStore.ts

@@ -36,6 +36,7 @@ export class PostgreSQLThreadStore implements ThreadStore {
   private migrationManager: MigrationManager;
   private initialized: boolean = false;
   private ownsPool: boolean = true;
+  private initPromise: Promise<void> | null = null; // Promise-based lock to prevent concurrent initialization
 
   constructor(options: PostgreSQLThreadStoreOptions) {
     // Create Pool from config
@@ -69,14 +70,31 @@ export class PostgreSQLThreadStore implements ThreadStore {
 
   /**
    * Initialize the store and run migrations
+   * Uses a promise-based lock to prevent concurrent initialization
    */
   async initialize(): Promise<void> {
+    // If already initialized, return immediately
     if (this.initialized) {
       return;
     }
 
-    await this.migrationManager.migrate();
-    this.initialized = true;
+    // If initialization is in progress, wait for it
+    if (this.initPromise) {
+      return this.initPromise;
+    }
+
+    // Start initialization and store the promise
+    this.initPromise = (async () => {
+      try {
+        await this.migrationManager.migrate();
+        this.initialized = true;
+      } finally {
+        // Clear the promise after completion (success or failure)
+        this.initPromise = null;
+      }
+    })();
+
+    return this.initPromise;
   }
 
   /**