git.m6w6.name Git - awesomized/libmemcached/blob - clients/memstat.c

   1 /* LibMemcached
   2  * Copyright (C) 2006-2009 Brian Aker
   3  * All rights reserved.
   4  *
   5  * Use and distribution licensed under the BSD license.  See
   6  * the COPYING file in the parent directory for full text.
   7  *
   8  * Summary:
   9  *
  10  * Authors:
  11  *          Brian Aker
  12  *          Toru Maesaka
  13  */
  14
  15 #include "libmemcached/common.h"
  16 #include <stdio.h>
  17 #include <sys/types.h>
  18 #include <sys/stat.h>
  19 #include <sys/types.h>
  20 #include <sys/mman.h>
  21 #include <fcntl.h>
  22 #include <string.h>
  23 #include <getopt.h>
  24 #include <sys/time.h>
  25
  26 #include <libmemcached/memcached.h>
  27
  28 #include "client_options.h"
  29 #include "utilities.h"
  30
  31 #define PROGRAM_NAME "memstat"
  32 #define PROGRAM_DESCRIPTION "Output the state of a memcached cluster."
  33
  34 /* Prototypes */
  35 static void options_parse(int argc, char *argv[]);
  36 static void run_analyzer(memcached_st *memc, memcached_stat_st *memc_stat,
  37                          memcached_server_st *server_list);
  38 static void print_server_listing(memcached_st *memc, memcached_stat_st *memc_stat,
  39                                  memcached_server_st *server_list);
  40 static void print_analysis_report(memcached_st *memc,
  41                                   memcached_analysis_st *report,
  42                                   memcached_server_st *server_list);
  43
  44 static int opt_verbose= 0;
  45 static int opt_displayflag= 0;
  46 static int opt_analyze= 0;
  47 static char *opt_servers= NULL;
  48 static char *analyze_mode= NULL;
  49
  50 static struct option long_options[]=
  51 {
  52   {(OPTIONSTRING)"version", no_argument, NULL, OPT_VERSION},
  53   {(OPTIONSTRING)"help", no_argument, NULL, OPT_HELP},
  54   {(OPTIONSTRING)"verbose", no_argument, &opt_verbose, OPT_VERBOSE},
  55   {(OPTIONSTRING)"debug", no_argument, &opt_verbose, OPT_DEBUG},
  56   {(OPTIONSTRING)"servers", required_argument, NULL, OPT_SERVERS},
  57   {(OPTIONSTRING)"flag", no_argument, &opt_displayflag, OPT_FLAG},
  58   {(OPTIONSTRING)"analyze", optional_argument, NULL, OPT_ANALYZE},
  59   {0, 0, 0, 0},
  60 };
  61
  62 int main(int argc, char *argv[])
  63 {
  64   memcached_return_t rc;
  65   memcached_st *memc;
  66   memcached_stat_st *memc_stat;
  67   memcached_server_st *servers;
  68   memcached_server_st *server_list;
  69
  70   options_parse(argc, argv);
  71
  72   if (!opt_servers)
  73   {
  74     char *temp;
  75
  76     if ((temp= getenv("MEMCACHED_SERVERS")))
  77       opt_servers= strdup(temp);
  78     else
  79     {
  80       fprintf(stderr, "No Servers provided\n\n");
  81       help_command(PROGRAM_NAME, PROGRAM_DESCRIPTION, long_options, 0);
  82       exit(1);
  83     }
  84   }
  85
  86   memc= memcached_create(NULL);
  87
  88   servers= memcached_servers_parse(opt_servers);
  89   memcached_server_push(memc, servers);
  90   memcached_server_list_free(servers);
  91
  92   memc_stat= memcached_stat(memc, NULL, &rc);
  93
  94   if (rc != MEMCACHED_SUCCESS && rc != MEMCACHED_SOME_ERRORS)
  95   {
  96     printf("Failure to communicate with servers (%s)\n",
  97            memcached_strerror(memc, rc));
  98     exit(1);
  99   }
 100
 101   server_list= memcached_server_list(memc);
 102
 103   if (opt_analyze)
 104     run_analyzer(memc, memc_stat, server_list);
 105   else
 106     print_server_listing(memc, memc_stat, server_list);
 107
 108   free(memc_stat);
 109   free(opt_servers);
 110
 111   memcached_free(memc);
 112
 113   return 0;
 114 }
 115
 116 static void run_analyzer(memcached_st *memc, memcached_stat_st *memc_stat,
 117                          memcached_server_st *server_list)
 118 {
 119   memcached_return_t rc;
 120
 121   if (analyze_mode == NULL)
 122   {
 123     memcached_analysis_st *report;
 124     report= memcached_analyze(memc, memc_stat, &rc);
 125     if (rc != MEMCACHED_SUCCESS || report == NULL)
 126     {
 127       printf("Failure to analyze servers (%s)\n",
 128              memcached_strerror(memc, rc));
 129       exit(1);
 130     }
 131     print_analysis_report(memc, report, server_list);
 132     free(report);
 133   }
 134   else if (strcmp(analyze_mode, "latency") == 0)
 135   {
 136     memcached_st **servers;
 137     uint32_t flags, server_count= memcached_server_count(memc);
 138     uint32_t num_of_tests= 32;
 139     const char *test_key= "libmemcached_test_key";
 140
 141     servers= malloc(sizeof(memcached_st*) * server_count);
 142     if (!servers)
 143     {
 144       fprintf(stderr, "Failed to allocate memory\n");
 145       return;
 146     }
 147
 148     for (uint32_t x= 0; x < server_count; x++)
 149     {
 150       if((servers[x]= memcached_create(NULL)) == NULL)
 151       {
 152         fprintf(stderr, "Failed to memcached_create()\n");
 153         if (x > 0)
 154           memcached_free(servers[0]);
 155         x--;
 156         for (; x > 0; x--)
 157           memcached_free(servers[x]);
 158
 159         free(servers);
 160         return;
 161       }
 162       memcached_server_add(servers[x],
 163                            memcached_server_name(&server_list[x]),
 164                            memcached_server_port(&server_list[x]));
 165     }
 166
 167     printf("Network Latency Test:\n\n");
 168     struct timeval start_time, end_time;
 169     long elapsed_time, slowest_time= 0, slowest_server= 0;
 170
 171     for (uint32_t x= 0; x < server_count; x++)
 172     {
 173       gettimeofday(&start_time, NULL);
 174       for (uint32_t y= 0; y < num_of_tests; y++)
 175       {
 176         size_t vlen;
 177         char *val= memcached_get(servers[x], test_key, strlen(test_key),
 178                                  &vlen, &flags, &rc);
 179         if (rc != MEMCACHED_NOTFOUND && rc != MEMCACHED_SUCCESS)
 180           break;
 181         free(val);
 182       }
 183       gettimeofday(&end_time, NULL);
 184
 185       elapsed_time= (long) timedif(end_time, start_time);
 186       elapsed_time /= (long) num_of_tests;
 187
 188       if (elapsed_time > slowest_time)
 189       {
 190         slowest_server= (long)x;
 191         slowest_time= elapsed_time;
 192       }
 193
 194       if (rc != MEMCACHED_NOTFOUND && rc != MEMCACHED_SUCCESS)
 195       {
 196         printf("\t %s (%d)  =>  failed to reach the server\n",
 197                memcached_server_name(&server_list[x]),
 198                memcached_server_port(&server_list[x]));
 199       }
 200       else
 201       {
 202         printf("\t %s (%d)  =>  %ld.%ld seconds\n",
 203                memcached_server_name(&server_list[x]),
 204                memcached_server_port(&server_list[x]),
 205                elapsed_time / 1000, elapsed_time % 1000);
 206       }
 207     }
 208
 209     if (server_count > 1 && slowest_time > 0)
 210     {
 211       printf("---\n");
 212       printf("Slowest Server: %s (%d) => %ld.%ld seconds\n",
 213              memcached_server_name(&server_list[slowest_server]),
 214              memcached_server_port(&server_list[slowest_server]),
 215              slowest_time / 1000, slowest_time % 1000);
 216     }
 217     printf("\n");
 218
 219     for (uint32_t x= 0; x < server_count; x++)
 220       memcached_free(servers[x]);
 221
 222     free(servers);
 223     free(analyze_mode);
 224   }
 225   else
 226   {
 227     fprintf(stderr, "Invalid Analyzer Option provided\n");
 228     free(analyze_mode);
 229   }
 230 }
 231
 232 static void print_server_listing(memcached_st *memc, memcached_stat_st *memc_stat,
 233                                  memcached_server_st *server_list)
 234 {
 235   memcached_return_t rc;
 236
 237   printf("Listing %u Server\n\n", memcached_server_count(memc));
 238   for (uint32_t x= 0; x < memcached_server_count(memc); x++)
 239   {
 240     char **list;
 241     char **ptr;
 242
 243     list= memcached_stat_get_keys(memc, &memc_stat[x], &rc);
 244
 245     printf("Server: %s (%u)\n", memcached_server_name(&server_list[x]),
 246            (uint32_t)memcached_server_port(&server_list[x]));
 247     for (ptr= list; *ptr; ptr++)
 248     {
 249       char *value= memcached_stat_get_value(memc, &memc_stat[x], *ptr, &rc);
 250
 251       printf("\t %s: %s\n", *ptr, value);
 252       free(value);
 253     }
 254
 255     free(list);
 256     printf("\n");
 257   }
 258 }
 259
 260 static void print_analysis_report(memcached_st *memc,
 261                                   memcached_analysis_st *report,
 262                                   memcached_server_st *server_list)
 263 {
 264   uint32_t server_count= memcached_server_count(memc);
 265
 266   printf("Memcached Cluster Analysis Report\n\n");
 267
 268   printf("\tNumber of Servers Analyzed         : %u\n", server_count);
 269   printf("\tAverage Item Size (incl/overhead)  : %u bytes\n",
 270          report->average_item_size);
 271
 272   if (server_count == 1)
 273   {
 274     printf("\nFor a detailed report, you must supply multiple servers.\n");
 275     return;
 276   }
 277
 278   printf("\n");
 279   printf("\tNode with most memory consumption  : %s:%u (%llu bytes)\n",
 280          memcached_server_name(&server_list[report->most_consumed_server]),
 281          (uint32_t)memcached_server_port(&server_list[report->most_consumed_server]),
 282          (unsigned long long)report->most_used_bytes);
 283   printf("\tNode with least free space         : %s:%u (%llu bytes remaining)\n",
 284          memcached_server_name(&server_list[report->least_free_server]),
 285          (uint32_t)memcached_server_port(&server_list[report->least_free_server]),
 286          (unsigned long long)report->least_remaining_bytes);
 287   printf("\tNode with longest uptime           : %s:%u (%us)\n",
 288          memcached_server_name(&server_list[report->oldest_server]),
 289          (uint32_t)memcached_server_port(&server_list[report->oldest_server]),
 290          report->longest_uptime);
 291   printf("\tPool-wide Hit Ratio                : %1.f%%\n", report->pool_hit_ratio);
 292   printf("\n");
 293 }
 294
 295 static void options_parse(int argc, char *argv[])
 296 {
 297   memcached_programs_help_st help_options[]=
 298   {
 299     {0},
 300   };
 301
 302   int option_index= 0;
 303   int option_rv;
 304
 305   while (1)
 306   {
 307     option_rv= getopt_long(argc, argv, "Vhvds:a", long_options, &option_index);
 308     if (option_rv == -1) break;
 309     switch (option_rv)
 310     {
 311     case 0:
 312       break;
 313     case OPT_VERBOSE: /* --verbose or -v */
 314       opt_verbose = OPT_VERBOSE;
 315       break;
 316     case OPT_DEBUG: /* --debug or -d */
 317       opt_verbose = OPT_DEBUG;
 318       break;
 319     case OPT_VERSION: /* --version or -V */
 320       version_command(PROGRAM_NAME);
 321       break;
 322     case OPT_HELP: /* --help or -h */
 323       help_command(PROGRAM_NAME, PROGRAM_DESCRIPTION, long_options, help_options);
 324       break;
 325     case OPT_SERVERS: /* --servers or -s */
 326       opt_servers= strdup(optarg);
 327       break;
 328     case OPT_ANALYZE: /* --analyze or -a */
 329       opt_analyze= OPT_ANALYZE;
 330       analyze_mode= (optarg) ? strdup(optarg) : NULL;
 331       break;
 332     case '?':
 333       /* getopt_long already printed an error message. */
 334       exit(1);
 335     default:
 336       abort();
 337     }
 338   }
 339 }