try fix

Author: bymoye

frameworks/Python/blacksheep/app-socketify.py

@@ -111,21 +111,26 @@ async def db_updates_test(request,queries):
         random.sample(range(1, 10000), num_queries)
     ), key=lambda x: x[1])
     worlds = [{"id": row_id, "randomNumber": number} for row_id, number in updates]
-    async with db_pool.connection() as db_conn:
-        await db_conn.execute("SET TRANSACTION ISOLATION LEVEL READ COMMITTED")
-        async with db_conn.cursor() as cursor:
-            for row_id, number in updates:
-                await cursor.execute(READ_ROW_SQL, (row_id,))
-                await cursor.fetchone()
-            for _ in range(5):
-                try:
-                    await cursor.executemany(WRITE_ROW_SQL + " NOWAIT", [(number, row_id) for row_id, number in updates])
-                    break
-                except psycopg.errors.DeadlockDetected:
-                    await db_conn.rollback()
-                    continue
-            # await cursor.executemany(WRITE_ROW_SQL, [(number, row_id) for row_id, number in updates])
-    return bs.json(worlds)
+    for _ in range(5):
+        async with db_pool.connection() as db_conn:
+            try:
+                await db_conn.execute("SET TRANSACTION ISOLATION LEVEL READ COMMITTED")
+                async with db_conn.cursor() as cursor:
+                    for row_id, number in updates:
+                        await cursor.execute(READ_ROW_SQL, (row_id,))
+                        await cursor.fetchone()
+                    for _ in range(5):
+                        try:
+                            await cursor.executemany(WRITE_ROW_SQL, [(number, row_id) for row_id, number in updates])
+                            return bs.json(worlds)
+                        except psycopg.errors.DeadlockDetected:
+                            await db_conn.rollback()
+                            continue
+                    # await cursor.executemany(WRITE_ROW_SQL, [(number, row_id) for row_id, number in updates])
+            except (psycopg.errors.OperationalError, psycopg.errors.PipelineAborted):
+                await db_conn.rollback()
+                continue
+    raise Exception("connect error")
 
 @bs.get('/plaintext')
 async def plaintext_test(request):