47 lines
1.6 KiB
Python
47 lines
1.6 KiB
Python
"""Provides a pre-kill method to run on macOS."""
|
|
from __future__ import print_function
|
|
|
|
# system imports
|
|
import subprocess
|
|
import sys
|
|
|
|
# third-party module imports
|
|
import six
|
|
|
|
|
|
def do_pre_kill(process_id, runner_context, output_stream, sample_time=3):
|
|
"""Samples the given process id, and puts the output to output_stream.
|
|
|
|
@param process_id the local process to sample.
|
|
|
|
@param runner_context a dictionary of details about the architectures
|
|
and platform on which the given process is running. Expected keys are
|
|
archs (array of architectures), platform_name, platform_url, and
|
|
platform_working_dir.
|
|
|
|
@param output_stream file-like object that should be used to write the
|
|
results of sampling.
|
|
|
|
@param sample_time specifies the time in seconds that should be captured.
|
|
"""
|
|
|
|
# Validate args.
|
|
if runner_context is None:
|
|
raise Exception("runner_context argument is required")
|
|
if not isinstance(runner_context, dict):
|
|
raise Exception("runner_context argument must be a dictionary")
|
|
|
|
# We will try to run sample on the local host only if there is no URL
|
|
# to a remote.
|
|
if "platform_url" in runner_context and (
|
|
runner_context["platform_url"] is not None):
|
|
import pprint
|
|
sys.stderr.write(
|
|
"warning: skipping timeout pre-kill sample invocation because we "
|
|
"don't know how to run on a remote yet. runner_context={}\n"
|
|
.format(pprint.pformat(runner_context)))
|
|
|
|
output = subprocess.check_output(['sample', six.text_type(process_id),
|
|
str(sample_time)])
|
|
output_stream.write(output)
|