Commit 14498387 by Edward Zarecor

whitespace

parent 830dc9cf
...@@ -19,9 +19,8 @@ Originally written by 'Jharrod LaFon' ...@@ -19,9 +19,8 @@ Originally written by 'Jharrod LaFon'
class CallbackModule(object): class CallbackModule(object):
""" """
Ansible plugin get the time of each task and total time
Ansible plugin get the time of each task and total time to run the complete playbook to run the complete playbook
""" """
def __init__(self): def __init__(self):
self.stats = {} self.stats = {}
...@@ -36,7 +35,9 @@ class CallbackModule(object): ...@@ -36,7 +35,9 @@ class CallbackModule(object):
self.datadog_api_initialized = True self.datadog_api_initialized = True
def playbook_on_play_start(self, pattern): def playbook_on_play_start(self, pattern):
self.playbook_name, _ = os.path.splitext(os.path.basename(self.play.playbook.filename)) self.playbook_name, _ = os.path.splitext(
os.path.basename(self.play.playbook.filename)
)
def playbook_on_task_start(self, name, is_conditional): def playbook_on_task_start(self, name, is_conditional):
""" """
...@@ -54,15 +55,17 @@ class CallbackModule(object): ...@@ -54,15 +55,17 @@ class CallbackModule(object):
def playbook_on_stats(self, stats): def playbook_on_stats(self, stats):
""" """
Prints the timing of each task and total time to run the complete playbook Prints the timing of each task and total time to
run the complete playbook
""" """
# Record the timing of the very last task, we use it here, because we
# Record the timing of the very last task, we use it here, because we don't have stop task function by default # don't have stop task function by default
if self.current_task is not None: if self.current_task is not None:
self.stats[self.current_task] = (time.time(), time.time() - self.stats[self.current_task]) self.stats[self.current_task] = (time.time(), time.time() - self.stats[self.current_task])
# Sort the tasks by their running time # Sort the tasks by their running time
results = sorted(self.stats.items(), key=lambda value: value[1][1], reverse=True) results = sorted(self.stats.items(),
key=lambda value: value[1][1], reverse=True)
# Total time to run the complete playbook # Total time to run the complete playbook
total_seconds = sum([x[1][1] for x in self.stats.items()]) total_seconds = sum([x[1][1] for x in self.stats.items()])
...@@ -74,13 +77,16 @@ class CallbackModule(object): ...@@ -74,13 +77,16 @@ class CallbackModule(object):
metric="edx.ansible.task_duration", metric="edx.ansible.task_duration",
date_happened=[0], date_happened=[0],
points=points[1], points=points[1],
tags=["task:{0}".format(name.replace(" | ", ".").replace(" ", "-").lower())] tags=["task:{0}".format(
name.replace(" | ", ".").replace(" ", "-").lower())]
) )
datadog.api.Metric.send( datadog.api.Metric.send(
metric="edx.ansible.play_duration", metric="edx.ansible.play_duration",
date_happened=time.time(), date_happened=time.time(),
points=total_seconds, points=total_seconds,
tags=["play:{0}".format(self.playbook_name.replace(" | ", ".").replace(" ", "-").lower())] tags=["play:{0}".format(
self.playbook_name.replace(" | ", ".").
replace(" ", "-").lower())]
) )
# Log the time of each task # Log the time of each task
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment