Merge pull request #495 from Ungate-Ai/main

fix: Gracefully Handle Add Participants Unique Constraint Error in Postgres
elizaOS · Nov 22, 2024 · 68a4dcd · 68a4dcd
2 parents 5ea1551 + 2f06f15
commit 68a4dcd
Showing 1 changed file with 21 additions and 2 deletions.
diff --git a/packages/adapter-postgres/src/index.ts b/packages/adapter-postgres/src/index.ts
@@ -729,15 +729,34 @@ export class PostgresDatabaseAdapter
     async addParticipant(userId: UUID, roomId: UUID): Promise<boolean> {
         const client = await this.pool.connect();
         try {
+            // Check if the participant already exists
+            const existingParticipant = await client.query(
+                `SELECT * FROM participants WHERE "userId" = $1 AND "roomId" = $2`,
+                [userId, roomId]
+            );
+
+            if (existingParticipant.rows.length > 0) {
+                console.log(`Participant with userId ${userId} already exists in room ${roomId}.`);
+                return; // Exit early if the participant already exists
+            }
+
+            // Proceed to add the participant if they do not exist
             await client.query(
                 `INSERT INTO participants (id, "userId", "roomId") 
                 VALUES ($1, $2, $3)`,
                 [v4(), userId, roomId]
             );
             return true;
         } catch (error) {
-            console.log("Error adding participant", error);
-            return false;
+            // This is to prevent duplicate participant error in case of a race condition
+            // Handle unique constraint violation error (code 23505)
+            if (error.code === '23505') {
+                console.warn(`Participant with userId ${userId} already exists in room ${roomId}.`);               // Optionally, you can log this or handle it differently
+            } else {
+                // Handle other errors
+                console.error('Error adding participant:', error);
+                return false;
+            }
         } finally {
             client.release();
         }