1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
|
#include "swarmkv/swarmkv.h"
#include <gtest/gtest.h>
#include <string.h>
#include <pthread.h>
#include <asm/errno.h>
#include <sys/syscall.h>
#include <unistd.h>
#include <uuid/uuid.h>
#include <time.h>
#define FOREVER for(;;)
static void help()
{
fprintf(stderr, "Welcome to Swarmkv Node\n");
fprintf(stderr,
"swarmkv_simple_node <-n| -h | -p |-t> arg\n"
"Usage:\n"
" -n <Cluster Name>\n"
" -h <Ip:port>\n"
" -c <Ip:port>\n"
" -t <Thread Number>\n"
" -k <Key Number\n");
exit(1);
}
int main(int argc, char ** argv)
{
int ret=0;
long long i=0;
char host[32]={0};
char consul_host[32]={0};
long long key_number=0;
int worker_thread_number=1;
unsigned int consul_port=8500;
unsigned int cluster_port;
char cluster_name[256];
if(argc < 3) { help();}
strcpy(consul_host, "127.0.0.1");
strncpy(cluster_name, "swarmkv-simple-node", sizeof(cluster_name));
for (i = 1; i < argc; i++)
{
int lastarg = i==argc-1;
if (!strcmp(argv[i], "-n") && !lastarg)
{
strncpy(cluster_name, argv[++i], sizeof(cluster_name));
}
else if (!strcmp(argv[i], "-h") && !lastarg)
{
sscanf(argv[++i], "%[^:]:%u", host, &cluster_port);
}
else if(!strcmp(argv[i], "-c") && !lastarg)
{
sscanf(argv[++i], "%[^:]:%u", consul_host, &consul_port);
}
else if(!strcmp(argv[i], "-t") && !lastarg)
{
sscanf(argv[++i], "%u", &worker_thread_number);
}
else if(!strcmp(argv[i], "-k") && !lastarg)
{
sscanf(argv[++i], "%lld", &key_number);
}
else
{
help();
}
}
char *err=NULL;
struct swarmkv_options *opts=swarmkv_options_new();
swarmkv_options_set_cluster_port(opts, cluster_port);
swarmkv_options_set_bind_address(opts, host);
swarmkv_options_set_consul_host(opts, consul_host);
swarmkv_options_set_consul_port(opts, consul_port);
swarmkv_options_set_worker_thread_number(opts, worker_thread_number);
swarmkv_options_set_cluster_timeout_us(opts, 500000);
swarmkv_options_set_sync_interval_us(opts, 500);
//swarmkv_options_set_disable_run_for_leader(opts);
struct swarmkv *db=swarmkv_open(opts, cluster_name, &err);
if(err)
{
printf("swarmkv_open instance failed: %s\n", err);
free(err);
err=NULL;
return -1;
}
struct timeval start, end;
long long eplapsed_second=0;
gettimeofday(&start, NULL);
struct swarmkv_reply *reply=NULL;
if(key_number>0)
{
printf("Adding %lld keys:", key_number);
fflush(stdout);
for(i=0; i<key_number; i++)
{
reply=swarmkv_command_on(db, NULL, "TCFG tb-%d 100000 10000", i, i);
swarmkv_reply_free(reply);
if(i%(key_number/10)==0)
{
printf(" > %lld%%", i*100/key_number);
fflush(stdout);
}
}
gettimeofday(&end, NULL);
eplapsed_second=end.tv_sec-start.tv_sec;
if(eplapsed_second==0){ eplapsed_second=1;}
printf("> 100%%\nUse %lld seconds, %lld cmd/s.\n", (long long) end.tv_sec-start.tv_sec, key_number/eplapsed_second);
}
unsigned int round=0;
FOREVER
{
reply=swarmkv_command(db, "tconsume tb-%lld 1000", round%key_number);
swarmkv_reply_free(reply);
round++;
};
swarmkv_close(db);
return ret;
}
|
