grpc 1.73.0 → 1.74.1

data/src/ruby/ext/grpc/rb_channel.c

@@ -59,126 +59,26 @@ static VALUE grpc_rb_cChannel = Qnil;
 /* Used during the conversion of a hash to channel args during channel setup */
 static VALUE grpc_rb_cChannelArgs;
 
-typedef struct bg_watched_channel {
-  grpc_channel* channel;
-  // these fields must only be accessed under global_connection_polling_mu
-  struct bg_watched_channel* next;
-  int channel_destroyed;
-  int refcount;
-} bg_watched_channel;
-
 /* grpc_rb_channel wraps a grpc_channel. */
 typedef struct grpc_rb_channel {
-  VALUE credentials;
-  grpc_channel_args args;
-  /* The actual channel (protected in a wrapper to tell when it's safe to
-   * destroy) */
-  bg_watched_channel* bg_wrapped;
-} grpc_rb_channel;
-
-typedef enum { CONTINUOUS_WATCH, WATCH_STATE_API } watch_state_op_type;
-
-typedef struct watch_state_op {
-  watch_state_op_type op_type;
-  // from event.success
-  union {
-    struct {
-      int success;
-      // has been called back due to a cq next call
-      int called_back;
-    } api_callback_args;
-    struct {
-      bg_watched_channel* bg;
-    } continuous_watch_callback_args;
-  } op;
-} watch_state_op;
-
-static bg_watched_channel* bg_watched_channel_list_head = NULL;
-
-static void grpc_rb_channel_try_register_connection_polling(
-    bg_watched_channel* bg);
-static void* channel_init_try_register_connection_polling_without_gil(
-    void* arg);
-
-typedef struct channel_init_try_register_stack {
   grpc_channel* channel;
-  grpc_rb_channel* wrapper;
-} channel_init_try_register_stack;
-
-static grpc_completion_queue* g_channel_polling_cq;
-static gpr_mu global_connection_polling_mu;
-static gpr_cv global_connection_polling_cv;
-static int g_abort_channel_polling = 0;
-static gpr_once g_once_init = GPR_ONCE_INIT;
-static VALUE g_channel_polling_thread = Qnil;
-
-static int bg_watched_channel_list_lookup(bg_watched_channel* bg);
-static bg_watched_channel* bg_watched_channel_list_create_and_add(
-    grpc_channel* channel);
-static void bg_watched_channel_list_free_and_remove(bg_watched_channel* bg);
-static void run_poll_channels_loop_unblocking_func(void* arg);
-static void* run_poll_channels_loop_unblocking_func_wrapper(void* arg);
-
-// Needs to be called under global_connection_polling_mu
-static void grpc_rb_channel_watch_connection_state_op_complete(
-    watch_state_op* op, int success) {
-  GRPC_RUBY_ASSERT(!op->op.api_callback_args.called_back);
-  op->op.api_callback_args.called_back = 1;
-  op->op.api_callback_args.success = success;
-  // wake up the watch API call that's waiting on this op
-  gpr_cv_broadcast(&global_connection_polling_cv);
-}
-
-/* Avoids destroying a channel twice. */
-static void grpc_rb_channel_safe_destroy(bg_watched_channel* bg) {
-  gpr_mu_lock(&global_connection_polling_mu);
-  GRPC_RUBY_ASSERT(bg_watched_channel_list_lookup(bg));
-  if (!bg->channel_destroyed) {
-    grpc_channel_destroy(bg->channel);
-    bg->channel_destroyed = 1;
-  }
-  bg->refcount--;
-  if (bg->refcount == 0) {
-    bg_watched_channel_list_free_and_remove(bg);
-  }
-  gpr_mu_unlock(&global_connection_polling_mu);
-}
-
-static void* channel_safe_destroy_without_gil(void* arg) {
-  grpc_rb_channel_safe_destroy((bg_watched_channel*)arg);
-  return NULL;
-}
+  VALUE mark;
+} grpc_rb_channel;
 
-static void grpc_rb_channel_free_internal(void* p) {
-  grpc_rb_channel* ch = NULL;
+static void grpc_rb_channel_free(void* p) {
   if (p == NULL) {
     return;
   };
-  ch = (grpc_rb_channel*)p;
-  if (ch->bg_wrapped != NULL) {
-    /* assumption made here: it's ok to directly gpr_mu_lock the global
-     * connection polling mutex because we're in a finalizer,
-     * and we can count on this thread to not be interrupted or
-     * yield the gil. */
-    grpc_rb_channel_safe_destroy(ch->bg_wrapped);
-    grpc_rb_channel_args_destroy(&ch->args);
+  grpc_rb_channel* wrapper = (grpc_rb_channel*)p;
+  if (wrapper->channel != NULL) {
+    grpc_channel_destroy(wrapper->channel);
+    wrapper->channel = NULL;
   }
   xfree(p);
 }
 
-/* Destroys Channel instances. */
-static void grpc_rb_channel_free(void* p) { grpc_rb_channel_free_internal(p); }
-
-/* Protects the mark object from GC */
 static void grpc_rb_channel_mark(void* p) {
-  grpc_rb_channel* channel = NULL;
-  if (p == NULL) {
-    return;
-  }
-  channel = (grpc_rb_channel*)p;
-  if (channel->credentials != Qnil) {
-    rb_gc_mark(channel->credentials);
-  }
+  rb_gc_mark(((grpc_rb_channel*)p)->mark);
 }
 
 static rb_data_type_t grpc_channel_data_type = {"grpc_channel",
@@ -197,9 +97,8 @@ static rb_data_type_t grpc_channel_data_type = {"grpc_channel",
 static VALUE grpc_rb_channel_alloc(VALUE cls) {
   grpc_ruby_init();
   grpc_rb_channel* wrapper = ALLOC(grpc_rb_channel);
-  wrapper->bg_wrapped = NULL;
-  wrapper->credentials = Qnil;
-  MEMZERO(&wrapper->args, grpc_channel_args, 1);
+  wrapper->channel = NULL;
+  wrapper->mark = Qnil;
   return TypedData_Wrap_Struct(cls, &grpc_channel_data_type, wrapper);
 }
 
@@ -212,53 +111,46 @@ static VALUE grpc_rb_channel_alloc(VALUE cls) {
 
   Creates channel instances. */
 static VALUE grpc_rb_channel_init(int argc, VALUE* argv, VALUE self) {
-  VALUE channel_args = Qnil;
-  VALUE credentials = Qnil;
+  VALUE rb_channel_args = Qnil;
+  VALUE rb_credentials = Qnil;
   VALUE target = Qnil;
   grpc_rb_channel* wrapper = NULL;
-  grpc_channel* ch = NULL;
-  grpc_channel_credentials* creds = NULL;
   char* target_chars = NULL;
-  channel_init_try_register_stack stack;
-
   grpc_ruby_fork_guard();
   /* "3" == 3 mandatory args */
-  rb_scan_args(argc, argv, "3", &target, &channel_args, &credentials);
-
+  rb_scan_args(argc, argv, "3", &target, &rb_channel_args, &rb_credentials);
   TypedData_Get_Struct(self, grpc_rb_channel, &grpc_channel_data_type, wrapper);
   target_chars = StringValueCStr(target);
-  grpc_rb_hash_convert_to_channel_args(channel_args, &wrapper->args);
-  if (TYPE(credentials) == T_SYMBOL) {
-    if (id_insecure_channel != SYM2ID(credentials)) {
+  grpc_channel_args channel_args;
+  memset(&channel_args, 0, sizeof(channel_args));
+  grpc_rb_hash_convert_to_channel_args(rb_channel_args, &channel_args);
+  if (TYPE(rb_credentials) == T_SYMBOL) {
+    if (id_insecure_channel != SYM2ID(rb_credentials)) {
       rb_raise(rb_eTypeError,
                "bad creds symbol, want :this_channel_is_insecure");
       return Qnil;
     }
     grpc_channel_credentials* insecure_creds =
         grpc_insecure_credentials_create();
-    ch = grpc_channel_create(target_chars, insecure_creds, &wrapper->args);
+    wrapper->channel =
+        grpc_channel_create(target_chars, insecure_creds, &channel_args);
     grpc_channel_credentials_release(insecure_creds);
   } else {
-    wrapper->credentials = credentials;
-    if (grpc_rb_is_channel_credentials(credentials)) {
-      creds = grpc_rb_get_wrapped_channel_credentials(credentials);
-    } else if (grpc_rb_is_xds_channel_credentials(credentials)) {
-      creds = grpc_rb_get_wrapped_xds_channel_credentials(credentials);
+    grpc_channel_credentials* creds;
+    if (grpc_rb_is_channel_credentials(rb_credentials)) {
+      creds = grpc_rb_get_wrapped_channel_credentials(rb_credentials);
+    } else if (grpc_rb_is_xds_channel_credentials(rb_credentials)) {
+      creds = grpc_rb_get_wrapped_xds_channel_credentials(rb_credentials);
     } else {
       rb_raise(rb_eTypeError,
                "bad creds, want ChannelCredentials or XdsChannelCredentials");
       return Qnil;
     }
-    ch = grpc_channel_create(target_chars, creds, &wrapper->args);
+    wrapper->channel = grpc_channel_create(target_chars, creds, &channel_args);
+    wrapper->mark = rb_credentials;
   }
-
-  GRPC_RUBY_ASSERT(ch);
-  stack.channel = ch;
-  stack.wrapper = wrapper;
-  rb_thread_call_without_gvl(
-      channel_init_try_register_connection_polling_without_gil, &stack, NULL,
-      NULL);
-  if (ch == NULL) {
+  grpc_rb_channel_args_destroy(&channel_args);
+  if (wrapper->channel == NULL) {
     rb_raise(rb_eRuntimeError, "could not create an rpc channel to target:%s",
              target_chars);
     return Qnil;
@@ -268,27 +160,6 @@ static VALUE grpc_rb_channel_init(int argc, VALUE* argv, VALUE self) {
   return self;
 }
 
-typedef struct get_state_stack {
-  bg_watched_channel* bg;
-  int try_to_connect;
-  int out;
-} get_state_stack;
-
-static void* get_state_without_gil(void* arg) {
-  get_state_stack* stack = (get_state_stack*)arg;
-
-  gpr_mu_lock(&global_connection_polling_mu);
-  if (stack->bg->channel_destroyed) {
-    stack->out = GRPC_CHANNEL_SHUTDOWN;
-  } else {
-    stack->out = grpc_channel_check_connectivity_state(stack->bg->channel,
-                                                       stack->try_to_connect);
-  }
-  gpr_mu_unlock(&global_connection_polling_mu);
-
-  return NULL;
-}
-
 /*
   call-seq:
     ch.connectivity_state       -> state
@@ -302,68 +173,17 @@ static VALUE grpc_rb_channel_get_connectivity_state(int argc, VALUE* argv,
                                                     VALUE self) {
   VALUE try_to_connect_param = Qfalse;
   grpc_rb_channel* wrapper = NULL;
-  get_state_stack stack;
-
   /* "01" == 0 mandatory args, 1 (try_to_connect) is optional */
   rb_scan_args(argc, argv, "01", &try_to_connect_param);
-
   TypedData_Get_Struct(self, grpc_rb_channel, &grpc_channel_data_type, wrapper);
-  if (wrapper->bg_wrapped == NULL) {
+  if (wrapper->channel == NULL) {
     rb_raise(rb_eRuntimeError, "closed!");
     return Qnil;
   }
-
-  stack.bg = wrapper->bg_wrapped;
-  stack.try_to_connect = RTEST(try_to_connect_param) ? 1 : 0;
-  rb_thread_call_without_gvl(get_state_without_gil, &stack, NULL, NULL);
-
-  return LONG2NUM(stack.out);
-}
-
-typedef struct watch_state_stack {
-  bg_watched_channel* bg_wrapped;
-  gpr_timespec deadline;
-  int last_state;
-} watch_state_stack;
-
-static void* wait_for_watch_state_op_complete_without_gvl(void* arg) {
-  watch_state_stack* stack = (watch_state_stack*)arg;
-  watch_state_op* op = NULL;
-  void* success = (void*)0;
-
-  gpr_mu_lock(&global_connection_polling_mu);
-  // it's unsafe to do a "watch" after "channel polling abort" because the cq
-  // has been shut down.
-  if (g_abort_channel_polling || stack->bg_wrapped->channel_destroyed) {
-    gpr_mu_unlock(&global_connection_polling_mu);
-    return (void*)0;
-  }
-  op = gpr_zalloc(sizeof(watch_state_op));
-  op->op_type = WATCH_STATE_API;
-  grpc_channel_watch_connectivity_state(stack->bg_wrapped->channel,
-                                        stack->last_state, stack->deadline,
-                                        g_channel_polling_cq, op);
-
-  while (!op->op.api_callback_args.called_back) {
-    gpr_cv_wait(&global_connection_polling_cv, &global_connection_polling_mu,
-                gpr_inf_future(GPR_CLOCK_REALTIME));
-  }
-  if (op->op.api_callback_args.success) {
-    success = (void*)1;
-  }
-  gpr_free(op);
-  gpr_mu_unlock(&global_connection_polling_mu);
-
-  return success;
-}
-static void wait_for_watch_state_op_complete_unblocking_func(void* arg) {
-  bg_watched_channel* bg = (bg_watched_channel*)arg;
-  gpr_mu_lock(&global_connection_polling_mu);
-  if (!bg->channel_destroyed) {
-    grpc_channel_destroy(bg->channel);
-    bg->channel_destroyed = 1;
-  }
-  gpr_mu_unlock(&global_connection_polling_mu);
+  bool try_to_connect = RTEST(try_to_connect_param) ? true : false;
+  int state =
+      grpc_channel_check_connectivity_state(wrapper->channel, try_to_connect);
+  return LONG2NUM(state);
 }
 
 /* Wait until the channel's connectivity state becomes different from
@@ -375,79 +195,46 @@ static void wait_for_watch_state_op_complete_unblocking_func(void* arg) {
  * */
 static VALUE grpc_rb_channel_watch_connectivity_state(VALUE self,
                                                       VALUE last_state,
-                                                      VALUE deadline) {
+                                                      VALUE rb_deadline) {
   grpc_rb_channel* wrapper = NULL;
-  watch_state_stack stack;
-  void* op_success = 0;
-
   grpc_ruby_fork_guard();
   TypedData_Get_Struct(self, grpc_rb_channel, &grpc_channel_data_type, wrapper);
-
-  if (wrapper->bg_wrapped == NULL) {
+  if (wrapper->channel == NULL) {
     rb_raise(rb_eRuntimeError, "closed!");
     return Qnil;
   }
-
   if (!FIXNUM_P(last_state)) {
     rb_raise(
         rb_eTypeError,
         "bad type for last_state. want a GRPC::Core::ChannelState constant");
     return Qnil;
   }
-
-  stack.bg_wrapped = wrapper->bg_wrapped;
-  stack.deadline = grpc_rb_time_timeval(deadline, 0),
-  stack.last_state = NUM2LONG(last_state);
-
-  op_success = rb_thread_call_without_gvl(
-      wait_for_watch_state_op_complete_without_gvl, &stack,
-      wait_for_watch_state_op_complete_unblocking_func, wrapper->bg_wrapped);
-
-  return op_success ? Qtrue : Qfalse;
-}
-
-static void grpc_rb_channel_maybe_recreate_channel_after_fork(
-    grpc_rb_channel* wrapper, VALUE target) {
-  // TODO(apolcyn): maybe check if fork support is enabled here.
-  // The only way we can get bg->channel_destroyed without bg itself being
-  // NULL is if we destroyed the channel during GRPC::prefork.
-  bg_watched_channel* bg = wrapper->bg_wrapped;
-  if (bg->channel_destroyed) {
-    // There must be one ref at this point, held by the ruby-level channel
-    // object, drop this one here.
-    GRPC_RUBY_ASSERT(bg->refcount == 1);
-    rb_thread_call_without_gvl(channel_safe_destroy_without_gil, bg, NULL,
-                               NULL);
-    // re-create C-core channel
-    const char* target_str = StringValueCStr(target);
-    grpc_channel* channel;
-    if (wrapper->credentials == Qnil) {
-      grpc_channel_credentials* insecure_creds =
-          grpc_insecure_credentials_create();
-      channel = grpc_channel_create(target_str, insecure_creds, &wrapper->args);
-      grpc_channel_credentials_release(insecure_creds);
-    } else {
-      grpc_channel_credentials* creds;
-      if (grpc_rb_is_channel_credentials(wrapper->credentials)) {
-        creds = grpc_rb_get_wrapped_channel_credentials(wrapper->credentials);
-      } else if (grpc_rb_is_xds_channel_credentials(wrapper->credentials)) {
-        creds =
-            grpc_rb_get_wrapped_xds_channel_credentials(wrapper->credentials);
-      } else {
-        rb_raise(rb_eTypeError,
-                 "failed to re-create channel after fork: bad creds, want "
-                 "ChannelCredentials or XdsChannelCredentials");
-        return;
-      }
-      channel = grpc_channel_create(target_str, creds, &wrapper->args);
-    }
-    // re-register with channel polling thread
-    channel_init_try_register_stack stack;
-    stack.channel = channel;
-    stack.wrapper = wrapper;
-    rb_thread_call_without_gvl(
-        channel_init_try_register_connection_polling_without_gil, &stack, NULL,
-        NULL);
+  const void* tag = &wrapper;
+  gpr_timespec deadline = grpc_rb_time_timeval(rb_deadline, 0);
+  grpc_completion_queue* cq = grpc_completion_queue_create_for_pluck(NULL);
+  grpc_channel_watch_connectivity_state(wrapper->channel, NUM2LONG(last_state),
+                                        deadline, cq, tag);
+  grpc_event event =
+      rb_completion_queue_pluck(cq, tag, gpr_inf_future(GPR_CLOCK_REALTIME),
+                                "grpc_channel_watch_connectivity_state");
+  // TODO(apolcyn): this CQ would leak if the thread were killed
+  // while polling queue_pluck, e.g. with Thread#kill. One fix may be
+  // to make this CQ owned by the channel object. Another fix could be to
+  // busy-poll watch_connectivity_state with a short deadline, without
+  // the GIL, rather than just polling CQ pluck, and destroy the CQ
+  // before exitting the no-GIL block.
+  grpc_completion_queue_shutdown(cq);
+  grpc_rb_completion_queue_destroy(cq);
+  if (event.type == GRPC_OP_COMPLETE) {
+    return Qtrue;
+  } else if (event.type == GRPC_QUEUE_TIMEOUT) {
+    return Qfalse;
+  } else {
+    grpc_absl_log_int(
+        GPR_ERROR,
+        "GRPC_RUBY: unexpected grpc_channel_watch_connectivity_state result:",
+        event.type);
+    return Qfalse;
   }
 }
 
@@ -466,7 +253,6 @@ static VALUE grpc_rb_channel_create_call(VALUE self, VALUE parent, VALUE mask,
   grpc_slice host_slice;
   grpc_slice* host_slice_ptr = NULL;
   char* tmp_str = NULL;
-
   grpc_ruby_fork_guard();
   if (host != Qnil) {
     host_slice =
@@ -479,40 +265,29 @@ static VALUE grpc_rb_channel_create_call(VALUE self, VALUE parent, VALUE mask,
   if (parent != Qnil) {
     parent_call = grpc_rb_get_wrapped_call(parent);
   }
-
   TypedData_Get_Struct(self, grpc_rb_channel, &grpc_channel_data_type, wrapper);
-  if (wrapper->bg_wrapped == NULL) {
+  if (wrapper->channel == NULL) {
     rb_raise(rb_eRuntimeError, "closed!");
     return Qnil;
   }
-  // TODO(apolcyn): only do this check if fork support is enabled
-  rb_mutex_lock(rb_ivar_get(self, id_channel_recreation_mu));
-  grpc_rb_channel_maybe_recreate_channel_after_fork(
-      wrapper, rb_ivar_get(self, id_target));
-  rb_mutex_unlock(rb_ivar_get(self, id_channel_recreation_mu));
-
   cq = grpc_completion_queue_create_for_pluck(NULL);
   method_slice =
       grpc_slice_from_copied_buffer(RSTRING_PTR(method), RSTRING_LEN(method));
-  call = grpc_channel_create_call(wrapper->bg_wrapped->channel, parent_call,
-                                  flags, cq, method_slice, host_slice_ptr,
+  call = grpc_channel_create_call(wrapper->channel, parent_call, flags, cq,
+                                  method_slice, host_slice_ptr,
                                   grpc_rb_time_timeval(deadline,
                                                        /* absolute time */ 0),
                                   NULL);
-
   if (call == NULL) {
     tmp_str = grpc_slice_to_c_string(method_slice);
     rb_raise(rb_eRuntimeError, "cannot create call with method %s", tmp_str);
     return Qnil;
   }
-
   grpc_slice_unref(method_slice);
   if (host_slice_ptr != NULL) {
     grpc_slice_unref(host_slice);
   }
-
   res = grpc_rb_wrap_call(call, cq);
-
   /* Make this channel an instance attribute of the call so that it is not GCed
    * before the call. */
   rb_ivar_set(res, id_channel, self);
@@ -524,14 +299,11 @@ static VALUE grpc_rb_channel_create_call(VALUE self, VALUE parent, VALUE mask,
  * this */
 static VALUE grpc_rb_channel_destroy(VALUE self) {
   grpc_rb_channel* wrapper = NULL;
-
   TypedData_Get_Struct(self, grpc_rb_channel, &grpc_channel_data_type, wrapper);
-  if (wrapper->bg_wrapped != NULL) {
-    rb_thread_call_without_gvl(channel_safe_destroy_without_gil,
-                               wrapper->bg_wrapped, NULL, NULL);
-    wrapper->bg_wrapped = NULL;
+  if (wrapper->channel != NULL) {
+    grpc_channel_destroy(wrapper->channel);
+    wrapper->channel = NULL;
   }
-
   return Qnil;
 }
 
@@ -540,271 +312,13 @@ static VALUE grpc_rb_channel_get_target(VALUE self) {
   grpc_rb_channel* wrapper = NULL;
   VALUE res = Qnil;
   char* target = NULL;
-
   TypedData_Get_Struct(self, grpc_rb_channel, &grpc_channel_data_type, wrapper);
-  target = grpc_channel_get_target(wrapper->bg_wrapped->channel);
+  target = grpc_channel_get_target(wrapper->channel);
   res = rb_str_new2(target);
   gpr_free(target);
-
   return res;
 }
 
-/* Needs to be called under global_connection_polling_mu */
-static int bg_watched_channel_list_lookup(bg_watched_channel* target) {
-  bg_watched_channel* cur = bg_watched_channel_list_head;
-
-  while (cur != NULL) {
-    if (cur == target) {
-      return 1;
-    }
-    cur = cur->next;
-  }
-
-  return 0;
-}
-
-/* Needs to be called under global_connection_polling_mu */
-static bg_watched_channel* bg_watched_channel_list_create_and_add(
-    grpc_channel* channel) {
-  bg_watched_channel* watched = gpr_zalloc(sizeof(bg_watched_channel));
-
-  watched->channel = channel;
-  watched->next = bg_watched_channel_list_head;
-  watched->refcount = 1;
-  bg_watched_channel_list_head = watched;
-  return watched;
-}
-
-/* Needs to be called under global_connection_polling_mu */
-static void bg_watched_channel_list_free_and_remove(
-    bg_watched_channel* target) {
-  bg_watched_channel* bg = NULL;
-
-  GRPC_RUBY_ASSERT(bg_watched_channel_list_lookup(target));
-  GRPC_RUBY_ASSERT(target->channel_destroyed && target->refcount == 0);
-  if (bg_watched_channel_list_head == target) {
-    bg_watched_channel_list_head = target->next;
-    gpr_free(target);
-    return;
-  }
-  bg = bg_watched_channel_list_head;
-  while (bg != NULL && bg->next != NULL) {
-    if (bg->next == target) {
-      bg->next = bg->next->next;
-      gpr_free(target);
-      return;
-    }
-    bg = bg->next;
-  }
-  GRPC_RUBY_ASSERT(0);
-}
-
-/* Initialize a grpc_rb_channel's "protected grpc_channel" and try to push
- * it onto the background thread for constant watches. */
-static void* channel_init_try_register_connection_polling_without_gil(
-    void* arg) {
-  channel_init_try_register_stack* stack =
-      (channel_init_try_register_stack*)arg;
-
-  gpr_mu_lock(&global_connection_polling_mu);
-  stack->wrapper->bg_wrapped =
-      bg_watched_channel_list_create_and_add(stack->channel);
-  grpc_rb_channel_try_register_connection_polling(stack->wrapper->bg_wrapped);
-  gpr_mu_unlock(&global_connection_polling_mu);
-  return NULL;
-}
-
-// Needs to be called under global_connection_poolling_mu
-static void grpc_rb_channel_try_register_connection_polling(
-    bg_watched_channel* bg) {
-  grpc_connectivity_state conn_state;
-  watch_state_op* op = NULL;
-  if (bg->refcount == 0) {
-    GRPC_RUBY_ASSERT(bg->channel_destroyed);
-    bg_watched_channel_list_free_and_remove(bg);
-    return;
-  }
-  GRPC_RUBY_ASSERT(bg->refcount == 1);
-  if (bg->channel_destroyed || g_abort_channel_polling) {
-    return;
-  }
-  conn_state = grpc_channel_check_connectivity_state(bg->channel, 0);
-  if (conn_state == GRPC_CHANNEL_SHUTDOWN) {
-    return;
-  }
-  GRPC_RUBY_ASSERT(bg_watched_channel_list_lookup(bg));
-  // prevent bg from being free'd by GC while background thread is watching it
-  bg->refcount++;
-  op = gpr_zalloc(sizeof(watch_state_op));
-  op->op_type = CONTINUOUS_WATCH;
-  op->op.continuous_watch_callback_args.bg = bg;
-  grpc_channel_watch_connectivity_state(bg->channel, conn_state,
-                                        gpr_inf_future(GPR_CLOCK_REALTIME),
-                                        g_channel_polling_cq, op);
-}
-
-// Note this loop breaks out with a single call of
-// "run_poll_channels_loop_no_gil".
-// This assumes that a ruby call the unblocking func
-// indicates process shutdown.
-// In the worst case, this stops polling channel connectivity
-// early and falls back to current behavior.
-static void* run_poll_channels_loop_no_gil(void* arg) {
-  grpc_event event;
-  watch_state_op* op = NULL;
-  bg_watched_channel* bg = NULL;
-  (void)arg;
-  grpc_absl_log(GPR_DEBUG, "GRPC_RUBY: run_poll_channels_loop_no_gil - begin");
-
-  gpr_mu_lock(&global_connection_polling_mu);
-  gpr_cv_broadcast(&global_connection_polling_cv);
-  gpr_mu_unlock(&global_connection_polling_mu);
-
-  for (;;) {
-    event = grpc_completion_queue_next(
-        g_channel_polling_cq, gpr_inf_future(GPR_CLOCK_REALTIME), NULL);
-    if (event.type == GRPC_QUEUE_SHUTDOWN) {
-      break;
-    }
-    gpr_mu_lock(&global_connection_polling_mu);
-    if (event.type == GRPC_OP_COMPLETE) {
-      op = (watch_state_op*)event.tag;
-      if (op->op_type == CONTINUOUS_WATCH) {
-        bg = (bg_watched_channel*)op->op.continuous_watch_callback_args.bg;
-        bg->refcount--;
-        grpc_rb_channel_try_register_connection_polling(bg);
-        gpr_free(op);
-      } else if (op->op_type == WATCH_STATE_API) {
-        grpc_rb_channel_watch_connection_state_op_complete(
-            (watch_state_op*)event.tag, event.success);
-      } else {
-        GRPC_RUBY_ASSERT(0);
-      }
-    }
-    gpr_mu_unlock(&global_connection_polling_mu);
-  }
-  grpc_completion_queue_destroy(g_channel_polling_cq);
-  grpc_absl_log(
-      GPR_DEBUG,
-      "GRPC_RUBY: run_poll_channels_loop_no_gil - exit connection polling "
-      "loop");
-  return NULL;
-}
-
-static void run_poll_channels_loop_unblocking_func(void* arg) {
-  run_poll_channels_loop_unblocking_func_wrapper(arg);
-}
-
-// Notify the channel polling loop to cleanup and shutdown.
-static void* run_poll_channels_loop_unblocking_func_wrapper(void* arg) {
-  bg_watched_channel* bg = NULL;
-  (void)arg;
-
-  gpr_mu_lock(&global_connection_polling_mu);
-  grpc_absl_log(
-      GPR_DEBUG,
-      "GRPC_RUBY: run_poll_channels_loop_unblocking_func - begin aborting "
-      "connection polling");
-  // early out after first time through
-  if (g_abort_channel_polling) {
-    gpr_mu_unlock(&global_connection_polling_mu);
-    return NULL;
-  }
-  g_abort_channel_polling = 1;
-
-  // force pending watches to end by switching to shutdown state
-  bg = bg_watched_channel_list_head;
-  while (bg != NULL) {
-    if (!bg->channel_destroyed) {
-      grpc_channel_destroy(bg->channel);
-      bg->channel_destroyed = 1;
-    }
-    bg = bg->next;
-  }
-
-  grpc_absl_log_int(
-      GPR_DEBUG,
-      "GRPC_RUBY: cq shutdown on global polling cq. pid: ", getpid());
-  grpc_completion_queue_shutdown(g_channel_polling_cq);
-  gpr_cv_broadcast(&global_connection_polling_cv);
-  gpr_mu_unlock(&global_connection_polling_mu);
-  grpc_absl_log(
-      GPR_DEBUG,
-      "GRPC_RUBY: run_poll_channels_loop_unblocking_func - end aborting "
-      "connection polling");
-  return NULL;
-}
-
-// Poll channel connectivity states in background thread without the GIL.
-static VALUE run_poll_channels_loop(void* arg) {
-  (void)arg;
-  grpc_absl_log(
-      GPR_DEBUG,
-      "GRPC_RUBY: run_poll_channels_loop - create connection polling thread");
-  rb_thread_call_without_gvl(run_poll_channels_loop_no_gil, NULL,
-                             run_poll_channels_loop_unblocking_func, NULL);
-  return Qnil;
-}
-
-static void* set_abort_channel_polling_without_gil(void* arg) {
-  (void)arg;
-  gpr_mu_lock(&global_connection_polling_mu);
-  g_abort_channel_polling = 1;
-  gpr_cv_broadcast(&global_connection_polling_cv);
-  gpr_mu_unlock(&global_connection_polling_mu);
-  return NULL;
-}
-
-static void do_basic_init() {
-  gpr_mu_init(&global_connection_polling_mu);
-  gpr_cv_init(&global_connection_polling_cv);
-}
-
-/* Temporary fix for
- * https://github.com/GoogleCloudPlatform/google-cloud-ruby/issues/899.
- * Transports in idle channels can get destroyed. Normally c-core re-connects,
- * but in grpc-ruby core never gets a thread until an RPC is made, because ruby
- * only calls c-core's "completion_queu_pluck" API.
- * This uses a global background thread that calls
- * "completion_queue_next" on registered "watch_channel_connectivity_state"
- * calls - so that c-core can reconnect if needed, when there aren't any RPC's.
- * TODO(apolcyn) remove this when core handles new RPCs on dead connections.
- */
-void grpc_rb_channel_polling_thread_start() {
-  gpr_once_init(&g_once_init, do_basic_init);
-  GRPC_RUBY_ASSERT(!RTEST(g_channel_polling_thread));
-  GRPC_RUBY_ASSERT(!g_abort_channel_polling);
-  GRPC_RUBY_ASSERT(g_channel_polling_cq == NULL);
-
-  g_channel_polling_cq = grpc_completion_queue_create_for_next(NULL);
-  g_channel_polling_thread = rb_thread_create(run_poll_channels_loop, NULL);
-
-  if (!RTEST(g_channel_polling_thread)) {
-    grpc_absl_log(GPR_ERROR,
-                  "GRPC_RUBY: failed to spawn channel polling thread");
-    rb_thread_call_without_gvl(set_abort_channel_polling_without_gil, NULL,
-                               NULL, NULL);
-    return;
-  }
-}
-
-void grpc_rb_channel_polling_thread_stop() {
-  if (!RTEST(g_channel_polling_thread)) {
-    grpc_absl_log(
-        GPR_ERROR,
-        "GRPC_RUBY: channel polling thread stop: thread was not started");
-    return;
-  }
-  rb_thread_call_without_gvl(run_poll_channels_loop_unblocking_func_wrapper,
-                             NULL, NULL, NULL);
-  rb_funcall(g_channel_polling_thread, rb_intern("join"), 0);
-  // state associated with the channel polling thread is destroyed, reset so
-  // we can start again later
-  g_channel_polling_thread = Qnil;
-  g_abort_channel_polling = false;
-  g_channel_polling_cq = NULL;
-}
-
 static void Init_grpc_propagate_masks() {
   /* Constants representing call propagation masks in grpc.h */
   VALUE grpc_rb_mPropagateMasks =
@@ -838,7 +352,6 @@ static void Init_grpc_connectivity_states() {
 }
 
 void Init_grpc_channel() {
-  rb_global_variable(&g_channel_polling_thread);
   grpc_rb_cChannelArgs = rb_define_class("TmpChannelArgs", rb_cObject);
   rb_undef_alloc_func(grpc_rb_cChannelArgs);
   grpc_rb_cChannel =
@@ -883,5 +396,5 @@ void Init_grpc_channel() {
 grpc_channel* grpc_rb_get_wrapped_channel(VALUE v) {
   grpc_rb_channel* wrapper = NULL;
   TypedData_Get_Struct(v, grpc_rb_channel, &grpc_channel_data_type, wrapper);
-  return wrapper->bg_wrapped->channel;
+  return wrapper->channel;
 }