osm_mon/core/message_bus/producer.py

   1 # Copyright 2018 Whitestack, LLC
   2 # *************************************************************
   3
   4 # This file is part of OSM Monitoring module
   5 # All Rights Reserved to Whitestack, LLC
   6
   7 # Licensed under the Apache License, Version 2.0 (the "License"); you may
   8 # not use this file except in compliance with the License. You may obtain
   9 # a copy of the License at
  10
  11 #         http://www.apache.org/licenses/LICENSE-2.0
  12
  13 # Unless required by applicable law or agreed to in writing, software
  14 # distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
  15 # WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
  16 # License for the specific language governing permissions and limitations
  17 # under the License.
  18
  19 # For those usages not covered by the Apache License, Version 2.0 please
  20 # contact: bdiaz@whitestack.com or glavado@whitestack.com
  21 ##
  22 from kafka import KafkaProducer
  23
  24 from osm_mon.core.settings import Config
  25
  26
  27 class Producer(KafkaProducer):
  28     def __init__(self):
  29         cfg = Config.instance()
  30         super().__init__(bootstrap_servers=cfg.BROKER_URI,
  31                          key_serializer=str.encode,
  32                          value_serializer=str.encode)
  33
  34     def send(self, topic, value=None, key=None, partition=None, timestamp_ms=None):
  35         return super().send(topic, value, key, partition, timestamp_ms)