def build_marathon_task_dict(
marathon_task: MarathonTask) -> MutableMapping[str, Any]:
task_dict = {
"id": get_short_task_id(marathon_task.id),
"host": marathon_task.host.split(".")[0],
"port": marathon_task.ports[0],
"deployed_timestamp": marathon_task.staged_at.timestamp(),
}
if marathon_task.health_check_results:
task_dict["is_healthy"] = marathon_tools.is_task_healthy(marathon_task)
return task_dict
async def get_mesos_running_task_dict(
task: Task, num_tail_lines: int) -> MutableMapping[str, Any]:
short_hostname_future = asyncio.ensure_future(
results_or_unknown(get_short_hostname_from_task(task)))
mem_limit_future = asyncio.ensure_future(
_task_result_or_error(task.mem_limit()))
rss_future = asyncio.ensure_future(_task_result_or_error(task.rss()))
cpu_shares_future = asyncio.ensure_future(
_task_result_or_error(get_cpu_shares(task)))
cpu_time_future = asyncio.ensure_future(
_task_result_or_error(task.cpu_time()))
futures = [
short_hostname_future,
mem_limit_future,
rss_future,
cpu_shares_future,
cpu_time_future,
]
if num_tail_lines > 0:
tail_lines_future = asyncio.ensure_future(
get_tail_lines_for_mesos_task(task, get_short_task_id,
num_tail_lines))
futures.append(tail_lines_future)
else:
tail_lines_future = None
await asyncio.wait(futures)
task_dict = {
"id": get_short_task_id(task["id"]),
"hostname": short_hostname_future.result(),
"mem_limit": mem_limit_future.result(),
"rss": rss_future.result(),
"cpu_shares": cpu_shares_future.result(),
"cpu_used_seconds": cpu_time_future.result(),
"tail_lines": tail_lines_future.result() if tail_lines_future else {},
}
task_start_time = get_first_status_timestamp(task)
if task_start_time is not None:
task_dict["deployed_timestamp"] = task_start_time
current_time = int(datetime.datetime.now().strftime("%s"))
task_dict["duration_seconds"] = current_time - round(task_start_time)
return task_dict
async def get_mesos_non_running_task_dict(
task: Task, num_tail_lines: int) -> MutableMapping[str, Any]:
if num_tail_lines > 0:
tail_lines = await get_tail_lines_for_mesos_task(
task, get_short_task_id, num_tail_lines)
else:
tail_lines = {}
task_dict = {
"id": get_short_task_id(task["id"]),
"hostname": await
results_or_unknown(get_short_hostname_from_task(task)),
"state": task["state"],
"tail_lines": tail_lines,
}
task_start_time = get_first_status_timestamp(task)
if task_start_time is not None:
task_dict["deployed_timestamp"] = task_start_time
return task_dict
def test_get_short_task_id():
task_id = 'service.instance.githash.confighash.uuid'
assert marathon_serviceinit.get_short_task_id(task_id) == 'uuid'
def test_get_short_task_id():
task_id = 'service.instance.githash.confighash.uuid'
assert marathon_serviceinit.get_short_task_id(task_id) == 'uuid'
