import logging
import os
import sys
+import yaml
logging.basicConfig(stream=sys.stdout,
format='%(asctime)s %(message)s',
for message in common_consumer:
log.info("Message arrived: %s", message)
try:
- values = json.loads(message.value)
+ try:
+ values = json.loads(message.value)
+ except:
+ values = yaml.safe_load(message.value)
# Check the message topic
if message.topic == "metric_request":
# Check the vim desired by the message