diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-21 11:54:28 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-21 11:54:28 +0000 |
commit | e6918187568dbd01842d8d1d2c808ce16a894239 (patch) | |
tree | 64f88b554b444a49f656b6c656111a145cbbaa28 /src/pybind/mgr/telegraf/protocol.py | |
parent | Initial commit. (diff) | |
download | ceph-e6918187568dbd01842d8d1d2c808ce16a894239.tar.xz ceph-e6918187568dbd01842d8d1d2c808ce16a894239.zip |
Adding upstream version 18.2.2.upstream/18.2.2
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'src/pybind/mgr/telegraf/protocol.py')
-rw-r--r-- | src/pybind/mgr/telegraf/protocol.py | 50 |
1 files changed, 50 insertions, 0 deletions
diff --git a/src/pybind/mgr/telegraf/protocol.py b/src/pybind/mgr/telegraf/protocol.py new file mode 100644 index 000000000..7cf8bbe9e --- /dev/null +++ b/src/pybind/mgr/telegraf/protocol.py @@ -0,0 +1,50 @@ +from typing import Dict, Optional, Union + +from telegraf.utils import format_string, format_value, ValueType + + +class Line(object): + def __init__(self, + measurement: ValueType, + values: Union[Dict[str, ValueType], ValueType], + tags: Optional[Dict[str, str]] = None, + timestamp: Optional[int] = None) -> None: + self.measurement = measurement + self.values = values + self.tags = tags + self.timestamp = timestamp + + def get_output_measurement(self) -> str: + return format_string(self.measurement) + + def get_output_values(self) -> str: + if not isinstance(self.values, dict): + metric_values = {'value': self.values} + else: + metric_values = self.values + + sorted_values = sorted(metric_values.items()) + sorted_values = [(k, v) for k, v in sorted_values if v is not None] + + return ','.join('{0}={1}'.format(format_string(k), format_value(v)) for k, v in sorted_values) + + def get_output_tags(self) -> str: + if not self.tags: + self.tags = dict() + + sorted_tags = sorted(self.tags.items()) + + return ','.join('{0}={1}'.format(format_string(k), format_string(v)) for k, v in sorted_tags) + + def get_output_timestamp(self) -> str: + return ' {0}'.format(self.timestamp) if self.timestamp else '' + + def to_line_protocol(self) -> str: + tags = self.get_output_tags() + + return '{0}{1} {2}{3}'.format( + self.get_output_measurement(), + "," + tags if tags else '', + self.get_output_values(), + self.get_output_timestamp() + ) |