Skip to content
Projects
Groups
Snippets
Help
Loading...
Sign in
Toggle navigation
B
beego
Project
Project
Details
Activity
Cycle Analytics
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
Issues
0
Issues
0
List
Board
Labels
Milestones
Merge Requests
0
Merge Requests
0
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Charts
Create a new issue
Commits
Issue Boards
Open sidebar
go
beego
Commits
63f19974
Commit
63f19974
authored
Jan 11, 2014
by
astaxie
Browse files
Options
Browse Files
Download
Plain Diff
Merge pull request #460 from pengfei-xue/develop
use connection pool for redis, support auto connection
parents
6e9ba0ea
cb55009c
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
38 additions
and
116 deletions
+38
-116
redis.go
cache/redis.go
+38
-116
No files found.
cache/redis.go
View file @
63f19974
...
...
@@ -3,7 +3,7 @@ package cache
import
(
"encoding/json"
"errors"
"
io
"
"
time
"
"github.com/beego/redigo/redis"
)
...
...
@@ -15,7 +15,7 @@ var (
// Redis cache adapter.
type
RedisCache
struct
{
c
redis
.
Conn
p
*
redis
.
Pool
// redis connection pool
conninfo
string
key
string
}
...
...
@@ -25,23 +25,17 @@ func NewRedisCache() *RedisCache {
return
&
RedisCache
{
key
:
DefaultKey
}
}
// Get cache from redis.
func
(
rc
*
RedisCache
)
Get
(
key
string
)
interface
{}
{
if
rc
.
c
==
nil
{
var
err
error
rc
.
c
,
err
=
rc
.
connectInit
()
if
err
!=
nil
{
return
nil
}
}
// actually do the redis cmds
func
(
rc
*
RedisCache
)
do
(
commandName
string
,
args
...
interface
{})
(
reply
interface
{},
err
error
)
{
c
:=
rc
.
p
.
Get
()
defer
c
.
Close
()
v
,
err
:=
rc
.
c
.
Do
(
"HGET"
,
rc
.
key
,
key
)
// write to closed socket, reset rc.c to nil
if
err
==
io
.
EOF
{
rc
.
c
=
nil
return
nil
}
return
c
.
Do
(
commandName
,
args
...
)
}
// Get cache from redis.
func
(
rc
*
RedisCache
)
Get
(
key
string
)
interface
{}
{
v
,
err
:=
rc
.
do
(
"HGET"
,
rc
.
key
,
key
)
if
err
!=
nil
{
return
nil
}
...
...
@@ -52,61 +46,19 @@ func (rc *RedisCache) Get(key string) interface{} {
// put cache to redis.
// timeout is ignored.
func
(
rc
*
RedisCache
)
Put
(
key
string
,
val
interface
{},
timeout
int64
)
error
{
if
rc
.
c
==
nil
{
var
err
error
rc
.
c
,
err
=
rc
.
connectInit
()
if
err
!=
nil
{
return
err
}
}
_
,
err
:=
rc
.
c
.
Do
(
"HSET"
,
rc
.
key
,
key
,
val
)
// write to closed socket, reset rc.c to nil
if
err
==
io
.
EOF
{
rc
.
c
=
nil
return
err
}
_
,
err
:=
rc
.
do
(
"HSET"
,
rc
.
key
,
key
,
val
)
return
err
}
// delete cache in redis.
func
(
rc
*
RedisCache
)
Delete
(
key
string
)
error
{
if
rc
.
c
==
nil
{
var
err
error
rc
.
c
,
err
=
rc
.
connectInit
()
if
err
!=
nil
{
return
err
}
}
_
,
err
:=
rc
.
c
.
Do
(
"HDEL"
,
rc
.
key
,
key
)
// write to closed socket, reset rc.c to nil
if
err
==
io
.
EOF
{
rc
.
c
=
nil
return
err
}
_
,
err
:=
rc
.
do
(
"HDEL"
,
rc
.
key
,
key
)
return
err
}
// check cache exist in redis.
func
(
rc
*
RedisCache
)
IsExist
(
key
string
)
bool
{
if
rc
.
c
==
nil
{
var
err
error
rc
.
c
,
err
=
rc
.
connectInit
()
if
err
!=
nil
{
return
false
}
}
v
,
err
:=
redis
.
Bool
(
rc
.
c
.
Do
(
"HEXISTS"
,
rc
.
key
,
key
))
// write to closed socket, reset rc.c to nil
if
err
==
io
.
EOF
{
rc
.
c
=
nil
return
false
}
v
,
err
:=
redis
.
Bool
(
rc
.
do
(
"HEXISTS"
,
rc
.
key
,
key
))
if
err
!=
nil
{
return
false
}
...
...
@@ -116,59 +68,19 @@ func (rc *RedisCache) IsExist(key string) bool {
// increase counter in redis.
func
(
rc
*
RedisCache
)
Incr
(
key
string
)
error
{
if
rc
.
c
==
nil
{
var
err
error
rc
.
c
,
err
=
rc
.
connectInit
()
if
err
!=
nil
{
return
err
}
}
_
,
err
:=
redis
.
Bool
(
rc
.
c
.
Do
(
"HINCRBY"
,
rc
.
key
,
key
,
1
))
// write to closed socket
if
err
==
io
.
EOF
{
rc
.
c
=
nil
}
_
,
err
:=
redis
.
Bool
(
rc
.
do
(
"HINCRBY"
,
rc
.
key
,
key
,
1
))
return
err
}
// decrease counter in redis.
func
(
rc
*
RedisCache
)
Decr
(
key
string
)
error
{
if
rc
.
c
==
nil
{
var
err
error
rc
.
c
,
err
=
rc
.
connectInit
()
if
err
!=
nil
{
return
err
}
}
_
,
err
:=
redis
.
Bool
(
rc
.
c
.
Do
(
"HINCRBY"
,
rc
.
key
,
key
,
-
1
))
// write to closed socket
if
err
==
io
.
EOF
{
rc
.
c
=
nil
}
_
,
err
:=
redis
.
Bool
(
rc
.
do
(
"HINCRBY"
,
rc
.
key
,
key
,
-
1
))
return
err
}
// clean all cache in redis. delete this redis collection.
func
(
rc
*
RedisCache
)
ClearAll
()
error
{
if
rc
.
c
==
nil
{
var
err
error
rc
.
c
,
err
=
rc
.
connectInit
()
if
err
!=
nil
{
return
err
}
}
_
,
err
:=
rc
.
c
.
Do
(
"DEL"
,
rc
.
key
)
// write to closed socket
if
err
==
io
.
EOF
{
rc
.
c
=
nil
}
_
,
err
:=
rc
.
do
(
"DEL"
,
rc
.
key
)
return
err
}
...
...
@@ -179,32 +91,42 @@ func (rc *RedisCache) ClearAll() error {
func
(
rc
*
RedisCache
)
StartAndGC
(
config
string
)
error
{
var
cf
map
[
string
]
string
json
.
Unmarshal
([]
byte
(
config
),
&
cf
)
if
_
,
ok
:=
cf
[
"key"
];
!
ok
{
cf
[
"key"
]
=
DefaultKey
}
if
_
,
ok
:=
cf
[
"conn"
];
!
ok
{
return
errors
.
New
(
"config has no conn key"
)
}
rc
.
key
=
cf
[
"key"
]
rc
.
conninfo
=
cf
[
"conn"
]
var
err
error
rc
.
c
,
err
=
rc
.
connectInit
()
if
err
!=
nil
{
rc
.
connectInit
()
c
:=
rc
.
p
.
Get
()
defer
c
.
Close
()
if
err
:=
c
.
Err
();
err
!=
nil
{
return
err
}
if
rc
.
c
==
nil
{
return
errors
.
New
(
"dial tcp conn error"
)
}
return
nil
}
// connect to redis.
func
(
rc
*
RedisCache
)
connectInit
()
(
redis
.
Conn
,
error
)
{
c
,
err
:=
redis
.
Dial
(
"tcp"
,
rc
.
conninfo
)
if
err
!=
nil
{
return
nil
,
err
func
(
rc
*
RedisCache
)
connectInit
()
{
// initialize a new pool
rc
.
p
=
&
redis
.
Pool
{
MaxIdle
:
3
,
IdleTimeout
:
180
*
time
.
Second
,
Dial
:
func
()
(
redis
.
Conn
,
error
)
{
c
,
err
:=
redis
.
Dial
(
"tcp"
,
rc
.
conninfo
)
if
err
!=
nil
{
return
nil
,
err
}
return
c
,
nil
},
}
return
c
,
nil
}
func
init
()
{
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment