10 #include <libmemcached/memcached.h>
12 #include "client_options.h"
13 #include "utilities.h"
15 #define PROGRAM_NAME "memstat"
16 #define PROGRAM_DESCRIPTION "Output the state of a memcached cluster."
19 static void options_parse(int argc
, char *argv
[]);
20 static void run_analyzer(memcached_st
*memc
, memcached_stat_st
*stat
,
21 memcached_server_st
*server_list
);
22 static void print_server_listing(memcached_st
*memc
, memcached_stat_st
*stat
,
23 memcached_server_st
*server_list
);
24 static void print_analysis_report(memcached_st
*memc
,
25 memcached_analysis_st
*report
,
26 memcached_server_st
*server_list
);
28 static int opt_verbose
= 0;
29 static int opt_displayflag
= 0;
30 static int opt_analyze
= 0;
31 static char *opt_servers
= NULL
;
32 static char *analyze_mode
= NULL
;
34 static struct option long_options
[]=
36 {"version", no_argument
, NULL
, OPT_VERSION
},
37 {"help", no_argument
, NULL
, OPT_HELP
},
38 {"verbose", no_argument
, &opt_verbose
, OPT_VERBOSE
},
39 {"debug", no_argument
, &opt_verbose
, OPT_DEBUG
},
40 {"servers", required_argument
, NULL
, OPT_SERVERS
},
41 {"flag", no_argument
, &opt_displayflag
, OPT_FLAG
},
42 {"analyze", optional_argument
, NULL
, OPT_ANALYZE
},
46 int main(int argc
, char *argv
[])
50 memcached_stat_st
*stat
;
51 memcached_server_st
*servers
;
52 memcached_server_st
*server_list
;
54 options_parse(argc
, argv
);
60 if ((temp
= getenv("MEMCACHED_SERVERS")))
61 opt_servers
= strdup(temp
);
64 fprintf(stderr
, "No Servers provided\n\n");
65 help_command(PROGRAM_NAME
, PROGRAM_DESCRIPTION
, long_options
, 0);
70 memc
= memcached_create(NULL
);
72 servers
= memcached_servers_parse(opt_servers
);
73 memcached_server_push(memc
, servers
);
74 memcached_server_list_free(servers
);
76 stat
= memcached_stat(memc
, NULL
, &rc
);
78 if (rc
!= MEMCACHED_SUCCESS
&& rc
!= MEMCACHED_SOME_ERRORS
)
80 printf("Failure to communicate with servers (%s)\n",
81 memcached_strerror(memc
, rc
));
85 server_list
= memcached_server_list(memc
);
88 run_analyzer(memc
, stat
, server_list
);
90 print_server_listing(memc
, stat
, server_list
);
100 static void run_analyzer(memcached_st
*memc
, memcached_stat_st
*stat
,
101 memcached_server_st
*server_list
)
105 if (analyze_mode
== NULL
)
107 memcached_analysis_st
*report
;
108 report
= memcached_analyze(memc
, stat
, &rc
);
109 if (rc
!= MEMCACHED_SUCCESS
|| report
== NULL
)
111 printf("Failure to analyze servers (%s)\n",
112 memcached_strerror(memc
, rc
));
115 print_analysis_report(memc
, report
, server_list
);
118 else if (strcmp(analyze_mode
, "latency") == 0)
120 memcached_st
**servers
;
121 uint32_t x
, y
, flags
, server_count
= memcached_server_count(memc
);
122 uint32_t num_of_tests
= 32;
123 const char *test_key
= "libmemcached_test_key";
125 servers
= malloc(sizeof(memcached_st
*) * server_count
);
128 fprintf(stderr
, "Failed to allocate memory\n");
132 for (x
= 0; x
< server_count
; x
++)
134 if((servers
[x
]= memcached_create(NULL
)) == NULL
)
136 fprintf(stderr
, "Failed to memcached_create()\n");
139 memcached_free(servers
[x
]);
143 memcached_server_add(servers
[x
],
144 memcached_server_name(memc
, server_list
[x
]),
145 memcached_server_port(memc
, server_list
[x
]));
148 printf("Network Latency Test:\n\n");
149 struct timeval start_time
, end_time
;
150 long elapsed_time
, slowest_time
= 0, slowest_server
= 0;
152 for (x
= 0; x
< server_count
; x
++)
154 gettimeofday(&start_time
, NULL
);
155 for (y
= 0; y
< num_of_tests
; y
++)
158 char *val
= memcached_get(servers
[x
], test_key
, strlen(test_key
),
160 if (rc
!= MEMCACHED_NOTFOUND
&& rc
!= MEMCACHED_SUCCESS
)
164 gettimeofday(&end_time
, NULL
);
166 elapsed_time
= timedif(end_time
, start_time
);
167 elapsed_time
/= num_of_tests
;
169 if (elapsed_time
> slowest_time
)
172 slowest_time
= elapsed_time
;
175 if (rc
!= MEMCACHED_NOTFOUND
&& rc
!= MEMCACHED_SUCCESS
)
177 printf("\t %s (%d) => failed to reach the server\n",
178 memcached_server_name(memc
, server_list
[x
]),
179 memcached_server_port(memc
, server_list
[x
]));
183 printf("\t %s (%d) => %ld.%ld seconds\n",
184 memcached_server_name(memc
, server_list
[x
]),
185 memcached_server_port(memc
, server_list
[x
]),
186 elapsed_time
/ 1000, elapsed_time
% 1000);
190 if (server_count
> 1 && slowest_time
> 0)
193 printf("Slowest Server: %s (%d) => %ld.%ld seconds\n",
194 memcached_server_name(memc
, server_list
[slowest_server
]),
195 memcached_server_port(memc
, server_list
[slowest_server
]),
196 slowest_time
/ 1000, slowest_time
% 1000);
200 for (x
= 0; x
< server_count
; x
++)
201 memcached_free(servers
[x
]);
208 fprintf(stderr
, "Invalid Analyzer Option provided\n");
213 static void print_server_listing(memcached_st
*memc
, memcached_stat_st
*stat
,
214 memcached_server_st
*server_list
)
219 printf("Listing %u Server\n\n", memcached_server_count(memc
));
220 for (x
= 0; x
< memcached_server_count(memc
); x
++)
225 list
= memcached_stat_get_keys(memc
, &stat
[x
], &rc
);
227 printf("Server: %s (%u)\n", memcached_server_name(memc
, server_list
[x
]),
228 memcached_server_port(memc
, server_list
[x
]));
229 for (ptr
= list
; *ptr
; ptr
++)
232 char *value
= memcached_stat_get_value(memc
, &stat
[x
], *ptr
, &rc
);
234 printf("\t %s: %s\n", *ptr
, value
);
243 static void print_analysis_report(memcached_st
*memc
,
244 memcached_analysis_st
*report
,
245 memcached_server_st
*server_list
)
247 uint32_t server_count
= memcached_server_count(memc
);
249 printf("Memcached Cluster Analysis Report\n\n");
251 printf("\tNumber of Servers Analyzed : %d\n", server_count
);
252 printf("\tAverage Item Size (incl/overhead) : %u bytes\n",
253 report
->average_item_size
);
255 if (server_count
== 1)
257 printf("\nFor a detailed report, you must supply multiple servers.\n");
262 printf("\tNode with most memory consumption : %s:%u (%llu bytes)\n",
263 memcached_server_name(memc
, server_list
[report
->most_consumed_server
]),
264 memcached_server_port(memc
, server_list
[report
->most_consumed_server
]),
265 (unsigned long long)report
->most_used_bytes
);
266 printf("\tNode with least free space : %s:%u (%llu bytes remaining)\n",
267 memcached_server_name(memc
, server_list
[report
->least_free_server
]),
268 memcached_server_port(memc
, server_list
[report
->least_free_server
]),
269 (unsigned long long)report
->least_remaining_bytes
);
270 printf("\tNode with longest uptime : %s:%u (%us)\n",
271 memcached_server_name(memc
, server_list
[report
->oldest_server
]),
272 memcached_server_port(memc
, server_list
[report
->oldest_server
]),
273 report
->longest_uptime
);
274 printf("\tPool-wide Hit Ratio : %1.f%%\n", report
->pool_hit_ratio
);
278 static void options_parse(int argc
, char *argv
[])
280 memcached_programs_help_st help_options
[]=
290 option_rv
= getopt_long(argc
, argv
, "Vhvds:a", long_options
, &option_index
);
291 if (option_rv
== -1) break;
296 case OPT_VERBOSE
: /* --verbose or -v */
297 opt_verbose
= OPT_VERBOSE
;
299 case OPT_DEBUG
: /* --debug or -d */
300 opt_verbose
= OPT_DEBUG
;
302 case OPT_VERSION
: /* --version or -V */
303 version_command(PROGRAM_NAME
);
305 case OPT_HELP
: /* --help or -h */
306 help_command(PROGRAM_NAME
, PROGRAM_DESCRIPTION
, long_options
, help_options
);
308 case OPT_SERVERS
: /* --servers or -s */
309 opt_servers
= strdup(optarg
);
311 case OPT_ANALYZE
: /* --analyze or -a */
312 opt_analyze
= OPT_ANALYZE
;
313 analyze_mode
= (optarg
) ? strdup(optarg
) : NULL
;
316 /* getopt_long already printed an error message. */