53 lines
1.1 KiB
Go
53 lines
1.1 KiB
Go
package cache
|
||
|
||
import (
|
||
"strings"
|
||
"sync"
|
||
"time"
|
||
)
|
||
|
||
// memoryFallback Redis 不可用时的内存备用缓存，保证服务可用
|
||
var (
|
||
memoryMu sync.RWMutex
|
||
memoryData = make(map[string]*memoryEntry)
|
||
)
|
||
|
||
type memoryEntry struct {
|
||
Data []byte
|
||
Expiry time.Time
|
||
}
|
||
|
||
func memoryGet(key string) ([]byte, bool) {
|
||
memoryMu.RLock()
|
||
defer memoryMu.RUnlock()
|
||
e, ok := memoryData[key]
|
||
if !ok || e == nil || time.Now().After(e.Expiry) {
|
||
return nil, false
|
||
}
|
||
return e.Data, true
|
||
}
|
||
|
||
func memorySet(key string, data []byte, ttl time.Duration) {
|
||
memoryMu.Lock()
|
||
defer memoryMu.Unlock()
|
||
memoryData[key] = &memoryEntry{Data: data, Expiry: time.Now().Add(ttl)}
|
||
}
|
||
|
||
func memoryDel(key string) {
|
||
memoryMu.Lock()
|
||
defer memoryMu.Unlock()
|
||
delete(memoryData, key)
|
||
}
|
||
|
||
// memoryDelPattern 按前缀删除（pattern 如 soul:book:chapters-by-part:* 转为前缀 soul:book:chapters-by-part:）
|
||
func memoryDelPattern(pattern string) {
|
||
prefix := strings.TrimSuffix(pattern, "*")
|
||
memoryMu.Lock()
|
||
defer memoryMu.Unlock()
|
||
for k := range memoryData {
|
||
if strings.HasPrefix(k, prefix) {
|
||
delete(memoryData, k)
|
||
}
|
||
}
|
||
}
|