2013-09-26 10:07:00 +00:00
|
|
|
package session
|
|
|
|
|
|
|
|
import (
|
|
|
|
"container/list"
|
2014-01-05 06:48:36 +00:00
|
|
|
"net/http"
|
2013-09-26 10:07:00 +00:00
|
|
|
"sync"
|
|
|
|
"time"
|
|
|
|
)
|
|
|
|
|
|
|
|
var mempder = &MemProvider{list: list.New(), sessions: make(map[string]*list.Element)}
|
|
|
|
|
|
|
|
type MemSessionStore struct {
|
2014-01-05 06:48:36 +00:00
|
|
|
sid string //session id
|
|
|
|
timeAccessed time.Time //last access time
|
|
|
|
value map[interface{}]interface{} //session store
|
2013-09-26 10:07:00 +00:00
|
|
|
lock sync.RWMutex
|
|
|
|
}
|
|
|
|
|
|
|
|
func (st *MemSessionStore) Set(key, value interface{}) error {
|
|
|
|
st.lock.Lock()
|
|
|
|
defer st.lock.Unlock()
|
|
|
|
st.value[key] = value
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func (st *MemSessionStore) Get(key interface{}) interface{} {
|
|
|
|
st.lock.RLock()
|
|
|
|
defer st.lock.RUnlock()
|
|
|
|
if v, ok := st.value[key]; ok {
|
|
|
|
return v
|
|
|
|
} else {
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func (st *MemSessionStore) Delete(key interface{}) error {
|
|
|
|
st.lock.Lock()
|
|
|
|
defer st.lock.Unlock()
|
|
|
|
delete(st.value, key)
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func (st *MemSessionStore) Flush() error {
|
|
|
|
st.lock.Lock()
|
|
|
|
defer st.lock.Unlock()
|
|
|
|
st.value = make(map[interface{}]interface{})
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func (st *MemSessionStore) SessionID() string {
|
|
|
|
return st.sid
|
|
|
|
}
|
|
|
|
|
2014-01-05 06:48:36 +00:00
|
|
|
func (st *MemSessionStore) SessionRelease(w http.ResponseWriter) {
|
2013-09-26 10:07:00 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
type MemProvider struct {
|
|
|
|
lock sync.RWMutex //用来锁
|
|
|
|
sessions map[string]*list.Element //用来存储在内存
|
|
|
|
list *list.List //用来做gc
|
|
|
|
maxlifetime int64
|
|
|
|
savePath string
|
|
|
|
}
|
|
|
|
|
|
|
|
func (pder *MemProvider) SessionInit(maxlifetime int64, savePath string) error {
|
|
|
|
pder.maxlifetime = maxlifetime
|
|
|
|
pder.savePath = savePath
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func (pder *MemProvider) SessionRead(sid string) (SessionStore, error) {
|
|
|
|
pder.lock.RLock()
|
|
|
|
if element, ok := pder.sessions[sid]; ok {
|
|
|
|
go pder.SessionUpdate(sid)
|
|
|
|
pder.lock.RUnlock()
|
|
|
|
return element.Value.(*MemSessionStore), nil
|
|
|
|
} else {
|
|
|
|
pder.lock.RUnlock()
|
|
|
|
pder.lock.Lock()
|
|
|
|
newsess := &MemSessionStore{sid: sid, timeAccessed: time.Now(), value: make(map[interface{}]interface{})}
|
|
|
|
element := pder.list.PushBack(newsess)
|
|
|
|
pder.sessions[sid] = element
|
|
|
|
pder.lock.Unlock()
|
|
|
|
return newsess, nil
|
|
|
|
}
|
|
|
|
return nil, nil
|
|
|
|
}
|
|
|
|
|
2013-11-05 14:23:48 +00:00
|
|
|
func (pder *MemProvider) SessionExist(sid string) bool {
|
|
|
|
pder.lock.RLock()
|
|
|
|
defer pder.lock.RUnlock()
|
|
|
|
if _, ok := pder.sessions[sid]; ok {
|
|
|
|
return true
|
|
|
|
} else {
|
|
|
|
return false
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-09-26 10:07:00 +00:00
|
|
|
func (pder *MemProvider) SessionRegenerate(oldsid, sid string) (SessionStore, error) {
|
|
|
|
pder.lock.RLock()
|
|
|
|
if element, ok := pder.sessions[oldsid]; ok {
|
|
|
|
go pder.SessionUpdate(oldsid)
|
|
|
|
pder.lock.RUnlock()
|
|
|
|
pder.lock.Lock()
|
|
|
|
element.Value.(*MemSessionStore).sid = sid
|
|
|
|
pder.sessions[sid] = element
|
|
|
|
delete(pder.sessions, oldsid)
|
|
|
|
pder.lock.Unlock()
|
|
|
|
return element.Value.(*MemSessionStore), nil
|
|
|
|
} else {
|
|
|
|
pder.lock.RUnlock()
|
|
|
|
pder.lock.Lock()
|
|
|
|
newsess := &MemSessionStore{sid: sid, timeAccessed: time.Now(), value: make(map[interface{}]interface{})}
|
|
|
|
element := pder.list.PushBack(newsess)
|
|
|
|
pder.sessions[sid] = element
|
|
|
|
pder.lock.Unlock()
|
|
|
|
return newsess, nil
|
|
|
|
}
|
|
|
|
return nil, nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func (pder *MemProvider) SessionDestroy(sid string) error {
|
|
|
|
pder.lock.Lock()
|
|
|
|
defer pder.lock.Unlock()
|
|
|
|
if element, ok := pder.sessions[sid]; ok {
|
|
|
|
delete(pder.sessions, sid)
|
|
|
|
pder.list.Remove(element)
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func (pder *MemProvider) SessionGC() {
|
|
|
|
pder.lock.RLock()
|
|
|
|
for {
|
|
|
|
element := pder.list.Back()
|
|
|
|
if element == nil {
|
|
|
|
break
|
|
|
|
}
|
|
|
|
if (element.Value.(*MemSessionStore).timeAccessed.Unix() + pder.maxlifetime) < time.Now().Unix() {
|
|
|
|
pder.lock.RUnlock()
|
|
|
|
pder.lock.Lock()
|
|
|
|
pder.list.Remove(element)
|
|
|
|
delete(pder.sessions, element.Value.(*MemSessionStore).sid)
|
|
|
|
pder.lock.Unlock()
|
|
|
|
pder.lock.RLock()
|
|
|
|
} else {
|
|
|
|
break
|
|
|
|
}
|
|
|
|
}
|
|
|
|
pder.lock.RUnlock()
|
|
|
|
}
|
|
|
|
|
2013-11-01 16:16:10 +00:00
|
|
|
func (pder *MemProvider) SessionAll() int {
|
|
|
|
return pder.list.Len()
|
|
|
|
}
|
|
|
|
|
2013-09-26 10:07:00 +00:00
|
|
|
func (pder *MemProvider) SessionUpdate(sid string) error {
|
|
|
|
pder.lock.Lock()
|
|
|
|
defer pder.lock.Unlock()
|
|
|
|
if element, ok := pder.sessions[sid]; ok {
|
|
|
|
element.Value.(*MemSessionStore).timeAccessed = time.Now()
|
|
|
|
pder.list.MoveToFront(element)
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func init() {
|
|
|
|
Register("memory", mempder)
|
|
|
|
}
|