mirror of
https://github.com/open-webui/open-webui
synced 2025-04-27 17:51:31 +00:00
57 lines
2.0 KiB
Python
57 lines
2.0 KiB
Python
import socketio
|
|
import redis
|
|
from redis import asyncio as aioredis
|
|
from urllib.parse import urlparse
|
|
|
|
|
|
def parse_redis_service_url(redis_url):
|
|
parsed_url = urlparse(redis_url)
|
|
if parsed_url.scheme != "redis":
|
|
raise ValueError("Invalid Redis URL scheme. Must be 'redis'.")
|
|
|
|
return {
|
|
"username": parsed_url.username or None,
|
|
"password": parsed_url.password or None,
|
|
"service": parsed_url.hostname or "mymaster",
|
|
"port": parsed_url.port or 6379,
|
|
"db": int(parsed_url.path.lstrip("/") or 0),
|
|
}
|
|
|
|
|
|
def get_redis_connection(redis_url, redis_sentinels, decode_responses=True):
|
|
if redis_sentinels:
|
|
redis_config = parse_redis_service_url(redis_url)
|
|
sentinel = redis.sentinel.Sentinel(
|
|
redis_sentinels,
|
|
port=redis_config["port"],
|
|
db=redis_config["db"],
|
|
username=redis_config["username"],
|
|
password=redis_config["password"],
|
|
decode_responses=decode_responses,
|
|
)
|
|
|
|
# Get a master connection from Sentinel
|
|
return sentinel.master_for(redis_config["service"])
|
|
else:
|
|
# Standard Redis connection
|
|
return redis.Redis.from_url(redis_url, decode_responses=decode_responses)
|
|
|
|
|
|
def get_sentinels_from_env(sentinel_hosts_env, sentinel_port_env):
|
|
if sentinel_hosts_env:
|
|
sentinel_hosts = sentinel_hosts_env.split(",")
|
|
sentinel_port = int(sentinel_port_env)
|
|
return [(host, sentinel_port) for host in sentinel_hosts]
|
|
return []
|
|
|
|
|
|
def get_sentinel_url_from_env(redis_url, sentinel_hosts_env, sentinel_port_env):
|
|
redis_config = parse_redis_service_url(redis_url)
|
|
username = redis_config["username"] or ""
|
|
password = redis_config["password"] or ""
|
|
auth_part = ""
|
|
if username or password:
|
|
auth_part = f"{username}:{password}@"
|
|
hosts_part = ",".join(f"{host}:{sentinel_port_env}" for host in sentinel_hosts_env.split(","))
|
|
return f"redis+sentinel://{auth_part}{hosts_part}/{redis_config['db']}/{redis_config['service']}"
|