Return PIPELINE_SYNC result alone in generators.fetch_many()

diff --git a/psycopg/psycopg/generators.py b/psycopg/psycopg/generators.py
index 76583bbf367d9e225f288b790a0a6217d59b1d7c..2a4aeaf5df2d3a024f4bff689b151a6ad9678cf3 100644 (file)
--- a/psycopg/psycopg/generators.py
+++ b/psycopg/psycopg/generators.py
@@ -125,6 +125,12 @@ def fetch_many(pgconn: PGconn) -> PQGen[List[PGresult]]:
             # for every request so let's break the endless loop.
             break
 
+        if res.status == pq.ExecStatus.PIPELINE_SYNC:
+            # PIPELINE_SYNC is not followed by a NULL, but we return it alone
+            # similarly to other result sets.
+            assert len(results) == 1, results
+            break
+
     return results
 
 

diff --git a/psycopg_c/psycopg_c/_psycopg/generators.pyx b/psycopg_c/psycopg_c/_psycopg/generators.pyx
index 239494974ddf74d8fd74ce2091a9f8b713bbec16..86461024e30f784bc5d68431702a899aa261b635 100644 (file)
--- a/psycopg_c/psycopg_c/_psycopg/generators.pyx
+++ b/psycopg_c/psycopg_c/_psycopg/generators.pyx
@@ -137,6 +137,12 @@ def fetch_many(pq.PGconn pgconn) -> PQGen[List[PGresult]]:
             # for every request so let's break the endless loop.
             break
 
+        if status == libpq.PGRES_PIPELINE_SYNC:
+            # PIPELINE_SYNC is not followed by a NULL, but we return it alone
+            # similarly to other result sets.
+            assert len(results) == 1, results
+            break
+
     return results