1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
|
##############################################################################
# Copyright (c) 2016 ZTE Corporation and others.
#
# All rights reserved. This program and the accompanying materials
# are made available under the terms of the Apache License, Version 2.0
# which accompanies this distribution, and is available at
# http://www.apache.org/licenses/LICENSE-2.0
##############################################################################
"""
PoC of performance profiler for OPNFV doctor project
Usage:
Export environment variables to set timestamp at each checkpoint in millisecond.
Valid check points are: DOCTOR_PROFILER_T{00-09}
See also: https://goo.gl/98Osig
"""
import json
import os
OUTPUT = 'doctor_profiling_output'
PREFIX = 'DOCTOR_PROFILER'
TOTAL_CHECK_POINTS = 10
MODULE_CHECK_POINTS = ['T00', 'T01', 'T04', 'T05', 'T06', 'T09']
TAG_FORMAT = "{:<5}"
# Inspired by https://github.com/reorx/httpstat
TEMPLATE = """
Total time cost: {total}(ms)
==============================================================================>
|Monitor|Inspector |Controller|Notifier|Evaluator |
|{M00} |{M01} |{M02} |{M03} |{M04} |
| | | | | | | | | |
link down:{T00}| | | | | | | | |
raw failure:{T01}| | | | | | | |
found affected:{T02}| | | | | | |
set VM error:{T03}| | | | | |
marked host down:{T04}| | | | |
notified VM error:{T05} | | | |
transformed event:{T06}| | |
evaluated event:{T07}| |
fired alarm:{T08}|
received alarm:{T09}
"""
def main():
check_points = ["T{:02d}".format(i) for i in range(TOTAL_CHECK_POINTS)]
module_map = {"M{:02d}".format(i):
(MODULE_CHECK_POINTS[i], MODULE_CHECK_POINTS[i + 1])
for i in range(len(MODULE_CHECK_POINTS) - 1)}
# check point tags
elapsed_ms = {cp: os.getenv("{}_{}".format(PREFIX, cp))
for cp in check_points}
def format_tag(tag):
return TAG_FORMAT.format(tag or '?')
tags = {cp: format_tag(ms) for cp, ms in elapsed_ms.iteritems()}
def time_cost(cp):
if elapsed_ms[cp[0]] and elapsed_ms[cp[1]]:
return int(elapsed_ms[cp[1]]) - int(elapsed_ms[cp[0]])
else:
return None
# module time cost tags
modules_cost_ms = {module: time_cost(cp)
for module, cp in module_map.iteritems()}
tags.update({module: format_tag(cost)
for module, cost in modules_cost_ms.iteritems()})
tags.update({'total': time_cost((check_points[0], check_points[-1]))})
profile = TEMPLATE.format(**tags)
logfile = open('{}.json'.format(OUTPUT), 'w')
logfile.write(json.dumps(tags))
print profile
if __name__ == '__main__':
main()
|
