test/acl
1
0
Fork 0
mirror of https://gitee.com/acl-dev/acl.git synced 2024-11-30 02:47:56 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

optimize redis_pipeline

Browse Source
This commit is contained in:
zhengshuxin 2021-01-02 20:48:36 +08:00
parent 91a9d26c69
commit 6554585b29
3 changed files with 22 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
lib_acl_cpp/include/acl_cpp/redis/redis_client_pipeline.hpp
View File

@ -107,6 +107,7 @@ public:
const redis_result* run(redis_pipeline_message& msg);
public:
redis_client_pipeline& set_password(const char* passwd);
redis_client_pipeline& set_timeout(int conn_timeout, int rw_timeout);
redis_client_pipeline& set_retry(bool on);
redis_client_pipeline& set_channels(size_t n);

20
lib_acl_cpp/src/redis/redis_client_pipeline.cpp
View File

@ -33,13 +33,13 @@ redis_pipeline_channel & redis_pipeline_channel::set_passwd(const char *passwd)
bool redis_pipeline_channel::start_thread(void)
{
if (!passwd_.empty()) {
conn_->set_password(passwd_);
}
if (!((connect_client*) conn_)->open()) {
logger_error("open %s error %s", addr_.c_str(), last_serror());
return false;
}
if (!passwd_.empty()) {
conn_->set_password(passwd_);
}
this->start();
return true;
}
@ -173,6 +173,14 @@ redis_client_pipeline& redis_client_pipeline::set_channels(size_t n)
return *this;
}
redis_client_pipeline& redis_client_pipeline::set_password(const char* passwd)
{
if (passwd && *passwd) {
passwd_ = passwd;
}
return *this;
}
redis_client_pipeline & redis_client_pipeline::set_max_slot(size_t max_slot) {
max_slot_ = max_slot;
return *this;
@ -324,6 +332,9 @@ void redis_client_pipeline::start_channels(void) {
cit != addrs_.end(); ++cit) {
redis_pipeline_channel* channel = NEW redis_pipeline_channel(
*cit, conn_timeout_, rw_timeout_, retry_);
if (!passwd_.empty()) {
channel->set_passwd(passwd_);
}
if (channel->start_thread()) {
channels_->insert(*cit, channel);
} else {
@ -337,6 +348,9 @@ void redis_client_pipeline::start_channels(void) {
redis_pipeline_channel* channel = NEW redis_pipeline_channel(
addr_, conn_timeout_, rw_timeout_, retry_);
if (!passwd_.empty()) {
channel->set_passwd(passwd_);
}
if (channel->start_thread()) {
channels_->insert(addr_, channel);
} else {

5
lib_fiber/samples/redis_pipeline/redis_pipeline.cpp
View File

@ -191,7 +191,7 @@ protected:
break;
}
if (i > 0 && i % 1000 == 0) {
if (i > 0 && i % 5000 == 0) {
char tmp[128];
acl::safe_snprintf(tmp, sizeof(tmp), "%d", i);
acl::meter_time(__FILE__, __LINE__, tmp);
@ -319,6 +319,9 @@ int main(int argc, char* argv[])
acl::redis_client_pipeline pipeline(addrs);
pipeline.set_timeout(conn_timeout, rw_timeout)
.set_channels((size_t) pipe_channels);
if (!passwd.empty()) {
pipeline.set_password(passwd);
}
pipeline.start();
struct timeval begin;