particle-os/debian-koji
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

defer message sending until after commit

Browse source
This commit is contained in:
Mike McLean 2016-05-25 16:36:35 -04:00
parent d73a0f256e
commit ec53e01547
1 changed files with 17 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

20
plugins/hub/messagebus.py
View file

@ -5,6 +5,7 @@
# Mike Bonnet <mikeb@redhat.com>
from koji import PluginError
from koji.context import context
from koji.plugin import callbacks, callback, ignore_error, convert_datetime
import ConfigParser
import logging
@ -204,9 +205,8 @@ def get_message_headers(msgtype, *args, **kws):
and c != 'postCommit'])
@ignore_error
@convert_datetime
def send_message(cbtype, *args, **kws):
def prep_message(cbtype, *args, **kws):
global config
sender = get_sender()
if cbtype.startswith('post'):
msgtype = cbtype[4:]
else:
@ -226,5 +226,19 @@ def send_message(cbtype, *args, **kws):
else:
raise PluginError('unsupported exchange type: %s' % exchange_type)
sender.send(message, sync=True, timeout=config.getfloat('broker', 'timeout'))
messages = getattr(context, 'messagebus_plugin_messages', [])
messages.append(message)
context.messagebus_plugin_messages = messages
@callback('postCommit')
@ignore_error
def send_messages(cbtype, *args, **kws):
'''Send the cached message from the other callback'''
global config
messages = getattr(context, 'messagebus_plugin_messages', [])
sender = get_sender()
for message in messages:
sender.send(message, sync=True, timeout=config.getfloat('broker', 'timeout'))
sender.close(timeout=config.getfloat('broker', 'timeout'))