PluginSQL.hpp

Go to the documentation of this file.

#pragma once
 
#include <iostream>
#include <chrono>
#include "Connection.hpp"
#include "SchemeSQL.hpp"
 
#define LOG std::cout << "PluginSQL: "
#define ERR std::cerr << "PluginSQL: "
 
namespace PluginSQL {
    template<typename cell_t>
    class PluginSQL {
        std::string host;
        int port;
        std::string user;
        std::string password;
        std::string database;
        
        Scheme scheme;
        Connection conn;
 
    public:
 
        PluginSQL(nlohmann::json const &config, bool dropTables = false) :
                host(config["sql"]["host"]),
                port(config["sql"]["port"]),
                user(config["sql"]["user"]),
                password(config["sql"]["password"]),
                database(config["sql"]["database"]),
                scheme(config["id"].get<std::string>()),
                conn(user, password, host, port, database){
            CreateTables(dropTables);
        }
 
        template<typename world_t>
        void endUpdate(world_t *w) {
            conn.Execute("BEGIN;");
            conn.Execute("SET LOCAL synchronous_commit = OFF;");
            pushCellsCopy(w);
            pushWorldQuery(w);
            conn.Execute("COMMIT;");
        }
 
    protected:
 
        template<typename world_t>
        void pushCellsCopy(world_t *w) {
            if (w->cells.size() > 0) {
                conn.CopyBinary(scheme.CellTable, scheme.CellTableDataLayout, [&]() {
                        conn.headerCopy(w->cells.size(),Scheme::CellRowSize);
                        for (auto* c : w->cells) {
                            Scheme::CellValueBinary(conn,c, w);
                        }
                        conn.footerCopy();
                    }
                );
            }
        }
 
        template<typename world_t>
        void pushWorldCopy(world_t *w) {
            conn.CopyBinary(scheme.WorldTable, scheme.WorldTableDataLayout, [&]() {
                    conn.headerCopy(1,Scheme::WorldRowSize);
                    Scheme::WorldValueBinary(conn, w);
                    conn.footerCopy();
                }
            );
        }
 
        template<typename world_t>
        void pushCellsQuery(world_t *w) {
            if (w->newCells.size() > 0) {
                std::stringstream cellInsertStatement;
                cellInsertStatement << "INSERT INTO " << scheme.CellTable << " " << scheme.CellTableDataLayout
                                    << " VALUES ";
                for (size_t i = 0; i < std::max(int(w->newCells.size() - 1),0); i++) {
                    scheme.CellValueStatement(cellInsertStatement,w->newCells[i], w);
                    cellInsertStatement << ", ";
                }
                if (w->newCells.size() > 0)
                    scheme.CellValueStatement(cellInsertStatement,w->newCells[w->newCells.size() - 1], w);
                cellInsertStatement << ";";
                conn.Execute(cellInsertStatement.str());
            }
        }
 
        template<typename world_t>
        void pushWorldQuery(world_t *w) {
            std::stringstream worldInsertStatement;
            worldInsertStatement << "INSERT INTO " << scheme.WorldTable << " " << scheme.WorldTableDataLayout
                                 << " VALUES " << scheme.WorldValueStatement(w) << ";";
            conn.Execute(worldInsertStatement.str());
        }
 
        void CreateTables(bool dropTablesIfExist) {
            if (dropTablesIfExist) {
                std::string dropCellTable = "DROP TABLE IF EXISTS " + scheme.CellTable + ";";
                conn.Execute(dropCellTable);
            }
            conn.Execute(scheme.CellTableCreationStatement+" CREATE INDEX ON " + scheme.CellTable + " (step);");
 
            if (dropTablesIfExist) {
                std::string dropWorldTable = "DROP TABLE IF EXISTS " + scheme.WorldTable + ";";
                conn.Execute(dropWorldTable);
            }
            conn.Execute(scheme.WorldTableCreationStatement+" CREATE INDEX ON " + scheme.WorldTable + " (step);");
        }
    };
}