first upload
This commit is contained in:
bryanqiu
commit
d91b804411
228
cache.go
Normal file
228
cache.go
Normal file
@ -0,0 +1,228 @@
|
||||
package cache
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"fmt"
|
||||
"time"
|
||||
|
||||
"github.com/gomodule/redigo/redis"
|
||||
)
|
||||
|
||||
var (
|
||||
ErrNotFound = errors.New("not found")
|
||||
)
|
||||
|
||||
type Cacher[T any] interface {
|
||||
GetFromCache(id string) (dat *T, err error)
|
||||
SetIntoCache(id string, dat *T) (err error)
|
||||
}
|
||||
|
||||
type Getter[T any] interface {
|
||||
GetById(id string) (dat *T, err error)
|
||||
}
|
||||
|
||||
type Config struct {
|
||||
UseLocalCache bool
|
||||
LocalCacheLifetimeSecond int64
|
||||
UseRedisCache bool
|
||||
RedisCacheConn redis.Conn
|
||||
RedisCacheKeyPrefix string
|
||||
RedisCacheLifetimeSecond int64
|
||||
}
|
||||
|
||||
func NewCache[T any](getter Getter[T], cfg Config) Cacher[T] {
|
||||
var c = new(cacher[T])
|
||||
// Getter
|
||||
c.getter = getter
|
||||
|
||||
// Local cache
|
||||
if cfg.UseLocalCache {
|
||||
c.localCache = new(localCacher[T])
|
||||
c.localCache.cacheItems = map[string]*cacheItem[T]{}
|
||||
c.localCache.cacheSecond = time.Duration(cfg.LocalCacheLifetimeSecond) * time.Second
|
||||
if c.localCache.cacheSecond.Nanoseconds() <= 0 {
|
||||
c.localCache.cacheSecond = 60 * time.Second
|
||||
}
|
||||
}
|
||||
|
||||
// Redis cache
|
||||
if cfg.UseRedisCache && cfg.RedisCacheConn != nil {
|
||||
c.redisCache = new(redisCacher[T])
|
||||
c.redisCache.rds = cfg.RedisCacheConn
|
||||
c.redisCache.cachePrefix = cfg.RedisCacheKeyPrefix
|
||||
c.redisCache.cacheSecond = time.Duration(cfg.RedisCacheLifetimeSecond) * time.Second
|
||||
} else if cfg.UseRedisCache {
|
||||
panic("want to user redis cache, but redis.Conn is nil")
|
||||
}
|
||||
|
||||
return c
|
||||
}
|
||||
|
||||
type cacheItem[T any] struct {
|
||||
Data T //cache data
|
||||
CreateTime time.Time //cache create time
|
||||
}
|
||||
|
||||
type localCacher[T any] struct {
|
||||
cacheItems map[string]*cacheItem[T]
|
||||
cacheSecond time.Duration
|
||||
}
|
||||
|
||||
type redisCacher struct {
|
||||
rds redis.Conn
|
||||
cachePrefix string
|
||||
cacheSecond time.Duration
|
||||
}
|
||||
|
||||
type cacher[T any] struct {
|
||||
getter Getter[T]
|
||||
localCache *localCacher[T]
|
||||
redisCache *redisCacher[T]
|
||||
}
|
||||
|
||||
func (c *localCacher) GetFromCache(id string) (t *T, err error) {
|
||||
var earliestCreateTime = time.Now().Add(-c.cacheSecond)
|
||||
if a, ok := c.cacheItems[appid]; !ok {
|
||||
return nil, ErrAppNotExist
|
||||
} else if a.CreateTime.Before(earliestCreateTime) {
|
||||
return a, nil
|
||||
} else if ok {
|
||||
delete(c.apps, appid)
|
||||
return nil, ErrAppNotExist
|
||||
}
|
||||
}
|
||||
|
||||
func (c *localCacher) SetIntoCache(id string, t *T) (err error) {
|
||||
c.apps[id] = t
|
||||
return nil
|
||||
}
|
||||
|
||||
func (c *redisCacher) GetFromCache(id string) (*T, error) {
|
||||
var (
|
||||
rds = c.rds
|
||||
redisKey = fmt.Sprintf("appconfig-by-appid:%s", appid)
|
||||
redisValue = ""
|
||||
err error = nil
|
||||
)
|
||||
|
||||
// Step 1. read from redis
|
||||
if redisValue, err = redis.String(rds.Do("GET", redisKey)); err != nil {
|
||||
log.Infof("get appconfig cache failed: 'redis return<%s>'", err.Error())
|
||||
return nil, ErrAppNotExist
|
||||
}
|
||||
|
||||
// Step 2. decode from string
|
||||
a := &App{}
|
||||
if err = json.Unmarshal([]byte(redisValue), a); err != nil {
|
||||
log.Errorf("get appconfig cache failed: 'json unmarshal failed <%s>'", err.Error())
|
||||
return nil, ErrAppNotExist
|
||||
}
|
||||
|
||||
// Step 3. check expire time
|
||||
var d = c.cacheSecond
|
||||
if d.Nanoseconds() > 0 {
|
||||
var earliestCreateTime = time.Now().Add(-c.cacheSecond)
|
||||
if a.CreateTime.Before(earliestCreateTime) {
|
||||
log.Infof("app(%s) is in redis cache but expired", appid)
|
||||
return nil, ErrAppNotExist
|
||||
}
|
||||
}
|
||||
|
||||
return a, nil
|
||||
}
|
||||
|
||||
func (c *redisCacher) SetIntoCache(appid string, a *App) error {
|
||||
var (
|
||||
rds = c.rds
|
||||
redisKey = fmt.Sprintf("appconfig-by-appid:%s", appid)
|
||||
redisValue = []byte{}
|
||||
err error = nil
|
||||
)
|
||||
|
||||
// Step 1. decode from string
|
||||
if redisValue, err = json.Marshal(a); err != nil {
|
||||
log.Errorf("set appconfig cache failed: 'json marshal failed <%s>'", err.Error())
|
||||
return errors.New("set appconfig cache failed: 'marshal failed'")
|
||||
}
|
||||
|
||||
// Step 2. read from redis
|
||||
if _, err = rds.Do("SET", redisKey, string(redisValue)); err != nil {
|
||||
log.Errorf("set appconfig cache failed: 'redis failed <%s>'", err.Error())
|
||||
return errors.New("set appconfig cache failed: 'redis failed'")
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (c *cacher[T]) GetFromCache(id string) (dat *T, err error) {
|
||||
// Step 1. check id
|
||||
if len(id) < 8 {
|
||||
return nil, ErrNotFound
|
||||
}
|
||||
|
||||
// Step 2. get from [local]
|
||||
if c.localCache != nil {
|
||||
if dat, err := c.localCache.GetFromCache(id); err == nil {
|
||||
log.Infof("get cache(id:%s) from localCacher success", id)
|
||||
return dat, nil
|
||||
} else {
|
||||
log.Infof("get cache(id:%s) from localCacher failed, try next", id)
|
||||
}
|
||||
}
|
||||
|
||||
// Step 3. get from [redis]
|
||||
if c.redisCache != nil {
|
||||
if dat, err := c.redisCache.GetFromCache(id); err == nil {
|
||||
log.Infof("get cache(id:%s) from redisCacher success", id)
|
||||
return dat, nil
|
||||
} else if c.getter == nil {
|
||||
log.Warningf("get cache(id:%s) from local failed, and storager is nil, "+
|
||||
"trade as not found. may be you forgot set in db ???", id)
|
||||
return nil, ErrAppNotExist
|
||||
}
|
||||
}
|
||||
if c.getter == nil {
|
||||
log.Warninf("get cache(id:%s) from all cache failed, and storager is nil, "+
|
||||
"trade as not found. may be you forgot set in db ???", id)
|
||||
return nil, ErrAppNotExist
|
||||
}
|
||||
|
||||
// Step 4. get from [storager(database or somewhere)]
|
||||
dat, err := c.getter.GetAppId(id)
|
||||
if err != nil {
|
||||
log.Errorf("cache(id:%s) is not in database, something error: %s", id, err)
|
||||
return nil, ErrAppNotExist
|
||||
}
|
||||
dat.CreateTime = time.Now()
|
||||
log.Infof("get cache(id:%s) from exportStorager(maybe database) success", id)
|
||||
|
||||
// Step 5. set to cacha
|
||||
if c.localCache != nil {
|
||||
c.localCache.SetIntoCache(id, dat)
|
||||
}
|
||||
if c.redisCache != nil {
|
||||
c.redisCache.SetIntoCache(id, dat)
|
||||
}
|
||||
|
||||
// Step 6. return
|
||||
log.PrintPretty("success cached/return app:", dat)
|
||||
return dat, nil
|
||||
}
|
||||
|
||||
func (c *cacher[T]) SetIntoCache(id string, dat *T) (err error) {
|
||||
// localCache
|
||||
if c.localCache != nil {
|
||||
if err = c.localCache.SetIntoCache(id, dat); err != nil {
|
||||
log.Infof("set data into localCache failed, err:%s", err)
|
||||
}
|
||||
}
|
||||
|
||||
// redisCache
|
||||
if c.redisCache != nil {
|
||||
if err = c.redisCache.SetIntoCache(id, dat); err != nil {
|
||||
log.Infof("set data into redisCache failed, err:%s", err)
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
Loading…
Reference in New Issue
Block a user