[coro_rpc] allow user executor customize post op (#245)

alibaba · Mar 22, 2023 · 69235e3 · 69235e3
1 parent a4181ba
commit 69235e3
Showing 1 changed file with 14 additions and 2 deletions.
diff --git a/include/asio_util/asio_coro_util.hpp b/include/asio_util/asio_coro_util.hpp
@@ -43,13 +43,25 @@ class ExecutorWrapper : public async_simple::Executor {
   using context_t = std::remove_cvref_t<decltype(executor_.context())>;
 
   virtual bool schedule(Func func) override {
-    asio::post(executor_, std::move(func));
+    if constexpr (requires(ExecutorImpl e) { e.post(std::move(func)); }) {
+      executor_.post(std::move(func));
+    }
+    else {
+      asio::post(executor_, std::move(func));
+    }
+
     return true;
   }
 
   virtual bool checkin(Func func, void *ctx) override {
     using context_t = std::remove_cvref_t<decltype(executor_.context())>;
-    asio::post(*(context_t *)ctx, func);
+    auto &executor = *(context_t *)ctx;
+    if constexpr (requires(ExecutorImpl e) { e.post(std::move(func)); }) {
+      executor.post(std::move(func));
+    }
+    else {
+      asio::post(executor, std::move(func));
+    }
     return true;
   }
   virtual void *checkout() override { return &executor_.context(); }