wangbo/ComfyUI-Manager
1
0
Fork 0
mirror of https://github.com/Comfy-Org/ComfyUI-Manager.git synced 2026-07-10 16:37:15 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

perf(cnr): add cache 7-day expiration logic

Browse Source 
* for github star, download counts.
This commit is contained in:
craftingmod 2026-06-30 07:19:49 +09:00
parent 9a827e409a
commit 5427efda58
1 changed files with 26 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
glob/cnr_utils.py
View File

@ -3,7 +3,7 @@ import json
import os import os
import platform import platform
from dataclasses import dataclass from dataclasses import dataclass
from datetime import datetime, timedelta from datetime import datetime, timedelta, timezone
from typing import List from typing import List
from urllib.parse import urlencode from urllib.parse import urlencode
@ -99,27 +99,42 @@ async def _get_cnr_data(sync_mode=None, dont_wait=True, **kwargs):
cached_data = None cached_data = None
last_updated = None last_updated = None
full_nodes = {} full_nodes = {}
is_cache_expired = True
cache_built_at = None
if normalized_mode != 'force' and manager_util.get_cache_state(uri, expired_days=None) == 'cached': if normalized_mode != 'force' and manager_util.get_cache_state(uri, expired_days=None) == 'cached':
try: try:
with open(cache_path, 'r', encoding="UTF-8", errors="ignore") as json_file: with open(cache_path, 'r', encoding="UTF-8", errors="ignore") as json_file:
cached_data = json.load(json_file) cached_data = json.load(json_file)
if (cached_data.get('comfyui_ver') == comfyui_ver and cache_built_at = cached_data.get('cache_built_at')
if cache_built_at:
try:
built_dt = datetime.fromisoformat(cache_built_at.replace('Z', '+00:00'))
current_dt = datetime.now(timezone.utc)
delta_dt = current_dt - built_dt
if timedelta(seconds=0) <= delta_dt and delta_dt < timedelta(days=7):
is_cache_expired = False
except Exception:
pass
if not is_cache_expired and (cached_data.get('comfyui_ver') == comfyui_ver and
cached_data.get('form_factor') == form_factor): cached_data.get('form_factor') == form_factor):
last_updated = cached_data.get('last_updated') last_updated = cached_data.get('last_updated')
for node in cached_data.get('nodes', []): for node in cached_data.get('nodes', []):
full_nodes[node['id']] = node full_nodes[node['id']] = node
else: else:
logging.info("[ComfyUI-Manager] Environment change detected. Invalidating local ComfyRegistry cache.") logging.info("[ComfyUI-Manager] Registry cache expired or environment changed. Invalidating local cache.")
cached_data = None cached_data = None
full_nodes = {} full_nodes = {}
last_updated = None last_updated = None
is_cache_expired = True
except Exception as e: except Exception as e:
logging.error(f"[ComfyUI-Manager] Failed to read cached data: {e}") logging.error(f"[ComfyUI-Manager] Failed to read cached data: {e}")
cached_data = None cached_data = None
full_nodes = {} full_nodes = {}
last_updated = None last_updated = None
is_cache_expired = True
if normalized_mode == 'cache': if normalized_mode == 'cache':
is_cache_loading = True is_cache_loading = True
@ -130,7 +145,7 @@ async def _get_cnr_data(sync_mode=None, dont_wait=True, **kwargs):
return cached_data.get('nodes', []) return cached_data.get('nodes', [])
return [] return []
if cached_data is not None and manager_util.get_cache_state(uri, expired_days=1) == 'cached': if cached_data is not None and not is_cache_expired:
is_cache_loading = False is_cache_loading = False
return cached_data.get('nodes', []) return cached_data.get('nodes', [])
@ -191,11 +206,17 @@ async def _get_cnr_data(sync_mode=None, dont_wait=True, **kwargs):
else: else:
new_timestamp = last_updated new_timestamp = last_updated
if is_cache_expired or normalized_mode == 'force' or not cache_built_at:
new_cache_built_at = datetime.now(timezone.utc).strftime("%Y-%m-%dT%H:%M:%SZ")
else:
new_cache_built_at = cache_built_at
cache_to_save = { cache_to_save = {
'nodes': json_obj['nodes'], 'nodes': json_obj['nodes'],
'comfyui_ver': comfyui_ver, 'comfyui_ver': comfyui_ver,
'form_factor': form_factor, 'form_factor': form_factor,
'last_updated': new_timestamp 'last_updated': new_timestamp,
'cache_built_at': new_cache_built_at
} }
manager_util.save_to_cache(uri, cache_to_save) manager_util.save_to_cache(uri, cache_to_save)
return json_obj['nodes'] return json_obj['nodes']