From 3533145133f8b3daee36019a0e4ce13492185f80 Mon Sep 17 00:00:00 2001 From: Relintai Date: Thu, 6 Jan 2022 19:30:02 +0100 Subject: [PATCH] Added a new database based settings class. --- core/settings/db_settings.cpp | 80 +++++++++++++++++++++++++++++++++++ core/settings/db_settings.h | 27 ++++++++++++ 2 files changed, 107 insertions(+) create mode 100644 core/settings/db_settings.cpp create mode 100644 core/settings/db_settings.h diff --git a/core/settings/db_settings.cpp b/core/settings/db_settings.cpp new file mode 100644 index 0000000..be6f9de --- /dev/null +++ b/core/settings/db_settings.cpp @@ -0,0 +1,80 @@ +#include "db_settings.h" + +#include "core/database/database.h" +#include "core/database/database_manager.h" +#include "core/database/query_builder.h" +#include "core/database/query_result.h" +#include "core/database/table_builder.h" + +void DBSettings::set_value(const String &key, const Variant &value) { + const Variant &v = _data[key]; + + int id = 0; + + if (!v.is_null()) { + id = _key_map[key]; + } + + _data[key] = value; + + Ref qb = DatabaseManager::get_singleton()->ddb->get_query_builder(); + + if (id == 0) { + qb->insert(_table, "key,value")->values()->val(key)->val(value.to_string())->cvalues(); + qb->select_last_insert_id(); + Ref res = qb->run(); + + id = res->get_last_insert_rowid(); + + _key_map[key] = id; + } else { + qb->update(_table)->set()->setp("key", key)->setp("value", value.to_string())->cset()->where()->wp("id", id); + qb->run_query(); + } +} + +void DBSettings::load() { + _data.clear(); + _key_map.clear(); + + Ref qb = DatabaseManager::get_singleton()->ddb->get_query_builder(); + qb->select("id,key,value")->from(_table); + + Ref res = qb->run(); + + while (res->next_row()) { + int id = res->get_cell_int(0); + String key = res->get_cell_str(1); + String value = res->get_cell_str(2); + + _data[key] = Variant::parse_string(value); + _key_map[key] = id; + } +} + +void DBSettings::migrate() { + Ref tb = DatabaseManager::get_singleton()->ddb->get_table_builder(); + tb->drop_table_if_exists(_table); + + tb->create_table(_table); + tb->integer("id")->auto_increment()->next_row(); + tb->varchar("key", 300)->not_null()->next_row(); + tb->text("value")->not_null()->next_row(); + tb->primary_key("id"); + tb->ccreate_table(); + + tb->run_query(); +} + +void DBSettings::set_table(const String &table) { + _table = table; +} + +DBSettings::DBSettings(const bool singleton) : + Settings(singleton) { + + _table = "settings"; +} + +DBSettings::~DBSettings() { +} diff --git a/core/settings/db_settings.h b/core/settings/db_settings.h new file mode 100644 index 0000000..8291569 --- /dev/null +++ b/core/settings/db_settings.h @@ -0,0 +1,27 @@ +#ifndef DB_SETTINGS_H +#define DB_SETTINGS_H + +#include "settings.h" + +class DBSettings : public Settings { + RCPP_OBJECT(DBSettings, Settings); + +public: + virtual void set_value(const String &key, const Variant &value); + + void load(); + + void migrate(); + + void set_table(const String &table); + + DBSettings(const bool singleton = false); + virtual ~DBSettings(); + +protected: + String _table; + + std::map _key_map; +}; + +#endif \ No newline at end of file