mirror of
https://github.com/alerta/alerta-contrib.git
synced 2025-03-17 05:52:38 +00:00
54 lines
1.7 KiB
Python
54 lines
1.7 KiB
Python
import os
|
|
import logging
|
|
|
|
from kombu import BrokerConnection, Exchange, Producer
|
|
from kombu.utils.debug import setup_logging
|
|
|
|
from alerta.app import app
|
|
from alerta.plugins import PluginBase
|
|
|
|
LOG = logging.getLogger('alerta.plugins.amqp')
|
|
|
|
DEFAULT_AMQP_URL = 'mongodb://localhost:27017/kombu'
|
|
DEFAULT_AMQP_TOPIC = 'notify'
|
|
|
|
AMQP_URL = os.environ.get('REDIS_URL') or os.environ.get('AMQP_URL') or app.config.get('AMQP_URL', DEFAULT_AMQP_URL)
|
|
AMQP_TOPIC = os.environ.get('AMQP_TOPIC') or app.config.get('AMQP_TOPIC', DEFAULT_AMQP_TOPIC)
|
|
|
|
|
|
class FanoutPublisher(PluginBase):
|
|
|
|
def __init__(self, name=None):
|
|
|
|
if app.debug:
|
|
setup_logging(loglevel='DEBUG', loggers=[''])
|
|
|
|
self.connection = BrokerConnection(AMQP_URL)
|
|
try:
|
|
self.connection.connect()
|
|
except Exception as e:
|
|
LOG.error('Failed to connect to AMQP transport %s: %s', AMQP_URL, e)
|
|
raise RuntimeError
|
|
|
|
self.channel = self.connection.channel()
|
|
self.exchange_name = AMQP_TOPIC
|
|
|
|
self.exchange = Exchange(name=self.exchange_name, type='fanout', channel=self.channel)
|
|
self.producer = Producer(exchange=self.exchange, channel=self.channel)
|
|
|
|
super(FanoutPublisher, self).__init__(name)
|
|
|
|
LOG.info('Configured fanout publisher on topic "%s"', AMQP_TOPIC)
|
|
|
|
def pre_receive(self, alert):
|
|
return alert
|
|
|
|
def post_receive(self, alert):
|
|
|
|
LOG.info('Sending message %s to AMQP topic "%s"', alert.get_id(), AMQP_TOPIC)
|
|
LOG.debug('Message: %s', alert.get_body())
|
|
|
|
self.producer.publish(alert.get_body(), declare=[self.exchange], retry=True)
|
|
|
|
def status_change(self, alert, status, text):
|
|
return
|