prometheus-mumble/mprom/server.py
David Soria Parra 0271920b60 Initial version
2016-05-24 04:39:09 +02:00

53 lines
1.4 KiB
Python

from __future__ import print_function
import sys
import prometheus_client as node
from contextlib import contextmanager
import argparse, time
import Ice, Murmur
@contextmanager
def ice_connect(host, port):
prxstr = "Meta:tcp -h %s -p %d -t 1000" % (host, port)
ic = Ice.initialize(sys.argv)
base = ic.stringToProxy(prxstr)
meta = Murmur.MetaPrx.checkedCast(base)
if not meta:
print('cannot establish connection', file=sys.stderr)
sys.exit(1)
yield meta
if ic:
ic.destroy()
def main():
parser = argparse.ArgumentParser(description='Prometheus statistics for a Mumble ICE interface')
parser.add_argument('-H', '--host', help='Host of the Ice interface', default='127.0.0.1')
parser.add_argument('-p', '--port', help='Port of the Ice interface', default=6502)
parser.add_argument('-i', '--interval', help='Interval in seconds', default=60)
args = parser.parse_args()
node.start_http_server(8001)
g = node.Gauge('users_connected', 'Number of connected users')
while True:
t1 = time.time()
print('gathering statistics')
with ice_connect(args.host, args.port) as meta:
for server in meta.getBootedServers():
g.set(len(server.getUsers()))
time_to_wait = args.interval - (time.time() - t1)
if time_to_wait > 0:
time.sleep(time_to_wait)
return 0
if __name__ == '__main__':
sys.exit(main())