A script to flood a Ceilometer metering queue

flood-metering-queue.py

from oslo.config import cfg

from ceilometer.openstack.common import context
from ceilometer.openstack.common import log
import ceilometer.publisher as publisher
import ceilometer.sample as sample
import ceilometer.messaging as messaging

# use with --config-file /etc/ceilometer/ceilometer.conf

cfg.CONF.register_cli_opts([cfg.IntOpt('messages',
                                       short='m',
                                       default=1,
                                       help='number of messages to send'),
                            cfg.IntOpt('samples-per-message',
                                       short='spm',
                                       default=1,
                                       help='number of samples per '
                                            'message to send'),
                            cfg.StrOpt('publisher',
                                       short='p',
                                       default='rpc://',
                                       help='publisher to use, probably '
                                            'rpc:// for Icehouse and '
                                            'notifier:// for Juno')])

cfg.CONF()

log.setup('flood-collector')
messaging.setup()
pub = publisher.get_publisher(cfg.CONF.publisher)
s = sample.Sample('test', sample.TYPE_CUMULATIVE, 'MB', 5.0,
                  '72baba3da6b9439e97fbe5334ad92b46',
                  'd3a2040216864f35ac46735d5ab6ba94',
                  'xxx', '2014-09-22T14:25:05.604471', None)
ctxt = context.RequestContext('admin', 'admin', is_admin=True)

for _ in range(cfg.CONF.messages):
        samples = [s for _ in range(cfg.CONF.samples_per_message)]
        pub.publish_samples(ctxt, samples)