test/evsbench.c

   1 #define _BSD_SOURCE
   2 /*
   3  * Copyright (c) 2004 MontaVista Software, Inc.
   4  *
   5  * All rights reserved.
   6  *
   7  * Author: Steven Dake (sdake@mvista.com)
   8  *
   9  * This software licensed under BSD license, the text of which follows:
  10  *
  11  * Redistribution and use in source and binary forms, with or without
  12  * modification, are permitted provided that the following conditions are met:
  13  *
  14  * - Redistributions of source code must retain the above copyright notice,
  15  *   this list of conditions and the following disclaimer.
  16  * - Redistributions in binary form must reproduce the above copyright notice,
  17  *   this list of conditions and the following disclaimer in the documentation
  18  *   and/or other materials provided with the distribution.
  19  * - Neither the name of the MontaVista Software, Inc. nor the names of its
  20  *   contributors may be used to endorse or promote products derived from this
  21  *   software without specific prior written permission.
  22  *
  23  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
  24  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  25  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  26  * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
  27  * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  28  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  29  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  30  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  31  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  32  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
  33  * THE POSSIBILITY OF SUCH DAMAGE.
  34  */
  35
  36 #include <stdio.h>
  37 #include <stdlib.h>
  38 #include <signal.h>
  39 #include <unistd.h>
  40 #include <errno.h>
  41 #include <unistd.h>
  42 #include <time.h>
  43 #include <sys/time.h>
  44 #include <sys/types.h>
  45 #include <sys/socket.h>
  46 #include <sys/select.h>
  47 #include <sys/un.h>
  48 #include <sys/socket.h>
  49 #include <netinet/in.h>
  50 #include <arpa/inet.h>
  51
  52 #include "saAis.h"
  53 #include "evs.h"
  54
  55 #ifdef OPENAIS_SOLARIS
  56 #define timersub(a, b, result)                                          \
  57     do {                                                                \
  58         (result)->tv_sec = (a)->tv_sec - (b)->tv_sec;                   \
  59         (result)->tv_usec = (a)->tv_usec - (b)->tv_usec;                \
  60         if ((result)->tv_usec < 0) {                                    \
  61             --(result)->tv_sec;                                         \
  62             (result)->tv_usec += 1000000;                               \
  63         }                                                               \
  64     } while (0)
  65 #endif
  66
  67 volatile static int alarm_notice = 0;
  68
  69 int outstanding = 0;
  70
  71
  72
  73 void evs_deliver_fn (
  74         unsigned int nodeid,
  75         void *msg,
  76         int msg_len)
  77 {
  78         outstanding--;
  79 // printf ("Delivering message %s\n", msg);
  80 }
  81
  82 void evs_confchg_fn (
  83         unsigned int *member_list, int member_list_entries,
  84         unsigned int *left_list, int left_list_entries,
  85         unsigned int *joined_list, int joined_list_entries)
  86 {
  87         int i;
  88
  89         printf ("CONFIGURATION CHANGE\n");
  90         printf ("--------------------\n");
  91         printf ("New configuration\n");
  92         for (i = 0; i < member_list_entries; i++) {
  93                 printf ("%x\n", member_list[i]);
  94         }
  95         printf ("Members Left:\n");
  96         for (i = 0; i < left_list_entries; i++) {
  97                 printf ("%x\n", left_list[i]);
  98         }
  99         printf ("Members Joined:\n");
 100         for (i = 0; i < joined_list_entries; i++) {
 101                 printf ("%x\n", joined_list[i]);
 102         }
 103 }
 104
 105 evs_callbacks_t callbacks = {
 106         evs_deliver_fn,
 107         evs_confchg_fn
 108 };
 109
 110 struct evs_group groups[3] = {
 111         { "key1" },
 112         { "key2" },
 113         { "key3" }
 114 };
 115
 116 char buffer[200000];
 117
 118 struct iovec iov = {
 119         .iov_base = buffer,
 120         .iov_len = sizeof (buffer)
 121 };
 122
 123 void evs_benchmark (evs_handle_t handle,
 124         int write_size)
 125 {
 126         struct timeval tv1, tv2, tv_elapsed;
 127         evs_error_t result;
 128         int write_count = 0;
 129
 130         /*
 131          * Run benchmark for 10 seconds
 132          */
 133         alarm (10);
 134         gettimeofday (&tv1, NULL);
 135
 136         iov.iov_len = write_size;
 137         do {
 138                 sprintf (buffer, "This is message %d\n", write_count);
 139                 if (outstanding < 10) {
 140                         result = evs_mcast_joined (handle, EVS_TYPE_AGREED, &iov, 1);
 141
 142                         if (result != EVS_ERR_TRY_AGAIN) {
 143                                 write_count += 1;
 144                                 outstanding++;
 145                         }
 146                 }
 147                 result = evs_dispatch (handle, EVS_DISPATCH_ALL);
 148         } while (alarm_notice == 0);
 149         gettimeofday (&tv2, NULL);
 150         timersub (&tv2, &tv1, &tv_elapsed);
 151
 152         printf ("%5d Writes ", write_count);
 153         printf ("%5d bytes per write ", write_size);
 154         printf ("%7.3f Seconds runtime ",
 155                 (tv_elapsed.tv_sec + (tv_elapsed.tv_usec / 1000000.0)));
 156         printf ("%9.3f TP/s ",
 157                 ((float)write_count) /  (tv_elapsed.tv_sec + (tv_elapsed.tv_usec / 1000000.0)));
 158         printf ("%7.3f MB/s.\n",
 159                 ((float)write_count) * ((float)write_size) /  ((tv_elapsed.tv_sec + (tv_elapsed.tv_usec / 1000000.0)) * 1000000.0));
 160
 161         alarm_notice = 0;
 162 }
 163
 164 void sigalrm_handler (int num)
 165 {
 166         alarm_notice = 1;
 167 }
 168
 169 void sigintr_handler (int num)
 170 {
 171         exit (1);
 172 }
 173
 174 int main (void) {
 175         int size;
 176         int i;
 177         evs_error_t result;
 178         evs_handle_t handle;
 179
 180         signal (SIGALRM, sigalrm_handler);
 181         signal (SIGINT, sigintr_handler);
 182
 183
 184         result = evs_initialize (&handle, &callbacks);
 185         printf ("Init result %d\n", result);
 186         result = evs_join (handle, groups, 3);
 187         printf ("Join result %d\n", result);
 188         result = evs_leave (handle, &groups[0], 1);
 189         printf ("Leave result %d\n", result);
 190
 191         size = 1;
 192
 193         for (i = 0; i < 225; i++) { /* number of repetitions - up to 50k */
 194                 evs_benchmark (handle, size);
 195                 /*
 196                  * Adjust count to 95% of previous count
 197                  * Adjust bytes to write per checkpoint up by 1500
 198                  */
 199                 size += 1000;
 200         }
 201         return (0);
 202 }