git.m6w6.name Git - m6w6/libmemcached/blob - clients/utilities.c

   1 #include "libmemcached/common.h"
   2 #include <stdio.h>
   3 #include <ctype.h>
   4 #include <string.h>
   5 #include "utilities.h"
   6
   7
   8 long int timedif(struct timeval a, struct timeval b)
   9 {
  10   register int us, s;
  11
  12   us = a.tv_usec - b.tv_usec;
  13   us /= 1000;
  14   s = a.tv_sec - b.tv_sec;
  15   s *= 1000;
  16   return s + us;
  17 }
  18
  19 void version_command(char *command_name)
  20 {
  21   printf("%s v%u.%u\n", command_name, 1, 0);
  22   exit(0);
  23 }
  24
  25 static char *lookup_help(memcached_options option)
  26 {
  27   switch (option)
  28   {
  29   case OPT_SERVERS: return("List which servers you wish to connect to.");
  30   case OPT_VERSION: return("Display the version of the application and then exit.");
  31   case OPT_HELP: return("Diplay this message and then exit.");
  32   case OPT_VERBOSE: return("Give more details on the progression of the application.");
  33   case OPT_DEBUG: return("Provide output only useful for debugging.");
  34   case OPT_FLAG: return("Provide flag information for storage operation.");
  35   case OPT_EXPIRE: return("Set the expire option for the object.");
  36   case OPT_SET: return("Use set command with memcached when storing.");
  37   case OPT_REPLACE: return("Use replace command with memcached when storing.");
  38   case OPT_ADD: return("Use add command with memcached when storing.");
  39   case OPT_SLAP_EXECUTE_NUMBER: return("Number of times to execute the given test.");
  40   case OPT_SLAP_INITIAL_LOAD: return("Number of key pairs to load before executing tests.");
  41   case OPT_SLAP_TEST: return("Test to run (currently \"get\" or \"set\").");
  42   case OPT_SLAP_CONCURRENCY: return("Number of users to simulate with load.");
  43   case OPT_SLAP_NON_BLOCK: return("Set TCP up to use non-blocking IO.");
  44   case OPT_SLAP_TCP_NODELAY: return("Set TCP socket up to use nodelay.");
  45   case OPT_FLUSH: return("Flush servers before running tests.");
  46   case OPT_HASH: return("Select hash type.");
  47   case OPT_BINARY: return("Switch to binary protocol.");
  48   case OPT_ANALYZE: return("Analyze the provided servers.");
  49   case OPT_UDP: return("Use UDP protocol when communicating with server.");
  50   };
  51
  52   WATCHPOINT_ASSERT(0);
  53   return "forgot to document this function :)";
  54 }
  55
  56 void help_command(char *command_name, char *description,
  57                   const struct option *long_options,
  58                   memcached_programs_help_st *options __attribute__((unused)))
  59 {
  60   unsigned int x;
  61
  62   printf("%s v%u.%u\n\n", command_name, 1, 0);
  63   printf("\t%s\n\n", description);
  64   printf("Current options. A '=' means the option takes a value.\n\n");
  65
  66   for (x= 0; long_options[x].name; x++)
  67   {
  68     char *help_message;
  69
  70     printf("\t --%s%c\n", long_options[x].name,
  71            long_options[x].has_arg ? '=' : ' ');
  72     if ((help_message= lookup_help(long_options[x].val)))
  73       printf("\t\t%s\n", help_message);
  74   }
  75
  76   printf("\n");
  77   exit(0);
  78 }
  79
  80 void process_hash_option(memcached_st *memc, char *opt_hash)
  81 {
  82   uint64_t set;
  83   memcached_return rc;
  84
  85   if (opt_hash == NULL)
  86     return;
  87
  88   set= MEMCACHED_HASH_DEFAULT; /* Just here to solve warning */
  89   if (!strcasecmp(opt_hash, "CRC"))
  90     set= MEMCACHED_HASH_CRC;
  91   else if (!strcasecmp(opt_hash, "FNV1_64"))
  92     set= MEMCACHED_HASH_FNV1_64;
  93   else if (!strcasecmp(opt_hash, "FNV1A_64"))
  94     set= MEMCACHED_HASH_FNV1A_64;
  95   else if (!strcasecmp(opt_hash, "FNV1_32"))
  96     set= MEMCACHED_HASH_FNV1_32;
  97   else if (!strcasecmp(opt_hash, "FNV1A_32"))
  98     set= MEMCACHED_HASH_FNV1A_32;
  99   else
 100   {
 101     fprintf(stderr, "hash: type not recognized %s\n", opt_hash);
 102     exit(1);
 103   }
 104
 105   rc= memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_HASH, set);
 106   if (rc != MEMCACHED_SUCCESS)
 107   {
 108     fprintf(stderr, "hash: memcache error %s\n", memcached_strerror(memc, rc));
 109     exit(1);
 110   }
 111 }
 112