1 files changed, 80 insertions, 0 deletions
diff --git a/glances/exports/opentsdb/__init__.py b/glances/exports/opentsdb/__init__.py
new file mode 100644
index 00000000..1f864a80
--- /dev/null
+++ b/glances/exports/opentsdb/__init__.py
@@ -0,0 +1,80 @@
+# -*- coding: utf-8 -*-
+#
+# This file is part of Glances.
+#
+# SPDX-FileCopyrightText: 2022 Nicolas Hennion <nicolas@nicolargo.com>
+#
+# SPDX-License-Identifier: LGPL-3.0-only
+#
+
+"""OpenTSDB interface class."""
+
+import sys
+from numbers import Number
+
+from glances.logger import logger
+from glances.exports.export import GlancesExport
+
+import potsdb
+
+
+class Export(GlancesExport):
+
+    """This class manages the OpenTSDB export module."""
+
+    def __init__(self, config=None, args=None):
+        """Init the OpenTSDB export IF."""
+        super(Export, self).__init__(config=config, args=args)
+
+        # Mandatory configuration keys (additional to host and port)
+        # N/A
+
+        # Optionals configuration keys
+        self.prefix = None
+        self.tags = None
+
+        # Load the configuration file
+        self.export_enable = self.load_conf('opentsdb', mandatories=['host', 'port'], options=['prefix', 'tags'])
+        if not self.export_enable:
+            exit('Missing OPENTSDB config')
+
+        # Default prefix for stats is 'glances'
+        if self.prefix is None:
+            self.prefix = 'glances'
+
+        # Init the OpenTSDB client
+        self.client = self.init()
+
+    def init(self):
+        """Init the connection to the OpenTSDB server."""
+        if not self.export_enable:
+            return None
+
+        try:
+            db = potsdb.Client(self.host, port=int(self.port), check_host=True)
+        except Exception as e:
+            logger.critical("Cannot connect to OpenTSDB server %s:%s (%s)" % (self.host, self.port, e))
+            sys.exit(2)
+
+        return db
+
+    def export(self, name, columns, points):
+        """Export the stats to the Statsd server."""
+        for i in range(len(columns)):
+            if not isinstance(points[i], Number):
+                continue
+            stat_name = '{}.{}.{}'.format(self.prefix, name, columns[i])
+            stat_value = points[i]
+            tags = self.parse_tags(self.tags)
+            try:
+                self.client.send(stat_name, stat_value, **tags)
+            except Exception as e:
+                logger.error("Can not export stats %s to OpenTSDB (%s)" % (name, e))
+        logger.debug("Export {} stats to OpenTSDB".format(name))
+
+    def exit(self):
+        """Close the OpenTSDB export module."""
+        # Waits for all outstanding metrics to be sent and background thread closes
+        self.client.wait()
+        # Call the father method
+        super(Export, self).exit()