sample_benchmark_std_couroutine_task_reuse_generator_no_trivial.cpp

Go to the documentation of this file.

// Copyright 2023 owent
// std coroutine no trivial task_future benchmark
 
#include <libcopp/coroutine/callable_promise.h>
#include <libcopp/coroutine/generator_promise.h>
#include <libcotask/task_promise.h>
 
#include <inttypes.h>
#include <stdint.h>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <ctime>
#include <vector>
 
#if defined(LIBCOPP_MACRO_ENABLE_STD_COROUTINE) && LIBCOPP_MACRO_ENABLE_STD_COROUTINE
 
#  ifdef LIBCOTASK_MACRO_ENABLED
 
#    if defined(PROJECT_LIBCOPP_SAMPLE_HAS_CHRONO) && PROJECT_LIBCOPP_SAMPLE_HAS_CHRONO
#      include <chrono>
#      define CALC_CLOCK_T std::chrono::system_clock::time_point
#      define CALC_CLOCK_NOW() std::chrono::system_clock::now()
#      define CALC_MS_CLOCK(x) static_cast<int>(std::chrono::duration_cast<std::chrono::milliseconds>(x).count())
#      define CALC_NS_AVG_CLOCK(x, y) \
        static_cast<long long>(std::chrono::duration_cast<std::chrono::nanoseconds>(x).count() / (y ? y : 1))
#    else
#      define CALC_CLOCK_T clock_t
#      define CALC_CLOCK_NOW() clock()
#      define CALC_MS_CLOCK(x) static_cast<int>((x) / (CLOCKS_PER_SEC / 1000))
#      define CALC_NS_AVG_CLOCK(x, y) (1000000LL * static_cast<long long>((x) / (CLOCKS_PER_SEC / 1000)) / (y ? y : 1))
#    endif
 
struct benchmark_no_trivial_message_t {
  ~benchmark_no_trivial_message_t() {}
 
  // For task
  benchmark_no_trivial_message_t() : data(0) {}
  template <class TFirst, class... TRest, typename = std::enable_if_t<!std::is_same_v<std::decay_t<TFirst>, int64_t>>>
  benchmark_no_trivial_message_t(TFirst &&, TRest &&...) : data(0) {}
 
  // For generator
  benchmark_no_trivial_message_t(int64_t input) : data(input) {}
  benchmark_no_trivial_message_t(const benchmark_no_trivial_message_t &) = default;
  benchmark_no_trivial_message_t(benchmark_no_trivial_message_t &&) = default;
  benchmark_no_trivial_message_t &operator=(const benchmark_no_trivial_message_t &) = default;
  benchmark_no_trivial_message_t &operator=(benchmark_no_trivial_message_t &&) = default;
 
  int64_t data;
};
 
LIBCOPP_COPP_NAMESPACE_BEGIN
template <>
struct std_coroutine_default_error_transform<benchmark_no_trivial_message_t> {
  using type = benchmark_no_trivial_message_t;
  type operator()(promise_status in) const { return benchmark_no_trivial_message_t{static_cast<int64_t>(in)}; }
};
LIBCOPP_COPP_NAMESPACE_END
 
using benchmark_task_future_type = cotask::task_future<benchmark_no_trivial_message_t, void>;
using benchmark_generator_future_type = copp::generator_future<benchmark_no_trivial_message_t>;
 
std::vector<benchmark_task_future_type> g_benchmark_task_list;
std::vector<benchmark_generator_future_type::context_pointer_type> g_benchmark_generator_list;
 
benchmark_task_future_type run_benchmark(size_t idx, int left_switch_count) {
  int64_t result = 0;
 
  benchmark_generator_future_type generator{[idx](benchmark_generator_future_type::context_pointer_type ctx) {
    g_benchmark_generator_list[idx] = std::move(ctx);
  }};
  while (left_switch_count-- >= 0) {
    generator.get_context()->reset_value();
    auto offset = co_await generator;
    result += offset.data;
  }
 
  co_return benchmark_no_trivial_message_t{result};
}
 
int switch_count = 100;
int max_task_number = 100000;
 
static void benchmark_round(int index) {
  g_benchmark_task_list.reserve(static_cast<size_t>(max_task_number));
  g_benchmark_generator_list.resize(static_cast<size_t>(max_task_number), nullptr);
 
  printf("### Round: %d ###\n", index);
 
  time_t begin_time = time(nullptr);
  CALC_CLOCK_T begin_clock = CALC_CLOCK_NOW();
 
  // create coroutines task
  while (g_benchmark_task_list.size() < static_cast<size_t>(max_task_number)) {
    g_benchmark_task_list.push_back(run_benchmark(g_benchmark_task_list.size(), switch_count));
  }
  for (auto &task_inst : g_benchmark_task_list) {
    task_inst.start();
  }
 
  time_t end_time = time(nullptr);
  CALC_CLOCK_T end_clock = CALC_CLOCK_NOW();
  printf("create %d task(s) and generator(s), cost time: %d s, clock time: %d ms, avg: %lld ns\n", max_task_number,
         static_cast<int>(end_time - begin_time), CALC_MS_CLOCK(end_clock - begin_clock),
         CALC_NS_AVG_CLOCK(end_clock - begin_clock, max_task_number));
 
  begin_time = end_time;
  begin_clock = end_clock;
 
  // yield & resume from runner
  bool continue_flag = true;
  long long real_switch_times = static_cast<long long>(0);
  int32_t round = 0;
 
  while (continue_flag) {
    ++round;
    continue_flag = false;
    for (auto &generator_context : g_benchmark_generator_list) {
      benchmark_generator_future_type::context_pointer_type move_context;
      move_context.swap(generator_context);
      if (move_context) {
        move_context->set_value(benchmark_no_trivial_message_t{round});
        ++real_switch_times;
        continue_flag = true;
      }
    }
  }
 
  end_time = time(nullptr);
  end_clock = CALC_CLOCK_NOW();
  printf("resume %d task(s) and generator(s) for %lld times, cost time: %d s, clock time: %d ms, avg: %lld ns\n",
         max_task_number, real_switch_times, static_cast<int>(end_time - begin_time),
         CALC_MS_CLOCK(end_clock - begin_clock), CALC_NS_AVG_CLOCK(end_clock - begin_clock, real_switch_times));
 
  begin_time = end_time;
  begin_clock = end_clock;
 
  g_benchmark_task_list.clear();
  g_benchmark_generator_list.clear();
 
  end_time = time(nullptr);
  end_clock = CALC_CLOCK_NOW();
  printf("remove %d task(s), cost time: %d s, clock time: %d ms, avg: %lld ns\n", max_task_number,
         static_cast<int>(end_time - begin_time), CALC_MS_CLOCK(end_clock - begin_clock),
         CALC_NS_AVG_CLOCK(end_clock - begin_clock, max_task_number));
}
 
int main(int argc, char *argv[]) {
  puts("###################### std task - reuse generator - no trivial ###################");
  printf("########## Cmd:");
  for (int i = 0; i < argc; ++i) {
    printf(" %s", argv[i]);
  }
  puts("");
 
  if (argc > 1) {
    max_task_number = atoi(argv[1]);
  }
 
  if (argc > 2) {
    switch_count = atoi(argv[2]);
  }
 
  for (int i = 1; i <= 5; ++i) {
    benchmark_round(i);
  }
  return 0;
}
#  else
int main() {
  puts("task_future disabled.");
  return 0;
}
#  endif
#else
int main() {
  puts("std coroutine is not supported by current compiler.");
  return 0;
}
#endif