diff --git a/.gitignore b/.gitignore index 31a4ad7..263cfcc 100644 --- a/.gitignore +++ b/.gitignore @@ -108,4 +108,8 @@ dmypy.json .pyre/ # Pycharm -.idea/ \ No newline at end of file +.idea/ +.venv/ + +# config +config.json diff --git a/contrib/init/linux-systemd/ejabberd-influxdb.service b/contrib/init/linux-systemd/ejabberd-influxdb.service new file mode 100644 index 0000000..bade80f --- /dev/null +++ b/contrib/init/linux-systemd/ejabberd-influxdb.service @@ -0,0 +1,14 @@ +[Unit] +Description=ejabberd2influxdb + +[Service] +Type=simple +User=nobody +Group=nobody +ExecStart=/opt/ejabberd-metrics/influx.py +Restart=always +RestartSec=5s +Environment=PATH=/usr/bin:/usr/local/bin + +[Install] +WantedBy=multi-user.target diff --git a/ejabberdrpc.py b/ejabberdrpc.py index b2cc908..cee769b 100755 --- a/ejabberdrpc.py +++ b/ejabberdrpc.py @@ -109,7 +109,11 @@ class EjabberdMetrics(): def fetch_muc(self, vhost=None): host = "global" if vhost is not None: - host = "conference." + vhost + version = self._cmd("status", {}) + if "19.09" in version: + host = "conference." + vhost + else: + host = vhost result = self._cmd("muc_online_rooms", {"host": host}) if "rooms" in result: return len(result["rooms"]) diff --git a/influx.py b/influx.py new file mode 100644 index 0000000..34ee600 --- /dev/null +++ b/influx.py @@ -0,0 +1,91 @@ +#!/usr/bin/env python3 +# -*- coding: utf-8 -*- + +import time +import json + +from influxdb import InfluxDBClient + +from ejabberdrpc import EjabberdMetrics + + +class Influx(): + def __init__(self, metrics): + self._metrics = metrics + + @staticmethod + def _timestamp(): + return int(time.time() * 1000) + + def _parse(self, name, key, value, ts, tags=None): + output = name + + # check if tags is a dict + if isinstance(tags, dict): + + # create tag_key=tag_value pairs for all elements and append them to name + for k, v in tags.items(): + output += ",{}={}".format(k, v) + + # append key=value to name + + if " " in str(key): + key = key.replace(" ", "-") + + output += ' {}={}i {}'.format(key, value, ts) + return output + + def writeMetrics(self): + name = "ejabberd" + data = list() + + # global values + cur_ts = self._timestamp() + data.append("{m} registered={v}i {ts}".format(m=name, v=self._metrics.get_registered(), ts=cur_ts)) + data.append("{m} muc={v}i {ts}".format(m=name, v=self._metrics.get_muc(), ts=cur_ts)) + data.append("{m} s2s_in={v}i {ts}".format(m=name, v=self._metrics.get_s2s_in(), ts=cur_ts)) + data.append("{m} s2s_out={v}i {ts}".format(m=name, v=self._metrics.get_s2s_out(), ts=cur_ts)) + + # node values + for node in self._metrics.get_nodes(): + cur_ts = self._timestamp() + for k, v in self._metrics.get_online_by_status(node=node).items(): + data.append(self._parse("ejabberd_online_status", k, v, cur_ts, {"node": node})) + for k, v in self._metrics.get_online_by_client(node=node).items(): + data.append(self._parse("ejabberd_online_client", k, v, cur_ts, {"node": node})) + for k, v in self._metrics.get_online_by_ipversion(node=node).items(): + data.append(self._parse("ejabberd_online_ipversion", k, v, cur_ts, {"node": node})) + for k, v in self._metrics.get_online_by_connection(node=node).items(): + data.append(self._parse("ejabberd_online_connection", k, v, cur_ts, {"node": node})) + for k, v in self._metrics.get_online_by_vhost(node=node).items(): + data.append(self._parse("ejabberd_online_node", k, v, cur_ts, {"node": node})) + + # vhost values + # todo + + # output + client = InfluxDBClient(host='localhost', port=8086) + + # test code + client.create_database('writetest') + client_write_start_time = time.perf_counter() + + client.write_points(data, database='writetest', time_precision='ms', batch_size=10000, protocol='line') + + # test code + client_write_end_time = time.perf_counter() + print("Client Library Write: {time}s".format(time=client_write_end_time - client_write_start_time)) + + +if __name__ == "__main__": + with open("config.json", "r", encoding="utf-8") as f: + login = json.load(f) + + metrics = EjabberdMetrics("http://[::1]:4560", login) + influx = Influx(metrics) + + while True: + metrics.update() + influx.writeMetrics() + + time.sleep(10) \ No newline at end of file