lbsadmin/Beego
1
0
Fork 0
mirror of https://github.com/astaxie/beego.git synced 2025-07-11 00:01:02 +00:00
Code Issues Releases Activity

test done

Browse Source
This commit is contained in:
lcbluestorm
2016-05-19 21:00:04 +08:00
parent e50f4f5631
commit 6234b50111
2 changed files with 88 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

50
session/ssdb/sess_ssdb.go
View File

@ -15,7 +15,7 @@ var ssdbProvider = &SsdbProvider{}
type SsdbProvider struct {
client *ssdb.Client
host string
port int32
port int
maxLifetime int64
}
@ -29,15 +29,15 @@ func (p *SsdbProvider) connectInit() error {
}
func (p *SsdbProvider) SessionInit(maxLifetime int64, savePath string) error {
var e error = nil
p.maxLifetime = maxLifetime
address := strings.Split(savePath, ":")
p.host = address[0]
port, e := strconv.Atoi(address[1])
p.port, e = strconv.Atoi(address[1])
if e != nil {
return e
}
p.port = address[1]
err := p.connectinit()
err := p.connectInit()
if err != nil {
return err
}
@ -55,10 +55,10 @@ func (p *SsdbProvider) SessionRead(sid string) (session.Store, error) {
if err != nil {
return nil, err
}
if value == nil || len(value) == 0 {
if value == nil || len(value.(string)) == 0 {
kv = make(map[interface{}]interface{})
} else {
kv, err = session.DecodeGob([]byte(value))
kv, err = session.DecodeGob([]byte(value.(string)))
if err != nil {
return nil, err
}
@ -70,10 +70,14 @@ func (p *SsdbProvider) SessionRead(sid string) (session.Store, error) {
func (p *SsdbProvider) SessionExist(sid string) bool {
if p.client == nil {
if err := p.connectInit(); err != nil {
return nil, err
panic(err)
}
}
if value == nil || len(value) == 0 {
value, err := p.client.Get(sid)
if err != nil {
panic(err)
}
if value == nil || len(value.(string)) == 0 {
return false
}
return true
@ -87,24 +91,26 @@ func (p *SsdbProvider) SessionRegenerate(oldsid, sid string) (session.Store, err
}
}
value, err := p.client.Get(oldsid)
if err != nil || len(value) == 0 {
value = ""
} else {
err = p.client.Del(sid)
}
_, e := p.client.Do("setx", sid, value, p.maxLifetime)
if e != nil {
return nil, e
if err != nil {
return nil, err
}
var kv map[interface{}]interface{}
if value == nil || len(value) == 0 {
if value == nil || len(value.(string)) == 0 {
kv = make(map[interface{}]interface{})
} else {
var err error
kv, err = session.DecodeGob(value)
kv, err = session.DecodeGob([]byte(value.(string)))
if err != nil {
return nil, err
}
_, err = p.client.Del(oldsid)
if err != nil {
return nil, err
}
}
_, e := p.client.Do("setx", sid, value.(string), p.maxLifetime)
if e != nil {
return nil, e
}
rs := &SessionStore{sid: sid, values: kv, maxLifetime: p.maxLifetime, client: p.client}
return rs, nil
@ -113,10 +119,10 @@ func (p *SsdbProvider) SessionRegenerate(oldsid, sid string) (session.Store, err
func (p *SsdbProvider) SessionDestroy(sid string) error {
if p.client == nil {
if err := p.connectInit(); err != nil {
return nil, err
return err
}
}
flag, err := p.client.Del(sid)
_, err := p.client.Del(sid)
if err != nil {
return err
}
@ -148,7 +154,7 @@ func (s *SessionStore) Set(key, value interface{}) error {
func (s *SessionStore) Get(key interface{}) interface{} {
s.lock.Lock()
defer s.lock.Unlock()
if value, ok := rs.values[key]; ok {
if value, ok := s.values[key]; ok {
return value
}
return nil
@ -175,7 +181,7 @@ func (s *SessionStore) SessionRelease(w http.ResponseWriter) {
if err != nil {
return
}
s.client.Do("setx", s.sid, s.values, s.maxLifetime)
s.client.Do("setx", s.sid, string(b), s.maxLifetime)
}
func init() {