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

@@ -9,8 +9,8 @@ from starlette.staticfiles import StaticFiles
import routers
from constants import API_ROOT
from external.kafka import producer, consume_messages
from external.kafka.producer import init_producer
from external.kafka import consume_messages
from external.kafka.producer import init_producer, get_producer
origins = [
'http://localhost:5173'
@@ -21,6 +21,7 @@ origins = [
async def lifespan(app: FastAPI):
try:
await init_producer()
producer = await get_producer()
if producer:
await producer.start()
except KafkaConnectionError as e:
@@ -29,6 +30,7 @@ async def lifespan(app: FastAPI):
consumer_task = asyncio.create_task(consume_messages())
yield
producer = await get_producer()
if producer:
await producer.stop()
consumer_task.cancel()