virtual void* run()
{
bool ret;
acl::redis_client* conn;
acl::redis_key redis;
for (int i = 0; i < n_; i++)
{
conn = (acl::redis_client*) pool_.peek();
if (conn == NULL)
{
printf("peek redis_client failed\r\n");
break;
}
redis.set_client(conn);
if (cmd_ == "del")
ret = test_del(redis, i);
else if (cmd_ == "expire")
ret = test_expire(redis, i);
else if (cmd_ == "ttl")
ret = test_ttl(redis, i);
else if (cmd_ == "exists")
ret = test_exists(redis, i);
else if (cmd_ == "type")
ret = test_type(redis, i);
else if (cmd_ == "all")
{
if (test_expire(redis, i) == false
|| test_ttl(redis, i) == false
|| test_exists(redis, i) == false
|| test_type(redis, i) == false
|| test_del(redis, i) == false)
{
ret = false;
}
else
ret = true;
}
else
{
printf("unknown cmd: %s\r\n", cmd_.c_str());
break;
}
pool_.put(conn, ret);
if (ret == false)
break;
}
return NULL;
}
int
main(int argc, char *argv[])
{
int raw_sock, tcp_sock, udp_sock;
if (geteuid() != 0)
errx(-1, "FAIL: root privilege required");
raw_sock = socket(PF_INET, SOCK_RAW, 0);
if (raw_sock == -1)
err(-1, "FAIL: socket(PF_INET, SOCK_RAW)");
tcp_sock = socket(PF_INET, SOCK_STREAM, 0);
if (raw_sock == -1)
err(-1, "FAIL: socket(PF_INET, SOCK_STREAM)");
udp_sock = socket(PF_INET, SOCK_DGRAM, 0);
if (raw_sock == -1)
err(-1, "FAIL: socket(PF_INET, SOCK_DGRAM)");
test_ttl(raw_sock, tcp_sock, udp_sock);
test_loop(raw_sock, tcp_sock, udp_sock);
test_if(raw_sock, tcp_sock, udp_sock);
test_addr(raw_sock, tcp_sock, udp_sock);
close(udp_sock);
close(tcp_sock);
close(raw_sock);
test_udp();
return (0);
}
virtual void* run()
{
bool ret;
acl::redis cmd;
cmd.set_cluster(&cluster_, max_conns_);
for (int i = 0; i < n_; i++)
{
if (cmd_ == "set")
ret = test_set(cmd, i);
else if (cmd_ == "get")
ret = test_get(cmd, i);
else if (cmd_ == "del")
ret = test_del(cmd, i);
else if (cmd_ == "expire")
ret = test_expire(cmd, i);
else if (cmd_ == "ttl")
ret = test_ttl(cmd, i);
else if (cmd_ == "exists")
ret = test_exists(cmd, i);
else if (cmd_ == "type")
ret = test_type(cmd, i);
else if (cmd_ == "all")
{
if (test_set(cmd, i) == false
|| test_get(cmd, i) == false
|| test_exists(cmd, i) == false
|| test_type(cmd, i) == false
|| test_expire(cmd, i) == false
|| test_ttl(cmd, i) == false
|| test_del(cmd, i) == false)
{
ret = false;
}
else
ret = true;
}
else
{
printf("unknown cmd: %s\r\n", cmd_.c_str());
break;
}
if (ret == false)
{
printf("cmd: %s error, tid: %lu\r\n",
cmd_.c_str(), thread_self());
break;
}
if (i > 0 && i % 1000 == 0)
{
char tmp[128];
acl::safe_snprintf(tmp, sizeof(tmp), "%d", i);
acl::meter_time(__FILE__, __LINE__, tmp);
}
}
locker_.lock();
__threads_exit++;
locker_.unlock();
return NULL;
}
virtual void* run()
{
bool ret;
acl::redis_client_pool* pool;
acl::redis_client* conn;
acl::redis_key option;
for (int i = 0; i < n_; i++)
{
pool = (acl::redis_client_pool*) manager_.peek();
if (pool == NULL)
{
printf("peek connection pool failed\r\n");
break;
}
conn = (acl::redis_client*) pool->peek();
if (conn == NULL)
{
printf("peek redis_client failed\r\n");
break;
}
option.set_client(conn);
if (cmd_ == "del")
ret = test_del(option, i);
else if (cmd_ == "expire")
ret = test_expire(option, i);
else if (cmd_ == "ttl")
ret = test_ttl(option, i);
else if (cmd_ == "exists")
ret = test_exists(option, i);
else if (cmd_ == "type")
ret = test_type(option, i);
else if (cmd_ == "all")
{
if (test_expire(option, i) == false
|| test_ttl(option, i) == false
|| test_exists(option, i) == false
|| test_type(option, i) == false
|| test_del(option, i) == false)
{
ret = false;
}
else
ret = true;
}
else
{
printf("unknown cmd: %s\r\n", cmd_.c_str());
break;
}
pool->put(conn, ret);
if (ret == false)
break;
}
return NULL;
}
int main(int argc, char* argv[])
{
int ch, n = 1, conn_timeout = 10, rw_timeout = 10;
acl::string addr("127.0.0.1:6379"), cmd;
while ((ch = getopt(argc, argv, "hs:n:C:T:a:")) > 0)
{
switch (ch)
{
case 'h':
usage(argv[0]);
return 0;
case 's':
addr = optarg;
break;
case 'n':
n = atoi(optarg);
break;
case 'C':
conn_timeout = atoi(optarg);
break;
case 'T':
rw_timeout = atoi(optarg);
break;
case 'a':
cmd = optarg;
break;
default:
break;
}
}
acl::acl_cpp_init();
acl::redis_client client(addr.c_str(), conn_timeout, rw_timeout);
acl::redis_key redis(&client);
bool ret;
if (cmd == "del")
ret = test_del(redis, n);
else if (cmd == "expire")
ret = test_expire(redis, n);
else if (cmd == "ttl")
ret = test_ttl(redis, n);
else if (cmd == "exists")
ret = test_exists(redis, n);
else if (cmd == "type")
ret = test_type(redis, n);
else if (cmd == "all")
{
ret = test_expire(redis, n)
&& test_ttl(redis, n)
&& test_exists(redis, n)
&& test_type(redis, n)
&& test_del(redis, n);
}
else
{
ret = false;
printf("unknown cmd: %s\r\n", cmd.c_str());
}
if (ret == true)
printf("test OK!\r\n");
else
printf("test failed!\r\n");
#ifdef WIN32
printf("enter any key to exit\r\n");
getchar();
#endif
return 0;
}
