测试用例编写

test/testcase.c

@@ -163,7 +163,7 @@ void do_batch_test(int fd, int op, const char *key, const char *value){
 	char bkey[15]={0}, bval[15]={0};
 
     // 组 batch（最多 64 条）
-	for(int i = 0;i < 48; ++ i){
+	for(int i = 0;i < 100; ++ i){
 		if(value == NULL){
 			int klen = sprintf(bkey, "%s%d", key, i);
 	    	kvs_batch_add(&batch, op, bkey, klen, NULL, 0);
@@ -190,6 +190,64 @@ void do_batch_test(int fd, int op, const char *key, const char *value){
 	}
 }
 
+void batch_qps(int connfd) {
+    const int N = 1000000;
+    const int B = 100;                 // do_batch_test() 里写死 50
+    const char *valA = "va";
+    const char *valB = "vb";
+    const char *valC = "vc";
+
+    struct timeval tv_begin, tv_end;
+    gettimeofday(&tv_begin, NULL);
+
+    // ---------------- Phase 1: ADD 两条 DEL 一条 (100w) ----------------
+    // 每轮：ADD/SET A_i, ADD/SET B_i, DEL A_i
+    // 用 batch：每批处理 idx..idx+49
+    for (int base = 0; base < N; base += B) {
+        // prefix 必须短，避免 do_batch_test 里 bkey[15] 溢出
+        // do_batch_test 会生成: prefix + i (0..49)
+        // 我们把 base 编进 prefix，保证全局 key 唯一
+        char preA[16], preB[16];
+
+        // 例：A123450_0..A123450_49（base=123450）
+        // 注意：这里 prefix 长度要尽量 <= 10 左右
+        snprintf(preA, sizeof(preA), "A%d_", base/100);
+        snprintf(preB, sizeof(preB), "B%d_", base/100);
+
+        do_batch_test(connfd, KVS_CMD_RSET, preA, valA); // 50次 RSET A
+        do_batch_test(connfd, KVS_CMD_RSET, preB, valB); // 50次 RSET B
+        do_batch_test(connfd, KVS_CMD_RDEL, preA, NULL); // 50次 RDEL A
+
+        if (base % 10000 == 0) printf("P1 base:%d\n", base);
+    }
+    printf("phase 1 end\n");
+
+    // ---------------- Phase 2: ADD 一条 DEL 两条 (100w) ----------------
+    // 每轮：ADD/SET C_i, DEL B_i, DEL C_i
+    for (int base = 0; base < N; base += B) {
+        char preB[16], preC[16];
+        snprintf(preB, sizeof(preB), "B%d_", base/100);
+        snprintf(preC, sizeof(preC), "C%d_", base/100);
+
+        do_batch_test(connfd, KVS_CMD_RSET, preC, valC); // 50次 RSET C
+        do_batch_test(connfd, KVS_CMD_RDEL, preB, NULL); // 50次 RDEL B
+        do_batch_test(connfd, KVS_CMD_RDEL, preC, NULL); // 50次 RDEL C
+
+        if (base % 10000 == 0) printf("P2 base:%d\n", base);
+    }
+    printf("phase 2 end\n");
+
+    gettimeofday(&tv_end, NULL);
+    int time_used = TIME_SUB_MS(tv_end, tv_begin);
+
+    // 真实总 ops 还是 6*N （每轮 6 个操作）
+    long long ops = (long long)N * 6;
+    long long qps = (time_used > 0) ? (ops * 1000 / time_used) : 0;
+
+    printf("BATCH(do_batch_test) ADD2-DEL1 then ADD1-DEL2 (N=%d) --> time_used=%d ms, ops=%lld, qps=%lld\n",
+           N, time_used, ops, qps);
+}
+
 void save(int connfd){
 	testcase(connfd, KVS_CMD_SAVE, NULL, 0, NULL, 0, KVS_STATUS_OK, NULL, 0, "SAVE");
 }
@@ -303,6 +361,8 @@ int main(int argc, char *argv[]) {
 		do_batch_test(connfd, KVS_CMD_HEXIST, "hash_set", NULL);
 	}else if(mode == -1){
 		save(connfd);
+	}else if(mode == 4){
+		batch_qps(connfd);
 	}
 	return 0;
 }