King_DuckZ/dindexer
1
0
Fork 0
mirror of https://github.com/KingDuckZ/dindexer.git synced 2025-07-04 14:24:10 +00:00
Code Issues Releases Wiki Activity

Add optional database selection parameter in yml file.

Browse source
This commit is contained in:
King_DuckZ 2016-06-10 09:55:37 +02:00
parent 8bdb1f237f
commit 04b667485e
4 changed files with 20 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

18
src/backends/redis/backend_redis.cpp
View file

@ -29,6 +29,7 @@ namespace dindb {
struct RedisConnectionSettings {
std::string address;
uint16_t port;
uint16_t database;
};
} //unnamed namespace
} //namespace dindb
@ -40,6 +41,7 @@ namespace YAML {
Node node;
node["address"] = parSettings.address;
node["port"] = parSettings.port;
node["database"] = parSettings.database;
return node;
}
@ -50,22 +52,33 @@ namespace YAML {
parSettings.address = parNode["address"].as<std::string>();
parSettings.port = parNode["port"].as<uint16_t>();
if (parNode["database"])
parSettings.database = parNode["database"].as<uint16_t>();
return true;
}
};
} //namespace YAML
namespace dindb {
BackendRedis::BackendRedis(std::string &&parAddress, uint16_t parPort, bool parConnect) :
m_redis(std::move(parAddress), parPort, parConnect)
BackendRedis::BackendRedis(std::string &&parAddress, uint16_t parPort, uint16_t parDatabase, bool parConnect) :
m_redis(std::move(parAddress), parPort),
m_database(parDatabase)
{
if (parConnect)
this->connect();
}
BackendRedis::~BackendRedis() noexcept {
}
void BackendRedis::connect() {
using boost::lexical_cast;
m_redis.connect();
if (m_redis.is_connected() and m_database > 0) {
const std::string command = "SELECT " + lexical_cast<std::string>(m_database);
m_redis.run(command.c_str());
}
}
void BackendRedis::disconnect() {
@ -146,6 +159,7 @@ extern "C" dindb::Backend* dindexer_create_backend (const YAML::Node* parConfig)
return new dindb::BackendRedis(
std::move(config.address),
config.port,
config.database,
true
);
}