to_ck/common.py
2021-04-21 18:02:03 +08:00

14 lines
292 B
Python

from kafka import KafkaConsumer
from settings import settings
__all__ = 'consumer',
def consumer():
c = KafkaConsumer(**settings.KAFKA_CONSUMER_CONF)
c.subscribe(settings.SUBSCRIBE_TOPIC)
for msg in c:
topic = msg.topic
val = msg.value
yield topic, val