优化代码,添加订单处理
This commit is contained in:
@@ -7,6 +7,7 @@ KAFKA_BOOTSTRAP_SERVERS = '127.0.0.1:19092'
|
||||
class KafkaProducerSingleton:
|
||||
_instance = None
|
||||
_producer = None
|
||||
_producer2 = None
|
||||
|
||||
def __new__(cls):
|
||||
if cls._instance is None:
|
||||
@@ -23,20 +24,34 @@ class KafkaProducerSingleton:
|
||||
)
|
||||
await self._producer.start()
|
||||
return self._producer
|
||||
|
||||
async def get_producer2(self):
|
||||
if self._producer2 is None:
|
||||
self._producer2 = AIOKafkaProducer(
|
||||
bootstrap_servers=KAFKA_BOOTSTRAP_SERVERS,
|
||||
key_serializer=lambda k: k.encode('utf-8'),
|
||||
value_serializer=lambda v: v.encode('utf-8'),
|
||||
)
|
||||
await self._producer2.start()
|
||||
return self._producer2
|
||||
|
||||
async def stop_producer(self):
|
||||
if self._producer is not None:
|
||||
await self._producer.stop()
|
||||
self._producer = None
|
||||
if self._producer2 is not None:
|
||||
await self._producer2.stop()
|
||||
self._producer2 = None
|
||||
|
||||
|
||||
|
||||
# 初始化单例实例
|
||||
_kafka_producer_singleton = KafkaProducerSingleton()
|
||||
|
||||
async def send_to_kafka(topic, data, key=None):
|
||||
async def send_to_kafka(topic, data, key=None, send_now=False):
|
||||
"""将数据发送到 Kafka 主题"""
|
||||
producer = await _kafka_producer_singleton.get_producer()
|
||||
#print(topic, data, key)
|
||||
try:
|
||||
await producer.send(topic, data,key=key)
|
||||
except Exception as e:
|
||||
print("error: ", topic, data, key,e)
|
||||
if send_now:
|
||||
producer = await _kafka_producer_singleton.get_producer2()
|
||||
else:
|
||||
producer = await _kafka_producer_singleton.get_producer()
|
||||
await producer.send(topic, data,key=key)
|
||||
Reference in New Issue
Block a user