69 lines
2.7 KiB
Python
69 lines
2.7 KiB
Python
# (C) Copyright 2015-2016 Hewlett Packard Enterprise Development LP
|
|
# Copyright 2017 Fujitsu LIMITED
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
|
|
# implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
import time
|
|
|
|
from oslo_config import cfg
|
|
from oslo_log import log as logging
|
|
|
|
from monasca_common.kafka import consumer
|
|
from monasca_common.kafka import producer
|
|
from monasca_notification.common.utils import get_statsd_client
|
|
from monasca_notification.processors import alarm_processor as ap
|
|
from monasca_notification.processors import notification_processor as np
|
|
|
|
log = logging.getLogger(__name__)
|
|
CONF = cfg.CONF
|
|
|
|
|
|
class NotificationEngine(object):
|
|
def __init__(self):
|
|
self._statsd = get_statsd_client()
|
|
self._consumer = consumer.KafkaConsumer(
|
|
CONF.kafka.url,
|
|
','.join(CONF.zookeeper.url),
|
|
CONF.zookeeper.notification_path,
|
|
CONF.kafka.group,
|
|
CONF.kafka.alarm_topic)
|
|
self._producer = producer.KafkaProducer(CONF.kafka.url)
|
|
self._alarms = ap.AlarmProcessor()
|
|
self._notifier = np.NotificationProcessor()
|
|
|
|
def _add_periodic_notifications(self, notifications):
|
|
for notification in notifications:
|
|
topic = notification.periodic_topic
|
|
if topic in CONF.kafka.periodic and notification.type == "webhook":
|
|
notification.notification_timestamp = time.time()
|
|
self._producer.publish(CONF.kafka.periodic[topic],
|
|
[notification.to_json()])
|
|
|
|
def run(self):
|
|
finished_count = self._statsd.get_counter(name='alarms_finished_count')
|
|
for alarm in self._consumer:
|
|
log.debug('Received alarm >|%s|<', str(alarm))
|
|
notifications, partition, offset = self._alarms.to_notification(alarm)
|
|
if notifications:
|
|
self._add_periodic_notifications(notifications)
|
|
|
|
sent, failed = self._notifier.send(notifications)
|
|
self._producer.publish(CONF.kafka.notification_topic,
|
|
[i.to_json() for i in sent])
|
|
self._producer.publish(CONF.kafka.notification_retry_topic,
|
|
[i.to_json() for i in failed])
|
|
|
|
self._consumer.commit()
|
|
finished_count.increment()
|