changed output to be non-cascarding when using -n switch
[yaz-moved-to-github.git] / zoom / zoom-benchmark.c
index 34efe75..396cb67 100644 (file)
@@ -1,5 +1,5 @@
 /*
- * $Id: zoom-benchmark.c,v 1.9 2005-09-19 14:14:01 marc Exp $
+ * $Id: zoom-benchmark.c,v 1.17 2007-01-04 14:43:08 marc Exp $
  *
  * Asynchronous multi-target client doing search and piggyback retrieval
  */
@@ -23,11 +23,15 @@ static int zoom_progress[10];
 
 /* commando line parameters */
 static struct parameters_t { 
-    char host[4096];
-    char query[4096];
-    char progress[4096];
+    char host[1024];
+    char query[1024];
+    int progress[4096];
     int concurrent;
+    int repeat;
     int timeout;
+    char proxy[1024];
+    int piggypack;
+    int gnuplot;
 } parameters;
 
 struct  event_line_t 
@@ -52,27 +56,8 @@ void print_event_line(struct event_line_t *pel)
             pel->error, pel->errmsg);
 }
 
-//void update_event_line(struct event_line_t *pel,
-//                       int conn,
-//                       long sec,
-//                       long usec,
-//                       int prog,
-//                       int event,
-//                       const char * eventmsg,
-//                       int error,
-//                       const char * errmsg)
-//{
-//    pel->connection = conn;
-//    pel->time_sec = sec;
-//    pel->time_usec = usec;
-//    pel->progress = prog;
-//    pel->event = event;
-//    strcpy(pel->zoom_event, eventmsg);
-//    pel->error = error;
-//    strcpy(pel->errmsg, errmsg);
-//}
-
 void  update_events(int *elc, struct event_line_t *els,
+                    int repeat,
                     int conn,
                     long sec,
                     long usec,
@@ -81,28 +66,43 @@ void  update_events(int *elc, struct event_line_t *els,
                     const char * eventmsg,
                     int error,
                     const char * errmsg){
-    
-    els[conn * 10 + elc[conn]].connection = conn;
-    els[conn * 10 + elc[conn]].time_sec = sec;
-    els[conn * 10 + elc[conn]].time_usec = usec;
-    els[conn * 10 + elc[conn]].progress = prog;
-    els[conn * 10 + elc[conn]].event = event;
-    strcpy(els[conn * 10 + elc[conn]].zoom_event, eventmsg);
-    els[conn * 10 + elc[conn]].error = error;
-    strcpy(els[conn * 10 + elc[conn]].errmsg, errmsg);
-
-    //print_event_line(&els[conn*10 + elc[conn]]);
-
-    elc[conn] += 1;
+
+    int ielc = repeat*parameters.concurrent + conn;
+    int iels = repeat*parameters.concurrent*10 + conn*10 + elc[ielc];
+
+    els[iels].connection = conn;
+    els[iels].time_sec = sec;
+    els[iels].time_usec = usec;
+    els[iels].progress = prog;
+    els[iels].event = event;
+
+    if (eventmsg)
+        strcpy(els[iels].zoom_event, eventmsg);
+    else
+        strcpy(els[iels].zoom_event, "---");
+
+    els[iels].error = error;
+    strcpy(els[iels].errmsg, errmsg);
+    /* print_event_line(&els[iels]); */
+    elc[ielc] += 1;
 }
 
 void  print_events(int *elc,  struct event_line_t *els, 
                    int connections){
     int i;
     int j;
-    for (i=0; i < connections; i++){
-        for (j=0; j < elc[i]; j++){
-            print_event_line(&els[i*10 + j]);
+    int k;
+    int ielc;
+    int iels;
+
+    for (k=0; k < parameters.repeat; k++){
+        for (i=0; i < connections; i++){
+            ielc = k * parameters.concurrent + i;
+            for (j=0; j < elc[ielc]; j++){
+                iels = k * parameters.concurrent * 10 + i * 10 + j;
+                print_event_line(&els[iels]);
+            }
+            printf("\n");
         }
         printf("\n");
     }
@@ -110,8 +110,11 @@ void  print_events(int *elc,  struct event_line_t *els,
 
 
 
-void init_statics()
+void init_statics(void)
 {
+    int i;
+    char nullstring[1] = "";
+
     /* naming events */
     zoom_events[ZOOM_EVENT_NONE] = "ZOOM_EVENT_NONE";
     zoom_events[ZOOM_EVENT_CONNECT] = "ZOOM_EVENT_CONNECT";
@@ -123,25 +126,30 @@ void init_statics()
     zoom_events[ZOOM_EVENT_RECV_APDU] = "ZOOM_EVENT_RECV_APDU";
     zoom_events[ZOOM_EVENT_RECV_RECORD] = "ZOOM_EVENT_RECV_RECORD";
     zoom_events[ZOOM_EVENT_RECV_SEARCH] = "ZOOM_EVENT_RECV_SEARCH";
+    zoom_events[ZOOM_EVENT_END] = "ZOOM_EVENT_END";
 
     /* re-sorting event numbers to progress numbers */
     zoom_progress[ZOOM_EVENT_NONE] = 0;
     zoom_progress[ZOOM_EVENT_CONNECT] = 1;
     zoom_progress[ZOOM_EVENT_SEND_DATA] = 3;
     zoom_progress[ZOOM_EVENT_RECV_DATA] = 4;
-    zoom_progress[ZOOM_EVENT_TIMEOUT] = 8;
-    zoom_progress[ZOOM_EVENT_UNKNOWN] = 9;
+    zoom_progress[ZOOM_EVENT_TIMEOUT] = 9;
+    zoom_progress[ZOOM_EVENT_UNKNOWN] = 10;
     zoom_progress[ZOOM_EVENT_SEND_APDU] = 2;
     zoom_progress[ZOOM_EVENT_RECV_APDU] = 5;
     zoom_progress[ZOOM_EVENT_RECV_RECORD] = 7;
     zoom_progress[ZOOM_EVENT_RECV_SEARCH] = 6;
+    zoom_progress[ZOOM_EVENT_END] = 8;
 
     /* parameters */
     parameters.concurrent = 1;
     parameters.timeout = 0;
+    parameters.repeat = 1;
+    strcpy(parameters.proxy, nullstring);
+    parameters.gnuplot = 0;
+    parameters.piggypack = 0;
 
     /* progress initializing */
-    int i;
     for (i = 0; i < 4096; i++){
         parameters.progress[i] = 0;
     }
@@ -185,12 +193,16 @@ long time_usec(struct time_type *ptime)
     return ptime->usec;
 }
 
-void print_option_error()
+void print_option_error(void)
 {
     fprintf(stderr, "zoom-benchmark:  Call error\n");
     fprintf(stderr, "zoom-benchmark -h host:port -q pqf-query "
-            "[-c no_concurrent] "
-            "[-t timeout] \n");
+            "[-c no_concurrent (max 4096)] "
+            "[-n no_repeat] "
+            "[-b (piggypack)] "
+            "[-g (gnuplot outfile)] "
+            "[-p proxy] \n");
+    /* "[-t timeout] \n"); */
     exit(1);
 }
 
@@ -198,7 +210,7 @@ void print_option_error()
 void read_params(int argc, char **argv, struct parameters_t *p_parameters){    
     char *arg;
     int ret;
-    while ((ret = options("h:q:c:t:", argv, argc, &arg)) != -2)
+    while ((ret = options("h:q:c:t:p:bgn:", argv, argc, &arg)) != -2)
     {
         switch (ret)
         {
@@ -208,11 +220,25 @@ void read_params(int argc, char **argv, struct parameters_t *p_parameters){
         case 'q':
             strcpy(p_parameters->query, arg);
             break;
+        case 'p':
+            strcpy(p_parameters->proxy, arg);
+            break;
         case 'c':
             p_parameters->concurrent = atoi(arg);
             break;
-        case 't':
+#if 0
+            case 't':
             p_parameters->timeout = atoi(arg);
+            break;
+#endif
+        case 'b':
+            p_parameters->piggypack = 1;
+                    break;
+        case 'g':
+            p_parameters->gnuplot = 1;
+                    break;
+        case 'n':
+            p_parameters->repeat = atoi(arg);
                     break;
         case 0:
             print_option_error();
@@ -227,7 +253,13 @@ void read_params(int argc, char **argv, struct parameters_t *p_parameters){
         printf("   host:       %s \n", p_parameters->host);
         printf("   query:      %s \n", p_parameters->query);
         printf("   concurrent: %d \n", p_parameters->concurrent);
-        printf("   timeout:    %d \n\n", p_parameters->timeout);
+        printf("   repeat:     %d \n", p_parameters->repeat);
+#if 0
+        printf("   timeout:    %d \n", p_parameters->timeout);
+#endif
+        printf("   proxy:      %s \n", p_parameters->proxy);
+        printf("   piggypack:  %d \n\n", p_parameters->piggypack);
+        printf("   gnuplot:    %d \n\n", p_parameters->gnuplot);
     }
 
     if (! strlen(p_parameters->host))
@@ -236,12 +268,18 @@ void read_params(int argc, char **argv, struct parameters_t *p_parameters){
         print_option_error();
     if (! (p_parameters->concurrent > 0))
         print_option_error();
+    if (! (p_parameters->repeat > 0))
+        print_option_error();
     if (! (p_parameters->timeout >= 0))
         print_option_error();
+    if (! ( p_parameters->concurrent <= 4096))
+        print_option_error();
 }
 
-void print_table_header()
+void print_table_header(void)
 {
+    if (parameters.gnuplot)
+        printf("#");
     printf ("target\tsecond.usec\tprogress\tevent\teventname\t");
     printf("error\terrorname\n");
 }
@@ -256,6 +294,7 @@ int main(int argc, char **argv)
     struct event_line_t *els;
     ZOOM_options o;
     int i;
+    int k;
 
     init_statics();
 
@@ -263,15 +302,22 @@ int main(int argc, char **argv)
 
     z = xmalloc(sizeof(*z) * parameters.concurrent);
     r = xmalloc(sizeof(*r) * parameters.concurrent);
-    elc = xmalloc(sizeof(*elc) * parameters.concurrent);
-    els = xmalloc(sizeof(*els) * parameters.concurrent * 10);
+    elc = xmalloc(sizeof(*elc) * parameters.concurrent * parameters.repeat);
+    els = xmalloc(sizeof(*els) 
+                  * parameters.concurrent * parameters.repeat * 10);
     o = ZOOM_options_create();
 
     /* async mode */
     ZOOM_options_set (o, "async", "1");
 
-    /* get first record of result set (using piggyback) */
-    ZOOM_options_set (o, "count", "1");
+    /* get first record of result set (using piggypack) */
+    if (parameters.piggypack)
+        ZOOM_options_set (o, "count", "1");
+
+    /* set proxy */
+    if (strlen(parameters.proxy))
+        ZOOM_options_set (o, "proxy", parameters.proxy);
+
 
     /* preferred record syntax */
     if (0){
@@ -279,76 +325,100 @@ int main(int argc, char **argv)
         ZOOM_options_set (o, "elementSetName", "F");
     }
     
+    time_init(&time);
+    /* repeat loop */
+    for (k = 0; k < parameters.repeat; k++){
 
-    /* connect to all concurrent connections*/
-    for ( i = 0; i < parameters.concurrent; i++){
-        /* set event count to zero */
-        elc[i] = 0;
+        /* progress zeroing */
+        for (i = 0; i < 4096; i++){
+            parameters.progress[i] = k * 5 -1;
+        }
 
-        /* create connection - pass options (they are the same for all) */
-        z[i] = ZOOM_connection_create(o);
+        /* connect to all concurrent connections*/
+        for ( i = 0; i < parameters.concurrent; i++){
+            /* set event count to zero */
+            elc[k * parameters.concurrent + i] = 0;
+
+            /* create connection - pass options (they are the same for all) */
+            z[i] = ZOOM_connection_create(o);
+            
+            /* connect and init */
+            ZOOM_connection_connect(z[i], parameters.host, 0);
+        }
+        /* search all */
+        for (i = 0; i < parameters.concurrent; i++)
+            r[i] = ZOOM_connection_search_pqf (z[i], parameters.query);
+
+        /* network I/O. pass number of connections and array of connections */
+        while ((i = ZOOM_event (parameters.concurrent, z))){ 
+            int event = ZOOM_connection_last_event(z[i-1]);
+            const char *errmsg;
+            const char *addinfo;
+            int error = 0;
+            int progress = zoom_progress[event];
+            
+            if (event == ZOOM_EVENT_SEND_DATA || event == ZOOM_EVENT_RECV_DATA)
+                continue;
+
+            time_stamp(&time);
+
+            /* updating events and event list */
+            error = ZOOM_connection_error(z[i-1] , &errmsg, &addinfo);
+            if (error)
+                parameters.progress[i] = zoom_progress[ZOOM_EVENT_UNKNOWN];
+            //parameters.progress[i] = zoom_progress[ZOOM_EVENT_NONE];
+            else if (event == ZOOM_EVENT_CONNECT)
+                parameters.progress[i] = zoom_progress[event];
+            else
+                //parameters.progress[i] = zoom_progress[event];
+                parameters.progress[i] += 1;
+            
+            update_events(elc, els,
+                          k, i-1, 
+                          time_sec(&time), time_usec(&time), 
+                          parameters.progress[i],
+                          event, zoom_events[event], 
+                          error, errmsg);
+        }
+
+        /* destroy connections */
+        for (i = 0; i<parameters.concurrent; i++)
+            {
+                ZOOM_resultset_destroy (r[i]);
+                ZOOM_connection_destroy (z[i]);
+            }
 
-        /* connect and init */
-        ZOOM_connection_connect(z[i], parameters.host, 0);
-    }
-    /* search all */
-    for (i = 0; i < parameters.concurrent; i++)
-        r[i] = ZOOM_connection_search_pqf (z[i], parameters.query);
 
-    time_init(&time);
-    /* network I/O. pass number of connections and array of connections */
-    while ((i = ZOOM_event (parameters.concurrent, z)))
-    { 
-        int event = ZOOM_connection_last_event(z[i-1]);
-        const char *errmsg;
-        const char *addinfo;
-        int error = 0;
-        int progress = zoom_progress[event];
-        struct event_line_t el;
-        
-        if (event == ZOOM_EVENT_SEND_DATA || event == ZOOM_EVENT_RECV_DATA)
-            continue;
-
-        time_stamp(&time);
-
-        /* updating events and event list */
-        error = ZOOM_connection_error(z[i-1] , &errmsg, &addinfo);
-        if (error)
-            parameters.progress[i] = -progress;
-        else
-            parameters.progress[i] += 1;
-
-
-        //if (0){
-        // printf ("%d\t%ld.%06ld\t%d\t%d\t%s\t%d\t%s\n",
-        //update_event_line(& el,
-        //                  i-1, time_sec(&time), time_usec(&time), 
-        //                  parameters.progress[i],
-        //                  event, zoom_events[event], 
-        //                  error, errmsg);
-        //print_event_line(&el);
-        //}
-        
-        update_events(elc, els,
-                      i-1, time_sec(&time), time_usec(&time), 
-                      parameters.progress[i],
-                      event, zoom_events[event], 
-                      error, errmsg);
-        
-        
-    }
 
+    } /* for (k = 0; k < parameters.repeat; k++) repeat loop */
 
     /* output */
+
+    if (parameters.gnuplot){
+        printf("# gnuplot data and instruction file \n");
+        printf("# gnuplot thisfile \n");
+        printf("\n");
+        printf("set title \"Z39.50 connection plot\"\n");
+        printf("set xlabel \"Connection\"\n");
+        printf("set ylabel \"Time Seconds\"\n");
+        printf("set zlabel \"Progress\"\n");
+        printf("set ticslevel 0\n");
+        printf("set grid\n");
+        printf("set pm3d\n");
+        printf("splot '-' using ($1):($2):($3) t '' with points\n");
+        printf("\n");
+        printf("\n");
+    }
+    
     print_table_header();    
     print_events(elc,  els, parameters.concurrent);
     
-    /* destroy and exit */
-    for (i = 0; i<parameters.concurrent; i++)
-    {
-        ZOOM_resultset_destroy (r[i]);
-        ZOOM_connection_destroy (z[i]);
+    if (parameters.gnuplot){
+        printf("end\n");
+        printf("pause -1 \"Hit ENTER to return\"\n");
     }
+
+    /* destroy data structures and exit */
     xfree(z);
     xfree(r);
     xfree(elc);