• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 // Copyright 2021 gRPC authors.
2 //
3 // Licensed under the Apache License, Version 2.0 (the "License");
4 // you may not use this file except in compliance with the License.
5 // You may obtain a copy of the License at
6 //
7 //     http://www.apache.org/licenses/LICENSE-2.0
8 //
9 // Unless required by applicable law or agreed to in writing, software
10 // distributed under the License is distributed on an "AS IS" BASIS,
11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 // See the License for the specific language governing permissions and
13 // limitations under the License.
14 
15 #ifndef GRPC_SRC_CORE_LIB_PROMISE_EXEC_CTX_WAKEUP_SCHEDULER_H
16 #define GRPC_SRC_CORE_LIB_PROMISE_EXEC_CTX_WAKEUP_SCHEDULER_H
17 
18 #include <grpc/support/port_platform.h>
19 
20 #include "absl/status/status.h"
21 #include "src/core/lib/iomgr/closure.h"
22 #include "src/core/lib/iomgr/error.h"
23 #include "src/core/lib/iomgr/exec_ctx.h"
24 #include "src/core/util/debug_location.h"
25 
26 namespace grpc_core {
27 
28 // A callback scheduler for activities that works by scheduling callbacks on the
29 // exec ctx.
30 class ExecCtxWakeupScheduler {
31  public:
32   template <typename ActivityType>
33   class BoundScheduler {
34    protected:
BoundScheduler(ExecCtxWakeupScheduler)35     explicit BoundScheduler(ExecCtxWakeupScheduler) {}
36     BoundScheduler(const BoundScheduler&) = delete;
37     BoundScheduler& operator=(const BoundScheduler&) = delete;
ScheduleWakeup()38     void ScheduleWakeup() {
39       GRPC_CLOSURE_INIT(
40           &closure_,
41           [](void* arg, grpc_error_handle) {
42             static_cast<ActivityType*>(arg)->RunScheduledWakeup();
43           },
44           static_cast<ActivityType*>(this), nullptr);
45       ExecCtx::Run(DEBUG_LOCATION, &closure_, absl::OkStatus());
46     }
47 
48    private:
49     grpc_closure closure_;
50   };
51 };
52 
53 }  // namespace grpc_core
54 
55 #endif  // GRPC_SRC_CORE_LIB_PROMISE_EXEC_CTX_WAKEUP_SCHEDULER_H
56