mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-03-20 03:57:30 +08:00
refactor:重构配置热加载
This commit is contained in:
@@ -1,5 +1,6 @@
|
||||
from typing import Any, Union
|
||||
|
||||
from app.core.config import ConfigChangeEvent, config_notifier, ConfigChangeType
|
||||
from app.db import DbOper
|
||||
from app.db.models.systemconfig import SystemConfig
|
||||
from app.schemas.types import SystemConfigKey
|
||||
@@ -24,17 +25,32 @@ class SystemConfigOper(DbOper, metaclass=Singleton):
|
||||
"""
|
||||
if isinstance(key, SystemConfigKey):
|
||||
key = key.value
|
||||
# 旧值
|
||||
old_value = self.__SYSTEMCONF.get(key)
|
||||
# 更新内存
|
||||
self.__SYSTEMCONF[key] = value
|
||||
conf = SystemConfig.get_by_key(self._db, key)
|
||||
if conf:
|
||||
if value:
|
||||
conf.update(self._db, {"value": value})
|
||||
# 发送配置变更通知
|
||||
if old_value != value:
|
||||
event = ConfigChangeEvent(key, old_value=old_value, new_value=value,
|
||||
change_type=ConfigChangeType.UPDATE)
|
||||
config_notifier.notify(event)
|
||||
else:
|
||||
conf.delete(self._db, conf.id)
|
||||
# 发送配置删除通知
|
||||
event = ConfigChangeEvent(key, old_value=old_value, new_value=None,
|
||||
change_type=ConfigChangeType.DELETE)
|
||||
config_notifier.notify(event)
|
||||
else:
|
||||
conf = SystemConfig(key=key, value=value)
|
||||
conf.create(self._db)
|
||||
# 发送配置变更通知
|
||||
event = ConfigChangeEvent(key, old_value=None, new_value=value,
|
||||
change_type=ConfigChangeType.ADD)
|
||||
config_notifier.notify(event)
|
||||
|
||||
def get(self, key: Union[str, SystemConfigKey] = None) -> Any:
|
||||
"""
|
||||
@@ -59,11 +75,15 @@ class SystemConfigOper(DbOper, metaclass=Singleton):
|
||||
if isinstance(key, SystemConfigKey):
|
||||
key = key.value
|
||||
# 更新内存
|
||||
self.__SYSTEMCONF.pop(key, None)
|
||||
old_value = self.__SYSTEMCONF.pop(key, None)
|
||||
# 写入数据库
|
||||
conf = SystemConfig.get_by_key(self._db, key)
|
||||
if conf:
|
||||
conf.delete(self._db, conf.id)
|
||||
# 发送配置变更通知
|
||||
event = ConfigChangeEvent(key, old_value=old_value, new_value=None,
|
||||
change_type=ConfigChangeType.ADD)
|
||||
config_notifier.notify(event)
|
||||
return True
|
||||
|
||||
def __del__(self):
|
||||
|
||||
Reference in New Issue
Block a user