diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/database/database.cpp | 33 | ||||
| -rw-r--r-- | src/database/database.hpp | 6 | ||||
| -rw-r--r-- | src/main.cpp | 19 |
3 files changed, 27 insertions, 31 deletions
diff --git a/src/database/database.cpp b/src/database/database.cpp index 0c7f425..61e1b47 100644 --- a/src/database/database.cpp +++ b/src/database/database.cpp @@ -2,8 +2,6 @@ #ifdef USE_DATABASE #include <database/database.hpp> -#include <config/config.hpp> -#include <utils/xdg.hpp> #include <logger/logger.hpp> #include <string> @@ -11,37 +9,36 @@ using namespace std::string_literals; std::unique_ptr<db::BibouDB> Database::db; -db::BibouDB& Database::get_db() +void Database::open(const std::string& filename, const std::string& db_type) { - if (!Database::db) + try { - const std::string db_filename = Config::get("db_name", - xdg_data_path("biboumi.sqlite")); - Database::db = std::make_unique<db::BibouDB>("sqlite3", - "database="s + db_filename); + auto new_db = std::make_unique<db::BibouDB>(db_type, + "database="s + filename); + if (new_db->needsUpgrade()) + new_db->upgrade(); + Database::db.reset(new_db.release()); + } catch (const litesql::DatabaseError& e) { + log_error("Failed to open database ", filename, ". ", e.what()); + throw; } - - if (Database::db->needsUpgrade()) - Database::db->upgrade(); - - return *Database::db.get(); } void Database::set_verbose(const bool val) { - Database::get_db().verbose = val; + Database::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(), + auto options = litesql::select<db::IrcServerOptions>(*Database::db, db::IrcServerOptions::Owner == owner && db::IrcServerOptions::Server == server).one(); return options; } catch (const litesql::NotFound& e) { - db::IrcServerOptions options(Database::get_db()); + db::IrcServerOptions options(*Database::db); options.owner = owner; options.server = server; // options.update(); @@ -54,13 +51,13 @@ db::IrcChannelOptions Database::get_irc_channel_options(const std::string& owner const std::string& channel) { try { - auto options = litesql::select<db::IrcChannelOptions>(Database::get_db(), + auto options = litesql::select<db::IrcChannelOptions>(*Database::db, db::IrcChannelOptions::Owner == owner && db::IrcChannelOptions::Server == server && db::IrcChannelOptions::Channel == channel).one(); return options; } catch (const litesql::NotFound& e) { - db::IrcChannelOptions options(Database::get_db()); + db::IrcChannelOptions options(*Database::db); options.owner = owner; options.server = server; options.channel = channel; diff --git a/src/database/database.hpp b/src/database/database.hpp index 0131669..7173bcd 100644 --- a/src/database/database.hpp +++ b/src/database/database.hpp @@ -26,7 +26,7 @@ public: template<typename PersistentType> static size_t count() { - return litesql::select<PersistentType>(Database::get_db()).count(); + return litesql::select<PersistentType>(*Database::db).count(); } /** * Return the object from the db. Create it beforehand (with all default @@ -42,11 +42,11 @@ public: const std::string& channel); static void close(); + static void open(const std::string& filename, const std::string& db_type="sqlite3"); + private: static std::unique_ptr<db::BibouDB> db; - - static db::BibouDB& get_db(); }; #endif /* USE_DATABASE */ diff --git a/src/main.cpp b/src/main.cpp index ed05d36..422219a 100644 --- a/src/main.cpp +++ b/src/main.cpp @@ -3,22 +3,15 @@ #include <network/poller.hpp> #include <config/config.hpp> #include <logger/logger.hpp> -#include <utils/reload.hpp> #include <utils/xdg.hpp> - -#include <iostream> -#include <memory> -#include <atomic> - -#include <signal.h> +#include <utils/reload.hpp> #ifdef CARES_FOUND # include <network/dns_handler.hpp> #endif -#ifdef SYSTEMD_FOUND -# include <systemd/sd-daemon.h> -#endif +#include <atomic> +#include <signal.h> // A flag set by the SIGINT signal handler. static volatile std::atomic<bool> stop(false); @@ -71,6 +64,12 @@ int main(int ac, char** av) if (hostname.empty()) return config_help("hostname"); + try { + open_database(); + } catch (...) { + return 1; + } + // Block the signals we want to manage. They will be unblocked only during // the epoll_pwait or ppoll calls. This avoids some race conditions, // explained in man 2 pselect on linux |