workload-production.c

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

// Create a new item for the database
static char *create_unique_item_prod(uint64_t uid, uint64_t max_uid) {
  size_t item_size;
  if (uid * 100LU / max_uid < 1) // 1%
    item_size = 100;
  else if (uid * 100LU / max_uid < 82) // 81% + 1%
    item_size = 400;
  else if (uid * 100LU / max_uid < 98)
    item_size = 1024;
  else
    item_size = 4096;

  return create_unique_item(item_size, uid);
}

static void launch_prod(struct workload *w, bench_t b) {
  declare_periodic_count;
  random_gen_t rand_next =
      (b == prod1) ? (production_random1) : (production_random2);

  uint64_t nb_requests = w->nb_requests_per_thread;
  for (size_t i = 0; i < nb_requests; i++) {
    struct slab_callback *cb = bench_cb();
    cb->item = create_unique_item_prod(rand_next(), w->nb_items_in_db);

    // 58% write 40% read 2% scan
    long random = uniform_next() % 100;
    if (random < 58) {
      kv_update_async(cb);
    } else if (random < 98) {
      kv_read_async(cb);
    } else {
      tree_scan_res_t scan_res =
          kv_init_scan(cb->item, uniform_next() % 99 + 1);
      free(cb->item);
      free(cb);
      for (size_t j = 0; j < scan_res.nb_entries; j++) {
        cb = bench_cb();
        cb->item =
            create_unique_item_prod(scan_res.hashes[j], w->nb_items_in_db);
        kv_read_async_no_lookup(cb, scan_res.entries[j].slab,
                                scan_res.entries[j].slab_idx);
      }
      free(scan_res.hashes);
      free(scan_res.entries);
    }
    periodic_count(1000, "Production Load Injector");
  }
}

/* Pretty printing */
static const char *name_prod(bench_t w) {
  switch (w) {
  case prod1:
    return "Production 1";
  case ycsb_b_uniform:
    return "Production 2";
  default:
    return "??";
  }
}

static int handles_prod(bench_t w) {
  switch (w) {
  case prod1:
  case prod2:
    return 1;
  default:
    return 0;
  }
}

static const char *api_name_prod(void) { return "PRODUCTION"; }

struct workload_api PRODUCTION = {
    .handles = handles_prod,
    .launch = launch_prod,
    .name = name_prod,
    .api_name = api_name_prod,
    .create_unique_item = create_unique_item_prod,
};