f657509a by astaxie

Merge pull request #693 from fuxiaohei/develop

code style simplify
2 parents 15759f60 77c40e6f
...@@ -7,7 +7,7 @@ ...@@ -7,7 +7,7 @@
7 // @license http://github.com/astaxie/beego/blob/master/LICENSE 7 // @license http://github.com/astaxie/beego/blob/master/LICENSE
8 // 8 //
9 // @authors astaxie 9 // @authors astaxie
10 package cache 10 package redis
11 11
12 import ( 12 import (
13 "encoding/json" 13 "encoding/json"
...@@ -46,23 +46,21 @@ func (rc *RedisCache) do(commandName string, args ...interface{}) (reply interfa ...@@ -46,23 +46,21 @@ func (rc *RedisCache) do(commandName string, args ...interface{}) (reply interfa
46 46
47 // Get cache from redis. 47 // Get cache from redis.
48 func (rc *RedisCache) Get(key string) interface{} { 48 func (rc *RedisCache) Get(key string) interface{} {
49 v, err := rc.do("GET", key) 49 if v, err := rc.do("GET", key); err == nil {
50 if err != nil {
51 return nil
52 }
53
54 return v 50 return v
51 }
52 return nil
55 } 53 }
56 54
57 // put cache to redis. 55 // put cache to redis.
58 func (rc *RedisCache) Put(key string, val interface{}, timeout int64) error { 56 func (rc *RedisCache) Put(key string, val interface{}, timeout int64) error {
59 _, err := rc.do("SET", key, val) 57 var err error
60 if err != nil { 58 if _, err = rc.do("SET", key, val); err != nil {
61 return nil 59 return err
62 } 60 }
63 _, err = rc.do("HSET", rc.key, key, true) 61
64 if err != nil { 62 if _, err = rc.do("HSET", rc.key, key, true); err != nil {
65 return nil 63 return err
66 } 64 }
67 _, err = rc.do("EXPIRE", key, timeout) 65 _, err = rc.do("EXPIRE", key, timeout)
68 return err 66 return err
...@@ -70,9 +68,9 @@ func (rc *RedisCache) Put(key string, val interface{}, timeout int64) error { ...@@ -70,9 +68,9 @@ func (rc *RedisCache) Put(key string, val interface{}, timeout int64) error {
70 68
71 // delete cache in redis. 69 // delete cache in redis.
72 func (rc *RedisCache) Delete(key string) error { 70 func (rc *RedisCache) Delete(key string) error {
73 _, err := rc.do("DEL", key) 71 var err error
74 if err != nil { 72 if _, err = rc.do("DEL", key); err != nil {
75 return nil 73 return err
76 } 74 }
77 _, err = rc.do("HDEL", rc.key, key) 75 _, err = rc.do("HDEL", rc.key, key)
78 return err 76 return err
...@@ -85,8 +83,7 @@ func (rc *RedisCache) IsExist(key string) bool { ...@@ -85,8 +83,7 @@ func (rc *RedisCache) IsExist(key string) bool {
85 return false 83 return false
86 } 84 }
87 if v == false { 85 if v == false {
88 _, err := rc.do("HDEL", rc.key, key) 86 if _, err = rc.do("HDEL", rc.key, key); err != nil {
89 if err != nil {
90 return false 87 return false
91 } 88 }
92 } 89 }
...@@ -108,10 +105,12 @@ func (rc *RedisCache) Decr(key string) error { ...@@ -108,10 +105,12 @@ func (rc *RedisCache) Decr(key string) error {
108 // clean all cache in redis. delete this redis collection. 105 // clean all cache in redis. delete this redis collection.
109 func (rc *RedisCache) ClearAll() error { 106 func (rc *RedisCache) ClearAll() error {
110 cachedKeys, err := redis.Strings(rc.do("HKEYS", rc.key)) 107 cachedKeys, err := redis.Strings(rc.do("HKEYS", rc.key))
111 for _, str := range cachedKeys {
112 _, err := rc.do("DEL", str)
113 if err != nil { 108 if err != nil {
114 return nil 109 return err
110 }
111 for _, str := range cachedKeys {
112 if _, err = rc.do("DEL", str); err != nil {
113 return err
115 } 114 }
116 } 115 }
117 _, err = rc.do("DEL", rc.key) 116 _, err = rc.do("DEL", rc.key)
...@@ -140,26 +139,21 @@ func (rc *RedisCache) StartAndGC(config string) error { ...@@ -140,26 +139,21 @@ func (rc *RedisCache) StartAndGC(config string) error {
140 139
141 c := rc.p.Get() 140 c := rc.p.Get()
142 defer c.Close() 141 defer c.Close()
143 if err := c.Err(); err != nil {
144 return err
145 }
146 142
147 return nil 143 return c.Err()
148 } 144 }
149 145
150 // connect to redis. 146 // connect to redis.
151 func (rc *RedisCache) connectInit() { 147 func (rc *RedisCache) connectInit() {
148 dialFunc := func() (c redis.Conn, err error) {
149 c, err = redis.Dial("tcp", rc.conninfo)
150 return
151 }
152 // initialize a new pool 152 // initialize a new pool
153 rc.p = &redis.Pool{ 153 rc.p = &redis.Pool{
154 MaxIdle: 3, 154 MaxIdle: 3,
155 IdleTimeout: 180 * time.Second, 155 IdleTimeout: 180 * time.Second,
156 Dial: func() (redis.Conn, error) { 156 Dial: dialFunc,
157 c, err := redis.Dial("tcp", rc.conninfo)
158 if err != nil {
159 return nil, err
160 }
161 return c, nil
162 },
163 } 157 }
164 } 158 }
165 159
......
...@@ -8,7 +8,7 @@ ...@@ -8,7 +8,7 @@
8 8
9 // @authors astaxie 9 // @authors astaxie
10 10
11 package cache 11 package redis
12 12
13 import ( 13 import (
14 "testing" 14 "testing"
......
Styling with Markdown is supported
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!