"""Example avro message consumer."""

from helpers import debug_msk_message
from helpers import load_file_json

from kafka_utils.consumer.deserializer.simple_json import JSONDeserializer
from kafka_utils.consumer.deserializer.string import StringDeserializer
from kafka_utils.consumer.source.mapping import EventSourceMessage

EVENT_JSON = load_file_json('simple_json.json')


def main():
    """Process message."""
    string_deserializer = StringDeserializer()
    json_deserializer = JSONDeserializer()
    event = EventSourceMessage(EVENT_JSON)
    for batch_key, msk_message in event:
        debug_msk_message(
            batch_key,
            msk_message,
            key_deserializer=string_deserializer,
            value_deserializer=json_deserializer)


if __name__ == '__main__':
    main()