- self.producer.send(topic='alarm_request', key='delete_alarm_request', value=json.dumps(msg))
- self.producer.flush()
- consumer = KafkaConsumer(bootstrap_servers=self.kafka_server,
- key_deserializer=bytes.decode,
- value_deserializer=bytes.decode,
- consumer_timeout_ms=10000)
- consumer.subscribe(['alarm_response'])
- for message in consumer:
- if message.key == 'delete_alarm_response':
- content = json.loads(message.value)
- log.info("Received delete_alarm_response %s", content)
- if self._is_alarm_response_correlation_id_eq(cor_id, content):
- if not content['alarm_delete_response']['status']:
- raise ValueError("Error deleting alarm in MON")
- alarm_uuid = content['alarm_delete_response']['alarm_uuid']
- return alarm_uuid
-
- raise ValueError('Timeout: No alarm creation response from MON. Is MON up?')
+ producer = AIOKafkaProducer(loop=self.loop,
+ bootstrap_servers=self.kafka_server,
+ key_serializer=str.encode,
+ value_serializer=str.encode)
+ await producer.start()
+ consumer = AIOKafkaConsumer(
+ "alarm_response",
+ loop=self.loop,
+ bootstrap_servers=self.kafka_server,
+ group_id="pol-consumer-" + str(uuid.uuid4()),
+ enable_auto_commit=False,
+ key_deserializer=bytes.decode,
+ value_deserializer=bytes.decode,
+ consumer_timeout_ms=10000)
+ await consumer.start()
+ try:
+ await producer.send_and_wait("alarm_request", key="delete_alarm_request", value=json.dumps(msg))
+ finally:
+ await producer.stop()
+ try:
+ async for message in consumer:
+ if message.key == 'delete_alarm_response':
+ content = json.loads(message.value)
+ log.info("Received delete_alarm_response %s", content)
+ if self._is_alarm_response_correlation_id_eq(cor_id, content):
+ if not content['alarm_delete_response']['status']:
+ raise ValueError("Error deleting alarm in MON")
+ alarm_uuid = content['alarm_delete_response']['alarm_uuid']
+ await consumer.stop()
+ return alarm_uuid
+ finally:
+ await consumer.stop()
+ raise ValueError('Timeout: No alarm deletion response from MON. Is MON up?')