feat: fix
This commit is contained in:
1
external/kafka/__init__.py
vendored
1
external/kafka/__init__.py
vendored
@@ -1,2 +1 @@
|
||||
from .producer import producer
|
||||
from .consumer import consume_messages
|
||||
|
||||
13
external/kafka/producer.py
vendored
13
external/kafka/producer.py
vendored
@@ -5,13 +5,18 @@ from aiokafka import AIOKafkaProducer
|
||||
from backend.config import KAFKA_URL
|
||||
from external.kafka.context import context
|
||||
|
||||
producer: Optional[AIOKafkaProducer] = None
|
||||
_producer: Optional[AIOKafkaProducer] = None
|
||||
|
||||
|
||||
async def init_producer():
|
||||
global producer
|
||||
producer = AIOKafkaProducer(
|
||||
global _producer
|
||||
_producer = AIOKafkaProducer(
|
||||
bootstrap_servers=KAFKA_URL,
|
||||
security_protocol='SSL',
|
||||
ssl_context=context,
|
||||
)
|
||||
)
|
||||
|
||||
|
||||
async def get_producer() -> Optional[AIOKafkaProducer]:
|
||||
global _producer
|
||||
return _producer
|
||||
|
||||
3
external/kafka/services/producer_service.py
vendored
3
external/kafka/services/producer_service.py
vendored
@@ -4,8 +4,8 @@ from uuid import UUID
|
||||
from aiohttp import ClientConnectorError
|
||||
|
||||
from backend.config import KAFKA_PRODUCER_TOPIC, CHAT_CONNECTOR_API_KEY
|
||||
from external.kafka import producer
|
||||
from external.kafka.enums import KafkaMessageType
|
||||
from external.kafka.producer import get_producer
|
||||
from external.kafka.schemas.producer import *
|
||||
from services.base import BaseService
|
||||
|
||||
@@ -14,6 +14,7 @@ class ProducerService(BaseService):
|
||||
@staticmethod
|
||||
async def _send_message(request: BaseConnectorRequest):
|
||||
try:
|
||||
producer = await get_producer()
|
||||
await producer.send(KAFKA_PRODUCER_TOPIC, value=pickle.dumps(request.model_dump()))
|
||||
except ClientConnectorError:
|
||||
return False, 'Ошибка подключения к сервису'
|
||||
|
||||
Reference in New Issue
Block a user