def retrieve_instance_sandbox_directory(instance, job):
"""Given an instance and its parent job, determines the Mesos agent sandbox directory"""
# Check if we've simply been handed the sandbox directory
if 'sandbox_directory' in instance:
logging.debug('found sandbox directory directly on instance')
return instance['sandbox_directory']
# Try parsing it from the output url if present
if 'output_url' in instance:
output_url = instance['output_url']
url = urlparse(output_url)
query_dict = parse_qs(url.query)
if 'path' in query_dict:
path_list = query_dict['path']
if len(path_list) == 1:
logging.debug('parsed sandbox directory from output url')
return path_list[0]
# As a last resort, query the Mesos agent state
agent_url = instance_to_agent_url(instance)
resp = http.__get(f'{agent_url}/state')
if resp.status_code != 200:
logging.error(
f'mesos agent returned status code {resp.status_code} and body {resp.text}'
)
raise Exception(
'Encountered error when querying Mesos agent for the sandbox directory.'
)
# Parse the Mesos agent state and look for a matching executor
resp_json = resp.json()
frameworks = resp_json['completed_frameworks'] + resp_json['frameworks']
cook_framework = next(f for f in frameworks
if f['id'] == job['framework_id'])
cook_executors = cook_framework['completed_executors'] + cook_framework[
'executors']
instance_id = instance['task_id']
directories = [
e['directory'] for e in cook_executors if e['id'] == instance_id
]
if len(directories) == 0:
raise CookRetriableException(
f'Unable to retrieve sandbox directory for job instance {instance_id}.'
)
if len(directories) > 1:
# This should not happen, but we'll be defensive anyway
raise Exception(
f'Found more than one Mesos executor with ID {instance_id}')
return directories[0]
def download_file(instance, sandbox_dir, path):
"""Calls the Mesos agent files/download API for the given path"""
logging.info(f'downloading file from sandbox {sandbox_dir} with path {path}')
agent_url = instance_to_agent_url(instance)
params = {'path': os.path.join(sandbox_dir, path)}
resp = http.__get(f'{agent_url}/files/download', params=params, stream=True)
if resp.status_code == 404:
raise Exception(f"Cannot download '{path}' (file was not found).")
if resp.status_code != 200:
logging.error(f'mesos agent returned status code {resp.status_code} and body {resp.text}')
raise Exception('Could not download the file.')
return resp
def browse_files(instance, sandbox_dir, path):
"""
Calls the files/browse endpoint on the Mesos agent corresponding to
the given instance, and for the provided sandbox directory and path
"""
agent_url = mesos.instance_to_agent_url(instance)
resp = http.__get(f'{agent_url}/files/browse', params={'path': os.path.join(sandbox_dir, path or '')})
if resp.status_code == 404:
raise Exception(f"Cannot access '{path}' (no such file or directory).")
if resp.status_code != 200:
logging.error(f'mesos agent returned status code {resp.status_code} and body {resp.text}')
raise Exception('Encountered error when querying Mesos agent for its files.')
return resp.json()
def read_file(instance, sandbox_dir, path, offset=None, length=None):
"""Calls the Mesos agent files/read API for the given path, offset, and length"""
logging.info(f'reading file from sandbox {sandbox_dir} with path {path} at offset {offset} and length {length}')
agent_url = instance_to_agent_url(instance)
params = {'path': os.path.join(sandbox_dir, path)}
if offset is not None:
params['offset'] = offset
if length is not None:
params['length'] = length
resp = http.__get(f'{agent_url}/files/read', params=params)
if resp.status_code == 404:
raise Exception(f"Cannot open '{path}' for reading (file was not found).")
if resp.status_code != 200:
logging.error(f'mesos agent returned status code {resp.status_code} and body {resp.text}')
raise Exception('Could not read the file.')
return resp.json()
