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