An absurdly simple, non-blocking emitter for Square's Cube data collection framework.

emitter.py

"""
Simple, asynchronous, nonblocking UDP emitter for Cube metrics.
"""

import json
import socket
from datetime import datetime


class Emitter(object):
    def __init__(self, destination='127.0.0.1', port=1180):
        self.destination = destination
        self.port = port

    def send(self, event_type, event_data={}, **kwargs):
        if not isinstance(event_data, dict):
            event_data = {"value": event_data}
        event = dict(type=event_type, data=event_data)
        event["time"] = kwargs.get("time", datetime.utcnow().isoformat())
        if 'id' in kwargs:
            event["id"] = kwargs.get("id")
        self.socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM, 0)
        self.socket.connect((self.destination, self.port))
        self.socket.send(json.dumps(event))
        self.socket.close()