60_activemq-monitor.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-

from xml.etree import ElementTree as ET
from itertools import chain
import time
import json
import socket
import urllib2
import requests
import commands

url = 'http://127.0.0.1:8161/admin/xml/queues.jsp'
username = ''
password = ''

def get_queues_info():
    req = urllib2.HTTPPasswordMgrWithDefaultRealm()
    req.add_password(None, url, username, password)
    handler = urllib2.HTTPBasicAuthHandler(req)
    opener = urllib2.build_opener(handler)
    urllib2.install_opener(opener)
    page = urllib2.urlopen(url,timeout=3)
    html =page.read()
    per = ET.fromstring(html)
    name=per.findall('queue')
    othercolumn=per.findall('queue/stats')
    info=list(chain(*zip(name, othercolumn)))
#    print(info)
    return info
    
def get_ip_address(key):
    if key=='ip':
        return socket.gethostbyname(socket.gethostname())
    elif key=='hostname':
        return socket.gethostname()
    elif key=='endpoint':
        endpoint = commands.getoutput('''ifconfig `route|grep '^default'|awk '{print $NF}'`|grep inet|awk '{print $2}'|awk -F ':' '{print $NF}'|head -n 1 ''')
        return endpoint

def get_xml_size():
    output = []
    qinfo=get_queues_info()
    l=len(qinfo)
    column=[('pending','size'),('consumers','consumerCount'),('enqueue','enqueueCount'),('dequeued','dequeueCount')]
#    column=[('pending','size')]
    counterType=['GAUGE','COUNTER']
    i=0
    while i<l:
        dic=dict((qinfo[i+1].attrib).items()+(qinfo[i].attrib).items())
        i=i+2
        for ctype in counterType:
            for item in column:
                t = {}
                t['endpoint'] = get_ip_address('endpoint')
                t['timestamp'] = int(time.time())
                t['step'] = 60
                t['tags'] = 'name=%s' % dic["name"]
                if ctype=='COUNTER':
                    if item[0]=='consumers' or item[0]=='pending':
                        continue
                    t['metric'] = 'activemq_%s_rate' % item[0]
                elif ctype=='GAUGE':
                    t['metric'] = 'activemq_%s' % item[0]
                t['value'] = eval(dic[item[1]])
                t['counterType'] = ctype
                output.append(t)
    return output

if __name__ == "__main__":
    d = get_xml_size()
    if d:
        print(json.dumps(d))
        #requests.post("http://127.0.0.1:2058/api/collector/push", data=json.dumps(d),timeout=5)