#include "test_client.h"
#include <arpa/inet.h>


int connect_tcpserver(const char *ip, unsigned short port) {

	int connfd = socket(AF_INET, SOCK_STREAM, 0);

	struct sockaddr_in server_addr;
	memset(&server_addr, 0, sizeof(struct sockaddr_in));

	server_addr.sin_family = AF_INET;
	server_addr.sin_addr.s_addr = inet_addr(ip);
	server_addr.sin_port = htons(port);

	if (0 !=  connect(connfd, (struct sockaddr*)&server_addr, sizeof(struct sockaddr_in))) {
		perror("connect");
		return -1;
	}
	
	return connfd;
	
}


int send_msg(int connfd, char *msg, int length) {

	int res = send(connfd, msg, length, 0);
	if (res < 0) {
		perror("send");
		exit(1);
	}
	return res;
}

int recv_msg(int connfd, char *msg, int length) {

	int res = recv(connfd, msg, length, 0);
	if (res < 0) {
		perror("recv");
		exit(1);
	}
	return res;

}

void testcase(int connfd, uint8_t op, const char* key, const char* value, rsp_ret_status_e st, const char* rsp_value, const char* command_name){
	uint8_t buf[CMD_SIZE];
	uint8_t result[CMD_SIZE];
	kvs_response_t rsp;
	int len, recv_len;

	len = getcmd(op, key, value, buf);
	send_msg(connfd, buf, len);
	recv_len = recv_msg(connfd, result, CMD_SIZE);
	if (parse_response(result, recv_len, &rsp) > 0) {
		PRESP(command_name, &rsp);
		if(!verify_response(&rsp, op, st, rsp_value)) printf("%s\n", command_name);
	}else{
		printf("parser error\n");
	}

	return ;
}


void array_testcase_1w(int connfd) {

	int count = 1000;
	int i = 0;

	struct timeval tv_begin;
	gettimeofday(&tv_begin, NULL);

	for (i = 0;i < count;i ++) {
		testcase(connfd, KVS_CMD_SET, "name", "lian", KVS_STATUS_OK, NULL, "SET NAME");
		testcase(connfd, KVS_CMD_GET, "name", NULL, KVS_STATUS_OK, "lian", "GET NAME");
		testcase(connfd, KVS_CMD_MOD, "name", "liu", KVS_STATUS_OK, NULL, "MOD NAME");
		testcase(connfd, KVS_CMD_GET, "name", NULL, KVS_STATUS_OK, "liu", "GET NAME");
		testcase(connfd, KVS_CMD_EXIST, "name", NULL, KVS_STATUS_EXIST, NULL, "EXIST NAME");
		testcase(connfd, KVS_CMD_DEL, "name", NULL, KVS_STATUS_OK, NULL, "DEL NAME");
		testcase(connfd, KVS_CMD_EXIST, "name", NULL, KVS_STATUS_NO_EXIST, NULL, "NOT EXIST NAME");
	
		testcase(connfd, KVS_CMD_MOD, "stu", "liu", KVS_STATUS_NO_EXIST, NULL, "MOD NAME");
		testcase(connfd, KVS_CMD_DEL, "stu", NULL, KVS_STATUS_NO_EXIST, NULL, "DEL SUT");
	}

	struct timeval tv_end;
	gettimeofday(&tv_end, NULL);

	int time_used = TIME_SUB_MS(tv_end, tv_begin); // ms

	printf("array testcase --> time_used: %d, qps: %d\n", time_used, 9000 * 1000 / time_used);

}

void rbtree_testcase_1w(int connfd) {

	int count = 1000;
	int i = 0;

	struct timeval tv_begin;
	gettimeofday(&tv_begin, NULL);

	for (i = 0;i < count;i ++) {
		testcase(connfd, KVS_CMD_RSET, "name", "lian", KVS_STATUS_OK, NULL, "SET NAME");
		testcase(connfd, KVS_CMD_RGET, "name", NULL, KVS_STATUS_OK, "lian", "GET NAME");
		testcase(connfd, KVS_CMD_RMOD, "name", "liu", KVS_STATUS_OK, NULL, "MOD NAME");
		testcase(connfd, KVS_CMD_RGET, "name", NULL, KVS_STATUS_OK, "liu", "GET NAME");
		testcase(connfd, KVS_CMD_REXIST, "name", NULL, KVS_STATUS_EXIST, NULL, "EXIST NAME");
		testcase(connfd, KVS_CMD_RDEL, "name", NULL, KVS_STATUS_OK, NULL, "DEL NAME");
		testcase(connfd, KVS_CMD_REXIST, "name", NULL, KVS_STATUS_NO_EXIST, NULL, "NOT EXIST NAME");
	
		testcase(connfd, KVS_CMD_RMOD, "stu", "liu", KVS_STATUS_NO_EXIST, NULL, "MOD NAME");
		testcase(connfd, KVS_CMD_RDEL, "stu", NULL, KVS_STATUS_NO_EXIST, NULL, "DEL SUT");
	}

	struct timeval tv_end;
	gettimeofday(&tv_end, NULL);

	int time_used = TIME_SUB_MS(tv_end, tv_begin); // ms

	printf("array testcase --> time_used: %d, qps: %d\n", time_used, 9000 * 1000 / time_used);

}

void hash_testcase_1w(int connfd) {

	int count = 1000;
	int i = 0;

	struct timeval tv_begin;
	gettimeofday(&tv_begin, NULL);

	for (i = 0;i < count;i ++) {
		testcase(connfd, KVS_CMD_HSET, "name", "lian", KVS_STATUS_OK, NULL, "SET NAME");
		testcase(connfd, KVS_CMD_HGET, "name", NULL, KVS_STATUS_OK, "lian", "GET NAME");
		testcase(connfd, KVS_CMD_HMOD, "name", "liu", KVS_STATUS_OK, NULL, "MOD NAME");
		testcase(connfd, KVS_CMD_HGET, "name", NULL, KVS_STATUS_OK, "liu", "GET NAME");
		testcase(connfd, KVS_CMD_HEXIST, "name", NULL, KVS_STATUS_EXIST, NULL, "EXIST NAME");
		testcase(connfd, KVS_CMD_HDEL, "name", NULL, KVS_STATUS_OK, NULL, "DEL NAME");
		testcase(connfd, KVS_CMD_HEXIST, "name", NULL, KVS_STATUS_NO_EXIST, NULL, "NOT EXIST NAME");
	
		testcase(connfd, KVS_CMD_HMOD, "stu", "liu", KVS_STATUS_NO_EXIST, NULL, "MOD NAME");
		testcase(connfd, KVS_CMD_HDEL, "stu", NULL, KVS_STATUS_NO_EXIST, NULL, "DEL SUT");
	}

	struct timeval tv_end;
	gettimeofday(&tv_end, NULL);

	int time_used = TIME_SUB_MS(tv_end, tv_begin); // ms

	printf("array testcase --> time_used: %d, qps: %d\n", time_used, 9000 * 1000 / time_used);

}

void do_batch_SET_example(int fd)
{
    kvs_batch_t batch;
    kvs_batch_init(&batch);

	char key[10]={0}, val[10]={0};

    // 组 batch（最多 64 条）
	for(int i = 0;i < 48; ++ i){
		int len = sprintf(key, "k%d", i);
		len = sprintf(val, "v%d", i);
	    kvs_batch_add(&batch, KVS_CMD_SET, key, val);
	}

    // 一次性发送
    kvs_batch_send(fd, &batch);

    // 一次性 recv + parse
    uint8_t recvbuf[BATCH_SIZE];
    kvs_response_t rsps[KVS_BATCH_MAX];

    int nrsp = kvs_batch_recv_parse(fd, &batch, rsps, recvbuf, sizeof(recvbuf));

    // 打印/处理
    for (int i = 0; i < nrsp; i++) {
		int len = sprintf(key, "SET%d", i);
		PRESP(key, &rsps[i]);
	}
}

void do_batch_GET_example(int fd)
{
    kvs_batch_t batch;
    kvs_batch_init(&batch);

	char key[10]={0}, val[10]={0};

    // 组 batch（最多 64 条）
	for(int i = 0;i < 48; ++ i){
		int len = sprintf(key, "k%d", i);
	    kvs_batch_add(&batch, KVS_CMD_GET, key, NULL);
	}

    // 一次性发送
    kvs_batch_send(fd, &batch);

    // 一次性 recv + parse
    uint8_t recvbuf[BATCH_SIZE];
    kvs_response_t rsps[KVS_BATCH_MAX];

    int nrsp = kvs_batch_recv_parse(fd, &batch, rsps, recvbuf, sizeof(recvbuf));

    // 打印/处理
    for (int i = 0; i < nrsp; i++) {
		int len = sprintf(key, "GET%d", i);
		PRESP(key, &rsps[i]);
	}
}

void do_batch_DEL_example(int fd)
{
    kvs_batch_t batch;
    kvs_batch_init(&batch);

	char key[10]={0}, val[10]={0};

    // 组 batch（最多 64 条）
	for(int i = 0;i < 48; ++ i){
		int len = sprintf(key, "k%d", i);
	    kvs_batch_add(&batch, KVS_CMD_DEL, key, NULL);
	}

    // 一次性发送
    kvs_batch_send(fd, &batch);

    // 一次性 recv + parse
    uint8_t recvbuf[BATCH_SIZE];
    kvs_response_t rsps[KVS_BATCH_MAX];

    int nrsp = kvs_batch_recv_parse(fd, &batch, rsps, recvbuf, sizeof(recvbuf));

    // 打印/处理
    for (int i = 0; i < nrsp; i++) {
		int len = sprintf(key, "DEL%d", i);
		PRESP(key, &rsps[i]);
	}
}

void save(int connfd){
	testcase(connfd, KVS_CMD_SAVE, NULL, NULL, KVS_STATUS_OK, NULL, "SAVE");
}


int main(int argc, char *argv[]) {
	if (argc != 4) {
		printf("arg error\n");
		return -1;
	}

	char *ip = argv[1];
	int port = atoi(argv[2]);
	int mode = atoi(argv[3]);

	int connfd = connect_tcpserver(ip, port);

	if(mode == 0){
		do_batch_SET_example(connfd);
	}else if(mode == 1){
		do_batch_GET_example(connfd);
	}else if(mode == 2){
		do_batch_DEL_example(connfd);
	}else if(mode == 10){
		array_testcase_1w(connfd);
	}else if(mode == 11){
		rbtree_testcase_1w(connfd);
	}else if(mode == 12){
		hash_testcase_1w(connfd);
	}else if(mode == -1){
		save(connfd);
	}
	return 0;
}