xref: /aosp_15_r20/external/grpc-grpc/src/ruby/end2end/sig_int_during_channel_watch_client.rb (revision cc02d7e222339f7a4f6ba5f422e6413f4bd931f2)
1#!/usr/bin/env ruby
2
3# Copyright 2015 gRPC authors.
4#
5# Licensed under the Apache License, Version 2.0 (the "License");
6# you may not use this file except in compliance with the License.
7# You may obtain a copy of the License at
8#
9#     http://www.apache.org/licenses/LICENSE-2.0
10#
11# Unless required by applicable law or agreed to in writing, software
12# distributed under the License is distributed on an "AS IS" BASIS,
13# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14# See the License for the specific language governing permissions and
15# limitations under the License.
16
17require_relative './end2end_common'
18
19# Start polling the channel state in both the main thread
20# and a child thread. Try to get the driver to send process-ending
21# interrupt while both a child thread and the main thread are in the
22# middle of a blocking connectivity_state call.
23def main
24  STDERR.puts 'sig_int_during_channel_watch_client.rb main'
25  parent_controller_port = ''
26  server_port = ''
27  OptionParser.new do |opts|
28    opts.on('--parent_controller_port=P', String) do |p|
29      parent_controller_port = p
30    end
31    opts.on('--server_port=P', String) do |p|
32      server_port = p
33    end
34  end.parse!
35
36  trap('SIGINT') { exit 0 }
37  STDERR.puts 'sig_int_during_channel_watch_client.rb: SIGINT trap has been set'
38  # notify the parent process that we're ready for signals
39  report_controller_port_to_parent(parent_controller_port, 0)
40
41  thd = Thread.new do
42    child_thread_channel = GRPC::Core::Channel.new("localhost:#{server_port}",
43                                                   {},
44                                                   :this_channel_is_insecure)
45    loop do
46      state = child_thread_channel.connectivity_state(false)
47      child_thread_channel.watch_connectivity_state(state, Time.now + 360)
48    end
49  end
50
51  main_channel = GRPC::Core::Channel.new("localhost:#{server_port}",
52                                         {},
53                                         :this_channel_is_insecure)
54  loop do
55    state = main_channel.connectivity_state(false)
56    main_channel.watch_connectivity_state(state, Time.now + 360)
57  end
58
59  thd.join
60end
61
62main
63