passthrough.py

import argparse
import subprocess
import os

from common import DbType

def main():
  parser = argparse.ArgumentParser(description='Start Native controller instance.')
  parser.add_argument('--db', help='db to use, one of {rocksdb,redis}', default=DbType.ROCKSDB, required=False, type=DbType)
  parser.add_argument('--db_address', help='db IP address for client to connect', default=None, required=False, type=str)
  parser.add_argument('--controller_address', help='controller IP address', default="127.0.0.1", required=False, type=str)
  parser.add_argument('--controller_port', help='controller port', default="1312", required=False, type=str)
  args = parser.parse_args()

  # Open the controller process
  process_args = [os.path.join(os.path.dirname(os.path.abspath(__file__)), '../controller/build/native_controller')]
  process_args += ['--db', args.db]
  if args.db_address:
    process_args += ['--db_address', args.db_address]
  process_args += ['--controller_address', args.controller_address]
  process_args += ['--controller_port', args.controller_port]
  controller = subprocess.Popen(process_args, stdin=subprocess.PIPE, stderr=subprocess.PIPE)
  
  # Wait for the controller process to exit
  controller.wait()
  
  # Read the stderr output
  stderr_output = controller.stderr.read()
  # Decode the output if it is in bytes format
  if isinstance(stderr_output, bytes):
    stderr_output = stderr_output.decode()
  # Print the stderr output if it is not empty
  if stderr_output:
    print(stderr_output)

  return

if __name__ == "__main__":
  main()