1#!/usr/bin/env ruby 2# 3# Copyright 2016 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 17ENV['GRPC_ENABLE_FORK_SUPPORT'] = "1" 18fail "forking only supported on linux" unless RUBY_PLATFORM =~ /linux/ 19 20this_dir = File.expand_path(File.dirname(__FILE__)) 21protos_lib_dir = File.join(this_dir, 'lib') 22grpc_lib_dir = File.join(File.dirname(this_dir), 'lib') 23$LOAD_PATH.unshift(grpc_lib_dir) unless $LOAD_PATH.include?(grpc_lib_dir) 24$LOAD_PATH.unshift(protos_lib_dir) unless $LOAD_PATH.include?(protos_lib_dir) 25$LOAD_PATH.unshift(this_dir) unless $LOAD_PATH.include?(this_dir) 26 27require 'grpc' 28require 'end2end_common' 29 30def do_rpc(stub) 31 stub.echo(Echo::EchoRequest.new(request: 'hello'), deadline: Time.now + 300) 32end 33 34def main 35 this_dir = File.expand_path(File.dirname(__FILE__)) 36 echo_server_path = File.join(this_dir, 'echo_server.rb') 37 to_child_r, _to_child_w = IO.pipe 38 to_parent_r, to_parent_w = IO.pipe 39 # Note gRPC has not yet been initialized, otherwise we would need to call prefork 40 # before spawn and postfork_parent after. 41 # TODO(apolcyn): consider redirecting server's stderr to a file 42 Process.spawn(RbConfig.ruby, echo_server_path, in: to_child_r, out: to_parent_w, err: "server_log") 43 to_child_r.close 44 to_parent_w.close 45 child_port = to_parent_r.gets.strip 46 STDERR.puts "server running on port: #{child_port}" 47 key = "grpc." * 100_000 # large enough to malloc backing storage 48 value = "testvalue" * 100_000 49 stub = Echo::EchoServer::Stub.new("localhost:#{child_port}", :this_channel_is_insecure, channel_args: { key => value }) 50 with_logging("parent: GRPC.prefork") { GRPC.prefork } 51 pid = fork do 52 with_logging("child: GRPC.postfork_child") { GRPC.postfork_child } 53 key.clear 54 value.clear 55 GC.compact 56 with_logging("child: post-fork RPC") { do_rpc(stub) } 57 STDERR.puts "child: done" 58 end 59 with_logging("parent: GRPC.postfork_parent") { GRPC.postfork_parent } 60 Process.wait pid 61 STDERR.puts "parent: done" 62end 63 64main 65