redis 常用示例

// src/lib/redis.ts
import Redis from "ioredis";

// 创建并导出 Redis 实例
const redis = new Redis(process.env.REDIS_URL!);

// 获取 Redis 实例
export const getRedisInstance = () => redis;

// 获取历史记录
export async function getHistory(sessionId: string, maxLength: number = 10) {
  return redis.lrange(`session:${sessionId}:history`, 0, maxLength - 1);
}

// 保存对话记录
export async function saveMessage(
  sessionId: string,
  role: string,
  content: string
) {
  await redis.lpush(
    `session:${sessionId}:history`,
    JSON.stringify({ role, content })
  );
  // 限制历史记录的长度
  await redis.ltrim(`session:${sessionId}:history`, 0, 9); // 保持最多10条记录
}

// 设置过期时间
export async function setExpire(sessionId: string, ttl: number = 86400) {
  await redis.expire(`session:${sessionId}:history`, ttl); // ttl 是过期时间,默认 1 天
}

// 设置暂停状态
export async function setPausedState(state: boolean) {
  await redis.set("isPaused", state ? "true" : "false");
}

// 获取暂停状态
export async function getPausedState(): Promise<boolean> {
  const state = await redis.get("isPaused");
  return state === "true";
}