src/peerstore/perf_peerstore_store.c

   1 /*
   2      This file is part of GNUnet.
   3      Copyright (C)
   4
   5      GNUnet is free software: you can redistribute it and/or modify it
   6      under the terms of the GNU Affero General Public License as published
   7      by the Free Software Foundation, either version 3 of the License,
   8      or (at your option) any later version.
   9
  10      GNUnet is distributed in the hope that it will be useful, but
  11      WITHOUT ANY WARRANTY; without even the implied warranty of
  12      MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13      Affero General Public License for more details.
  14
  15      You should have received a copy of the GNU Affero General Public License
  16      along with this program.  If not, see <http://www.gnu.org/licenses/>.
  17
  18      SPDX-License-Identifier: AGPL3.0-or-later
  19 */
  20 /**
  21  * @file peerstore/perf_peerstore_store.c
  22  * @brief performance test for peerstore store operation
  23  */
  24 #include "platform.h"
  25 #include "gnunet_util_lib.h"
  26 #include "gnunet_testing_lib.h"
  27 #include "gnunet_peerstore_service.h"
  28
  29 #define STORES 10000
  30
  31 static int ok = 1;
  32
  33 static struct GNUNET_PEERSTORE_Handle *h;
  34
  35 static char *ss = "test_peerstore_stress";
  36 static struct GNUNET_PeerIdentity p;
  37 static char *k = "test_peerstore_stress_key";
  38 static char *v = "test_peerstore_stress_val";
  39
  40 static int count = 0;
  41
  42 static void
  43 disconnect ()
  44 {
  45   if (NULL != h)
  46     GNUNET_PEERSTORE_disconnect (h, GNUNET_YES);
  47   GNUNET_SCHEDULER_shutdown ();
  48 }
  49
  50
  51 static void
  52 store ()
  53 {
  54   GNUNET_PEERSTORE_store (h, ss, &p, k, v, strlen (v) + 1,
  55                           GNUNET_TIME_UNIT_FOREVER_ABS,
  56                           (count ==
  57                            0) ? GNUNET_PEERSTORE_STOREOPTION_REPLACE :
  58                           GNUNET_PEERSTORE_STOREOPTION_MULTIPLE, NULL, NULL);
  59   count++;
  60 }
  61
  62
  63 static void
  64 watch_cb (void *cls, const struct GNUNET_PEERSTORE_Record *record,
  65           const char *emsg)
  66 {
  67   GNUNET_assert (NULL == emsg);
  68   if (STORES == count)
  69   {
  70     ok = 0;
  71     disconnect ();
  72   }
  73   else
  74     store ();
  75 }
  76
  77
  78 static void
  79 run (void *cls, const struct GNUNET_CONFIGURATION_Handle *cfg,
  80      struct GNUNET_TESTING_Peer *peer)
  81 {
  82   memset (&p, 5, sizeof (p));
  83   h = GNUNET_PEERSTORE_connect (cfg);
  84   GNUNET_assert (NULL != h);
  85   GNUNET_PEERSTORE_watch (h, ss, &p, k, &watch_cb, NULL);
  86   store ();
  87 }
  88
  89
  90 int
  91 main (int argc, char *argv[])
  92 {
  93   struct GNUNET_TIME_Absolute start;
  94   struct GNUNET_TIME_Relative diff;
  95
  96   start = GNUNET_TIME_absolute_get ();
  97   if (0 !=
  98       GNUNET_TESTING_service_run ("perf-peerstore-store", "peerstore",
  99                                   "test_peerstore_api_data.conf", &run, NULL))
 100     return 1;
 101   diff = GNUNET_TIME_absolute_get_duration (start);
 102   fprintf (stderr, "Stored and retrieved %d records in %s (%s).\n", STORES,
 103            GNUNET_STRINGS_relative_time_to_string (diff, GNUNET_YES),
 104            GNUNET_STRINGS_relative_time_to_string (diff, GNUNET_NO));
 105   return ok;
 106 }
 107
 108 /* end of perf_peerstore_store.c */