src/core/lib/iomgr/event_engine/iomgr.cc

   1 // Copyright 2021 The 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 #include <grpc/support/port_platform.h>
  15
  16 #ifdef GRPC_USE_EVENT_ENGINE
  17 #include "src/core/lib/iomgr/event_engine/iomgr.h"
  18
  19 #include <grpc/event_engine/event_engine.h>
  20
  21 #include "src/core/lib/debug/trace.h"
  22 #include "src/core/lib/iomgr/closure.h"
  23 #include "src/core/lib/iomgr/event_engine/promise.h"
  24 #include "src/core/lib/iomgr/iomgr_internal.h"
  25 #include "src/core/lib/iomgr/tcp_client.h"
  26 #include "src/core/lib/iomgr/tcp_server.h"
  27 #include "src/core/lib/iomgr/timer.h"
  28 #include "src/core/lib/surface/init.h"
  29
  30 extern grpc_tcp_client_vtable grpc_event_engine_tcp_client_vtable;
  31 extern grpc_tcp_server_vtable grpc_event_engine_tcp_server_vtable;
  32 extern grpc_timer_vtable grpc_event_engine_timer_vtable;
  33 extern grpc_pollset_vtable grpc_event_engine_pollset_vtable;
  34 extern grpc_pollset_set_vtable grpc_event_engine_pollset_set_vtable;
  35 extern grpc_address_resolver_vtable grpc_event_engine_resolver_vtable;
  36
  37 // Disabled by default. grpc_polling_trace must be defined in all iomgr
  38 // implementations due to its usage in lockfree_event.
  39 grpc_core::DebugOnlyTraceFlag grpc_polling_trace(false, "polling");
  40
  41 namespace {
  42
  43 using ::grpc_event_engine::experimental::DefaultEventEngineFactory;
  44 using ::grpc_event_engine::experimental::EventEngine;
  45 using ::grpc_event_engine::experimental::Promise;
  46
  47 // Note: This is a pointer to a shared_ptr, so it's trivially destructible.
  48 std::shared_ptr<EventEngine>* g_event_engine;
  49
  50 void iomgr_platform_init(void) {
  51   g_event_engine =
  52       new std::shared_ptr<EventEngine>(DefaultEventEngineFactory());
  53 }
  54
  55 void iomgr_platform_flush(void) {}
  56
  57 void iomgr_platform_shutdown(void) {
  58   Promise<absl::Status> shutdown_status_promise;
  59   (*g_event_engine)->Shutdown([&shutdown_status_promise](absl::Status status) {
  60     shutdown_status_promise.Set(std::move(status));
  61   });
  62   auto shutdown_status = shutdown_status_promise.Get();
  63   GPR_ASSERT(shutdown_status.ok());
  64   delete g_event_engine;
  65   g_event_engine = nullptr;
  66 }
  67
  68 void iomgr_platform_shutdown_background_closure(void) {}
  69
  70 bool iomgr_platform_is_any_background_poller_thread(void) {
  71   return (*g_event_engine)->IsWorkerThread();
  72 }
  73
  74 bool iomgr_platform_add_closure_to_background_poller(
  75     grpc_closure* /* closure */, grpc_error* /* error */) {
  76   return false;
  77 }
  78
  79 grpc_iomgr_platform_vtable vtable = {
  80     iomgr_platform_init,
  81     iomgr_platform_flush,
  82     iomgr_platform_shutdown,
  83     iomgr_platform_shutdown_background_closure,
  84     iomgr_platform_is_any_background_poller_thread,
  85     iomgr_platform_add_closure_to_background_poller};
  86
  87 }  // namespace
  88
  89 void grpc_set_default_iomgr_platform() {
  90   grpc_set_tcp_client_impl(&grpc_event_engine_tcp_client_vtable);
  91   grpc_set_tcp_server_impl(&grpc_event_engine_tcp_server_vtable);
  92   grpc_set_timer_impl(&grpc_event_engine_timer_vtable);
  93   grpc_set_pollset_vtable(&grpc_event_engine_pollset_vtable);
  94   grpc_set_pollset_set_vtable(&grpc_event_engine_pollset_set_vtable);
  95   grpc_set_resolver_impl(&grpc_event_engine_resolver_vtable);
  96   grpc_set_iomgr_platform_vtable(&vtable);
  97 }
  98
  99 bool grpc_iomgr_run_in_background() { return false; }
 100
 101 grpc_event_engine::experimental::EventEngine* grpc_iomgr_event_engine() {
 102   return g_event_engine->get();
 103 }
 104
 105 #endif  // GRPC_USE_EVENT_ENGINE