stats-to-datadog.py

import urllib2
import json
import sys
from statsd import statsd

statsd.connect('localhost', 8125)

topology = sys.argv[1]
toporoot = sys.argv[2]
topic = sys.argv[3]

state = urllib2.urlopen(
    "http://localhost:9000/api/status?toporoot={}&topic={}".format(
        toporoot, topic
    )
).read()

data = json.loads(state)

amount = 0
for looplord in data:
    if looplord['amount'] is not None:
        statsd.gauge(
            'razor.kafkamon.topology.partition',
            amount,
            tags = [
                "topic:{}".format(topic),
                "topology:{}".format(topology),
                "partition:{}".format(looplord['partition'])
            ]
        )
        amount += looplord['amount']

print "Got {} for {}".format(amount, topology)

statsd.gauge(
    'razor.kafkamon.total_delta',
    amount, tags = [
        "topic:{}".format(topic),
        "topology:{}".format(topology)
    ]
)