Function Worker

deps/v8/tools/testrunner/local/pool.py:63–96

Worker to be run in a child process. The worker stops when the poison pill "STOP" is reached.

(fn, work_queue, done_queue,
           process_context_fn=None, process_context_args=None)

def Worker(fn, work_queue, done_queue,
           process_context_fn=None, process_context_args=None):
  """Worker to be run in a child process.
  The worker stops when the poison pill "STOP" is reached.
  """
  # Install a default signal handler for SIGTERM that stops the processing
  # loop below on the next occasion. The job function "fn" is supposed to
  # register their own handler to avoid blocking, but still chain to this
  # handler on SIGTERM to terminate the loop quickly.
  stop = [False]
  def handler(signum, frame):
    stop[0] = True
  signal.signal(signal.SIGTERM, handler)

  try:
    kwargs = {}
    if process_context_fn and process_context_args is not None:
      kwargs.update(process_context=process_context_fn(*process_context_args))
    for args in iter(work_queue.get, "STOP"):
      if stop[0]:
        # SIGINT, SIGTERM or internal hard timeout caught outside the execution
        # of "fn".
        break
      try:
        done_queue.put(NormalResult(fn(*args, **kwargs)))
      except AbortException:
        # SIGINT, SIGTERM or internal hard timeout caught during execution of
        # "fn".
        break
      except Exception as e:
        logging.exception('Unhandled error during worker execution.')
        done_queue.put(ExceptionResult(e))
  except KeyboardInterrupt:
    assert False, 'Unreachable'


@contextmanager