carbon_collector.py

Go to the documentation of this file.

#!/usr/bin/env python

import graphitesend
import rospy
from pal_statistics_msgs.msg import Statistics

class CarbonCollector:
    def __init__(self, topics, system_name=None, dry_run=False):
        self.dry_run = dry_run
        self.gs = graphitesend.GraphiteClient(prefix='', graphite_server='localhost', \
                                              system_name=system_name, dryrun=dry_run)

        self.statistics_subs = []
        for topic in topics:
            statistics_sub = rospy.Subscriber(topic, Statistics, self.statistics_callback)
            self.statistics_subs.append(statistics_sub)

    def statistics_callback(self, msg):
        timestamp = rospy.Time(msg.header.stamp.secs, msg.header.stamp.nsecs)

        stats = []
        for stat in msg.statistics:
            stats.append((stat.name, stat.value))

        result = self.gs.send_list(stats, timestamp.to_sec())

        if self.dry_run:
            print result

        return result