/usr/share/pyshared/fedmsg/commands/collectd.py is in python-fedmsg 0.7.1-1.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 | # This file is part of fedmsg.
# Copyright (C) 2012 Red Hat, Inc.
#
# fedmsg is free software; you can redistribute it and/or
# modify it under the terms of the GNU Lesser General Public
# License as published by the Free Software Foundation; either
# version 2.1 of the License, or (at your option) any later version.
#
# fedmsg is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
# Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public
# License along with fedmsg; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
#
# Authors: Ralph Bean <rbean@redhat.com>
#
import datetime
import logging
import socket
import time
import sys
import pygments
import pygments.lexers
import pygments.formatters
import fedmsg
import fedmsg.encoding
import fedmsg.meta
from fedmsg.commands import BaseCommand
from fedmsg.consumers import FedmsgConsumer
from moksha.hub.api import PollingProducer
from kitchen.iterutils import iterate
class CollectdConsumer(FedmsgConsumer):
config_key = "fedmsg.commands.collectd.enabled"
validate_messages = False
def __init__(self, hub):
self.hub = hub
# The consumer should pick up *all* messages.
self.topic = self.hub.config.get('topic_prefix', 'org.fedoraproject')
if not self.topic.endswith('*'):
self.topic += '*'
super(CollectdConsumer, self).__init__(hub)
self._dict = dict([
(p.__name__.lower(), 0) for p in fedmsg.meta.processors
])
self.host = socket.gethostname().split('.')[0]
def consume(self, msg):
processor = fedmsg.meta.msg2processor(msg, **self.hub.config)
modname = processor.__name__.lower()
self._dict[modname] += 1
def dump(self):
""" Called by CollectdProducer every `n` seconds. """
# Print out the collectd feedback.
# This is sent to stdout while other log messages are sent to stderr.
for k, v in sorted(self._dict.items()):
print self.formatter(k, v)
# Reset each entry to zero
for k, v in sorted(self._dict.items()):
self._dict[k] = 0
def formatter(self, key, value):
""" Format messages for collectd to consume. """
template = "PUTVAL {host}/fedmsg/fedmsg_wallboard-{key} " +\
"interval={interval} {timestamp}:{value}"
timestamp = int(time.time())
interval = self.hub.config['collectd_interval']
return template.format(
host=self.host,
timestamp=timestamp,
value=value,
interval=interval,
key=key,
)
class CollectdProducer(PollingProducer):
# "Frequency" is set later at runtime.
def poll(self):
self.hub.consumers[0].dump()
class CollectdCommand(BaseCommand):
""" Print machine-readable information for collectd to monitor the bus. """
name = "fedmsg-collectd"
extra_args = [
(['--collectd-interval'], {
'dest': 'collectd_interval',
'type': int,
'help': 'Number of seconds to sleep between collectd updates.',
'default': 2,
}),
]
def run(self):
# Initialize the processors before CollectdConsumer is instantiated.
fedmsg.meta.make_processors(**self.config)
# Do just like in fedmsg.commands.hub and mangle fedmsg-config.py
# to work with moksha's expected configuration.
moksha_options = dict(
mute=True, # Disable some warnings.
zmq_subscribe_endpoints=','.join(
','.join(bunch) for bunch in self.config['endpoints'].values()
),
)
self.config.update(moksha_options)
self.config[CollectdConsumer.config_key] = True
CollectdProducer.frequency = datetime.timedelta(
seconds=self.config['collectd_interval']
)
from moksha.hub import main
main(self.config, [CollectdConsumer], [CollectdProducer],
framework=False)
def collectd():
command = CollectdCommand()
command.execute()
|