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 <getopt.h>
   8
   9 #include <memcached.h>
  10
  11 #include "client_options.h"
  12 #include "utilities.h"
  13
  14 /* Prototypes */
  15 void options_parse(int argc, char *argv[]);
  16
  17 static int opt_verbose= 0;
  18 static int opt_displayflag= 0;
  19 static char *opt_servers= NULL;
  20
  21 int main(int argc, char *argv[])
  22 {
  23   unsigned int x;
  24   memcached_return rc;
  25   memcached_st *memc;
  26   memcached_stat_st *stat;
  27   memcached_server_st *servers;
  28   memcached_server_st *server_list;
  29
  30   options_parse(argc, argv);
  31
  32   if (!opt_servers)
  33     return 0;
  34
  35   memc= memcached_init(NULL);
  36
  37   servers= parse_opt_servers(opt_servers);
  38   memcached_server_push(memc, servers);
  39   memcached_server_list_free(servers);
  40
  41   stat= memcached_stat(memc, NULL, &rc);
  42
  43   if (rc != MEMCACHED_SUCCESS || rc != MEMCACHED_SOME_ERRORS);
  44   {
  45     printf("Failure to communicate with servers (%s)\n",
  46            memcached_strerror(memc, rc));
  47     exit(1);
  48   }
  49
  50   server_list= memcached_server_list(memc);
  51
  52   printf("Listing %u Server\n\n", memcached_server_count(memc));
  53   for (x= 0; x < memcached_server_count(memc); x++)
  54   {
  55     char **list;
  56     char **ptr;
  57
  58     list= memcached_stat_get_keys(memc, &stat[x], &rc);
  59     assert(list);
  60     assert(rc == MEMCACHED_SUCCESS);
  61
  62     printf("Server: %s (%u)\n", memcached_server_name(memc, server_list[x]),
  63            memcached_server_port(memc, server_list[x]));
  64     for (ptr= list; *ptr; ptr++)
  65     {
  66       memcached_return rc;
  67       char *value= memcached_stat_get_value(memc, &stat[x], *ptr, &rc);
  68
  69       printf("\t %s: %s\n", *ptr, value);
  70       free(value);
  71     }
  72
  73     free(list);
  74     printf("\n");
  75   }
  76
  77   free(stat);
  78   free(opt_servers);
  79
  80   memcached_deinit(memc);
  81
  82   return 0;
  83 }
  84
  85 void options_parse(int argc, char *argv[])
  86 {
  87   static struct option long_options[]=
  88   {
  89     {"version", no_argument, NULL, OPT_VERSION},
  90     {"help", no_argument, NULL, OPT_HELP},
  91     {"verbose", no_argument, &opt_verbose, OPT_VERBOSE},
  92     {"debug", no_argument, &opt_verbose, OPT_DEBUG},
  93     {"servers", required_argument, NULL, OPT_SERVERS},
  94     {"flag", no_argument, &opt_displayflag, OPT_FLAG},
  95     {0, 0, 0, 0},
  96   };
  97
  98   int option_index= 0;
  99   int option_rv;
 100
 101   while (1)
 102   {
 103     option_rv= getopt_long(argc, argv, "Vhvds:", long_options, &option_index);
 104     if (option_rv == -1) break;
 105     switch (option_rv)
 106     {
 107     case 0:
 108       break;
 109     case OPT_VERBOSE: /* --verbose or -v */
 110       opt_verbose = OPT_VERBOSE;
 111       break;
 112     case OPT_DEBUG: /* --debug or -d */
 113       opt_verbose = OPT_DEBUG;
 114       break;
 115     case OPT_VERSION: /* --version or -V */
 116       printf("memcache tools, memcat, v1.0\n");
 117       exit(0);
 118       break;
 119     case OPT_HELP: /* --help or -h */
 120       printf("useful help messages go here\n");
 121       exit(0);
 122       break;
 123     case OPT_SERVERS: /* --servers or -s */
 124       opt_servers= strdup(optarg);
 125       break;
 126     case '?':
 127       /* getopt_long already printed an error message. */
 128       exit(1);
 129     default:
 130       abort();
 131     }
 132   }
 133 }