Introduce an optional Database module

Uses litesql
This commit is contained in:
Florent Le Coz
2015-09-18 22:09:26 +02:00
parent 33fa1dcd5a
commit 88ae2599f6
8 changed files with 267 additions and 0 deletions
+54
View File
@@ -0,0 +1,54 @@
#include <database/database.hpp>
#include <config/config.hpp>
#include <utils/xdg.hpp>
#include <logger/logger.hpp>
#include <string>
using namespace std::string_literals;
std::unique_ptr<db::BibouDB> Database::db;
db::BibouDB& Database::get_db()
{
if (!Database::db)
{
const std::string db_filename = Config::get("db_name",
xdg_data_path("biboumi.sqlite"));
// log_info("Opening database: " << db_filename);
std::cout << "Opening database: " << db_filename << std::endl;
Database::db = std::make_unique<db::BibouDB>("sqlite3",
"database="s + db_filename);
}
if (Database::db->needsUpgrade())
Database::db->upgrade();
return *Database::db.get();
}
void Database::set_verbose(const bool val)
{
Database::get_db().verbose = val;
}
db::IrcServerOptions Database::get_irc_server_options(const std::string& owner,
const std::string& server)
{
try {
auto options = litesql::select<db::IrcServerOptions>(Database::get_db(),
db::IrcServerOptions::Owner == owner &&
db::IrcServerOptions::Server == server).one();
return options;
} catch (const litesql::NotFound& e) {
db::IrcServerOptions options(Database::get_db());
options.owner = owner;
options.server = server;
// options.update();
return options;
}
}
void Database::close()
{
Database::db.reset(nullptr);
}
+47
View File
@@ -0,0 +1,47 @@
#ifndef DATABASE_HPP_INCLUDED
#define DATABASE_HPP_INCLUDED
#include <biboumi.h>
#ifdef USE_DATABASE
#include "biboudb.hpp"
#include <memory>
#include <litesql.hpp>
class Database
{
public:
Database() = default;
~Database() = default;
static void set_verbose(const bool val);
template<typename PersistentType>
static size_t count()
{
return litesql::select<PersistentType>(Database::get_db()).count();
}
/**
* Return the object from the db. Create it beforehand (with all default
* values) if it is not already present.
*/
static db::IrcServerOptions get_irc_server_options(const std::string& owner,
const std::string& server);
static void close();
private:
static std::unique_ptr<db::BibouDB> db;
static db::BibouDB& get_db();
Database(const Database&) = delete;
Database(Database&&) = delete;
Database& operator=(const Database&) = delete;
Database& operator=(Database&&) = delete;
};
#endif /* USE_DATABASE */
#endif /* DATABASE_HPP_INCLUDED */