33 lines
743 B
Python
33 lines
743 B
Python
"""Redis 初始化模块
|
|
|
|
独立部署时使用的 Redis 初始化，不依赖 Dify 扩展。
|
|
"""
|
|
import os
|
|
from flask import Flask
|
|
import redis
|
|
|
|
|
|
# Redis 客户端实例
|
|
redis_client = None
|
|
|
|
|
|
def init_redis(app: Flask) -> None:
|
|
"""初始化 Redis 连接。"""
|
|
global redis_client
|
|
|
|
redis_url = app.config.get('REDIS_URL', 'redis://localhost:6379/0')
|
|
|
|
try:
|
|
redis_client = redis.from_url(redis_url, decode_responses=True)
|
|
# 测试连接
|
|
redis_client.ping()
|
|
print(f"[redis] Redis 连接成功: {redis_url}")
|
|
except Exception as e:
|
|
print(f"[redis] Redis 连接失败: {e}")
|
|
redis_client = None
|
|
|
|
|
|
def get_redis():
|
|
"""获取 Redis 客户端实例。"""
|
|
return redis_client
|