mysql override

src/Business/PdnsSlave.cpp

@@ -3,6 +3,7 @@
 //
 
 #include <fstream>
+#include <DataAccess/MySql.h>
 #include "DataAccess/PdnsSlaveConfig.h"
 #include "DataAccess/HostsConfig.h"
 #include "PdnsSlave.h"
@@ -63,7 +64,16 @@ Result<Actions> PdnsSlave::readHosts()
 
 BResult PdnsSlave::overridePdns()
 {
-    return true;
+    MySql mysql(_masterConfig, _slaveConfig);
+    auto res = mysql.dump();
+    if (!res)
+        return res;
+    if (!(res = mysql.insert()))
+        return res;
+    std::string sql;
+    for (auto a : _actions)
+        sql += a->getSqlQuery();
+    return mysql.override(sql);
 }
 
 BResult PdnsSlave::overrideDhcp()

src/CMakeLists.txt

@@ -30,5 +30,5 @@ set(SOURCE_FILES
   DBO/Actions/ActionAddDomain.h
   )
 set(LIBS ${LIBS} jsoncpp)
-add_executable(pdns-slave ${SOURCE_FILES})
+add_executable(pdns-slave ${SOURCE_FILES} DataAccess/MySql.cpp DataAccess/MySql.h)
 target_link_libraries(pdns-slave ${LIBS})

src/DBO/Actions/Action.cpp

@@ -24,7 +24,7 @@ void Action::setDomain(const std::string &domain)
 
 const std::string Action::getSqlQuery() const
 {
-    return "SET @domain_id=(SELECT get_domain_id(\"" + _domain + "\"));\n" + getSql();
+    return "SET @domain_id=(SELECT id FROM domains WHERE name=\"" + _domain + "\");\n" + getSql();
 }
 
 const std::string Action::getDhcpConf() const

src/DBO/Actions/ActionAddDomain.cpp

@@ -11,6 +11,6 @@ ActionAddDomain::ActionAddDomain()
 const std::string ActionAddDomain::getSql() const
 {
     return "INSERT INTO domains (name, type) VALUES (\"" + _domain + "\", \"MASTER\");\n"
-            "INSERT INTO zones (domain_id, owner, zone_templ_id) VALUES ((SELECT get_domain_id(\""
-           + _domain + "\")), (SELECT id FROM users ORDER BY id LIMIT 1), 0);";
+            "INSERT INTO zones (domain_id, owner, zone_templ_id) VALUES ((SELECT id FROM domains WHERE name=\""
+           + _domain + "\"), (SELECT id FROM users ORDER BY id LIMIT 1), 0);";
 }

src/DBO/Actions/ActionAddHost.cpp

@@ -37,7 +37,7 @@ const std::string ActionAddHost::getSql() const
         auto reverseDomain = _reverseDomain.empty() ? "in-addr.arpa" : _reverseDomain + ".in-addr.arpa";
 
         query += "INSERT INTO records (domain_id, name, type, content, ttl, prio, change_date)\n"
-                 "    VALUES((SELECT get_domain_id(\"" + reverseDomain + "\")), \""
+                 "    VALUES((SELECT id FROM domains WHERE name=\"" + reverseDomain + "\"), \""
                  + reversedValue + "." + reverseDomain + "\", \"PTR\","
                  " \"" + _host + "." + _domain + "\", 84600, 0, " + std::to_string(time(nullptr)) + ");\n";
     }

src/DataAccess/MySql.cpp

@@ -0,0 +1,87 @@
+//
+// Created by robin on 8/9/15.
+//
+
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>
+#include <fstream>
+#include "MySql.h"
+
+#define tmpTemplate "/tmp/pdns-slave.XXXXXX"
+
+MySql::MySql(const SqlConfiguration &masterConfig, const SqlConfiguration &slaveConfig)
+    : _masterConfig(masterConfig)
+    , _slaveConfig(slaveConfig)
+{
+}
+
+MySql::~MySql()
+{
+    clean();
+}
+
+BResult MySql::dump()
+{
+    BResult res;
+    _dumpFilePath = getTempFile();
+    int status = system(std::string("mysqldump --routines '-h" + _masterConfig.getHost() + "'"
+                       + " '-u" + _masterConfig.getUser() + "'"
+                       + " '-p" + _masterConfig.getPassword() + "'"
+                       + " " + _masterConfig.getDatabase()
+                       + " > " + _dumpFilePath).c_str());
+    if (status == 0)
+        return res.ok(true);
+    return res.error("mysqldump exited with code " + std::to_string(status));
+}
+
+const std::string MySql::getTempFile()
+{
+    char tmp[sizeof(tmpTemplate)];
+    strcpy(tmp, tmpTemplate);
+    mkstemp(tmp);
+    return std::string(tmp);
+}
+
+void MySql::clean()
+{
+    if (!_dumpFilePath.empty())
+    {
+        unlink(_dumpFilePath.c_str());
+        _dumpFilePath = "";
+    }
+    if (!_overrideFilePath.empty())
+    {
+        unlink(_overrideFilePath.c_str());
+        _overrideFilePath = "";
+    }
+}
+
+BResult MySql::insert()
+{
+    _dumpFilePath = getTempFile();
+    return insertSlave(_dumpFilePath);
+}
+
+BResult MySql::override(const std::string &sql)
+{
+    _overrideFilePath = getTempFile();
+    std::ofstream file(_overrideFilePath);
+    if (!file)
+        return BResult().error("Could not open temp file " + _overrideFilePath);
+    file << sql;
+    file.close();
+    return insertSlave(_overrideFilePath);
+}
+
+BResult MySql::insertSlave(const std::string &file)
+{
+    BResult res;
+    int status = system(std::string("cat " + file + " | mysql '-h" + _slaveConfig.getHost() + "'"
+                                    + " '-u" + _slaveConfig.getUser() + "'"
+                                    + " '-p" + _slaveConfig.getPassword() + "'"
+                                    + " " + _slaveConfig.getDatabase()).c_str());
+    if (status == 0)
+        return res.ok(true);
+    return res.error("mysql exited with code " + std::to_string(status));
+}

src/DataAccess/MySql.h

@@ -0,0 +1,41 @@
+//
+// Created by robin on 8/9/15.
+//
+
+#ifndef PDNS_SLAVE_MYSQL_H
+#define PDNS_SLAVE_MYSQL_H
+
+#include <DBO/Result.h>
+#include <DBO/SqlConfiguration.h>
+# include "DBO/Result.h"
+
+class MySql
+{
+public:
+    MySql(const SqlConfiguration& masterConfig, const SqlConfiguration& slaveConfig);
+    virtual ~MySql();
+
+    BResult dump();
+
+    BResult insert();
+
+    BResult override(const std::string& sql);
+
+    void clean();
+
+private:
+    const std::string getTempFile();
+
+    BResult insertSlave(const std::string& file);
+
+    SqlConfiguration _masterConfig;
+
+    SqlConfiguration _slaveConfig;
+
+    std::string _dumpFilePath;
+
+    std::string _overrideFilePath;
+};
+
+
+#endif //PDNS_SLAVE_MYSQL_H