Beispiel #1
0
	virtual void* run()
	{
		bool ret;
		acl::redis_client* conn;
		acl::redis_zset redis;
		acl::redis_key key_redis;
		acl::string key;

		for (int i = 0; i < n_; i++)
		{
			// 从全局线程池中获取一个 redis 连接对象
			conn = (acl::redis_client*) pool_.peek();
			
			if (conn == NULL)
			{
				printf("peek redis_client failed\r\n");
				break;
			}

			// 每个线程一个 ID 号,做为键值组成部分
			key.format("%s_%d_%d", __keypre.c_str(), id_, i);

			redis.clear();
			// 将 redis 连接对象与 redis 命令操作类对象进行绑定关联
			redis.set_client(conn);

			if (cmd_ == "zadd")
				ret = test_zadd(redis, i, key.c_str(),
					__big_data, __big_data_length,
					__base_length);
			else if (cmd_ == "zcard")
				ret = test_zcard(redis, i, key);
			else if (cmd_ == "zrange")
				ret = test_zrange(redis, i, key, __hmac);
			else if (cmd_ == "del")
			{
				key_redis.set_client(conn);
				ret = test_del(key_redis, i, key);
			}
			else if (cmd_ != "all")
			{
				printf("unknown cmd: %s\r\n", cmd_.c_str());
				ret = false;
			}
			else if (test_zadd(redis, i, key.c_str(),
					__big_data, __big_data_length,
					__base_length) == false
				|| test_zcard(redis, i, key) == false
				|| test_zrange(redis, i, key, __hmac) == false)
			{
				ret = false;
			}
			else
				ret = true;

			// 将 redis 连接对象归还给连接池,是否保持该连接,
			// 通过判断该连接是否断开决定
			pool_.put(conn, !conn->eof());

			if (ret == false)
				break;
		}

		return NULL;
	}
Beispiel #2
0
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_zset redis(&client);

	bool ret;

	if (cmd == "zadd")
		ret = test_zadd(redis, n);
	else if (cmd == "zcard")
		ret = test_zcard(redis, n);
	else if (cmd == "zcount")
		ret = test_zcount(redis, n);
	else if (cmd == "zincrby")
		ret = test_zincrby(redis, n);
	else if (cmd == "zrange")
		ret = test_zrange(redis, n);
	else if (cmd == "zrangebyscore")
		ret = test_zrangebyscore(redis, n);
	else if (cmd == "zrank")
		ret = test_zrank(redis, n);
	else if (cmd == "zrem")
		ret = test_zrem(redis, n);
	else if (cmd == "zscore")
		ret = test_zscore(redis, n);
	else if (cmd == "zunionstore")
		ret = test_zunionstore(redis, n);
	else if (cmd == "zinterstore")
		ret = test_zinterstore(redis, n);
	else if (cmd == "zscan")
		ret = test_zscan(redis, n);
	else if (cmd == "zrangebylex")
		ret = test_zrangebylex(redis, n);
	else if (cmd == "zlexcount")
		ret = test_zlexcount(redis, n);
	else if (cmd == "zremrangebylex")
		ret = test_zremrangebylex(redis, n);
	else if (cmd == "all")
	{
		ret = test_zadd(redis, n)
			&& test_zcard(redis, n)
			&& test_zcount(redis, n)
			&& test_zincrby(redis, n)
			&& test_zrange(redis, n)
			&& test_zrangebyscore(redis, n)
			&& test_zrank(redis, n)
			&& test_zrem(redis, n)
			&& test_zscore(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;
}