瀏覽代碼

feat: 重启自动清理部分缓存

SongZihuan 2 年之前
父節點
當前提交
574f2d579c
共有 2 個文件被更改,包括 15 次插入0 次删除
  1. 4 0
      main.py
  2. 11 0
      sql/cache.py

+ 4 - 0
main.py

@@ -1,4 +1,5 @@
 from configure import configure
 from configure import configure
+
 import os
 import os
 import logging
 import logging
 
 
@@ -11,6 +12,9 @@ else:
     logging.info(f"Configure file {hblog_conf}")
     logging.info(f"Configure file {hblog_conf}")
     configure(hblog_conf)
     configure(hblog_conf)
 
 
+from sql.cache import restart_clear_cache
+restart_clear_cache()  # 清理缓存
+
 from app import HBlogFlask
 from app import HBlogFlask
 from waitress import serve
 from waitress import serve
 
 

+ 11 - 0
sql/cache.py

@@ -396,3 +396,14 @@ def write_role_operate_to_cache(role_id: int, operate: str, res: bool):
 def delete_role_operate_from_cache(role_id: int):
 def delete_role_operate_from_cache(role_id: int):
     for i in cache.keys(f"{CACHE_PREFIX}:operate:{role_id}:*"):
     for i in cache.keys(f"{CACHE_PREFIX}:operate:{role_id}:*"):
         cache.delete(i)
         cache.delete(i)
+
+
+@__try_redis(None)
+def restart_clear_cache():
+    """
+    重启服务时必须要清理的缓存
+    包括Hblog-Cache和Flask-Cache
+    """
+
+    # 删除inject_base, 重新载入conf作为缓存
+    cache.delete(f"flask_cache:inject_base")