mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-10-29 03:18:14 +08:00
96 lines
3.1 KiB
Python
96 lines
3.1 KiB
Python
"""
|
||
缓存 Adapter 的宿主组合根。
|
||
"""
|
||
|
||
from __future__ import annotations
|
||
|
||
from dataclasses import dataclass
|
||
from typing import Any, Optional
|
||
|
||
from app.adapters.cache.backends import configure_platform_cache
|
||
from app.runtime.settings import get_runtime_setting
|
||
|
||
# 未选用 Redis 时的本地缓存类型；平台缓存只区分 redis 与非 redis 两条路由。
|
||
_LOCAL_BACKEND_TYPE = "cachetools"
|
||
|
||
|
||
@dataclass(frozen=True, slots=True)
|
||
class CacheComposition:
|
||
"""
|
||
记录本进程启动时选定的缓存实现，供关闭与配置重载定位连接 owner。
|
||
|
||
:param redis_enabled: 本次进程启动是否选择 Redis 缓存；配置切换不直接改变此快照
|
||
:param sync_redis_owner: 同步 Redis 连接的生命周期 owner；未接入时为 None
|
||
:param async_redis_owner: 异步 Redis 连接的生命周期 owner；未接入时为 None
|
||
"""
|
||
|
||
redis_enabled: bool
|
||
sync_redis_owner: Optional[type[Any]] = None
|
||
async_redis_owner: Optional[type[Any]] = None
|
||
|
||
@property
|
||
def redis_owners(self) -> tuple[type[Any], ...]:
|
||
"""
|
||
返回已接入的 Redis 连接 owner 类型。
|
||
"""
|
||
return tuple(
|
||
owner
|
||
for owner in (self.sync_redis_owner, self.async_redis_owner)
|
||
if owner is not None
|
||
)
|
||
|
||
|
||
_composition = CacheComposition(redis_enabled=False)
|
||
|
||
|
||
def configure_cache_composition() -> None:
|
||
"""
|
||
在业务模块导入前登记平台缓存具体实现。
|
||
|
||
缓存设置属于重启生效的配置组，因此只在启动时按 ``CACHE_BACKEND_TYPE`` 选一次实现。
|
||
选用 Redis 时才导入 Redis 适配器，文件缓存实例不加载 redis 客户端库。启动时未接入
|
||
Redis 的进程在运行中改成 redis 会继续使用本地缓存，直到重启；已接入 Redis 的进程
|
||
在运行中改回本地缓存则立即切换，与之前一致。
|
||
"""
|
||
global _composition
|
||
if get_runtime_setting("CACHE_BACKEND_TYPE") != "redis":
|
||
configure_platform_cache(
|
||
backend_type_provider=_local_backend_type,
|
||
)
|
||
_composition = CacheComposition(redis_enabled=False)
|
||
return
|
||
|
||
from app.adapters.cache.redis import (
|
||
AsyncRedisBackend,
|
||
AsyncRedisHelper,
|
||
RedisBackend,
|
||
RedisHelper,
|
||
)
|
||
|
||
configure_platform_cache(
|
||
backend_type_provider=lambda: get_runtime_setting("CACHE_BACKEND_TYPE"),
|
||
redis_factory=lambda ttl: RedisBackend(ttl=ttl),
|
||
async_redis_factory=lambda ttl: AsyncRedisBackend(ttl=ttl),
|
||
redis_probe=lambda: RedisHelper().test(),
|
||
)
|
||
_composition = CacheComposition(
|
||
redis_enabled=True,
|
||
sync_redis_owner=RedisHelper,
|
||
async_redis_owner=AsyncRedisHelper,
|
||
)
|
||
|
||
|
||
def get_cache_composition() -> CacheComposition:
|
||
"""
|
||
返回本进程当前的缓存组合结果。
|
||
"""
|
||
return _composition
|
||
|
||
|
||
def _local_backend_type() -> str:
|
||
"""
|
||
启动时未接入 Redis，运行中的 redis 设置在重启前不生效。
|
||
"""
|
||
backend_type = get_runtime_setting("CACHE_BACKEND_TYPE")
|
||
return _LOCAL_BACKEND_TYPE if backend_type == "redis" else backend_type
|