| #!/usr/bin/env python |
| # Copyright 2021 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. |
| """Manage PHP child processes for the main PHP xDS Interop client""" |
| |
| import argparse |
| import fcntl |
| import os |
| import subprocess |
| |
| # This script is being launched from src/php/bin/run_xds_client.sh |
| # to manage PHP child processes which will send 1 RPC each |
| # asynchronously. This script keeps track of all those open |
| # processes and reports back to the main PHP interop client each |
| # of the child RPCs' status code. |
| |
| if __name__ == "__main__": |
| parser = argparse.ArgumentParser() |
| parser.add_argument("--tmp_file1", nargs="?", default="") |
| parser.add_argument("--tmp_file2", nargs="?", default="") |
| parser.add_argument("--bootstrap_path", nargs="?", default="") |
| args = parser.parse_args() |
| server_address = "" |
| rpcs_started = [] |
| open_processes = {} |
| client_env = dict(os.environ) |
| client_env["GRPC_XDS_BOOTSTRAP"] = args.bootstrap_path |
| while True: |
| # tmp_file1 contains a list of RPCs (and their spec) the parent process |
| # wants executed |
| f1 = open(args.tmp_file1, "r+") |
| fcntl.flock(f1, fcntl.LOCK_EX) |
| while True: |
| key = f1.readline() |
| if not key: |
| break |
| key = key.strip() |
| if key.startswith("server_address"): |
| if not server_address: |
| server_address = key[15:] |
| elif not key in rpcs_started: |
| # format here needs to be in sync with |
| # src/php/tests/interop/xds_client.php |
| items = key.split("|") |
| num = items[0] |
| metadata = items[2] |
| timeout_sec = items[3] |
| if items[1] == "UnaryCall": |
| p = subprocess.Popen( |
| [ |
| "php", |
| "-d", |
| "extension=grpc.so", |
| "-d", |
| "extension=pthreads.so", |
| "src/php/tests/interop/xds_unary_call.php", |
| "--server=" + server_address, |
| "--num=" + str(num), |
| "--metadata=" + metadata, |
| "--timeout_sec=" + timeout_sec, |
| ], |
| env=client_env, |
| ) |
| elif items[1] == "EmptyCall": |
| p = subprocess.Popen( |
| [ |
| "php", |
| "-d", |
| "extension=grpc.so", |
| "-d", |
| "extension=pthreads.so", |
| "src/php/tests/interop/xds_empty_call.php", |
| "--server=" + server_address, |
| "--num=" + str(num), |
| "--metadata=" + metadata, |
| "--timeout_sec=" + timeout_sec, |
| ], |
| env=client_env, |
| ) |
| else: |
| continue |
| rpcs_started.append(key) |
| open_processes[key] = p |
| f1.truncate(0) |
| fcntl.flock(f1, fcntl.LOCK_UN) |
| f1.close() |
| # tmp_file2 contains the RPC result of each key received from tmp_file1 |
| f2 = open(args.tmp_file2, "a") |
| fcntl.flock(f2, fcntl.LOCK_EX) |
| keys_to_delete = [] |
| for key, process in open_processes.items(): |
| result = process.poll() |
| if result is not None: |
| # format here needs to be in sync with |
| # src/php/tests/interop/xds_client.php |
| f2.write(key + "," + str(process.returncode) + "\n") |
| keys_to_delete.append(key) |
| for key in keys_to_delete: |
| del open_processes[key] |
| fcntl.flock(f2, fcntl.LOCK_UN) |
| f2.close() |