From 20cc5b261e860f7a63a2e08688eba27f0589b085 Mon Sep 17 00:00:00 2001 From: Meaglith Ma Date: Sun, 1 Mar 2015 12:59:34 +0800 Subject: [PATCH] Reform the ledis_session.go --- session/ledis/ledis_session.go | 216 ++++++++++++++++----------------- 1 file changed, 108 insertions(+), 108 deletions(-) diff --git a/session/ledis/ledis_session.go b/session/ledis/ledis_session.go index 60f98fba..643b8817 100644 --- a/session/ledis/ledis_session.go +++ b/session/ledis/ledis_session.go @@ -1,14 +1,14 @@ package session import ( - "net/http" - "strconv" - "strings" - "sync" + "net/http" + "strconv" + "strings" + "sync" - "github.com/astaxie/beego/session" - "github.com/siddontang/ledisdb/config" - "github.com/siddontang/ledisdb/ledis" + "github.com/astaxie/beego/session" + "github.com/siddontang/ledisdb/config" + "github.com/siddontang/ledisdb/ledis" ) var ledispder = &LedisProvider{} @@ -16,165 +16,165 @@ var c *ledis.DB // ledis session store type LedisSessionStore struct { - sid string - lock sync.RWMutex - values map[interface{}]interface{} - maxlifetime int64 + sid string + lock sync.RWMutex + values map[interface{}]interface{} + maxlifetime int64 } // set value in ledis session func (ls *LedisSessionStore) Set(key, value interface{}) error { - ls.lock.Lock() - defer ls.lock.Unlock() - ls.values[key] = value - return nil + ls.lock.Lock() + defer ls.lock.Unlock() + ls.values[key] = value + return nil } // get value in ledis session func (ls *LedisSessionStore) Get(key interface{}) interface{} { - ls.lock.RLock() - defer ls.lock.RUnlock() - if v, ok := ls.values[key]; ok { - return v - } else { - return nil - } + ls.lock.RLock() + defer ls.lock.RUnlock() + if v, ok := ls.values[key]; ok { + return v + } else { + return nil + } } // delete value in ledis session func (ls *LedisSessionStore) Delete(key interface{}) error { - ls.lock.Lock() - defer ls.lock.Unlock() - delete(ls.values, key) - return nil + ls.lock.Lock() + defer ls.lock.Unlock() + delete(ls.values, key) + return nil } // clear all values in ledis session func (ls *LedisSessionStore) Flush() error { - ls.lock.Lock() - defer ls.lock.Unlock() - ls.values = make(map[interface{}]interface{}) - return nil + ls.lock.Lock() + defer ls.lock.Unlock() + ls.values = make(map[interface{}]interface{}) + return nil } // get ledis session id func (ls *LedisSessionStore) SessionID() string { - return ls.sid + return ls.sid } // save session values to ledis func (ls *LedisSessionStore) SessionRelease(w http.ResponseWriter) { - b, err := session.EncodeGob(ls.values) - if err != nil { - return - } - c.Set([]byte(ls.sid), b) - c.Expire([]byte(ls.sid), ls.maxlifetime) + b, err := session.EncodeGob(ls.values) + if err != nil { + return + } + c.Set([]byte(ls.sid), b) + c.Expire([]byte(ls.sid), ls.maxlifetime) } // ledis session provider type LedisProvider struct { - maxlifetime int64 - savePath string - db int + maxlifetime int64 + savePath string + db int } // init ledis session // savepath like ledis server saveDataPath,pool size // e.g. 127.0.0.1:6379,100,astaxie func (lp *LedisProvider) SessionInit(maxlifetime int64, savePath string) error { - var err error - lp.maxlifetime = maxlifetime - configs := strings.Split(savePath, ",") - if len(configs) == 1 { - lp.savePath = configs[0] - } else if len(configs) == 2 { - lp.savePath = configs[0] - lp.db, err = strconv.Atoi(configs[1]) - if err != nil { - return err - } - } - cfg := new(config.Config) - cfg.DataDir = lp.savePath - nowLedis, err := ledis.Open(cfg) - c, err = nowLedis.Select(lp.db) - if err != nil { - println(err) - return nil - } - return nil + var err error + lp.maxlifetime = maxlifetime + configs := strings.Split(savePath, ",") + if len(configs) == 1 { + lp.savePath = configs[0] + } else if len(configs) == 2 { + lp.savePath = configs[0] + lp.db, err = strconv.Atoi(configs[1]) + if err != nil { + return err + } + } + cfg := new(config.Config) + cfg.DataDir = lp.savePath + nowLedis, err := ledis.Open(cfg) + c, err = nowLedis.Select(lp.db) + if err != nil { + println(err) + return nil + } + return nil } // read ledis session by sid func (lp *LedisProvider) SessionRead(sid string) (session.SessionStore, error) { - kvs, err := c.Get([]byte(sid)) - var kv map[interface{}]interface{} - if len(kvs) == 0 { - kv = make(map[interface{}]interface{}) - } else { - kv, err = session.DecodeGob(kvs) - if err != nil { - return nil, err - } - } - ls := &LedisSessionStore{sid: sid, values: kv, maxlifetime: lp.maxlifetime} - return ls, nil + kvs, err := c.Get([]byte(sid)) + var kv map[interface{}]interface{} + if len(kvs) == 0 { + kv = make(map[interface{}]interface{}) + } else { + kv, err = session.DecodeGob(kvs) + if err != nil { + return nil, err + } + } + ls := &LedisSessionStore{sid: sid, values: kv, maxlifetime: lp.maxlifetime} + return ls, nil } // check ledis session exist by sid func (lp *LedisProvider) SessionExist(sid string) bool { - count, _ := c.Exists([]byte(sid)) - if count == 0 { - return false - } else { - return true - } + count, _ := c.Exists([]byte(sid)) + if count == 0 { + return false + } else { + return true + } } // generate new sid for ledis session func (lp *LedisProvider) SessionRegenerate(oldsid, sid string) (session.SessionStore, error) { - count, _ := c.Exists([]byte(sid)) - if count == 0 { - // oldsid doesn't exists, set the new sid directly - // ignore error here, since if it return error - // the existed value will be 0 - c.Set([]byte(sid), []byte("")) - c.Expire([]byte(sid), lp.maxlifetime) - } else { - data, _ := c.Get([]byte(oldsid)) - c.Set([]byte(sid), data) - c.Expire([]byte(sid), lp.maxlifetime) - } - kvs, err := c.Get([]byte(sid)) - var kv map[interface{}]interface{} - if len(kvs) == 0 { - kv = make(map[interface{}]interface{}) - } else { - kv, err = session.DecodeGob([]byte(kvs)) - if err != nil { - return nil, err - } - } - ls := &LedisSessionStore{sid: sid, values: kv, maxlifetime: lp.maxlifetime} - return ls, nil + count, _ := c.Exists([]byte(sid)) + if count == 0 { + // oldsid doesn't exists, set the new sid directly + // ignore error here, since if it return error + // the existed value will be 0 + c.Set([]byte(sid), []byte("")) + c.Expire([]byte(sid), lp.maxlifetime) + } else { + data, _ := c.Get([]byte(oldsid)) + c.Set([]byte(sid), data) + c.Expire([]byte(sid), lp.maxlifetime) + } + kvs, err := c.Get([]byte(sid)) + var kv map[interface{}]interface{} + if len(kvs) == 0 { + kv = make(map[interface{}]interface{}) + } else { + kv, err = session.DecodeGob([]byte(kvs)) + if err != nil { + return nil, err + } + } + ls := &LedisSessionStore{sid: sid, values: kv, maxlifetime: lp.maxlifetime} + return ls, nil } // delete ledis session by id func (lp *LedisProvider) SessionDestroy(sid string) error { - c.Del([]byte(sid)) - return nil + c.Del([]byte(sid)) + return nil } // Impelment method, no used. func (lp *LedisProvider) SessionGC() { - return + return } // @todo func (lp *LedisProvider) SessionAll() int { - return 0 + return 0 } func init() { - session.Register("ledis", ledispder) + session.Register("ledis", ledispder) }