mirror of
https://github.com/fastapi-practices/fastapi-best-architecture.git
synced 2026-09-21 05:02:49 +00:00
Wu Clan
a6da0bfd41
* Refactor code generation and plugin hot reloading * Improve implementation * Update plugin config loader
31 lines
839 B
Python
31 lines
839 B
Python
from collections.abc import AsyncGenerator
|
|
from contextlib import asynccontextmanager
|
|
from typing import Any
|
|
|
|
import anyio
|
|
|
|
from backend.core.path_conf import RELOAD_LOCK_FILE
|
|
from backend.database.redis import redis_client
|
|
|
|
|
|
@asynccontextmanager
|
|
async def acquire_distributed_reload_lock() -> AsyncGenerator[None, Any]:
|
|
"""获取分布式热重载锁"""
|
|
lock = redis_client.lock(
|
|
'fba:reload_lock',
|
|
timeout=300, # 锁持有超时：5 分钟
|
|
blocking_timeout=60, # 获取锁等待超时：60 秒
|
|
)
|
|
await lock.acquire()
|
|
|
|
# 文件锁（通知文件监控器跳过重载）
|
|
lock_path = anyio.Path(RELOAD_LOCK_FILE)
|
|
await lock_path.touch()
|
|
|
|
try:
|
|
yield
|
|
finally:
|
|
await lock_path.unlink(missing_ok=True)
|
|
if await lock.owned():
|
|
await lock.release()
|