feat: fix

This commit is contained in:
2025-04-09 16:25:58 +03:00
parent 177f3a3db3
commit e6674bd576
4 changed files with 15 additions and 8 deletions

View File

@@ -1,2 +1 @@
from .producer import producer
from .consumer import consume_messages

View File

@@ -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

View File

@@ -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, 'Ошибка подключения к сервису'