Happy new year.
[idzebra-moved-to-github.git] / isamb / benchisamb.c
1 /* This file is part of the Zebra server.
2    Copyright (C) 1994-2011 Index Data
3
4 Zebra is free software; you can redistribute it and/or modify it under
5 the terms of the GNU General Public License as published by the Free
6 Software Foundation; either version 2, or (at your option) any later
7 version.
8
9 Zebra is distributed in the hope that it will be useful, but WITHOUT ANY
10 WARRANTY; without even the implied warranty of MERCHANTABILITY or
11 FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
12 for more details.
13
14 You should have received a copy of the GNU General Public License
15 along with this program; if not, write to the Free Software
16 Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
17
18 */
19
20 #include <yaz/options.h>
21 #if HAVE_SYS_TIMES_H
22 #include <sys/times.h>
23 #endif
24 #if HAVE_SYS_TIME_H
25 #include <sys/time.h>
26 #endif
27
28 #include <stdlib.h>
29 #include <string.h>
30 #include <yaz/log.h>
31 #include <yaz/xmalloc.h>
32 #include <yaz/timing.h>
33 #include <idzebra/isamb.h>
34 #include <assert.h>
35
36 static void log_item(int level, const void *b, const char *txt)
37 {
38     int x;
39     memcpy(&x, b, sizeof(int));
40     yaz_log(YLOG_LOG, "%s %d", txt, x);
41 }
42
43 static void log_pr(const char *txt)
44 {
45     yaz_log(YLOG_LOG, "%s", txt);
46 }
47
48 int compare_item(const void *a, const void *b)
49 {
50     int ia, ib;
51
52     memcpy(&ia, (const char *) a + 1, sizeof(int));
53     memcpy(&ib, (const char *) b + 1, sizeof(int));
54     if (ia > ib)
55         return 1;
56     if (ia < ib)
57         return -1;
58    return 0;
59 }
60
61 void *code_start(void)
62 {
63     return 0;
64 }
65
66 void code_item(void *p, char **dst, const char **src)
67 {
68     int sz = **src;
69     memcpy (*dst, *src, sz);
70     (*dst) += sz;
71     (*src) += sz;
72 }
73
74 void code_reset(void *p)
75 {
76 }
77 void code_stop(void *p)
78 {
79 }
80
81 struct read_info {
82     int val;
83     int step;
84
85     int no;
86     int max;
87     int insertMode;
88     int sz;
89 };
90
91 int code_read(void *vp, char **dst, int *insertMode)
92 {
93     struct read_info *ri = (struct read_info *)vp;
94     int x;
95
96     if (ri->no >= ri->max)
97         return 0;
98     ri->no++;
99
100     x = ri->val;
101     memset(*dst, 0, ri->sz);
102     **dst = ri->sz;
103     memcpy(*dst + 1, &x, sizeof(int));
104
105     (*dst) += ri->sz;
106
107     ri->val = ri->val + ri->step;
108     *insertMode = ri->insertMode;
109
110 #if 0
111     yaz_log(YLOG_LOG, "%d %5d", ri->insertMode, x);
112 #endif
113     return 1;
114 }
115
116 void bench_insert(ISAMB isb, int number_of_trees,
117                   int number_of_rounds, int number_of_elements,
118                   int extra_size)
119 {
120     ISAMC_I isamc_i;
121     ISAM_P *isamc_p = xmalloc(sizeof(ISAM_P) * number_of_trees);
122     struct read_info ri;
123     int round, i;
124
125     for (i = 0; i<number_of_trees; i++)
126         isamc_p[i] = 0; /* initially, is empty */
127
128     ri.val = 0;
129     ri.step = 1;
130     ri.insertMode = 1;
131     ri.sz = sizeof(int) + 1 + extra_size;
132     
133     for (round = 0; round < number_of_rounds; round++)
134     {
135         yaz_timing_t t = yaz_timing_create();
136
137         yaz_timing_start(t);
138         for (i = 0; i<number_of_trees; i++)
139         {
140
141             /* insert a number of entries */
142             ri.no = 0;
143           
144             ri.val = (rand());
145             if (RAND_MAX < 65536)  
146                 ri.val = ri.val + 65536*rand();
147
148             // ri.val = number_of_elements * round;
149             ri.max = number_of_elements;
150             
151             isamc_i.clientData = &ri;
152             isamc_i.read_item = code_read;
153             
154             isamb_merge (isb, &isamc_p[i] , &isamc_i);
155
156             if (0)
157                 isamb_dump(isb, isamc_p[i], log_pr);
158         }
159         yaz_timing_stop(t);
160         printf("%3d %8.6f %5.2f %5.2f\n",
161                round+1,
162                yaz_timing_get_real(t),
163                yaz_timing_get_user(t),
164                yaz_timing_get_sys(t));
165         yaz_timing_destroy(&t);
166     }
167     xfree(isamc_p);
168 }
169
170 void exit_usage(void)
171 {
172     fprintf(stderr, "benchisamb [-r rounds] [-n items] [-i isams]\n");
173     exit(1);
174 }
175
176 int main(int argc, char **argv)
177 {
178     BFiles bfs;
179     ISAMB isb;
180     ISAMC_M method;
181     int ret;
182     char *arg;
183     int number_of_rounds = 10;
184     int number_of_items = 1000;
185     int number_of_isams = 1000;
186     int extra_size = 0;
187     yaz_timing_t t = 0;
188     
189     while ((ret = options("z:r:n:i:", argv, argc, &arg)) != -2)
190     {
191         switch(ret)
192         {
193         case 'r':
194             number_of_rounds = atoi(arg);
195             break;
196         case 'n':
197             number_of_items = atoi(arg);
198             break;
199         case 'i':
200             number_of_isams = atoi(arg);
201             break;
202         case 'z':
203             extra_size = atoi(arg);
204             break;
205         case 0:
206             fprintf(stderr, "bad arg: %s\n", arg);
207             exit_usage();
208         default:
209             fprintf(stderr, "bad option.\n");
210             exit_usage();
211         }
212     }
213         
214     /* setup method (attributes) */
215     method.compare_item = compare_item;
216     method.log_item = log_item;
217     method.codec.start = code_start;
218     method.codec.encode = code_item;
219     method.codec.decode = code_item;
220     method.codec.reset = code_reset;
221     method.codec.stop = code_stop;
222
223     t = yaz_timing_create();
224     
225     yaz_timing_start(t);
226
227     /* create block system */
228     bfs = bfs_create(0, 0);
229     if (!bfs)
230     {
231         yaz_log(YLOG_WARN, "bfs_create failed");
232         exit(1);
233     }
234
235     bf_reset(bfs);
236
237     /* create isam handle */
238     isb = isamb_open (bfs, "isamb", 1, &method, 0);
239     if (!isb)
240     {
241         yaz_log(YLOG_WARN, "isamb_open failed");
242         exit(2);
243     }
244     bench_insert(isb, number_of_isams, number_of_rounds, number_of_items,
245                  extra_size);
246     
247     isamb_close(isb);
248
249     /* exit block system */
250     bfs_destroy(bfs);
251
252     yaz_timing_stop(t);
253     printf("Total %8.6f %5.2f %5.2f\n",
254            yaz_timing_get_real(t),
255            yaz_timing_get_user(t),
256            yaz_timing_get_sys(t));
257     yaz_timing_destroy(&t);
258     exit(0);
259     return 0;
260 }
261 /*
262  * Local variables:
263  * c-basic-offset: 4
264  * c-file-style: "Stroustrup"
265  * indent-tabs-mode: nil
266  * End:
267  * vim: shiftwidth=4 tabstop=8 expandtab
268  */
269