King_DuckZ/dindexer
1
0
Fork 0
mirror of https://github.com/KingDuckZ/dindexer.git synced 2024-11-29 01:33:46 +00:00
Code Issues Releases Wiki Activity

Add on_connect/on_disconnect hiredis callbacks.

Browse source
This commit is contained in:
King_DuckZ 2016-06-21 17:52:13 +01:00
parent 29e2375e3b
commit 40e189cf4d
2 changed files with 35 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

36
src/backends/redis/command.cpp
View file

@ -43,17 +43,34 @@ namespace redis {
#endif
} //unnamed namespace
void on_connect (const redisAsyncContext* parContext, int parStatus) {
assert(parContext and parContext->data);
Command& self = *static_cast<Command*>(parContext->data);
self.m_connection_lost = false;
self.m_connected = (parStatus == REDIS_OK);
}
void on_disconnect (const redisAsyncContext* parContext, int parStatus) {
assert(parContext and parContext->data);
Command& self = *static_cast<Command*>(parContext->data);
assert(self.m_connected);
self.m_connection_lost = (REDIS_ERR == parStatus);
self.m_connected = false;
};
Command::Command (std::string&& parAddress, uint16_t parPort) :
m_conn(nullptr, &redisAsyncDisconnect),
m_address(std::move(parAddress)),
m_port(parPort)
m_port(parPort),
m_connected(false),
m_connection_lost(false)
{
}
Command::Command (std::string&& parSocket) :
m_conn(nullptr, &redisAsyncDisconnect),
m_address(std::move(parSocket),
m_port(0)
Command(std::move(parSocket), 0)
{
}
@ -75,12 +92,19 @@ namespace redis {
oss << "Unable to connect to Redis server at " << m_address << ':' << m_port;
throw std::runtime_error(oss.str());
}
else {
conn->data = this;
}
if (conn->err) {
std::ostringstream oss;
oss << "Unable to connect to Redis server at " << m_address << ':' << m_port <<
": " << conn->errstr;
throw std::runtime_error(oss.str());
}
if (REDIS_OK != redisAsyncSetConnectCallback(conn.get(), &on_connect))
throw std::runtime_error("Unable to set \"on_connect()\" callback");
if (REDIS_OK != redisAsyncSetDisconnectCallback(conn.get(), &on_disconnect))
throw std::runtime_error("Unable to set \"on_disconnect()\" callback");
std::swap(conn, m_conn);
}
}
@ -90,7 +114,9 @@ namespace redis {
}
bool Command::is_connected() const {
return m_conn and not m_conn->err;
const bool connected = m_conn and not m_conn->err and m_connected;
assert(not m_connection_lost or connected);
return connected;
}
auto Command::scan() -> scan_range {

4
src/backends/redis/command.hpp
View file

@ -44,6 +44,8 @@ struct redisAsyncContext;
namespace redis {
class Command {
friend void on_connect ( const redisAsyncContext*, int );
friend void on_disconnect ( const redisAsyncContext*, int );
public:
typedef ScanIterator<ScanSingleValues<std::string>> scan_iterator;
typedef boost::iterator_range<scan_iterator> scan_range;
@ -91,6 +93,8 @@ namespace redis {
#endif
std::string m_address;
uint16_t m_port;
bool m_connected;
bool m_connection_lost;
};
template <typename... Args>