Alexander Polcyn | 27338de | 2017-03-23 18:23:49 -0700 | [diff] [blame] | 1 | #!/usr/bin/env ruby |
| 2 | |
Jan Tattermusch | 7897ae9 | 2017-06-07 22:57:36 +0200 | [diff] [blame] | 3 | # Copyright 2016 gRPC authors. |
Alexander Polcyn | 27338de | 2017-03-23 18:23:49 -0700 | [diff] [blame] | 4 | # |
Jan Tattermusch | 7897ae9 | 2017-06-07 22:57:36 +0200 | [diff] [blame] | 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 |
Alexander Polcyn | 27338de | 2017-03-23 18:23:49 -0700 | [diff] [blame] | 8 | # |
Jan Tattermusch | 7897ae9 | 2017-06-07 22:57:36 +0200 | [diff] [blame] | 9 | # http://www.apache.org/licenses/LICENSE-2.0 |
Alexander Polcyn | 27338de | 2017-03-23 18:23:49 -0700 | [diff] [blame] | 10 | # |
Jan Tattermusch | 7897ae9 | 2017-06-07 22:57:36 +0200 | [diff] [blame] | 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. |
Alexander Polcyn | 27338de | 2017-03-23 18:23:49 -0700 | [diff] [blame] | 16 | |
| 17 | require_relative './end2end_common' |
| 18 | |
Alexander Polcyn | 077f890 | 2017-03-24 09:53:40 -0700 | [diff] [blame] | 19 | # Service that sleeps for a long time upon receiving an 'echo request' |
Alexander Polcyn | 4364ded | 2017-03-24 10:41:37 -0700 | [diff] [blame] | 20 | # Also, this notifies @call_started_cv once it has received a request. |
Alexander Polcyn | 27338de | 2017-03-23 18:23:49 -0700 | [diff] [blame] | 21 | class SleepingEchoServerImpl < Echo::EchoServer::Service |
| 22 | def initialize(call_started, call_started_mu, call_started_cv) |
| 23 | @call_started = call_started |
| 24 | @call_started_mu = call_started_mu |
| 25 | @call_started_cv = call_started_cv |
| 26 | end |
Alexander Polcyn | 077f890 | 2017-03-24 09:53:40 -0700 | [diff] [blame] | 27 | |
Alexander Polcyn | 27338de | 2017-03-23 18:23:49 -0700 | [diff] [blame] | 28 | def echo(echo_req, _) |
| 29 | @call_started_mu.synchronize do |
| 30 | @call_started.set_true |
| 31 | @call_started_cv.signal |
| 32 | end |
| 33 | sleep 1000 |
| 34 | Echo::EchoReply.new(response: echo_req.request) |
| 35 | end |
| 36 | end |
| 37 | |
Alexander Polcyn | 077f890 | 2017-03-24 09:53:40 -0700 | [diff] [blame] | 38 | # Mutable boolean |
Alexander Polcyn | 27338de | 2017-03-23 18:23:49 -0700 | [diff] [blame] | 39 | class BoolHolder |
Alexander Polcyn | 077f890 | 2017-03-24 09:53:40 -0700 | [diff] [blame] | 40 | attr_reader :val |
| 41 | |
Alexander Polcyn | 27338de | 2017-03-23 18:23:49 -0700 | [diff] [blame] | 42 | def init |
| 43 | @val = false |
| 44 | end |
Alexander Polcyn | 077f890 | 2017-03-24 09:53:40 -0700 | [diff] [blame] | 45 | |
Alexander Polcyn | 27338de | 2017-03-23 18:23:49 -0700 | [diff] [blame] | 46 | def set_true |
| 47 | @val = true |
| 48 | end |
Alexander Polcyn | 27338de | 2017-03-23 18:23:49 -0700 | [diff] [blame] | 49 | end |
| 50 | |
Alexander Polcyn | 27338de | 2017-03-23 18:23:49 -0700 | [diff] [blame] | 51 | def main |
| 52 | STDERR.puts 'start server' |
| 53 | |
| 54 | call_started = BoolHolder.new |
| 55 | call_started_mu = Mutex.new |
| 56 | call_started_cv = ConditionVariable.new |
| 57 | |
Alexander Polcyn | 077f890 | 2017-03-24 09:53:40 -0700 | [diff] [blame] | 58 | service_impl = SleepingEchoServerImpl.new(call_started, |
| 59 | call_started_mu, |
| 60 | call_started_cv) |
| 61 | server_runner = ServerRunner.new(service_impl) |
Alexander Polcyn | 27338de | 2017-03-23 18:23:49 -0700 | [diff] [blame] | 62 | server_port = server_runner.run |
| 63 | |
| 64 | STDERR.puts 'start client' |
| 65 | _, client_pid = start_client('killed_client_thread_client.rb', |
Alexander Polcyn | 077f890 | 2017-03-24 09:53:40 -0700 | [diff] [blame] | 66 | server_port) |
Alexander Polcyn | 27338de | 2017-03-23 18:23:49 -0700 | [diff] [blame] | 67 | |
| 68 | call_started_mu.synchronize do |
Alexander Polcyn | 077f890 | 2017-03-24 09:53:40 -0700 | [diff] [blame] | 69 | call_started_cv.wait(call_started_mu) until call_started.val |
Alexander Polcyn | 27338de | 2017-03-23 18:23:49 -0700 | [diff] [blame] | 70 | end |
| 71 | |
Alexander Polcyn | 2f3e588 | 2017-09-21 13:47:20 -0700 | [diff] [blame] | 72 | # SIGTERM the child process now that it's |
Alexander Polcyn | 077f890 | 2017-03-24 09:53:40 -0700 | [diff] [blame] | 73 | # in the middle of an RPC (happening on a non-main thread) |
Alexander Polcyn | 2f3e588 | 2017-09-21 13:47:20 -0700 | [diff] [blame] | 74 | Process.kill('SIGTERM', client_pid) |
Alexander Polcyn | 27338de | 2017-03-23 18:23:49 -0700 | [diff] [blame] | 75 | STDERR.puts 'sent shutdown' |
| 76 | |
| 77 | begin |
| 78 | Timeout.timeout(10) do |
| 79 | Process.wait(client_pid) |
| 80 | end |
| 81 | rescue Timeout::Error |
| 82 | STDERR.puts "timeout wait for client pid #{client_pid}" |
| 83 | Process.kill('SIGKILL', client_pid) |
| 84 | Process.wait(client_pid) |
| 85 | STDERR.puts 'killed client child' |
Alexander Polcyn | 077f890 | 2017-03-24 09:53:40 -0700 | [diff] [blame] | 86 | raise 'Timed out waiting for client process. ' \ |
| 87 | 'It likely hangs when killed while in the middle of an rpc' |
Alexander Polcyn | 27338de | 2017-03-23 18:23:49 -0700 | [diff] [blame] | 88 | end |
| 89 | |
| 90 | client_exit_code = $CHILD_STATUS |
Alexander Polcyn | 2f3e588 | 2017-09-21 13:47:20 -0700 | [diff] [blame] | 91 | if client_exit_code.termsig != 15 # SIGTERM |
| 92 | fail 'expected client exit from SIGTERM ' \ |
Alexander Polcyn | 077f890 | 2017-03-24 09:53:40 -0700 | [diff] [blame] | 93 | "but got child status: #{client_exit_code}" |
Alexander Polcyn | 27338de | 2017-03-23 18:23:49 -0700 | [diff] [blame] | 94 | end |
| 95 | |
| 96 | server_runner.stop |
| 97 | end |
| 98 | |
| 99 | main |