47 lines
1.3 KiB
Python
47 lines
1.3 KiB
Python
|
#!/usr/bin/env python
|
||
|
|
|
||
|
|
import json
|
||
|
|
import os
|
||
|
|
import socket
|
||
|
|
import sys
|
||
|
|
import zmq
|
||
|
|
|
||
|
|
try:
|
||
|
|
service = sys.argv[1]
|
||
|
|
check_list = frozenset(sys.argv[2:])
|
||
|
|
if not check_list:
|
||
|
|
print "UNKNOWN - empty list of fedmsg consumers and producers to check"
|
||
|
|
sys.exit(3)
|
||
|
|
uninitialized_cp = False
|
||
|
|
fname = '/var/run/fedmsg/monitoring-%s.socket' % service
|
||
|
|
if not os.path.exists(fname):
|
||
|
|
print "UNKNOWN - %s does not exist" % fname
|
||
|
|
sys.exit(3)
|
||
|
|
connect_to = "ipc:///%s" % fname
|
||
|
|
ctx = zmq.Context()
|
||
|
|
s = ctx.socket(zmq.SUB)
|
||
|
|
s.connect(connect_to)
|
||
|
|
s.setsockopt(zmq.SUBSCRIBE, '')
|
||
|
|
msg = s.recv()
|
||
|
|
msg = json.loads(msg)
|
||
|
|
|
||
|
|
for consumer in msg['consumers']:
|
||
|
|
if consumer['name'] in check_list and not consumer['initialized']:
|
||
|
|
print 'ERROR: fedmsg consumer %s is not initialized' % consumer['name']
|
||
|
|
uninitialized_cp = True
|
||
|
|
|
||
|
|
for producer in msg['producers']:
|
||
|
|
if producer['name'] in check_list and not producer['initialized']:
|
||
|
|
print 'ERROR: fedmsg producer %s is not initialized' % producer['name']
|
||
|
|
uninitialized_cp = True
|
||
|
|
|
||
|
|
if uninitialized_cp:
|
||
|
|
sys.exit(2)
|
||
|
|
|
||
|
|
print "OK: fedmsg consumer(s) and producer(s) initialized"
|
||
|
|
sys.exit(0)
|
||
|
|
|
||
|
|
except Exception as err:
|
||
|
|
print "UNKNOWN:", str(err)
|
||
|
|
sys.exit(3)
|