Rolling tweaks, not yet running

[yaz-moved-to-github.git] / util / benchmark.c

diff --git a/util/benchmark.c b/util/benchmark.c
index 963e510..2d36326 100644 (file)
--- a/util/benchmark.c
+++ b/util/benchmark.c
@@ -1,18 +1,131 @@
-/* $Id: benchmark.c,v 1.1 2003-12-24 16:23:43 mike Exp $
-   Copyright (C) 2003
-   Index Data Aps
-
-   This file is part of the YAZ toolkit.
-
-   See the file LICENSE.
-*/
+/* $Id: benchmark.c,v 1.4 2004-01-26 21:05:34 mike Exp $
+ * Copyright (C) 2003-2004 Index Data Aps
+ *
+ * This file is part of the YAZ toolkit.
+ *
+ * See the file LICENSE.
+ *
+ * This is an elementary benchmarker for server performance.  It works
+ * by repeatedly connecting to, seaching in and retrieving from the
+ * specified server, and keeps statistics about the minimum, maximum
+ * and average times for each operation.
+ */
 
 #include <stdlib.h>
 #include <stdio.h>
+#include <unistd.h>
+#include <stdarg.h>
 
 #include <yaz/zoom.h>
 
 
+struct options {
+    int nconnect;              /* number of connections to make */
+    int nsearch;               /* number of searches on each connection */
+    int npresent;              /* number of presents for each search */
+    int full;                  /* 1 = fetch full records, 0 = brief */
+    int delay;                 /* number of ms to delay between ops */
+    int random;                        /* if true, delay is random 0-specified */
+    int verbosity;             /* 0 = quiet, higher => more verbose */
+} options = {
+    3,
+    3,
+    3,
+    0,
+    1000,
+    1,
+    0,
+};
+
+
+static int test(char *host, int port);
+static void dprintf(int level, char *fmt, ...);
+
+
 int main(int argc, char **argv)
 {
+    char *host;
+    int port;
+    int c;
+    int i;
+    int ok;
+    int nok = 0;
+
+    while ((c = getopt(argc, argv, "c:s:p:fbd:rv:")) != -1) {
+       switch (c) {
+       case 'c': options.nconnect = atoi(optarg); break;
+       case 's': options.nsearch = atoi(optarg); break;
+       case 'p': options.npresent = atoi(optarg); break;
+       case 'f': options.full = 1; break;
+       case 'b': options.full = 0; break;
+       case 'd': options.delay = atoi(optarg); break;
+       case 'r': options.random = 1; break;
+       case 'v': options.verbosity = atoi(optarg); break;
+       default: goto USAGE;
+       }
+    }
+
+    if (argc-optind != 2) {
+    USAGE:
+       fprintf(stderr, "Usage: %s [options] <host> <port>\n"
+"      -c <n>  Make <n> connection to the server [default: 3]\n"
+"      -s <n>  Perform <n> searches on each connection [3]\n"
+"      -p <n>  Make <n> present requests after each search [3]\n"
+"      -f      Fetch full records [default: brief]\n"
+"      -b      Fetch brief records\n"
+"      -d <n>  Delay <n> ms after each operation\n"
+"      -r      Delays are random between 0 and the specified number of ms\n"
+"      -v <n>  Set verbosity level to <n> [0, silent on success]\n"
+, argv[0]);
+       return 1;
+    }
+
+    host = argv[optind];
+    port = atoi(argv[optind+1]);
+
+    for (i = 0; i < options.nconnect; i++) {
+       dprintf(2, "iteration %d of %d", i+1, options.nconnect);
+       ok = test(host, port);
+       if (ok) nok++;
+    }
+
+    dprintf(1, "passed %d of %d tests", nok, options.nconnect);
+    if (nok < options.nconnect)
+       printf("Failed %d of %d tests\n",
+              options.nconnect-nok, options.nconnect);
+
+
+    return 0;
+}
+
+
+static int test(char *host, int port)
+{
+    ZOOM_connection conn;
+    int error;
+    const char *errmsg, *addinfo;
+
+    conn = ZOOM_connection_new(host, port);
+    if ((error = ZOOM_connection_error(conn, &errmsg, &addinfo))) {
+       fprintf(stderr, "ZOOM error: %s (%d): %s\n", errmsg, error, addinfo);
+       return 0;
+    }
+
+    ZOOM_connection_destroy(conn);
+    return 1;
+}
+
+
+static void dprintf(int level, char *fmt, ...)
+{
+    va_list ap;
+
+    if (level > options.verbosity)
+       return;
+
+    fprintf(stderr, "DEBUG(%d): ", level);
+    va_start(ap, fmt);
+    vfprintf(stderr, fmt, ap);
+    fputc('\n', stderr);
+    va_end(ap);
 }