54 lines
983 B
Go
54 lines
983 B
Go
|
package cachemanager
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"github.com/fthvgb1/wp-go/cache"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
var ctx = context.Background()
|
||
|
|
|
||
|
|
type flush interface {
|
||
|
|
Flush(ctx context.Context)
|
||
|
|
}
|
||
|
|
|
||
|
|
type clear interface {
|
||
|
|
ClearExpired(ctx context.Context)
|
||
|
|
}
|
||
|
|
|
||
|
|
var clears []clear
|
||
|
|
|
||
|
|
var flushes []flush
|
||
|
|
|
||
|
|
func Flush() {
|
||
|
|
for _, f := range flushes {
|
||
|
|
f.Flush(ctx)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func MapCacheBy[K comparable, V any](fn func(...any) (V, error), expireTime time.Duration) *cache.MapCache[K, V] {
|
||
|
|
m := cache.NewMemoryMapCacheByFn[K, V](fn, expireTime)
|
||
|
|
FlushPush(m)
|
||
|
|
ClearPush(m)
|
||
|
|
return m
|
||
|
|
}
|
||
|
|
func MapBatchCacheBy[K comparable, V any](fn func(...any) (map[K]V, error), expireTime time.Duration) *cache.MapCache[K, V] {
|
||
|
|
m := cache.NewMemoryMapCacheByBatchFn[K, V](fn, expireTime)
|
||
|
|
FlushPush(m)
|
||
|
|
ClearPush(m)
|
||
|
|
return m
|
||
|
|
}
|
||
|
|
|
||
|
|
func FlushPush(f ...flush) {
|
||
|
|
flushes = append(flushes, f...)
|
||
|
|
}
|
||
|
|
func ClearPush(c ...clear) {
|
||
|
|
clears = append(clears, c...)
|
||
|
|
}
|
||
|
|
|
||
|
|
func ClearExpired() {
|
||
|
|
for _, c := range clears {
|
||
|
|
c.ClearExpired(ctx)
|
||
|
|
}
|
||
|
|
}
|