main.c

#include "headers.h"
#include "workload-common.h"

int main(int argc, char **argv) {
  int nb_disks, nb_workers_per_disk;
  declare_timer;

  /* Definition of the workload, if changed you need to erase the DB before
   * relaunching */
  struct workload w = {
      .api = &YCSB, .nb_items_in_db = 100000000LU, .nb_load_injectors = 8,
      //.nb_load_injectors = 12, // For scans (see scripts/run-aws.sh and
      //OVERVIEW.md)
  };

  /* Parsing of the options */
  if (argc < 3)
    die("Usage: ./main <nb disks> <nb workers per disk>\n\tData is stored in "
        "%s\n",
        PATH);
  nb_disks = atoi(argv[1]);
  nb_workers_per_disk = atoi(argv[2]);

  /* Pretty printing useful info */
  printf("# Configuration:\n");
  printf("# \tPage cache size: %lu GB\n", PAGE_CACHE_SIZE / 1024 / 1024 / 1024);
  printf("# \tWorkers: %d working on %d disks\n",
         nb_disks * nb_workers_per_disk, nb_disks);
  printf(
      "# \tIO configuration: %d queue depth (capped: %s, extra waiting: %s)\n",
      QUEUE_DEPTH, NEVER_EXCEED_QUEUE_DEPTH ? "yes" : "no",
      WAIT_A_BIT_FOR_MORE_IOS ? "yes" : "no");
  printf("# \tQueue configuration: %d maximum pending callbaks per worker\n",
         MAX_NB_PENDING_CALLBACKS_PER_WORKER);
  printf("# \tDatastructures: %d (memory index) %d (pagecache)\n", MEMORY_INDEX,
         PAGECACHE_INDEX);
  printf("# \tThread pinning: %s\n", PINNING ? "yes" : "no");
  printf("# \tBench: %s (%lu elements)\n", w.api->api_name(), w.nb_items_in_db);

  /* Initialization of random library */
  start_timer {
    printf("Initializing random number generator (Zipf) -- this might take a "
           "while for large databases...\n");
    init_zipf_generator(0, w.nb_items_in_db -
                               1); /* This takes about 3s... not sure why, but
                                      this is legacy code :/ */
  }
  stop_timer("Initializing random number generator (Zipf)");

  /* Recover database */
  start_timer { slab_workers_init(nb_disks, nb_workers_per_disk); }
  stop_timer("Init found %lu elements", get_database_size());

  /* Add missing items if any */
  repopulate_db(&w);

  /* Launch benchs */
  bench_t workload, workloads[] = {
                        ycsb_a_uniform,
                        // ycsb_a_uniform, ycsb_b_uniform, ycsb_c_uniform,
                        // ycsb_a_zipfian, ycsb_b_zipfian, ycsb_c_zipfian,
                        // //ycsb_e_uniform, ycsb_e_zipfian, // Scans
                    };
  foreach (workload, workloads) {
    if (workload == ycsb_e_uniform || workload == ycsb_e_zipfian) {
      w.nb_requests =
          2000000LU; // requests for YCSB E are longer (scans) so we do less
    } else {
      w.nb_requests = 100000000LU;
    }
    run_workload(&w, workload);
  }
  return 0;
}