Mirrors/blakeblackshear.frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2025-09-05 17:51:36 +02:00
Code Issues Projects Releases Wiki Activity

chore: change mirror use env

Browse Source
This commit is contained in:
ZhaiSoul 2025-08-11 05:55:11 +00:00
parent 41376f1273
commit b49da13e88
1 changed files with 3 additions and 34 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
frigate/util/downloader.py Normal file → Executable file
View File

@ -26,18 +26,6 @@ MIRROR_MAPPING = {
FORCE_OFFICIAL_SOURCE = False FORCE_OFFICIAL_SOURCE = False
def measure_latency(host: str, port: int = 80, timeout: float = 2.0) -> Optional[float]:
"""
Measure the latency to a host using a TCP connection.
Returns the latency in milliseconds or None if the connection failed.
"""
try:
start_time = time.time()
with socket.create_connection((host, port), timeout=timeout):
end_time = time.time()
return (end_time - start_time) * 1000 # Convert to milliseconds
except (socket.timeout, socket.error):
return None
def set_force_official_source(force: bool = True) -> None: def set_force_official_source(force: bool = True) -> None:
@ -52,7 +40,7 @@ def set_force_official_source(force: bool = True) -> None:
def get_best_mirror(url: str, latency_threshold: int = 20) -> Tuple[str, bool]: def get_best_mirror(url: str, latency_threshold: int = 20) -> Tuple[str, bool]:
""" """
Determine the best URL to use based on latency measurements. Determine whether to use mirror based on environment variable USE_MIRROR_SOURCE.
Returns a tuple of (url_to_use, is_mirror). Returns a tuple of (url_to_use, is_mirror).
""" """
if FORCE_OFFICIAL_SOURCE: if FORCE_OFFICIAL_SOURCE:
@ -66,27 +54,8 @@ def get_best_mirror(url: str, latency_threshold: int = 20) -> Tuple[str, bool]:
if not mirror_host: if not mirror_host:
return url, False return url, False
# Measure latency to both hosts # Check environment variable to determine if mirror should be used
official_latency = measure_latency(host) use_mirror = os.environ.get("USE_MIRROR_SOURCE", "").lower() == "true"
mirror_latency = measure_latency(mirror_host)
# Log latency information
if official_latency is not None and mirror_latency is not None:
logger.info(
f"Latency - Official: {official_latency:.2f}ms, Mirror: {mirror_latency:.2f}ms"
)
# Determine which URL to use
use_mirror = False
if official_latency is None:
# Official site unreachable, try mirror
use_mirror = mirror_latency is not None
elif mirror_latency is None:
# Mirror unreachable, use official
use_mirror = False
else:
# Both reachable, compare latency
use_mirror = mirror_latency < (official_latency - latency_threshold)
if use_mirror: if use_mirror:
mirror_url = url.replace(host, mirror_host) mirror_url = url.replace(host, mirror_host)