063aaf5e379ec7af207588bdd74681fbb0488f63
[vlendec/samba-autobuild/.git] / ctdb / tests / ctdb_store.c
1 /* 
2    simple tool to create a lot of records on a tdb and to read them out
3
4    Copyright (C) Andrew Tridgell  2006
5         Ronnie sahlberg 2007
6
7    This program is free software; you can redistribute it and/or modify
8    it under the terms of the GNU General Public License as published by
9    the Free Software Foundation; either version 2 of the License, or
10    (at your option) any later version.
11    
12    This program is distributed in the hope that it will be useful,
13    but WITHOUT ANY WARRANTY; without even the implied warranty of
14    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15    GNU General Public License for more details.
16    
17    You should have received a copy of the GNU General Public License
18    along with this program; if not, write to the Free Software
19    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
20 */
21
22 #include "includes.h"
23 #include "lib/events/events.h"
24 #include "system/filesys.h"
25 #include "popt.h"
26 #include "cmdline.h"
27
28 #include <sys/time.h>
29 #include <time.h>
30
31 static int num_records = 10;
32
33
34 static void store_records(struct ctdb_context *ctdb, struct event_context *ev)
35 {
36         TDB_DATA key, data;
37         struct ctdb_db_context *ctdb_db;
38         TALLOC_CTX *tmp_ctx = talloc_new(ctdb);
39         int ret;
40         struct ctdb_record_handle *h;
41         uint32_t i;
42         
43         ctdb_db = ctdb_db_handle(ctdb, "test.tdb");
44
45         printf("creating %d records\n", num_records);
46         for (i=0;i<num_records;i++) {
47                 key.dptr = (uint8_t *)&i;
48                 key.dsize = sizeof(uint32_t); 
49
50                 h = ctdb_fetch_lock(ctdb_db, tmp_ctx, key, &data);
51                 if (h == NULL) {
52                         printf("Failed to fetch record '%s' on node %d\n", 
53                                (const char *)key.dptr, ctdb_get_vnn(ctdb));
54                         talloc_free(tmp_ctx);
55                         return;
56                 }
57
58                 data.dptr = (uint8_t *)&i;
59                 data.dsize = sizeof(uint32_t);
60
61                 ret = ctdb_record_store(h, data);
62                 talloc_free(h);
63                 if (ret != 0) {
64                         printf("Failed to store record\n");
65                 }
66                 if (i % 1000 == 0) {
67                         printf("%u\r", i);
68                         fflush(stdout);
69                 }
70         }
71
72         printf("fetching all %d records\n", num_records);
73         while (1) {
74                 for (i=0;i<num_records;i++) {
75                         key.dptr = (uint8_t *)&i;
76                         key.dsize = sizeof(uint32_t); 
77
78                         h = ctdb_fetch_lock(ctdb_db, tmp_ctx, key, &data);
79                         if (h == NULL) {
80                                 printf("Failed to fetch record '%s' on node %d\n", 
81                                        (const char *)key.dptr, ctdb_get_vnn(ctdb));
82                                 talloc_free(tmp_ctx);
83                                 return;
84                         }
85                         talloc_free(h);
86                 }
87                 sleep(1);
88                 printf(".");
89                 fflush(stdout);
90         }
91
92         talloc_free(tmp_ctx);
93 }
94
95 /*
96   main program
97 */
98 int main(int argc, const char *argv[])
99 {
100         struct ctdb_context *ctdb;
101         struct ctdb_db_context *ctdb_db;
102
103         struct poptOption popt_options[] = {
104                 POPT_AUTOHELP
105                 POPT_CTDB_CMDLINE
106                 { "num-records", 'r', POPT_ARG_INT, &num_records, 0, "num_records", "integer" },
107                 POPT_TABLEEND
108         };
109         int opt;
110         const char **extra_argv;
111         int extra_argc = 0;
112         poptContext pc;
113         struct event_context *ev;
114
115         pc = poptGetContext(argv[0], argc, argv, popt_options, POPT_CONTEXT_KEEP_FIRST);
116
117         while ((opt = poptGetNextOpt(pc)) != -1) {
118                 switch (opt) {
119                 default:
120                         fprintf(stderr, "Invalid option %s: %s\n", 
121                                 poptBadOption(pc, 0), poptStrerror(opt));
122                         exit(1);
123                 }
124         }
125
126         /* talloc_enable_leak_report_full(); */
127
128         /* setup the remaining options for the main program to use */
129         extra_argv = poptGetArgs(pc);
130         if (extra_argv) {
131                 extra_argv++;
132                 while (extra_argv[extra_argc]) extra_argc++;
133         }
134
135         ev = event_context_init(NULL);
136
137         ctdb = ctdb_cmdline_client(ev);
138
139         /* attach to a specific database */
140         ctdb_db = ctdb_attach(ctdb, "test.tdb");
141         if (!ctdb_db) {
142                 printf("ctdb_attach failed - %s\n", ctdb_errstr(ctdb));
143                 exit(1);
144         }
145
146         printf("Waiting for cluster\n");
147         while (1) {
148                 uint32_t recmode=1;
149                 ctdb_ctrl_getrecmode(ctdb, timeval_zero(), CTDB_CURRENT_NODE, &recmode);
150                 if (recmode == 0) break;
151                 event_loop_once(ev);
152         }
153
154         store_records(ctdb, ev);
155
156         return 0;
157 }