Controlled conversion from Odr_int to int
[yaz-moved-to-github.git] / util / benchmark.c
1 /* This file is part of the YAZ toolkit.
2  * Copyright (C) 1995-2010 Index Data
3  * See the file LICENSE for details.
4  */
5
6 #include <stdlib.h>
7 #include <stdio.h>
8 #include <unistd.h>
9 #include <yaz/options.h>
10 #include <stdarg.h>
11
12 #include <yaz/zoom.h>
13
14
15 struct boptions {
16     int nconnect;               /* number of connections to make */
17     int nsearch;                /* number of searches on each connection */
18     int npresent;               /* number of presents for each search */
19     int full;                   /* 1 = fetch full records, 0 = brief */
20     int delay;                  /* number of ms to delay between ops */
21     int random;                 /* if true, delay is random 0-specified */
22     int verbosity;              /* 0 = quiet, higher => more verbose */
23 } boptions = {
24     3,
25     3,
26     3,
27     0,
28     1000,
29     1,
30     0,
31 };
32
33
34 static int test(char *host, int port);
35 static void db_printf(int level, char *fmt, ...);
36 static void usage(const char *prog);
37
38 int main(int argc, char **argv)
39 {
40     char *host = 0;
41     int port = 0;
42     int c;
43     int i;
44     int ok;
45     int nok = 0;
46     char *arg;
47     
48     while ((c = options("c:s:p:fbd:rv:", argv, argc, &arg)) != -2) {
49         switch (c) {
50         case 0:
51             if (!host)
52                 host = arg;
53             else if (!port)
54                 port = atoi(arg);
55             else
56                 usage(*argv);
57             break;
58         case 'c': boptions.nconnect = atoi(arg); break;
59         case 's': boptions.nsearch = atoi(arg); break;
60         case 'p': boptions.npresent = atoi(arg); break;
61         case 'f': boptions.full = 1; break;
62         case 'b': boptions.full = 0; break;
63         case 'd': boptions.delay = atoi(arg); break;
64         case 'r': boptions.random = 1; break;
65         case 'v': boptions.verbosity = atoi(arg); break;
66         default: usage(*argv);
67         }
68     }
69
70     if (!host || !port)
71         usage(*argv);
72
73     for (i = 0; i < boptions.nconnect; i++) {
74         db_printf(2, "iteration %d of %d", i+1, boptions.nconnect);
75         ok = test(host, port);
76         if (ok) nok++;
77     }
78
79     db_printf(1, "passed %d of %d tests", nok, boptions.nconnect);
80     if (nok < boptions.nconnect)
81         printf("Failed %d of %d tests\n",
82                boptions.nconnect-nok, boptions.nconnect);
83
84     return 0;
85 }
86
87 static void usage(const char *prog)
88 {
89     fprintf(stderr, "Usage: %s [options] <host> <port>\n"
90 "       -c <n>  Make <n> connection to the server [default: 3]\n"
91 "       -s <n>  Perform <n> searches on each connection [3]\n"
92 "       -p <n>  Make <n> present requests after each search [3]\n"
93 "       -f      Fetch full records [default: brief]\n"
94 "       -b      Fetch brief records\n"
95 "       -d <n>  Delay <n> ms after each operation\n"
96 "       -r      Delays are random between 0 and the specified number of ms\n"
97 "       -v <n>  Set verbosity level to <n> [0, silent on success]\n"
98             , prog);
99     exit(1);
100 }
101
102 static int test(char *host, int port)
103 {
104     ZOOM_connection conn;
105     int error;
106     const char *errmsg, *addinfo;
107
108     conn = ZOOM_connection_new(host, port);
109     if ((error = ZOOM_connection_error(conn, &errmsg, &addinfo))) {
110         fprintf(stderr, "ZOOM error: %s (%d): %s\n", errmsg, error, addinfo);
111         return 0;
112     }
113
114     ZOOM_connection_destroy(conn);
115     return 1;
116 }
117
118 static void db_printf(int level, char *fmt, ...)
119 {
120     va_list ap;
121
122     if (level > boptions.verbosity)
123         return;
124
125     fprintf(stderr, "DEBUG(%d): ", level);
126     va_start(ap, fmt);
127     vfprintf(stderr, fmt, ap);
128     fputc('\n', stderr);
129     va_end(ap);
130 }
131 /*
132  * Local variables:
133  * c-basic-offset: 4
134  * c-file-style: "Stroustrup"
135  * indent-tabs-mode: nil
136  * End:
137  * vim: shiftwidth=4 tabstop=8 expandtab
138  */
139