Refactor

src/hydra-queue-runner/db.hh

@@ -0,0 +1,39 @@
+#pragma once
+
+#include <pqxx/pqxx>
+
+#include "util.hh"
+
+using namespace nix;
+
+
+struct Connection : pqxx::connection
+{
+    Connection() : pqxx::connection(getFlags()) { };
+
+    string getFlags()
+    {
+        string s = getEnv("HYDRA_DBI", "dbi:Pg:dbname=hydra;");
+        string prefix = "dbi:Pg:";
+        if (string(s, 0, prefix.size()) != prefix)
+            throw Error("$HYDRA_DBI does not denote a PostgreSQL database");
+        return concatStringsSep(" ", tokenizeString<Strings>(string(s, prefix.size()), ";"));
+    }
+};
+
+
+struct receiver : public pqxx::notification_receiver
+{
+    bool status = false;
+    receiver(pqxx::connection_base & c, const std::string & channel)
+        : pqxx::notification_receiver(c, channel) { }
+    void operator() (const string & payload, int pid) override
+    {
+        status = true;
+    };
+    bool get() {
+        bool b = status;
+        status = false;
+        return b;
+    }
+};