chenxin
f0d89b462b
- 删除 cache_manager/audio_proxy/device_key_pool 中 21 处 cacheMutex：database/sql 连接池本身并发安全，SQLite 已配 WAL + busy_timeout，PostgreSQL 无并发写限制 - contentRequestCount 三函数由互斥锁改为 atomic.Int64 - cache_manager 中 config.GetConfig() 改为 GetAppConfig()，消除配置未初始化时 log.Fatal 崩溃整个进程的隐患 - VACUUM 与写操作的低频并发冲突由 busy_timeout 兜底（已注释说明） - 新增 TestConcurrentCacheAccess：8 goroutine 并发读写缓存，-race 验证无死锁无竞争
130 lines
3.5 KiB
Go
130 lines
3.5 KiB
Go
package fqapi
|
|
|
|
import (
|
|
"database/sql"
|
|
"encoding/json"
|
|
"fmt"
|
|
"time"
|
|
)
|
|
|
|
// AudioCacheItem 音频缓存项
|
|
type AudioCacheItem struct {
|
|
OriginalURL string `json:"original_url"` // 原始CDN URL
|
|
DeviceID string `json:"device_id"` // 关联的设备ID
|
|
Headers map[string]string `json:"headers"` // 请求头信息
|
|
CachedAt time.Time `json:"cached_at"` // 缓存时间
|
|
}
|
|
|
|
// GenerateAudioProxyURL 生成音频代理URL
|
|
func GenerateAudioProxyURL(originalURL, deviceID string, headers map[string]string) string {
|
|
// 生成缓存key
|
|
key := generateRandomKey(16)
|
|
|
|
// Redis 模式：TTL 自动过期
|
|
if redisCache != nil {
|
|
saveAudioProxyEntry(key, &audioProxyEntry{
|
|
OriginalURL: originalURL,
|
|
DeviceID: deviceID,
|
|
Headers: headers,
|
|
})
|
|
return fmt.Sprintf("/audio/proxy/%s", key)
|
|
}
|
|
|
|
// 将headers序列化为JSON字符串
|
|
headersJSON, _ := json.Marshal(headers)
|
|
|
|
// 存储到数据库
|
|
if cacheDB != nil {
|
|
|
|
_, err := cacheDB.Exec(q(`
|
|
INSERT INTO audio_cache (key, original_url, device_id, headers, created_at)
|
|
VALUES (?, ?, ?, ?, ?)
|
|
`), key, originalURL, deviceID, string(headersJSON), time.Now().Unix())
|
|
if err != nil {
|
|
audioLogger.Error("保存音频缓存失败: " + err.Error())
|
|
}
|
|
}
|
|
|
|
// 返回代理URL
|
|
return fmt.Sprintf("/audio/proxy/%s", key)
|
|
}
|
|
|
|
// HandleAudioProxy 处理音频代理请求
|
|
func HandleAudioProxy(key string) ([]byte, string, error) {
|
|
var originalURL, deviceID string
|
|
var headers map[string]string
|
|
|
|
if redisCache != nil {
|
|
entry, err := loadAudioProxyEntry(key)
|
|
if err != nil {
|
|
return nil, "", err
|
|
}
|
|
originalURL, deviceID, headers = entry.OriginalURL, entry.DeviceID, entry.Headers
|
|
} else {
|
|
if cacheDB == nil {
|
|
return nil, "", fmt.Errorf("数据库未初始化")
|
|
}
|
|
|
|
// 从数据库获取音频信息
|
|
var headersJSON string
|
|
var createdAt int64
|
|
|
|
err := cacheDB.QueryRow(q(`
|
|
SELECT original_url, device_id, headers, created_at
|
|
FROM audio_cache WHERE key = ?
|
|
`), key).Scan(&originalURL, &deviceID, &headersJSON, &createdAt)
|
|
|
|
if err == sql.ErrNoRows {
|
|
return nil, "", fmt.Errorf("音频信息不存在，请先调用 /audio 接口")
|
|
}
|
|
if err != nil {
|
|
return nil, "", fmt.Errorf("查询音频缓存失败: %w", err)
|
|
}
|
|
|
|
// 检查缓存是否过期（2小时，匹配CDN链接有效期）
|
|
if time.Now().Unix()-createdAt > 7200 { // 2小时 = 7200秒
|
|
// 清理过期缓存
|
|
cacheDB.Exec(q("DELETE FROM audio_cache WHERE key = ?"), key)
|
|
return nil, "", fmt.Errorf("音频信息已过期，请重新调用 /audio 接口")
|
|
}
|
|
|
|
// 解析headers
|
|
json.Unmarshal([]byte(headersJSON), &headers)
|
|
}
|
|
if headers == nil {
|
|
headers = make(map[string]string)
|
|
}
|
|
|
|
// 下载音频内容，使用关联的设备信息
|
|
data, contentType, err := downloadMediaWithRetry(originalURL, deviceID, headers, 3)
|
|
if err != nil {
|
|
return nil, "", fmt.Errorf("下载音频失败: %w", err)
|
|
}
|
|
|
|
return data, contentType, nil
|
|
}
|
|
|
|
// CleanExpiredAudioCache 清理过期的音频缓存
|
|
func CleanExpiredAudioCache() {
|
|
// Redis 模式下由 TTL 自动过期，无需清理
|
|
if redisCache != nil {
|
|
return
|
|
}
|
|
|
|
if cacheDB == nil {
|
|
return
|
|
}
|
|
|
|
// 删除2小时前的缓存
|
|
result, err := cacheDB.Exec(q("DELETE FROM audio_cache WHERE created_at < ?"), time.Now().Unix()-7200)
|
|
if err != nil {
|
|
audioLogger.Error("清理过期音频缓存失败: " + err.Error())
|
|
return
|
|
}
|
|
|
|
rowsAffected, _ := result.RowsAffected()
|
|
if rowsAffected > 0 {
|
|
audioLogger.Info(fmt.Sprintf("已清理 %d 条过期音频缓存", rowsAffected))
|
|
}
|
|
}
|