christoph/fail
1
Fork 0
Code Releases Activity

Database: commandline option for alt. config file

Browse Source 
This change adds an optional command line argument "--database-option-file",
which can be used to override the default database configuration file ~/.my.cnf

Change-Id: I5c71523e1c31dead26f3fedb0ca7354ca99892d4
This commit is contained in:
Michael Lenz
2014-02-26 15:05:28 +01:00
parent 69ba9e0f94
commit c6344af189
1 changed files with 11 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
src/core/util/Database.cc
View File

@ -12,14 +12,21 @@ using namespace fail;
boost::mutex Database::m_global_lock; boost::mutex Database::m_global_lock;
#endif #endif
static CommandLine::option_handle DATABASE, HOSTNAME, USERNAME, DBDEFAULTS;
Database::Database(const std::string &username, const std::string &host, const std::string &database) { Database::Database(const std::string &username, const std::string &host, const std::string &database) {
#ifndef __puma #ifndef __puma
boost::lock_guard<boost::mutex> guard(m_global_lock); boost::lock_guard<boost::mutex> guard(m_global_lock);
#endif #endif
CommandLine &cmd = CommandLine::Inst();
std::string db_conf_file = "~/.my.cnf";
if (cmd[DBDEFAULTS].count()) {
db_conf_file = cmd[DBDEFAULTS].first()->arg;
}
handle = mysql_init(0); handle = mysql_init(0);
last_result = 0; last_result = 0;
mysql_options(handle, MYSQL_READ_DEFAULT_FILE, "~/.my.cnf"); mysql_options(handle, MYSQL_READ_DEFAULT_FILE, db_conf_file.c_str());
if (!mysql_real_connect(handle, host.c_str(), if (!mysql_real_connect(handle, host.c_str(),
username.c_str(), username.c_str(),
0, database.c_str(), 0, 0, 0)) { 0, database.c_str(), 0, 0, 0)) {
@ -231,8 +238,6 @@ std::string Database::escape_string(const std::string unescaped_string) {
return result; return result;
} }
static CommandLine::option_handle DATABASE, HOSTNAME, USERNAME;
void Database::cmdline_setup() { void Database::cmdline_setup() {
CommandLine &cmd = CommandLine::Inst(); CommandLine &cmd = CommandLine::Inst();
@ -241,7 +246,9 @@ void Database::cmdline_setup() {
HOSTNAME = cmd.addOption("H", "hostname", Arg::Required, HOSTNAME = cmd.addOption("H", "hostname", Arg::Required,
"-h/--hostname \tMYSQL Hostname (default: taken from ~/.my.cnf)"); "-h/--hostname \tMYSQL Hostname (default: taken from ~/.my.cnf)");
USERNAME = cmd.addOption("u", "username", Arg::Required, USERNAME = cmd.addOption("u", "username", Arg::Required,
"-u/--username \tMYSQL Username (default: taken from ~/.my.cnf, or your current user)\n"); "-u/--username \tMYSQL Username (default: taken from ~/.my.cnf, or your current user)");
DBDEFAULTS = cmd.addOption("", "database-option--file", Arg::Required,
"--database-option-file \toverride MySQL ~/.my.cnf option file (prepend with './' for files in the CWD)\n");
// should be called before any threads are spawned // should be called before any threads are spawned
mysql_library_init(0, NULL, NULL); mysql_library_init(0, NULL, NULL);