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; }