gomod/cache
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

v1.4.1 add pool get timeout

Browse Source
This commit is contained in:
bryanqiu 2024-05-16 17:54:09 +08:00
parent 5298a2cabf
commit 499d8b2f00
1 changed files with 19 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
cache.go
View File

@ -1,6 +1,7 @@
package cache package cache
import ( import (
"context"
"encoding/json" "encoding/json"
"errors" "errors"
"fmt" "fmt"
@ -15,6 +16,7 @@ import (
var ( var (
ErrExpired = errors.New("expired") ErrExpired = errors.New("expired")
ErrNotFound = errors.New("not found") ErrNotFound = errors.New("not found")
ErrTimeout = errors.New("not found(timeout)")
OptWithoutGetter = optWithoutGetter{} // for get only OptWithoutGetter = optWithoutGetter{} // for get only
OptWithCreateTime = optWithCreateTime{} // for set & get OptWithCreateTime = optWithCreateTime{} // for set & get
OptWithRedisConn = optWithRedisConn{} // for set & get OptWithRedisConn = optWithRedisConn{} // for set & get
@ -236,7 +238,14 @@ func (c *redisCacher[T]) GetFromCache(id string, options ...Option) (*T, error)
if opt.withRedisConn != nil && opt.withRedisConn.redisconn != nil { if opt.withRedisConn != nil && opt.withRedisConn.redisconn != nil {
rds = opt.withRedisConn.redisconn rds = opt.withRedisConn.redisconn
} else if c.rdsPool != nil { } else if c.rdsPool != nil {
rds = c.rdsPool.Get() // timeout context
ctx, cancel := context.WithTimeout(context.Background(), 15*time.Second)
defer cancel()
// redis conn from pool
rds, err = c.rdsPool.GetContext(ctx)
if err != nil {
return nil, ErrTimeout
}
defer rds.Close() defer rds.Close()
} else if c.rds != nil { } else if c.rds != nil {
rds = c.rds rds = c.rds
@ -296,7 +305,14 @@ func (c *redisCacher[T]) SetIntoCache(id string, t *T, options ...Option) error
if opt.withRedisConn != nil && opt.withRedisConn.redisconn != nil { if opt.withRedisConn != nil && opt.withRedisConn.redisconn != nil {
rds = opt.withRedisConn.redisconn rds = opt.withRedisConn.redisconn
} else if c.rdsPool != nil { } else if c.rdsPool != nil {
rds = c.rdsPool.Get() // timeout context
ctx, cancel := context.WithTimeout(context.Background(), 15*time.Second)
defer cancel()
// redis conn from pool
rds, err = c.rdsPool.GetContext(ctx)
if err != nil {
return errors.New("set cache failed: 'get redis conn timeout'")
}
defer rds.Close() defer rds.Close()
} else if c.rds != nil { } else if c.rds != nil {
rds = c.rds rds = c.rds
@ -364,7 +380,7 @@ func (c *cacher[T]) GetFromCache(id string, options ...Option) (dat *T, err erro
log.Infof("get cache(id:%s) from redisCacher success(but expired), need try next", id) log.Infof("get cache(id:%s) from redisCacher success(but expired), need try next", id)
lastExpiredDat = dat lastExpiredDat = dat
} else { } else {
log.Infof("get cache(id:%s) from redisCacher failed, need try next", id) log.Infof("get cache(id:%s) from redisCacher failed, need try from next cacher", id)
} }
} }