static bool test_zcard(acl::redis_zset& redis, int i, const char* key)
{
// 因为该协议数据比较小，所以在组装请求数据时不必采用分片方式
redis.get_client()->set_slice_request(false);
int ret = redis.zcard(key);
if (ret < 0)
{
printf("zcard key: %s error\r\n", key);
return false;
}
else if (i < 10)
printf("zcard ok, key: %s, count: %d\r\n", key, ret);
return true;
}
static bool test_zrange(acl::redis_zset& redis, int i, const char* key,
const char* hmac)
{
int start = 0, end = -1;
// 请求的数据量比较小，所以在组装请求协议时不必采用分片方式
redis.get_client()->set_slice_request(false);
// 对服务器返回的数据也不分片
redis.get_client()->set_slice_respond(false);
int ret = redis.zrange(key, start, end, NULL);
if (ret <= 0)
{
printf("zrange return: %d\r\n", ret);
return false;
}
// 获得数组元素结果集
const acl::redis_result* result = redis.get_result();
if (result == NULL)
{
printf("result null\r\n");
return false;
}
size_t size;
// 直接获得数组集合
const acl::redis_result** children = result->get_children(&size);
if (children == NULL || size == 0)
{
printf("no children: %s, size: %d\r\n",
children ? "no" : "yes", (int) size);
return false;
}
// 校验获得的所有数据片的 MD5 值，与传入的进行比较
acl::md5* md5;
if (hmac != NULL)
md5 = new acl::md5;
else
md5 = NULL;
const acl::redis_result* child;
size_t len, argc, n;
// 先遍历所有数组元素对象
for (size_t j = 0; j < size; j++)
{
child = children[j];
if (child == NULL)
continue;
// 因为前面设置了禁止对响应数据进行分片，所以只需取第一个元素
argc = child->get_size();
assert(argc == 1);
const char* ptr = child->get(0, &len);
if (ptr == NULL)
{
printf("first is null\r\n");
continue;
}
const char* dat = strchr(ptr, ':');
if (dat == NULL)
{
printf("invalid data, j: %d\n", (int) j);
continue;
}
dat++;
n = dat - ptr;
if (len < n)
{
printf("invalid data, j: %d\n", (int) j);
continue;
}
len -= n;
// 取出数据计算 md5 值
if (md5 != NULL)
md5->update(dat, len);
}
if (md5 != NULL)
md5->finish();
// 获得字符串方式的 MD5 值
if (md5 != NULL)
{
const char* ptr = md5->get_string();
if (strcmp(ptr, hmac) != 0)
{
printf("md5 error, hmac: %s, %s, key: %s\r\n",
hmac, ptr, key);
return false;
}
else if (i < 10)
printf("md5 ok, hmac: %s, %s, key: %s\r\n",
hmac, ptr, key);
delete md5;
}
else if (i < 10)
printf("ok, key: %s\r\n", key);
return true;
}
static bool test_zadd(acl::redis_zset& redis, int i, const char* key,
const char* big_data, size_t length, size_t base_length)
{
// 将大数据进行分割，计算出分割后的数据块个数
size_t nmember = length / base_length;
if (length % base_length != 0)
nmember++;
// 从连接对象中获得统一的内存池分配对象，分配小内存块
acl::dbuf_pool* pool = redis.get_pool();
// 动态分配数据块指针数组内存
const char** members = (const char**)
pool->dbuf_alloc(nmember * sizeof(char*));
// 动态分配数据块长度数组内存
size_t* lens = (size_t*) pool->dbuf_alloc(nmember * sizeof(size_t));
// 动态分配数据块分值数组内存
double* scores = (double*) pool->dbuf_alloc(nmember * sizeof(double));
// 将大数据切分成小数据，置入数据块数组中，使用递增的整数做为分值
size_t len;
const char* ptr = big_data;
char* buf, id[64];
int n;
for (size_t j = 0; j < nmember; j++)
{
len = length > base_length ? base_length : length;
// 在每个原始数据前面加唯一前缀，从而可以保证有序集合中对象中
// 的每个成员数据都是不同的
n = acl::safe_snprintf(id, sizeof(id),
"%lu:", (unsigned long) j);
buf = (char*) pool->dbuf_alloc(len + n);
memcpy(buf, id, n);
memcpy(buf + n, ptr, len);
members[j] = buf;
lens[j] = len + n; // 该数据块的总长度：唯一前缀+数据
scores[j] = (double) j;
// 剩余数据块长度
length -= len;
ptr += len;
}
// 要求 redis 连接对象采用内存链协议组装方式，避免内部组装请求协议时
// 再组装成大内存
redis.get_client()->set_slice_request(true);
// 开始向 redis 添加数据
int ret = redis.zadd(key, members, lens, scores, nmember);
if (ret < 0)
{
printf("add key: %s error\r\n", key);
return false;
}
else if (i < 10)
printf("add ok, key: %s, ret: %d\r\n", key, ret);
return true;
}