3eb169fac3
Squished the notification down to one process in preparation for making multiple engine instances able to talk to the same topic. Added HA capabilities to the KafkaConsumer object. Change-Id: Id4799d690c7caf10ae7ba89c2d4e0aeb76c54375
69 lines
2.6 KiB
Python
69 lines
2.6 KiB
Python
# Copyright (c) 2015 Hewlett-Packard Development Company, L.P.
|
|
#
|
|
# 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 monascastatsd
|
|
|
|
from processors.alarm_processor import AlarmProcessor
|
|
from processors.base import BaseProcessor
|
|
from processors.kafka_consumer import KafkaConsumer
|
|
from processors.kafka_producer import KafkaProducer
|
|
from processors.notification_processor import NotificationProcessor
|
|
|
|
|
|
class NotificationEngine(object):
|
|
def __init__(self, config):
|
|
self._topics = {}
|
|
self._topics['notification_topic'] = config['kafka']['notification_topic']
|
|
|
|
self._statsd = monascastatsd.Client(name='monasca',
|
|
dimensions=BaseProcessor.dimensions)
|
|
|
|
self._consumer = KafkaConsumer(config['kafka']['url'],
|
|
config['zookeeper']['url'],
|
|
config['zookeeper']['path'],
|
|
config['kafka']['group'],
|
|
config['kafka']['alarm_topic'])
|
|
|
|
self._producer = KafkaProducer(config['kafka']['url'])
|
|
|
|
if 'ssl' in config['mysql']:
|
|
ssl_config = config['mysql']['ssl']
|
|
else:
|
|
ssl_config = None
|
|
|
|
self._alarms = AlarmProcessor(config['processors']['alarm']['ttl'],
|
|
config['mysql']['host'],
|
|
config['mysql']['user'],
|
|
config['mysql']['passwd'],
|
|
config['mysql']['db'],
|
|
ssl_config)
|
|
|
|
self._notifier = NotificationProcessor(config['notification_types'])
|
|
|
|
def run(self):
|
|
finished_count = self._statsd.get_counter(name='alarms_finished_count')
|
|
|
|
for alarm in self._consumer:
|
|
|
|
notifications, partition, offset = self._alarms.to_notification(alarm)
|
|
|
|
if notifications:
|
|
sent, failed = self._notifier.send(notifications)
|
|
self._producer.publish(self._topics['notification_topic'], sent)
|
|
|
|
self._consumer.commit([partition])
|
|
|
|
finished_count.increment()
|