fded346295
- Add internal/pkg/safego/ utility for safe goroutine launching with panic recovery, structured logging (slog + zap), and optional error callback support - Replace 8 fire-and-forget goroutines in service layer: user_service.go (1), subscription_service.go (4), redeem_service.go (2), promo_service.go (1) — all now use safego.Go() - Add inline panic recovery to 6 long-running goroutines: pricing_service.go scheduler loop Redis PubSub subscribers: tls_fingerprint/error_passthrough/api_key caches OAuth session cleanup: openai/oauth/geminicli S3 stream upload in sora_s3_storage.go - All changes verified: go build PASS, service tests ALL PASS, repository tests ALL PASS
128 lines
3.0 KiB
Go
128 lines
3.0 KiB
Go
package repository
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"log/slog"
|
|
"sync"
|
|
"time"
|
|
|
|
"github.com/Wei-Shaw/sub2api/internal/model"
|
|
"github.com/Wei-Shaw/sub2api/internal/service"
|
|
"github.com/redis/go-redis/v9"
|
|
)
|
|
|
|
const (
|
|
tlsFPProfileCacheKey = "tls_fingerprint_profiles"
|
|
tlsFPProfilePubSubKey = "tls_fingerprint_profiles_updated"
|
|
tlsFPProfileCacheTTL = 24 * time.Hour
|
|
)
|
|
|
|
type tlsFingerprintProfileCache struct {
|
|
rdb *redis.Client
|
|
localCache []*model.TLSFingerprintProfile
|
|
localMu sync.RWMutex
|
|
}
|
|
|
|
// NewTLSFingerprintProfileCache 创建 TLS 指纹模板缓存
|
|
func NewTLSFingerprintProfileCache(rdb *redis.Client) service.TLSFingerprintProfileCache {
|
|
return &tlsFingerprintProfileCache{
|
|
rdb: rdb,
|
|
}
|
|
}
|
|
|
|
// Get 从缓存获取模板列表
|
|
func (c *tlsFingerprintProfileCache) Get(ctx context.Context) ([]*model.TLSFingerprintProfile, bool) {
|
|
c.localMu.RLock()
|
|
if c.localCache != nil {
|
|
profiles := c.localCache
|
|
c.localMu.RUnlock()
|
|
return profiles, true
|
|
}
|
|
c.localMu.RUnlock()
|
|
|
|
data, err := c.rdb.Get(ctx, tlsFPProfileCacheKey).Bytes()
|
|
if err != nil {
|
|
if err != redis.Nil {
|
|
slog.Warn("tls_fp_profile_cache_get_failed", "error", err)
|
|
}
|
|
return nil, false
|
|
}
|
|
|
|
var profiles []*model.TLSFingerprintProfile
|
|
if err := json.Unmarshal(data, &profiles); err != nil {
|
|
slog.Warn("tls_fp_profile_cache_unmarshal_failed", "error", err)
|
|
return nil, false
|
|
}
|
|
|
|
c.localMu.Lock()
|
|
c.localCache = profiles
|
|
c.localMu.Unlock()
|
|
|
|
return profiles, true
|
|
}
|
|
|
|
// Set 设置缓存
|
|
func (c *tlsFingerprintProfileCache) Set(ctx context.Context, profiles []*model.TLSFingerprintProfile) error {
|
|
data, err := json.Marshal(profiles)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if err := c.rdb.Set(ctx, tlsFPProfileCacheKey, data, tlsFPProfileCacheTTL).Err(); err != nil {
|
|
return err
|
|
}
|
|
|
|
c.localMu.Lock()
|
|
c.localCache = profiles
|
|
c.localMu.Unlock()
|
|
|
|
return nil
|
|
}
|
|
|
|
// Invalidate 使缓存失效
|
|
func (c *tlsFingerprintProfileCache) Invalidate(ctx context.Context) error {
|
|
c.localMu.Lock()
|
|
c.localCache = nil
|
|
c.localMu.Unlock()
|
|
|
|
return c.rdb.Del(ctx, tlsFPProfileCacheKey).Err()
|
|
}
|
|
|
|
// NotifyUpdate 通知其他实例刷新缓存
|
|
func (c *tlsFingerprintProfileCache) NotifyUpdate(ctx context.Context) error {
|
|
return c.rdb.Publish(ctx, tlsFPProfilePubSubKey, "refresh").Err()
|
|
}
|
|
|
|
// SubscribeUpdates 订阅缓存更新通知
|
|
func (c *tlsFingerprintProfileCache) SubscribeUpdates(ctx context.Context, handler func()) {
|
|
go func() {
|
|
defer func() {
|
|
if r := recover(); r != nil {
|
|
slog.Error("tls_fp_subscriber_panic", "panic", r)
|
|
}
|
|
}()
|
|
sub := c.rdb.Subscribe(ctx, tlsFPProfilePubSubKey)
|
|
defer func() { _ = sub.Close() }()
|
|
|
|
ch := sub.Channel()
|
|
for {
|
|
select {
|
|
case <-ctx.Done():
|
|
slog.Debug("tls_fp_profile_cache_subscriber_stopped", "reason", "context_done")
|
|
return
|
|
case msg := <-ch:
|
|
if msg == nil {
|
|
slog.Warn("tls_fp_profile_cache_subscriber_stopped", "reason", "channel_closed")
|
|
return
|
|
}
|
|
c.localMu.Lock()
|
|
c.localCache = nil
|
|
c.localMu.Unlock()
|
|
|
|
handler()
|
|
}
|
|
}
|
|
}()
|
|
}
|