psycopg/psycopg2
1
1
Fork 0
mirror of https://github.com/psycopg/psycopg2.git synced 2025-11-04 01:37:31 +03:00
Code Issues Packages Projects Releases Wiki Activity

Check for possible errors from PQflush.

Browse Source 
Noticed by Daniele Varrazzo.
This commit is contained in:
Jan Urbański 2010-04-11 03:46:15 +02:00 committed by Federico Di Gregorio
parent 61f890c317
commit 4574bde3a2
1 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
psycopg/pqpath.c
View File

@ -725,6 +725,8 @@ pq_execute(cursorObject *curs, const char *query, int async)
} }
else if (async == 1) { else if (async == 1) {
int ret;
Dprintf("pq_execute: executing ASYNC query:"); Dprintf("pq_execute: executing ASYNC query:");
Dprintf(" %-.200s", query); Dprintf(" %-.200s", query);
@ -738,15 +740,21 @@ pq_execute(cursorObject *curs, const char *query, int async)
} }
Dprintf("pq_execute: async query sent to backend"); Dprintf("pq_execute: async query sent to backend");
if (PQflush(curs->conn->pgconn) == 0) { ret = PQflush(curs->conn->pgconn);
if (ret == 0) {
/* the query got fully sent to the server */ /* the query got fully sent to the server */
Dprintf("pq_execute: query got flushed immediately"); Dprintf("pq_execute: query got flushed immediately");
/* the async status will be ASYNC_READ */ /* the async status will be ASYNC_READ */
async_status = ASYNC_READ; async_status = ASYNC_READ;
} }
else { else if (ret == 1) {
/* not all of the query got sent to the server */
async_status = ASYNC_WRITE; async_status = ASYNC_WRITE;
} }
else {
/* there was an error */
return -1;
}
} }
pthread_mutex_unlock(&(curs->conn->lock)); pthread_mutex_unlock(&(curs->conn->lock));