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

   1 #include <stdio.h>
   2 #include <sys/types.h>
   3 #include <sys/stat.h>
   4 #include <sys/types.h>
   5 #include <sys/mman.h>
   6 #include <fcntl.h>
   7 #include <string.h>
   8 #include <assert.h>
   9 #include <getopt.h>
  10
  11 #include <memcached.h>
  12
  13 #include "client_options.h"
  14 #include "utilities.h"
  15
  16 #define PROGRAM_NAME "memstat"
  17 #define PROGRAM_DESCRIPTION "Output the state of a memcached cluster."
  18
  19 /* Prototypes */
  20 void options_parse(int argc, char *argv[]);
  21
  22 static int opt_verbose= 0;
  23 static int opt_displayflag= 0;
  24 static char *opt_servers= NULL;
  25
  26 int main(int argc, char *argv[])
  27 {
  28   unsigned int x;
  29   memcached_return rc;
  30   memcached_st *memc;
  31   memcached_stat_st *stat;
  32   memcached_server_st *servers;
  33   memcached_server_st *server_list;
  34
  35   options_parse(argc, argv);
  36
  37   if (!opt_servers)
  38   {
  39     char *temp;
  40
  41     if ((temp= getenv("MEMCACHED_SERVERS")))
  42       opt_servers= strdup(temp);
  43     else
  44     {
  45       fprintf(stderr, "No Servers provided\n");
  46       exit(1);
  47     }
  48   }
  49
  50   memc= memcached_create(NULL);
  51
  52   servers= memcached_servers_parse(opt_servers);
  53   memcached_server_push(memc, servers);
  54   memcached_server_list_free(servers);
  55
  56   stat= memcached_stat(memc, NULL, &rc);
  57
  58   if (rc != MEMCACHED_SUCCESS && rc != MEMCACHED_SOME_ERRORS)
  59   {
  60     printf("Failure to communicate with servers (%s)\n",
  61            memcached_strerror(memc, rc));
  62     exit(1);
  63   }
  64
  65   server_list= memcached_server_list(memc);
  66
  67   printf("Listing %u Server\n\n", memcached_server_count(memc));
  68   for (x= 0; x < memcached_server_count(memc); x++)
  69   {
  70     char **list;
  71     char **ptr;
  72
  73     list= memcached_stat_get_keys(memc, &stat[x], &rc);
  74     assert(list);
  75     assert(rc == MEMCACHED_SUCCESS);
  76
  77     printf("Server: %s (%u)\n", memcached_server_name(memc, server_list[x]),
  78            memcached_server_port(memc, server_list[x]));
  79     for (ptr= list; *ptr; ptr++)
  80     {
  81       memcached_return rc;
  82       char *value= memcached_stat_get_value(memc, &stat[x], *ptr, &rc);
  83
  84       printf("\t %s: %s\n", *ptr, value);
  85       free(value);
  86     }
  87
  88     free(list);
  89     printf("\n");
  90   }
  91
  92   free(stat);
  93   free(opt_servers);
  94
  95   memcached_free(memc);
  96
  97   return 0;
  98 }
  99
 100 void options_parse(int argc, char *argv[])
 101 {
 102   memcached_programs_help_st help_options[]=
 103   {
 104     {0},
 105   };
 106
 107   static struct option long_options[]=
 108   {
 109     {"version", no_argument, NULL, OPT_VERSION},
 110     {"help", no_argument, NULL, OPT_HELP},
 111     {"verbose", no_argument, &opt_verbose, OPT_VERBOSE},
 112     {"debug", no_argument, &opt_verbose, OPT_DEBUG},
 113     {"servers", required_argument, NULL, OPT_SERVERS},
 114     {"flag", no_argument, &opt_displayflag, OPT_FLAG},
 115     {0, 0, 0, 0},
 116   };
 117
 118   int option_index= 0;
 119   int option_rv;
 120
 121   while (1)
 122   {
 123     option_rv= getopt_long(argc, argv, "Vhvds:", long_options, &option_index);
 124     if (option_rv == -1) break;
 125     switch (option_rv)
 126     {
 127     case 0:
 128       break;
 129     case OPT_VERBOSE: /* --verbose or -v */
 130       opt_verbose = OPT_VERBOSE;
 131       break;
 132     case OPT_DEBUG: /* --debug or -d */
 133       opt_verbose = OPT_DEBUG;
 134       break;
 135     case OPT_VERSION: /* --version or -V */
 136       version_command(PROGRAM_NAME);
 137       break;
 138     case OPT_HELP: /* --help or -h */
 139       help_command(PROGRAM_NAME, PROGRAM_DESCRIPTION, long_options, help_options);
 140       break;
 141     case OPT_SERVERS: /* --servers or -s */
 142       opt_servers= strdup(optarg);
 143       break;
 144     case '?':
 145       /* getopt_long already printed an error message. */
 146       exit(1);
 147     default:
 148       abort();
 149     }
 150   }
 151 }