Work inspired by matt knox
[m6w6/libmemcached] / clients / memflush.c
1 #include "libmemcached/common.h"
2 #include <stdio.h>
3 #include <unistd.h>
4 #include <string.h>
5 #include <getopt.h>
6 #include <libmemcached/memcached.h>
7 #include "client_options.h"
8 #include "utilities.h"
9
10 static int opt_binary= 0;
11 static int opt_verbose= 0;
12 static time_t opt_expire= 0;
13 static char *opt_servers= NULL;
14
15 #define PROGRAM_NAME "memflush"
16 #define PROGRAM_DESCRIPTION "Erase all data in a server of memcached servers."
17
18 /* Prototypes */
19 void options_parse(int argc, char *argv[]);
20
21 int main(int argc, char *argv[])
22 {
23 memcached_st *memc;
24 memcached_return rc;
25 memcached_server_st *servers;
26
27 options_parse(argc, argv);
28
29 if (!opt_servers)
30 {
31 char *temp;
32
33 if ((temp= getenv("MEMCACHED_SERVERS")))
34 opt_servers= strdup(temp);
35 else
36 {
37 fprintf(stderr, "No Servers provided\n");
38 exit(1);
39 }
40 }
41
42 memc= memcached_create(NULL);
43
44 servers= memcached_servers_parse(opt_servers);
45 memcached_server_push(memc, servers);
46 memcached_server_list_free(servers);
47 memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_BINARY_PROTOCOL,
48 (uint64_t) opt_binary);
49
50 rc = memcached_flush(memc, opt_expire);
51 if (rc != MEMCACHED_SUCCESS)
52 {
53 fprintf(stderr, "memflush: memcache error %s",
54 memcached_strerror(memc, rc));
55 if (memc->cached_errno)
56 fprintf(stderr, " system error %s", strerror(memc->cached_errno));
57 fprintf(stderr, "\n");
58 }
59
60 memcached_free(memc);
61
62 free(opt_servers);
63
64 return 0;
65 }
66
67
68 void options_parse(int argc, char *argv[])
69 {
70 memcached_programs_help_st help_options[]=
71 {
72 {0},
73 };
74
75 static struct option long_options[]=
76 {
77 {"version", no_argument, NULL, OPT_VERSION},
78 {"help", no_argument, NULL, OPT_HELP},
79 {"verbose", no_argument, &opt_verbose, OPT_VERBOSE},
80 {"debug", no_argument, &opt_verbose, OPT_DEBUG},
81 {"servers", required_argument, NULL, OPT_SERVERS},
82 {"expire", required_argument, NULL, OPT_EXPIRE},
83 {"binary", no_argument, NULL, OPT_BINARY},
84 {0, 0, 0, 0},
85 };
86 int option_index= 0;
87 int option_rv;
88
89 while (1)
90 {
91 option_rv= getopt_long(argc, argv, "Vhvds:", long_options, &option_index);
92 if (option_rv == -1) break;
93 switch (option_rv)
94 {
95 case 0:
96 break;
97 case OPT_BINARY:
98 opt_binary = 1;
99 break;
100 case OPT_VERBOSE: /* --verbose or -v */
101 opt_verbose = OPT_VERBOSE;
102 break;
103 case OPT_DEBUG: /* --debug or -d */
104 opt_verbose = OPT_DEBUG;
105 break;
106 case OPT_VERSION: /* --version or -V */
107 version_command(PROGRAM_NAME);
108 break;
109 case OPT_HELP: /* --help or -h */
110 help_command(PROGRAM_NAME, PROGRAM_DESCRIPTION, long_options, help_options);
111 break;
112 case OPT_SERVERS: /* --servers or -s */
113 opt_servers= strdup(optarg);
114 break;
115 case OPT_EXPIRE: /* --expire */
116 opt_expire= (time_t)strtoll(optarg, (char **)NULL, 10);
117 break;
118 case '?':
119 /* getopt_long already printed an error message. */
120 exit(1);
121 default:
122 abort();
123 }
124 }
125 }