HealthCheckCollector.py

import urllib2

import diamond.collector


AVAILABLE_METRIC = 'available'


class HealthCheckCollector(diamond.collector.Collector):

    """
    Publish 0 or 1 depending on status code of healthcheck endpoint.
    Lifted from https://github.com/BrightcoveOS/Diamond/tree/master/src/collectors/websitemonitor.
    """

    def collect(self):
        req = urllib2.Request(self.config['url'])
        try:
            resp = urllib2.urlopen(req)
            available = 1 if resp.code == 200 else 0
            # Publish metrics 20X code
            self.publish_gauge(AVAILABLE_METRIC, available)
        # urllib2 will puke on non HTTP 200/OK URLs
        except urllib2.URLError as e:
            # Publish metrics non 200 code
            self.publish_gauge(AVAILABLE_METRIC, 0)
        except IOError, e:
            self.log.error('Unable to open %s' % (self.config['url']))
        except Exception, e:
            self.log.error("Unknown error opening url: %s", e)