auth: Allocate new statements after reconnecting to PostgreSQL

(cherry picked from commit db737018dab727d42873da36dd0b3d1cba3286f1)

diff --git a/modules/gpgsqlbackend/gpgsqlbackend.cc b/modules/gpgsqlbackend/gpgsqlbackend.cc
index 473b5079ede9683d5ded5456bc5e0f560fde2d79..b708b12e2c5d4973ecd5b21e10296c00b2700096 100644 (file)
--- a/modules/gpgsqlbackend/gpgsqlbackend.cc
+++ b/modules/gpgsqlbackend/gpgsqlbackend.cc
@@ -60,6 +60,8 @@ void gPgSQLBackend::reconnect()
 
   if (d_db) {
     d_db->reconnect();
+
+    allocateStatements();
   }
 }
 

diff --git a/pdns/backends/gsql/gsqlbackend.hh b/pdns/backends/gsql/gsqlbackend.hh
index 7a6fcff9091655e080fe1b19abc27193c7e37cbd..9125f863e38610e1d815fe5946b106b290b546fb 100644 (file)
--- a/pdns/backends/gsql/gsqlbackend.hh
+++ b/pdns/backends/gsql/gsqlbackend.hh
@@ -52,6 +52,13 @@ public:
     d_db=db;
     if (d_db) {
       d_db->setLog(::arg().mustDo("query-logging"));
+      allocateStatements();
+    }
+  }
+
+  void allocateStatements()
+  {
+    if (d_db) {
       d_NoIdQuery_stmt = d_db->prepare(d_NoIdQuery, 2);
       d_IdQuery_stmt = d_db->prepare(d_IdQuery, 3);
       d_ANYNoIdQuery_stmt = d_db->prepare(d_ANYNoIdQuery, 1);