wlan-cloud-ucentralsec/src/framework/StorageClass.h

//
// Created by stephane bourque on 2021-10-06.
//

#pragma once

#include "Poco/Data/SQLite/Connector.h"
#include "Poco/Data/Session.h"
#include "Poco/Data/SessionPool.h"
#include "Poco/JSON/Object.h"

#ifndef SMALL_BUILD
#include "Poco/Data/MySQL/Connector.h"
#include "Poco/Data/PostgreSQL/Connector.h"
#endif

#include "framework/MicroServiceFuncs.h"
#include "framework/SubSystemServer.h"

namespace OpenWifi {
	enum DBType { sqlite, pgsql, mysql };

	class StorageClass : public SubSystemServer {
	  public:

        inline int Start() override {
			std::lock_guard Guard(Mutex_);

			Logger().notice("Starting.");
			std::string DBType = MicroServiceConfigGetString("storage.type", "");

			if (DBType == "sqlite") {
				Setup_SQLite();
			} else if (DBType == "postgresql") {
				Setup_PostgreSQL();
			} else if (DBType == "mysql") {
				Setup_MySQL();
			}
			return 0;
		}

		inline void Stop() override { Pool_->shutdown(); }

		DBType Type() const { return dbType_; };

        StorageClass() noexcept : SubSystemServer("StorageClass", "STORAGE-SVR", "storage") {

        }

		Poco::Data::SessionPool &Pool() { return *Pool_; }

	  private:
		inline int Setup_SQLite();
		inline int Setup_MySQL();
		inline int Setup_PostgreSQL();


    protected:
		std::shared_ptr<Poco::Data::SessionPool> Pool_;
		Poco::Data::SQLite::Connector SQLiteConn_;
		Poco::Data::PostgreSQL::Connector PostgresConn_;
		Poco::Data::MySQL::Connector MySQLConn_;
		DBType dbType_ = sqlite;
	};

#ifdef SMALL_BUILD
	int Service::Setup_MySQL() {
		Daemon()->exit(Poco::Util::Application::EXIT_CONFIG);
		return 0;
	}
	int Service::Setup_PostgreSQL() {
		Daemon()->exit(Poco::Util::Application::EXIT_CONFIG);
		return 0;
	}
#else

	inline int StorageClass::Setup_SQLite() {
		Logger().notice("SQLite StorageClass enabled.");
		dbType_ = sqlite;
		auto DBName = MicroServiceDataDirectory() + "/" +
					  MicroServiceConfigGetString("storage.type.sqlite.db", "");
		int NumSessions = (int)MicroServiceConfigGetInt("storage.type.sqlite.maxsessions", 64);
		int IdleTime = (int)MicroServiceConfigGetInt("storage.type.sqlite.idletime", 60);

		Poco::Data::SQLite::Connector::registerConnector();
		//        Pool_ = std::make_unique<Poco::Data::SessionPool>(new
		//        Poco::Data::SessionPool(SQLiteConn_.name(), DBName, 8,
		//                                                                                     (int)NumSessions,
		//                                                                                     (int)IdleTime));
		Pool_ = std::make_shared<Poco::Data::SessionPool>(SQLiteConn_.name(), DBName, 8,
														  (int)NumSessions, (int)IdleTime);
		return 0;
	}

	inline int StorageClass::Setup_MySQL() {
		Logger().notice("MySQL StorageClass enabled.");
		dbType_ = mysql;
		int NumSessions = (int)MicroServiceConfigGetInt("storage.type.mysql.maxsessions", 64);
		int IdleTime = (int)MicroServiceConfigGetInt("storage.type.mysql.idletime", 60);
		auto Host = MicroServiceConfigGetString("storage.type.mysql.host", "");
		auto Username = MicroServiceConfigGetString("storage.type.mysql.username", "");
		auto Password = MicroServiceConfigGetString("storage.type.mysql.password", "");
		auto Database = MicroServiceConfigGetString("storage.type.mysql.database", "");
		auto Port = MicroServiceConfigGetString("storage.type.mysql.port", "");

		std::string ConnectionStr = "host=" + Host + ";user=" + Username + ";password=" + Password +
									";db=" + Database + ";port=" + Port +
									";compress=true;auto-reconnect=true";

		Poco::Data::MySQL::Connector::registerConnector();
		Pool_ = std::make_shared<Poco::Data::SessionPool>(MySQLConn_.name(), ConnectionStr, 8,
														  NumSessions, IdleTime);

		return 0;
	}

	inline int StorageClass::Setup_PostgreSQL() {
		Logger().notice("PostgreSQL StorageClass enabled.");
		dbType_ = pgsql;
		int NumSessions = (int)MicroServiceConfigGetInt("storage.type.postgresql.maxsessions", 64);
		int IdleTime = (int)MicroServiceConfigGetInt("storage.type.postgresql.idletime", 60);
		auto Host = MicroServiceConfigGetString("storage.type.postgresql.host", "");
		auto Username = MicroServiceConfigGetString("storage.type.postgresql.username", "");
		auto Password = MicroServiceConfigGetString("storage.type.postgresql.password", "");
		auto Database = MicroServiceConfigGetString("storage.type.postgresql.database", "");
		auto Port = MicroServiceConfigGetString("storage.type.postgresql.port", "");
		auto ConnectionTimeout =
			MicroServiceConfigGetString("storage.type.postgresql.connectiontimeout", "");

		std::string ConnectionStr = "host=" + Host + " user=" + Username + " password=" + Password +
									" dbname=" + Database + " port=" + Port +
									" connect_timeout=" + ConnectionTimeout;

		Poco::Data::PostgreSQL::Connector::registerConnector();
		Pool_ = std::make_shared<Poco::Data::SessionPool>(PostgresConn_.name(), ConnectionStr, 8,
														  NumSessions, IdleTime);

		return 0;
	}
#endif

} // namespace OpenWifi