Introduce an optional Database module
Uses litesql
This commit is contained in:
@@ -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);
|
||||
}
|
||||
@@ -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 */
|
||||
Reference in New Issue
Block a user