mirror of
https://github.com/astaxie/beego.git
synced 2024-11-16 13:00:55 +00:00
commit
f657509a42
56
cache/redis/redis.go
vendored
56
cache/redis/redis.go
vendored
@ -7,7 +7,7 @@
|
|||||||
// @license http://github.com/astaxie/beego/blob/master/LICENSE
|
// @license http://github.com/astaxie/beego/blob/master/LICENSE
|
||||||
//
|
//
|
||||||
// @authors astaxie
|
// @authors astaxie
|
||||||
package cache
|
package redis
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
@ -46,23 +46,21 @@ func (rc *RedisCache) do(commandName string, args ...interface{}) (reply interfa
|
|||||||
|
|
||||||
// Get cache from redis.
|
// Get cache from redis.
|
||||||
func (rc *RedisCache) Get(key string) interface{} {
|
func (rc *RedisCache) Get(key string) interface{} {
|
||||||
v, err := rc.do("GET", key)
|
if v, err := rc.do("GET", key); err == nil {
|
||||||
if err != nil {
|
return v
|
||||||
return nil
|
|
||||||
}
|
}
|
||||||
|
return nil
|
||||||
return v
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// put cache to redis.
|
// put cache to redis.
|
||||||
func (rc *RedisCache) Put(key string, val interface{}, timeout int64) error {
|
func (rc *RedisCache) Put(key string, val interface{}, timeout int64) error {
|
||||||
_, err := rc.do("SET", key, val)
|
var err error
|
||||||
if err != nil {
|
if _, err = rc.do("SET", key, val); err != nil {
|
||||||
return nil
|
return err
|
||||||
}
|
}
|
||||||
_, err = rc.do("HSET", rc.key, key, true)
|
|
||||||
if err != nil {
|
if _, err = rc.do("HSET", rc.key, key, true); err != nil {
|
||||||
return nil
|
return err
|
||||||
}
|
}
|
||||||
_, err = rc.do("EXPIRE", key, timeout)
|
_, err = rc.do("EXPIRE", key, timeout)
|
||||||
return err
|
return err
|
||||||
@ -70,9 +68,9 @@ func (rc *RedisCache) Put(key string, val interface{}, timeout int64) error {
|
|||||||
|
|
||||||
// delete cache in redis.
|
// delete cache in redis.
|
||||||
func (rc *RedisCache) Delete(key string) error {
|
func (rc *RedisCache) Delete(key string) error {
|
||||||
_, err := rc.do("DEL", key)
|
var err error
|
||||||
if err != nil {
|
if _, err = rc.do("DEL", key); err != nil {
|
||||||
return nil
|
return err
|
||||||
}
|
}
|
||||||
_, err = rc.do("HDEL", rc.key, key)
|
_, err = rc.do("HDEL", rc.key, key)
|
||||||
return err
|
return err
|
||||||
@ -85,8 +83,7 @@ func (rc *RedisCache) IsExist(key string) bool {
|
|||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
if v == false {
|
if v == false {
|
||||||
_, err := rc.do("HDEL", rc.key, key)
|
if _, err = rc.do("HDEL", rc.key, key); err != nil {
|
||||||
if err != nil {
|
|
||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -108,10 +105,12 @@ func (rc *RedisCache) Decr(key string) error {
|
|||||||
// clean all cache in redis. delete this redis collection.
|
// clean all cache in redis. delete this redis collection.
|
||||||
func (rc *RedisCache) ClearAll() error {
|
func (rc *RedisCache) ClearAll() error {
|
||||||
cachedKeys, err := redis.Strings(rc.do("HKEYS", rc.key))
|
cachedKeys, err := redis.Strings(rc.do("HKEYS", rc.key))
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
for _, str := range cachedKeys {
|
for _, str := range cachedKeys {
|
||||||
_, err := rc.do("DEL", str)
|
if _, err = rc.do("DEL", str); err != nil {
|
||||||
if err != nil {
|
return err
|
||||||
return nil
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
_, err = rc.do("DEL", rc.key)
|
_, err = rc.do("DEL", rc.key)
|
||||||
@ -140,26 +139,21 @@ func (rc *RedisCache) StartAndGC(config string) error {
|
|||||||
|
|
||||||
c := rc.p.Get()
|
c := rc.p.Get()
|
||||||
defer c.Close()
|
defer c.Close()
|
||||||
if err := c.Err(); err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
return nil
|
return c.Err()
|
||||||
}
|
}
|
||||||
|
|
||||||
// connect to redis.
|
// connect to redis.
|
||||||
func (rc *RedisCache) connectInit() {
|
func (rc *RedisCache) connectInit() {
|
||||||
|
dialFunc := func() (c redis.Conn, err error) {
|
||||||
|
c, err = redis.Dial("tcp", rc.conninfo)
|
||||||
|
return
|
||||||
|
}
|
||||||
// initialize a new pool
|
// initialize a new pool
|
||||||
rc.p = &redis.Pool{
|
rc.p = &redis.Pool{
|
||||||
MaxIdle: 3,
|
MaxIdle: 3,
|
||||||
IdleTimeout: 180 * time.Second,
|
IdleTimeout: 180 * time.Second,
|
||||||
Dial: func() (redis.Conn, error) {
|
Dial: dialFunc,
|
||||||
c, err := redis.Dial("tcp", rc.conninfo)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
return c, nil
|
|
||||||
},
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
14
cache/redis/redis_test.go
vendored
14
cache/redis/redis_test.go
vendored
@ -8,7 +8,7 @@
|
|||||||
|
|
||||||
// @authors astaxie
|
// @authors astaxie
|
||||||
|
|
||||||
package cache
|
package redis
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"testing"
|
"testing"
|
||||||
@ -20,7 +20,7 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
func TestRedisCache(t *testing.T) {
|
func TestRedisCache(t *testing.T) {
|
||||||
bm, err := cache.NewCache("redis", `{"conn": "127.0.0.1:6379"}`)
|
bm, err := cache.NewCache("redis", `{"conn": "127.0.0.1:6379"}`)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Error("init err")
|
t.Error("init err")
|
||||||
}
|
}
|
||||||
@ -48,7 +48,7 @@ func TestRedisCache(t *testing.T) {
|
|||||||
t.Error("Incr Error", err)
|
t.Error("Incr Error", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
if v, _ := redis.Int(bm.Get("astaxie"), err); v != 2 {
|
if v, _ := redis.Int(bm.Get("astaxie"), err); v != 2 {
|
||||||
t.Error("get err")
|
t.Error("get err")
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -74,8 +74,8 @@ func TestRedisCache(t *testing.T) {
|
|||||||
if v, _ := redis.String(bm.Get("astaxie"), err); v != "author" {
|
if v, _ := redis.String(bm.Get("astaxie"), err); v != "author" {
|
||||||
t.Error("get err")
|
t.Error("get err")
|
||||||
}
|
}
|
||||||
// test clear all
|
// test clear all
|
||||||
if err = bm.ClearAll(); err != nil {
|
if err = bm.ClearAll(); err != nil {
|
||||||
t.Error("clear all err")
|
t.Error("clear all err")
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user