cache/gocache.go

92 lines
2.3 KiB
Go
Raw Normal View History

2024-06-21 14:30:59 +08:00
// Package cache ...
//
// Description : cache ...
//
// Author : go_developer@163.com<白茶清欢>
//
// Date : 2024-06-14 16:50
package cache
import (
"context"
"errors"
cacheAbstract "git.zhangdeman.cn/zhangdeman/cache/abstract"
gocache "github.com/patrickmn/go-cache"
"sync"
"time"
)
// Gocache 基于gocache的内存缓存
//
// Author : go_developer@163.com<白茶清欢>
//
// Date : 16:51 2024/6/14
type Gocache struct{}
func (g *Gocache) Exist(ctx context.Context, cacheInstanceFlag string, key string) (bool, error) {
_, exist := GocacheStorageClient.Get(cacheInstanceFlag).Get(key)
return exist, nil
}
func (g *Gocache) Set(ctx context.Context, cacheInstanceFlag string, key string, value string, ttl int64) error {
now := time.Now()
if now.Unix() < ttl {
ttl = ttl - now.Unix()
}
GocacheStorageClient.Get(cacheInstanceFlag).Set(key, value, time.Duration(ttl)*time.Second)
return nil
}
func (g *Gocache) Get(ctx context.Context, cacheInstanceFlag string, key string) (string, error) {
val, exist := GocacheStorageClient.Get(cacheInstanceFlag).Get(key)
if !exist {
return "", errors.New(key + " : not found")
}
return val.(string), nil
}
func (g *Gocache) GetWithParse(ctx context.Context, cacheInstanceFlag string, key string, parseFunc cacheAbstract.ResultParseFunc) (string, error) {
val, err := g.Get(ctx, cacheInstanceFlag, key)
if nil != err {
return "", err
}
if nil == parseFunc {
return val, nil
}
return val, parseFunc(val)
}
func (g *Gocache) Delete(ctx context.Context, cacheInstanceFlag string, key string) error {
GocacheStorageClient.Get(cacheInstanceFlag).Delete(key)
return nil
}
var GocacheStorageClient = &GocacheStorage{
lock: &sync.RWMutex{},
clientTable: make(map[string]*gocache.Cache),
}
type GocacheStorage struct {
lock *sync.RWMutex
clientTable map[string]*gocache.Cache
}
func (g *GocacheStorage) Add(cacheInstanceFlag string) {
_ = g.Get(cacheInstanceFlag)
}
func (g *GocacheStorage) Del(cacheInstanceFlag string) {
g.lock.Lock()
defer g.lock.Unlock()
delete(g.clientTable, cacheInstanceFlag)
}
func (g *GocacheStorage) Get(cacheInstanceFlag string) *gocache.Cache {
g.lock.Lock()
defer g.lock.Unlock()
if _, exist := g.clientTable[cacheInstanceFlag]; !exist {
g.clientTable[cacheInstanceFlag] = gocache.New(0, 0)
}
return g.clientTable[cacheInstanceFlag]
}