#!/usr/bin/env ruby # Copyright 2015 gRPC authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. # You may obtain a copy of the License at # # http://www.apache.org/licenses/LICENSE-2.0 # # Unless required by applicable law or agreed to in writing, software # distributed under the License is distributed on an "AS IS" BASIS, # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. require_relative './end2end_common' # Test client. Sends RPC's as normal but process also has signal handlers class SigHandlingClientController < ClientControl::ClientController::Service def initialize(srv, stub) @srv = srv @stub = stub end def do_echo_rpc(req, _) response = @stub.echo(Echo::EchoRequest.new(request: req.request)) fail 'bad response' unless response.response == req.request ClientControl::Void.new end def shutdown(_, _) # Spawn a new thread because RpcServer#stop is # synchronous and blocks until either this RPC has finished, # or the server's "poll_period" seconds have passed. @shutdown_thread = Thread.new do @srv.stop end ClientControl::Void.new end def join_shutdown_thread @shutdown_thread.join end end def main client_control_port = '' server_port = '' OptionParser.new do |opts| opts.on('--client_control_port=P', String) do |p| client_control_port = p end opts.on('--server_port=P', String) do |p| server_port = p end end.parse! Signal.trap('TERM') do STDERR.puts 'SIGTERM received' end Signal.trap('INT') do STDERR.puts 'SIGINT received' end # The "shutdown" RPC should end very quickly. # Allow a few seconds to be safe. srv = new_rpc_server_for_testing(poll_period: 3) srv.add_http2_port("0.0.0.0:#{client_control_port}", :this_port_is_insecure) stub = Echo::EchoServer::Stub.new("localhost:#{server_port}", :this_channel_is_insecure) control_service = SigHandlingClientController.new(srv, stub) srv.handle(control_service) server_thread = Thread.new do srv.run end srv.wait_till_running # send a first RPC to notify the parent process that we've started stub.echo(Echo::EchoRequest.new(request: 'client/child started')) server_thread.join control_service.join_shutdown_thread end main