add concept coke::IsCokeAwaitable, remove coke::GenericAwaiter<T>

The main purpose is to report compilation errors when other coroutines
are mixed with coke.

Since all awaitable objects must be movable, coke::GenericAwaiter<T>
that does not meet this condition has been removed, users should use
coke::BasicAwaiter<T> directly.

Author: kedixa

BUILD

@@ -41,7 +41,6 @@ cc_library(
         "include/coke/deque.h",
         "include/coke/fileio.h",
         "include/coke/future.h",
-        "include/coke/generic_awaiter.h",
         "include/coke/global.h",
         "include/coke/go.h",
         "include/coke/latch.h",

benchmark/BUILD

@@ -5,7 +5,7 @@ cc_library(
     hdrs = [
         "bench_common.h",
     ],
-    deps = ["//:tools"],
+    deps = ["//:tools", "//:common"],
 )
 
 create_benchmark_target("bench_exception")

benchmark/bench_common.h

@@ -26,6 +26,20 @@
 #include <vector>
 
 #include "coke/tools/option_parser.h"
+#include "coke/basic_awaiter.h"
+#include "workflow/WFTaskFactory.h"
+
+class RepeaterAwaiter : public coke::BasicAwaiter<void> {
+public:
+    RepeaterAwaiter(WFRepeaterTask *task) {
+        task->set_callback([info = this->get_info()](void *) {
+            auto *awaiter = info->get_awaiter<RepeaterAwaiter>();
+            awaiter->done();
+        });
+
+        this->set_task(task);
+    }
+};
 
 inline long long current_msec() {
     auto dur = std::chrono::steady_clock::now().time_since_epoch();

benchmark/bench_go.cpp

@@ -64,12 +64,8 @@ coke::Task<> bench_wf_go_name(int max) {
         return nullptr;
     };
 
-    coke::GenericAwaiter<void> g;
-    WFRepeaterTask *task = WFTaskFactory::create_repeater_task(creater,
-        [&g](WFRepeaterTask *) { g.done(); }
-    );
-    g.take_over(task);
-    co_await g;
+    auto *rep = WFTaskFactory::create_repeater_task(creater, nullptr);
+    co_await RepeaterAwaiter(rep);
 }
 
 coke::Task<> bench_go_name(int max) {

benchmark/bench_graph.cpp

@@ -236,12 +236,8 @@ coke::Task<> do_bench_wf(Creater &&creater) {
         return nullptr;
     };
 
-    coke::GenericAwaiter<void> g;
-    auto callback = [&g] (WFRepeaterTask *) { g.done(); };
-    auto *rep = WFTaskFactory::create_repeater_task(create, callback);
-    g.take_over(rep);
-
-    co_await g;
+    auto *rep = WFTaskFactory::create_repeater_task(create, nullptr);
+    co_await RepeaterAwaiter(rep);
 }
 
 // benchmarks

benchmark/bench_timer.cpp

@@ -58,7 +58,7 @@ bool next(long long &cur) {
 
 template<typename Awaiter>
 coke::Task<> detach(Awaiter awaiter) {
-    co_await awaiter;
+    co_await std::move(awaiter);
 }
 
 template<typename Awaiter>
@@ -72,20 +72,16 @@ coke::Task<> bench_wf_repeat() {
     std::mt19937_64 mt(current_msec());
     long long i;
 
-    coke::GenericAwaiter<void> g;
     auto create = [&] (WFRepeaterTask *) -> SubTask * {
         if (next(i)) {
             int nsec = dist(mt) * 1000;
             return WFTaskFactory::create_timer_task(0, nsec, nullptr);
         }
         return nullptr;
     };
-    auto callback = [&] (WFRepeaterTask *) { g.done(); };
 
-    auto *rep = WFTaskFactory::create_repeater_task(create, callback);
-    g.take_over(rep);
-
-    co_await g;
+    auto *rep = WFTaskFactory::create_repeater_task(create, nullptr);
+    co_await RepeaterAwaiter(rep);
 }
 
 coke::Task<> bench_default_timer() {
@@ -136,7 +132,7 @@ coke::Task<> bench_cancel_by_name() {
         name = std::to_string(i);
         auto awaiter = coke::sleep(name, microseconds(dist(mt)));
         coke::cancel_sleep_by_name(name);
-        co_await awaiter;
+        co_await std::move(awaiter);
     }
 }
 
@@ -241,7 +237,7 @@ coke::Task<> bench_cancel_by_id() {
         id = coke::get_unique_id();
         auto awaiter = coke::sleep(id, microseconds(dist(mt)));
         coke::cancel_sleep_by_id(id);
-        co_await awaiter;
+        co_await std::move(awaiter);
     }
 }
 
@@ -455,9 +451,10 @@ int main(int argc, char *argv[]) {
     DO_BENCHMARK(timer_by_addr);
     DO_BENCHMARK(cancel_by_id);
     DO_BENCHMARK(detach_by_id);
-    DO_BENCHMARK(detach3_by_id);
+    // disable this test case, it always make workflow deadlock
+    //DO_BENCHMARK(detach3_by_id);
     DO_BENCHMARK(detach_inf_by_id);
-    DO_BENCHMARK(detach3_inf_by_id);
+    //DO_BENCHMARK(detach3_inf_by_id);
     DO_BENCHMARK(one_id);
     DO_BENCHMARK(two_id);
     DO_BENCHMARK(ten_id);

include/coke/coke.h

@@ -35,7 +35,6 @@
 #include "coke/sleep.h"
 #include "coke/qps_pool.h"
 #include "coke/wait.h"
-#include "coke/generic_awaiter.h"
 #include "coke/series.h"
 #include "coke/semaphore.h"
 #include "coke/task.h"

include/coke/dag.h

@@ -82,7 +82,7 @@ class DagContextBase {
     DagContextBase(const DagContextBase &) = delete;
     ~DagContextBase() = default;
 
-    auto operator co_await() { return lt.wait(); }
+    auto wait() { return lt.wait(); }
 
     void count_down() { lt.count_down(); }
 
@@ -259,7 +259,7 @@ class DagGraph : public DagGraphBase<T> {
     Task<> run(T &data) {
         detail::DagContext<T> ctx(Base::nodes.size(), data);
         Base::start(ctx);
-        co_await ctx;
+        co_await ctx.wait();
     }
 
 private:
@@ -284,7 +284,7 @@ class DagGraph<void> : public DagGraphBase<void> {
     Task<> run() {
         detail::DagContext<void> ctx(Base::nodes.size());
         Base::start(ctx);
-        co_await ctx;
+        co_await ctx.wait();
     }
 
 private:

include/coke/detail/awaiter_base.h

@@ -37,6 +37,8 @@ class AwaiterBase {
     static void *create_series(SubTask *first);
 
 public:
+    constexpr static bool __is_coke_awaitable_type = true;
+
     AwaiterBase() = default;
 
     /**

include/coke/detail/basic_concept.h

@@ -58,6 +58,12 @@ concept Queueable = (
 template<typename T>
 concept IsCokePromise = T::__is_coke_promise_type;
 
+template<typename T>
+constexpr bool is_coke_awaitable_v = false;
+
+template<typename T>
+concept IsCokeAwaitable = T::__is_coke_awaitable_type || is_coke_awaitable_v<T>;
+
 } // namespace coke
 
 #endif // COKE_DETAIL_CONCEPT_H