132 lines
2.7 KiB
Go
132 lines
2.7 KiB
Go
package session
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/garyburd/redigo/redis"
|
|
)
|
|
|
|
var redispder = &RedisProvider{}
|
|
|
|
var MAX_POOL_SIZE = 20
|
|
|
|
var redisPool chan redis.Conn
|
|
|
|
type RedisSessionStore struct {
|
|
c redis.Conn
|
|
sid string
|
|
}
|
|
|
|
func (rs *RedisSessionStore) Set(key, value interface{}) error {
|
|
//_, err := rs.c.Do("HSET", rs.sid, key, value)
|
|
_, err := rs.c.Do("HSET", rs.sid, key, value)
|
|
return err
|
|
}
|
|
|
|
func (rs *RedisSessionStore) Get(key interface{}) interface{} {
|
|
reply, err := rs.c.Do("HGET", rs.sid, key)
|
|
if err != nil {
|
|
return nil
|
|
}
|
|
return reply
|
|
}
|
|
|
|
func (rs *RedisSessionStore) Delete(key interface{}) error {
|
|
//_, err := rs.c.Do("HDEL", rs.sid, key)
|
|
_, err := rs.c.Do("HDEL", rs.sid, key)
|
|
return err
|
|
}
|
|
|
|
func (rs *RedisSessionStore) Flush() error {
|
|
_, err := rs.c.Do("DEL", rs.sid)
|
|
return err
|
|
}
|
|
|
|
func (rs *RedisSessionStore) SessionID() string {
|
|
return rs.sid
|
|
}
|
|
|
|
func (rs *RedisSessionStore) SessionRelease() {
|
|
rs.c.Close()
|
|
}
|
|
|
|
type RedisProvider struct {
|
|
maxlifetime int64
|
|
savePath string
|
|
}
|
|
|
|
func (rp *RedisProvider) connectInit() redis.Conn {
|
|
/*c, err := redis.Dial("tcp", rp.savePath)
|
|
if err != nil {
|
|
return nil
|
|
}
|
|
return c*/
|
|
//if redisPool == nil {
|
|
redisPool = make(chan redis.Conn, MAX_POOL_SIZE)
|
|
//}
|
|
if len(redisPool) == 0 {
|
|
go func() {
|
|
for i := 0; i < MAX_POOL_SIZE/2; i++ {
|
|
c, err := redis.Dial("tcp", rp.savePath)
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
return
|
|
}
|
|
putRedis(c)
|
|
}
|
|
}()
|
|
}
|
|
return <-redisPool
|
|
}
|
|
|
|
func putRedis(conn redis.Conn) {
|
|
if redisPool == nil {
|
|
redisPool = make(chan redis.Conn, MAX_POOL_SIZE)
|
|
}
|
|
if len(redisPool) >= MAX_POOL_SIZE {
|
|
conn.Close()
|
|
return
|
|
}
|
|
redisPool <- conn
|
|
}
|
|
|
|
func (rp *RedisProvider) SessionInit(maxlifetime int64, savePath string) error {
|
|
rp.maxlifetime = maxlifetime
|
|
rp.savePath = savePath
|
|
return nil
|
|
}
|
|
|
|
func (rp *RedisProvider) SessionRead(sid string) (SessionStore, error) {
|
|
c := rp.connectInit()
|
|
//if str, err := redis.String(c.Do("GET", sid)); err != nil || str == "" {
|
|
if str, err := redis.String(c.Do("HGET", sid, sid)); err != nil || str == "" {
|
|
//c.Do("SET", sid, sid, rp.maxlifetime)
|
|
c.Do("HSET", sid, sid, rp.maxlifetime)
|
|
}
|
|
rs := &RedisSessionStore{c: c, sid: sid}
|
|
return rs, nil
|
|
}
|
|
|
|
func (rp *RedisProvider) SessionRegenerate(oldsid, sid string) (SessionStore, error) {
|
|
c := rp.connectInit()
|
|
if str, err := redis.String(c.Do("HGET", oldsid, oldsid)); err != nil || str == "" {
|
|
c.Do("HSET", oldsid, oldsid, rp.maxlifetime)
|
|
}
|
|
c.Do("RENAME", oldsid, sid)
|
|
rs := &RedisSessionStore{c: c, sid: sid}
|
|
return rs, nil
|
|
}
|
|
|
|
func (rp *RedisProvider) SessionDestroy(sid string) error {
|
|
c := rp.connectInit()
|
|
c.Do("DEL", sid)
|
|
return nil
|
|
}
|
|
|
|
func (rp *RedisProvider) SessionGC() {
|
|
return
|
|
}
|
|
|
|
func init() {
|
|
Register("redis", redispder)
|
|
}
|