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
88
89
90
91
92
93
94
95
96
97
98
99
|
# Copyright (c) 2022 Cisco and/or its affiliates.
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at:
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
"""Telemetry utility."""
from resources.libraries.python.model.ExportResult import append_telemetry
from resources.libraries.python.Constants import Constants
from resources.libraries.python.ssh import exec_cmd_no_error
from resources.libraries.python.topology import NodeType
__all__ = ["TelemetryUtil"]
class TelemetryUtil:
"""Class contains methods for telemetry utility."""
@staticmethod
def _run_telemetry(
node, profile, sid=None, spath=None, rate="", export=False):
"""Get telemetry read on node.
:param node: Node in the topology.
:param profile: Telemetry configuration profile.
:param sid: Socket ID used to describe recipient side of socket.
:param spath: Socket path.
:param rate: Telemetry load, unique within the test (optional).
:param export: If false, do not attempt JSON export (default false).
:type node: dict
:type profile: str
:type sid: str
:type spath: str
:type rate: str
:type export: bool
"""
config = ""
config += f"{Constants.REMOTE_FW_DIR}/"
config += f"{Constants.RESOURCES_TPL_TELEMETRY}/"
config += f"{profile}"
cd_cmd = ""
cd_cmd += f"sh -c \"cd {Constants.REMOTE_FW_DIR}/"
cd_cmd += f"{Constants.RESOURCES_TOOLS}"
if spath:
bin_cmd = f"python3 -m telemetry --config {config} --hook {spath}\""
else:
bin_cmd = f"python3 -m telemetry --config {config}\""
exec_cmd_no_error(node, f"{cd_cmd} && {bin_cmd}", sudo=True)
if not export:
return
hostname = exec_cmd_no_error(node, "hostname")[0].strip()
stdout, _ = exec_cmd_no_error(
node, "cat /tmp/metric.prom", sudo=True, log_stdout_err=False
)
prefix = "{"
prefix += f"hostname=\"{hostname}\","
if sid:
prefix += f"hook=\"{sid}\","
prefix += f"rate=\"{rate}\","
for line in stdout.splitlines():
if line and not line.startswith("#"):
append_telemetry(
prefix.join(line.rsplit("{", 1)).replace("\"", "'")
)
def run_telemetry_on_all_duts(self, nodes, profile, rate="", export=False):
"""Get telemetry read on all DUTs.
:param nodes: Nodes in the topology.
:param profile: Telemetry configuration profile.
:param rate: Telemetry load, unique within the test (optional).
:param export: If false, do not attempt JSON export (default false).
:type nodes: dict
:type profile: str
:type rate: str
:type export: bool
"""
for node in nodes.values():
if node["type"] == NodeType.DUT:
try:
for sid, spath in node["sockets"]["CLI"].items():
self._run_telemetry(
node, profile=profile, sid=sid, spath=spath,
rate=rate, export=export
)
except IndexError:
pass
|
