Skip to content

Commit

Permalink
make get conn state always safe to call
Browse files Browse the repository at this point in the history
  • Loading branch information
apolcyn committed May 18, 2017
1 parent c24d53b commit d7455ab
Show file tree
Hide file tree
Showing 2 changed files with 11 additions and 8 deletions.
7 changes: 7 additions & 0 deletions src/ruby/end2end/multiple_killed_watching_threads_driver.rb
Original file line number Diff line number Diff line change
Expand Up @@ -46,10 +46,17 @@ def watch_state(ch)
end

def main
channels = []
10.times do
ch = GRPC::Core::Channel.new('dummy_host',
nil, :this_channel_is_insecure)
watch_state(ch)
channels << ch
end

# checking state should still be safe to call
channels.each do |c|
fail unless c.connectivity_state(false) == FATAL_FAILURE
end
end

Expand Down
12 changes: 4 additions & 8 deletions src/ruby/ext/grpc/rb_channel.c
Original file line number Diff line number Diff line change
Expand Up @@ -273,7 +273,7 @@ static VALUE grpc_rb_channel_init(int argc, VALUE *argv, VALUE self) {
}

typedef struct get_state_stack {
grpc_channel *channel;
bg_watched_channel *bg;
int try_to_connect;
int out;
} get_state_stack;
Expand All @@ -283,14 +283,10 @@ static void *get_state_without_gil(void *arg) {

gpr_mu_lock(&global_connection_polling_mu);
GPR_ASSERT(abort_channel_polling || channel_polling_thread_started);
if (abort_channel_polling) {
// Assume that this channel has been destroyed by the
// background thread.
// The case in which the channel polling thread
// failed to start just always shows shutdown state.
if (stack->bg->channel_destroyed) {
stack->out = GRPC_CHANNEL_SHUTDOWN;
} else {
stack->out = grpc_channel_check_connectivity_state(stack->channel,
stack->out = grpc_channel_check_connectivity_state(stack->bg->channel,
stack->try_to_connect);
}
gpr_mu_unlock(&global_connection_polling_mu);
Expand Down Expand Up @@ -322,7 +318,7 @@ static VALUE grpc_rb_channel_get_connectivity_state(int argc, VALUE *argv,
return Qnil;
}

stack.channel = wrapper->bg_wrapped->channel;
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);

Expand Down

0 comments on commit d7455ab

Please sign in to comment.