xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-08-21 10:56:40 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: add cache headers for images

Browse Source 
Prevents unnecessary repeated requests, especially for 404s
This commit is contained in:
Arjan Singh 2025-08-25 12:36:22 -07:00
parent d28b39d93d
commit 3f948e3c73
No known key found for this signature in database
GPG Key ID: B1102A5F9699979D
1 changed files with 11 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
server.py
View File

@ -39,6 +39,11 @@ from typing import Optional, Union
from api_server.routes.internal.internal_routes import InternalRoutes
from protocol import BinaryEventTypes
# Time in seconds
ONE_HOUR: int = 3600
ONE_DAY: int = 86400
IMG_EXTENSIONS = ('.jpg', '.jpeg', '.png', '.ppm', '.bmp', '.pgm', '.tif', '.tiff', '.webp')
async def send_socket_catch_exception(function, message):
try:
await function(message)
@ -46,12 +51,16 @@ async def send_socket_catch_exception(function, message):
logging.warning("send error: {}".format(err))
@web.middleware
async def cache_control(request: web.Request, handler):
async def cache_control(request: web.Request, handler) -> web.Response:
response: web.Response = await handler(request)
if request.path.endswith('.js') or request.path.endswith('.css') or request.path.endswith('index.json'):
response.headers.setdefault('Cache-Control', 'no-cache')
return response
elif request.path.lower().endswith(IMG_EXTENSIONS):
max_age = ONE_DAY if response.status == 200 else ONE_HOUR
response.headers['Cache-Control'] = f"public, max-age={max_age}"
return response
@web.middleware
async def compress_body(request: web.Request, handler):