From 76fc3ef79e8313e99be4d6daa844c13e34721323 Mon Sep 17 00:00:00 2001
From: Alexey Kopytov <akopytov@gmail.com>
Date: Wed, 23 Sep 2015 16:59:04 +0300
Subject: [PATCH] Fixes #10: oltp: pgsql query execution failed

Handle deadlocks correctly in the PostgreSQL driver, i.e. return
SB_DB_ERROR_DEADLOCK to the upper layer to restart the transaction and
increment the deadlocks counter.
---
 sysbench/drivers/pgsql/Makefile.am |  2 +-
 sysbench/drivers/pgsql/drv_pgsql.c | 23 +++++++++++++++++++----
 2 files changed, 20 insertions(+), 5 deletions(-)

diff --git a/sysbench/drivers/pgsql/Makefile.am b/sysbench/drivers/pgsql/Makefile.am
index 6e47744..1bc0412 100644
--- a/sysbench/drivers/pgsql/Makefile.am
+++ b/sysbench/drivers/pgsql/Makefile.am
@@ -17,6 +17,6 @@
 noinst_LIBRARIES = libsbpgsql.a
 
 libsbpgsql_a_SOURCES = drv_pgsql.c
-libsbpgsql_a_CPPFLAGS = $(PGSQL_CFLAGS) $(AM_CPPFLAGS)
+libsbpgsql_a_CPPFLAGS = -g $(PGSQL_CFLAGS) $(AM_CPPFLAGS)
 
 EXTRA_DIST = CMakeLists.txt
diff --git a/sysbench/drivers/pgsql/drv_pgsql.c b/sysbench/drivers/pgsql/drv_pgsql.c
index a1b8331..24d08f5 100644
--- a/sysbench/drivers/pgsql/drv_pgsql.c
+++ b/sysbench/drivers/pgsql/drv_pgsql.c
@@ -512,12 +512,19 @@ int pgsql_drv_execute(db_stmt_t *stmt, db_result_set_t *rs)
     status = PQresultStatus(pgres);
     if (status != PGRES_TUPLES_OK && status != PGRES_COMMAND_OK)
     {
-      log_text(LOG_FATAL, "PQexecPrepared() failed: %d %s", status,
-               PQerrorMessage(pgcon));
+      const char * const errmsg = PQerrorMessage(pgcon);
+
+      if (strstr(errmsg, "deadlock detected"))
+      {
+        PQexec(pgcon, "ROLLBACK");
+        return SB_DB_ERROR_DEADLOCK;
+      }
+
+      log_text(LOG_FATAL, "PQexecPrepared() failed: %d %s", status, errmsg);
       return SB_DB_ERROR_FAILED;
     }
     rs->ptr = (void *)pgres;
-    
+
     return SB_DB_ERROR_NONE;
   }
 
@@ -581,8 +588,16 @@ int pgsql_drv_query(db_conn_t *sb_conn, const char *query,
   status = PQresultStatus(pgres);
   if (status != PGRES_TUPLES_OK && status != PGRES_COMMAND_OK)
   {
+    const char * const errmsg = PQerrorMessage(con);
+
+    if (strstr(errmsg, "deadlock detected"))
+    {
+      PQexec(con, "ROLLBACK");
+      return SB_DB_ERROR_DEADLOCK;
+    }
+
     log_text(LOG_ALERT, "failed to execute query: %s:", query);
-    log_text(LOG_ALERT, "error: %d %s", status, PQerrorMessage(con));
+    log_text(LOG_ALERT, "error: %d %s", status, errmsg);
     return SB_DB_ERROR_FAILED;
   }