chenxin
586b714f3c
- 新增 withCache 包装：命中直接返回，miss 回源后写入，仅缓存成功响应 - 详情/目录/批量详情/章节摘要缓存 24 小时；搜索/分类列表/分类详情短缓存 10 分钟 - 7 个 fqapi 处理函数原函数体不动、改名 fetchXxx，外层套缓存壳 - redisGetCache 补 3 秒读超时，与写路径一致，防远程 Redis 网络抖动拖垮请求 - 新增 TestWithCache：验证命中不回源、失败响应不缓存
89 lines
2.7 KiB
Go
89 lines
2.7 KiB
Go
package fqapi
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"time"
|
|
)
|
|
|
|
// audioProxyEntry 音频代理缓存条目（Redis 版）
|
|
type audioProxyEntry struct {
|
|
OriginalURL string `json:"original_url"`
|
|
DeviceID string `json:"device_id"`
|
|
Headers map[string]string `json:"headers"`
|
|
}
|
|
|
|
const (
|
|
redisCacheKeyPrefix = "fq:cache:"
|
|
redisAudioKeyPrefix = "fq:audio:"
|
|
// audioCacheTTL 音频代理缓存有效期（匹配 CDN 链接有效期）
|
|
audioCacheTTL = 2 * time.Hour
|
|
)
|
|
|
|
// redisGetCache 从 Redis 获取正文缓存
|
|
func redisGetCache(key string) string {
|
|
// 读操作同样加 3s 超时，防止远程 Redis 网络抖动拖垮请求（与写路径一致）
|
|
ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second)
|
|
defer cancel()
|
|
val, err := redisCache.Get(ctx, redisCacheKeyPrefix+key).Result()
|
|
if err != nil {
|
|
return "" // miss 或错误都视为未命中
|
|
}
|
|
|
|
var cacheData CacheData
|
|
if err := json.Unmarshal([]byte(val), &cacheData); err != nil {
|
|
cacheLogger.Error("解析Redis缓存数据失败: " + err.Error())
|
|
return ""
|
|
}
|
|
|
|
cacheLogger.Debug("Redis缓存命中: " + key)
|
|
if time.Now().Unix() < int64(cacheData.Expi) {
|
|
return cacheData.Data
|
|
}
|
|
return ""
|
|
}
|
|
|
|
// redisUpdateCache 写入 Redis 正文缓存（TTL 与过期时间一致）
|
|
func redisUpdateCache(key string, value string, expi int) error {
|
|
cacheData := CacheData{
|
|
Data: value,
|
|
Expi: float64(time.Now().Unix() + int64(expi)),
|
|
}
|
|
jsonData, err := json.Marshal(cacheData)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second)
|
|
defer cancel()
|
|
return redisCache.Set(ctx, redisCacheKeyPrefix+key, jsonData, time.Duration(expi)*time.Second).Err()
|
|
}
|
|
|
|
// saveAudioProxyEntry 将音频代理信息存入 Redis
|
|
func saveAudioProxyEntry(key string, entry *audioProxyEntry) {
|
|
ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second)
|
|
defer cancel()
|
|
jsonData, _ := json.Marshal(entry)
|
|
if err := redisCache.Set(ctx, redisAudioKeyPrefix+key, jsonData, audioCacheTTL).Err(); err != nil {
|
|
audioLogger.Error("保存音频缓存到Redis失败: " + err.Error())
|
|
}
|
|
}
|
|
|
|
// loadAudioProxyEntry 从 Redis 读取音频代理信息，未命中或过期返回错误
|
|
func loadAudioProxyEntry(key string) (*audioProxyEntry, error) {
|
|
ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second)
|
|
defer cancel()
|
|
|
|
val, err := redisCache.Get(ctx, redisAudioKeyPrefix+key).Result()
|
|
if err != nil {
|
|
return nil, fmt.Errorf("音频信息不存在或已过期，请重新调用 /audio 接口")
|
|
}
|
|
|
|
var entry audioProxyEntry
|
|
if err := json.Unmarshal([]byte(val), &entry); err != nil {
|
|
return nil, fmt.Errorf("解析音频缓存失败: %w", err)
|
|
}
|
|
return &entry, nil
|
|
}
|