/*
  This file is part of TALER
  Copyright (C) 2014-2022 Taler Systems SA

  TALER is free software; you can redistribute it and/or modify it under the
  terms of the GNU General Public License as published by the Free Software
  Foundation; either version 3, or (at your option) any later version.

  TALER is distributed in the hope that it will be useful, but WITHOUT ANY
  WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
  A PARTICULAR PURPOSE.  See the GNU General Public License for more details.

  You should have received a copy of the GNU General Public License along with
  TALER; see the file COPYING.  If not, see <http://www.gnu.org/licenses/>
*/
/**
 * @file exchangedb/test_exchangedb_by_j.c
 * @brief test cases for DB interaction functions
 * @author Joseph Xu
 */
#include "platform.h"
#include "taler_exchangedb_lib.h"
#include "taler_json_lib.h"
#include "taler_exchangedb_plugin.h"
#include "math.h"
#define ROUNDS 10

/**
 * Global result from the testcase.
 */
static int result;

/**
 * Report line of error if @a cond is true, and jump to label "drop".
 */
#define FAILIF(cond)                            \
  do {                                          \
    if (! (cond)) {break;}                    \
    GNUNET_break (0);                           \
    goto drop;                                  \
  } while (0)


/**
 * Initializes @a ptr with random data.
 */
#define RND_BLK(ptr)                                                    \
  GNUNET_CRYPTO_random_block (GNUNET_CRYPTO_QUALITY_WEAK, ptr, sizeof (*ptr))

/**
 * Initializes @a ptr with zeros.
 */
#define ZR_BLK(ptr) \
  memset (ptr, 0, sizeof (*ptr))


/**
 * Currency we use.  Must match test-exchange-db-*.conf.
 */
#define CURRENCY "EUR"

/**
 * Database plugin under test.
 */
static struct TALER_EXCHANGEDB_Plugin *plugin;


/**
 * Main function that will be run by the scheduler.
 *
 * @param cls closure with config
 */
static void
run (void *cls)
{
  static const unsigned int batches[] = {1, 2, 3, 4, 8, 16 };
  struct GNUNET_TIME_Relative times[sizeof (batches) / sizeof(*batches)];
  unsigned long long sqrs[sizeof (batches) / sizeof(*batches)];
  struct GNUNET_CONFIGURATION_Handle *cfg = cls;
  const uint32_t num_partitions = 10;

  if (NULL ==
      (plugin = TALER_EXCHANGEDB_plugin_load (cfg)))
  {
    GNUNET_break (0);
    result = 77;
    return;
  }


  if (GNUNET_OK !=
      plugin->create_tables (plugin->cls,
                             true,
                             num_partitions))
  {
    GNUNET_break (0);
    result = 77;
    goto cleanup;
  }

  memset (times, 0, sizeof (times));
  memset (sqrs, 0, sizeof (sqrs));
  for (unsigned int r = 0; r < ROUNDS; r++)
  {
    for (unsigned int i = 0; i< 6; i++)
    {
      const char *sndr = "payto://x-taler-bank/localhost:8080/1";
      struct TALER_Amount value;
      unsigned int batch_size = batches[i];
      unsigned int iterations = 16;  // 1024*10;
      struct TALER_ReservePublicKeyP reserve_pubs[iterations];
      struct GNUNET_TIME_Absolute now;
      struct GNUNET_TIME_Timestamp ts;
      struct GNUNET_TIME_Relative duration;
      struct TALER_EXCHANGEDB_ReserveInInfo reserves[iterations];
      enum GNUNET_DB_QueryStatus results[iterations];
      unsigned long long duration_sq;

      GNUNET_assert (GNUNET_OK ==
                     TALER_string_to_amount (CURRENCY ":1.000010",
                                             &value));
      now = GNUNET_TIME_absolute_get ();
      ts = GNUNET_TIME_timestamp_get ();
      for (unsigned int r = 0; r<iterations; r++)
      {
        RND_BLK (&reserve_pubs[r]);
        reserves[r].reserve_pub = &reserve_pubs[r];
        reserves[r].balance = &value;
        reserves[r].execution_time = ts;
        reserves[r].sender_account_details = sndr;
        reserves[r].exchange_account_name = "name";
        reserves[r].wire_reference = r;
      }
      FAILIF (iterations !=
              plugin->batch2_reserves_in_insert (plugin->cls,
                                                 reserves,
                                                 iterations,
                                                 batch_size,
                                                 results));
      duration = GNUNET_TIME_absolute_get_duration (now);
      times[i] = GNUNET_TIME_relative_add (times[i],
                                           duration);
      duration_sq = duration.rel_value_us * duration.rel_value_us;
      GNUNET_assert (duration_sq / duration.rel_value_us ==
                     duration.rel_value_us);
      GNUNET_assert (sqrs[i] + duration_sq >= sqrs[i]);
      sqrs[i] += duration_sq;
      fprintf (stdout,
               "for a batchsize equal to %d it took %s\n",
               batch_size,
               GNUNET_STRINGS_relative_time_to_string (duration,
                                                       GNUNET_NO) );

      system ("./test.sh");   // DELETE AFTER TIMER
    }
  }
  for (unsigned int i = 0; i< 6; i++)
  {
    struct GNUNET_TIME_Relative avg;
    double avg_dbl;
    double variance;

    avg = GNUNET_TIME_relative_divide (times[i],
                                       ROUNDS);
    avg_dbl = avg.rel_value_us;
    variance = sqrs[i] - (avg_dbl * avg_dbl * ROUNDS);
    fprintf (stdout,
             "Batch[%2u]: %8llu ± %6.0f\n",
             batches[i],
             (unsigned long long) avg.rel_value_us,
             sqrt (variance / (ROUNDS - 1)));
  }

  result = 0;
drop:
  GNUNET_break (GNUNET_OK ==
                plugin->drop_tables (plugin->cls));
cleanup:
  TALER_EXCHANGEDB_plugin_unload (plugin);
  plugin = NULL;
}


int
main (int argc,
      char *const argv[])
{
  const char *plugin_name;
  char *config_filename;
  char *testname;
  struct GNUNET_CONFIGURATION_Handle *cfg;
  (void) argc;
  result = -1;
  if (NULL == (plugin_name = strrchr (argv[0], (int) '-')))
  {
    GNUNET_break (0);
    return -1;
  }

  GNUNET_log_setup (argv[0],
                    "WARNING",
                    NULL);
  plugin_name++;
  (void) GNUNET_asprintf (&testname,
                          "test-exchange-db-%s",
                          plugin_name);
  (void) GNUNET_asprintf (&config_filename,
                          "%s.conf",
                          testname);
  fprintf (stdout,
           "Using config: %s\n",
           config_filename);
  cfg = GNUNET_CONFIGURATION_create ();
  if (GNUNET_OK !=
      GNUNET_CONFIGURATION_parse (cfg,
                                  config_filename))
  {
    GNUNET_break (0);
    GNUNET_free (config_filename);
    GNUNET_free (testname);
    return 2;
  }
  GNUNET_SCHEDULER_run (&run,
                        cfg);
  GNUNET_CONFIGURATION_destroy (cfg);
  GNUNET_free (config_filename);
  GNUNET_free (testname);
  return result;
}


/* end of test_exchangedb_by_j.c */