vndangkhoa/spotify-clone
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Optimize stream for NAS: prioritize iOS client, flush logs

Browse source
This commit is contained in:
Khoa Vo 2026-01-01 16:44:29 +07:00
parent 91c1cd5302
commit a7e1692bdb
1 changed files with 14 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

27
backend/api/routes.py
View file

@ -608,7 +608,8 @@ async def stream_audio(id: str):
""" """
try: try:
# Check Cache for stream URL # Check Cache for stream URL
cache_key = f"v4:stream:{id}" # v4 cache key for clean slate # Check Cache for stream URL
cache_key = f"v5:stream:{id}" # v5 cache key for ios client priority
cached_data = cache.get(cache_key) cached_data = cache.get(cache_key)
stream_url = None stream_url = None
@ -634,7 +635,7 @@ async def stream_audio(id: str):
'socket_timeout': 30, 'socket_timeout': 30,
'retries': 3, 'retries': 3,
'force_ipv4': True, 'force_ipv4': True,
'extractor_args': {'youtube': {'player_client': ['android', 'web', 'ios']}}, # Android often gives good progressive streams 'extractor_args': {'youtube': {'player_client': ['ios', 'web', 'android']}}, # Prioritize iOS for server-like environments
} }
try: try:
@ -679,30 +680,30 @@ async def stream_audio(id: str):
if 'Cookie' in base_headers: if 'Cookie' in base_headers:
req_headers['Cookie'] = base_headers['Cookie'] req_headers['Cookie'] = base_headers['Cookie']
print(f"DEBUG_STREAM [{id}]: Opening connection to {stream_url[:50]}...") print(f"DEBUG_STREAM [{id}]: Opening connection to {stream_url[:50]}...", flush=True)
print(f"DEBUG_STREAM [{id}]: Headers: User-Agent={req_headers.get('User-Agent')[:30]}..., Cookie Present={'Cookie' in req_headers}") print(f"DEBUG_STREAM [{id}]: Headers: User-Agent={req_headers.get('User-Agent')[:30]}..., Cookie Present={'Cookie' in req_headers}", flush=True)
# Disable SSL verify to match yt-dlp 'nocheckcertificate' (fixes NAS CA issues) # Disable SSL verify to match yt-dlp 'nocheckcertificate' (fixes NAS CA issues)
external_req = requests.get(stream_url, stream=True, timeout=30, headers=req_headers, verify=False) external_req = requests.get(stream_url, stream=True, timeout=30, headers=req_headers, verify=False)
print(f"DEBUG_STREAM [{id}]: Upstream Status: {external_req.status_code}") print(f"DEBUG_STREAM [{id}]: Upstream Status: {external_req.status_code}", flush=True)
external_req.raise_for_status() external_req.raise_for_status()
except requests.exceptions.HTTPError as http_err: except requests.exceptions.HTTPError as http_err:
error_details = f"Upstream error: {http_err.response.status_code}" error_details = f"Upstream error: {http_err.response.status_code}"
print(f"DEBUG_STREAM [{id}]: FAILED {error_details}") print(f"DEBUG_STREAM [{id}]: FAILED {error_details}", flush=True)
try: try:
print(f"DEBUG_STREAM [{id}]: Response Headers: {http_err.response.headers}") print(f"DEBUG_STREAM [{id}]: Response Headers: {http_err.response.headers}", flush=True)
print(f"DEBUG_STREAM [{id}]: Response Body keys: {http_err.response.text[:200]}") print(f"DEBUG_STREAM [{id}]: Response Body keys: {http_err.response.text[:200]}", flush=True)
except: except:
pass pass
# If 403/404/410, invalidate cache # If 403/404/410, invalidate cache
if http_err.response.status_code in [403, 404, 410]: if http_err.response.status_code in [403, 404, 410]:
print(f"DEBUG_STREAM [{id}]: Invalidating cache key {cache_key}") print(f"DEBUG_STREAM [{id}]: Invalidating cache key {cache_key}", flush=True)
cache.delete(cache_key) cache.delete(cache_key)
raise HTTPException(status_code=500, detail=error_details) raise HTTPException(status_code=500, detail=error_details)
except Exception as e: except Exception as e:
print(f"DEBUG_STREAM [{id}]: Connection Failed: {e}") print(f"DEBUG_STREAM [{id}]: Connection Failed: {e}", flush=True)
import traceback import traceback
traceback.print_exc() traceback.print_exc()
raise HTTPException(status_code=500, detail=f"Stream connection failed: {str(e)}") raise HTTPException(status_code=500, detail=f"Stream connection failed: {str(e)}")
@ -711,7 +712,7 @@ async def stream_audio(id: str):
headers = {} headers = {}
if "Content-Length" in external_req.headers: if "Content-Length" in external_req.headers:
headers["Content-Length"] = external_req.headers["Content-Length"] headers["Content-Length"] = external_req.headers["Content-Length"]
print(f"DEBUG_STREAM [{id}]: Content-Length: {headers['Content-Length']}") print(f"DEBUG_STREAM [{id}]: Content-Length: {headers['Content-Length']}", flush=True)
def iterfile(): def iterfile():
try: try:
@ -719,9 +720,9 @@ async def stream_audio(id: str):
for chunk in external_req.iter_content(chunk_size=64*1024): for chunk in external_req.iter_content(chunk_size=64*1024):
yield chunk yield chunk
external_req.close() external_req.close()
print(f"DEBUG_STREAM [{id}]: Stream finished successfully") print(f"DEBUG_STREAM [{id}]: Stream finished successfully", flush=True)
except Exception as e: except Exception as e:
print(f"DEBUG_STREAM [{id}]: Stream Iterator Error: {e}") print(f"DEBUG_STREAM [{id}]: Stream Iterator Error: {e}", flush=True)
pass pass
return StreamingResponse(iterfile(), media_type=mime_type, headers=headers) return StreamingResponse(iterfile(), media_type=mime_type, headers=headers)