update:完善数据库和Kafka相关结构

This commit is contained in:
2024-12-23 12:25:36 +08:00
parent aaf297717a
commit 562b1add2a
8 changed files with 78 additions and 9 deletions

13
dao/kafka/util.py Normal file
View File

@@ -0,0 +1,13 @@
from conf.util import get_config_object
from kafka import KafkaConsumer
conf = get_config_object()
# Kafka
def get_KafkaConsumer() -> KafkaConsumer:
""" 返回KafkaConsumer对象 """
consumer = KafkaConsumer(
bootstrap_servers=conf.get("kafka", "bootstrap_servers"),
group_id=conf.get("kafka", "group_id")
)
return consumer