// Copyright 2026 tamakyi. All rights reserved.
// Use of this source code is governed by a MIT-style
// license that can be found in the LICENSE file.

// Package cache 提供进程内、并发安全的短 TTL 缓存。
//
// 适用场景：结果聚合统计等"非强一致"数据，用短 TTL 换取数据库实时重算压力的显著降低
// （投票结果页、探索页热门投票等）。需注意：进程内存缓存在多实例横向扩容时不会共享，
// 因此仅用于可接受短暂陈旧的数据；强一致数据请勿使用本缓存。
package cache

import (
	"sync"
	"time"
)

type ttlEntry struct {
	value     interface{}
	expiresAt time.Time
}

// TTLCache 并发安全的带 TTL 进程内缓存。
type TTLCache struct {
	mu       sync.RWMutex
	items    map[string]ttlEntry
	maxItems int
}

// NewTTLCache 创建缓存实例。maxItems<=0 时使用默认容量 256；达到容量上限时，
// 先淘汰过期项，若仍满则整表清空（避免内存无限增长，短 TTL 下过期项通常已占多数）。
func NewTTLCache(maxItems int) *TTLCache {
	if maxItems <= 0 {
		maxItems = 256
	}
	return &TTLCache{
		items:    make(map[string]ttlEntry, maxItems),
		maxItems: maxItems,
	}
}

// Get 读取缓存；命中且未过期返回值，否则返回 (nil, false)。
func (c *TTLCache) Get(key string) (interface{}, bool) {
	c.mu.RLock()
	e, ok := c.items[key]
	c.mu.RUnlock()
	if !ok {
		return nil, false
	}
	if time.Now().After(e.expiresAt) {
		c.mu.Lock()
		// 双重检查，避免并发重复删除（无影响，仅兜底）
		if cur, still := c.items[key]; still && time.Now().After(cur.expiresAt) {
			delete(c.items, key)
		}
		c.mu.Unlock()
		return nil, false
	}
	return e.value, true
}

// Set 写入缓存并设定过期时间。
func (c *TTLCache) Set(key string, value interface{}, ttl time.Duration) {
	c.mu.Lock()
	defer c.mu.Unlock()
	if len(c.items) >= c.maxItems {
		// 先淘汰过期项
		now := time.Now()
		for k, v := range c.items {
			if now.After(v.expiresAt) {
				delete(c.items, k)
			}
		}
		// 仍满（多为短 TTL 下瞬时高频写入）：整表重建，保证写入成功且不 OOM
		if len(c.items) >= c.maxItems {
			c.items = make(map[string]ttlEntry, c.maxItems)
		}
	}
	c.items[key] = ttlEntry{value: value, expiresAt: time.Now().Add(ttl)}
}

// Delete 删除指定键的缓存项；键不存在时静默返回。
// 用于写穿式失效：数据变更（如新增投票提交）后立即清除对应缓存，避免结果页短暂陈旧。
func (c *TTLCache) Delete(key string) {
	c.mu.Lock()
	defer c.mu.Unlock()
	delete(c.items, key)
}