xref: /aosp_15_r20/external/grpc-grpc/src/ruby/end2end/prefork_postfork_loop_test.rb (revision cc02d7e222339f7a4f6ba5f422e6413f4bd931f2)
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 main
31  10_000.times do
32    # The prefork and postfork APIs are meant to be used before and after a
33    # fork. So this is not technically correct usage of the API. However, the
34    # current implementation doesn't actually care about a "fork" call happening
35    # in between prefork and postfork_parent, and this is unlikely to change anytime
36    # soon. Also note the goal of this test is mainly to test the background thread startup
37    # and shutdown that happens in prefork and postfork_parent. If we were to actually
38    # fork in this test, it would take much longer to run.
39    GRPC.prefork
40    GRPC.postfork_parent
41  end
42end
43
44main
45