Go session存儲

上一節(jié)我們介紹了Session管理器的實現(xiàn)原理，定義了存儲session的接口，這小節(jié)我們將示例一個基于內(nèi)存的session存儲接口的實現(xiàn)，其他的存儲方式，讀者可以自行參考示例來實現(xiàn)，內(nèi)存的實現(xiàn)請看下面的例子代碼

package memory

import (
    "container/list"
    "github.com/astaxie/session"
    "sync"
    "time"
)

var pder = &Provider{list: list.New()}

type SessionStore struct {
    sid          string                      //session id唯一標(biāo)示
    timeAccessed time.Time                   //最后訪問時間
    value        map[interface{}]interface{} //session里面存儲的值
}

func (st *SessionStore) Set(key, value interface{}) error {
    st.value[key] = value
    pder.SessionUpdate(st.sid)
    return nil
}

func (st *SessionStore) Get(key interface{}) interface{} {
    pder.SessionUpdate(st.sid)
    if v, ok := st.value[key]; ok {
        return v
    } else {
        return nil
    }
    return nil
}

func (st *SessionStore) Delete(key interface{}) error {
    delete(st.value, key)
    pder.SessionUpdate(st.sid)
    return nil
}

func (st *SessionStore) SessionID() string {
    return st.sid
}

type Provider struct {
    lock     sync.Mutex               //用來鎖
    sessions map[string]*list.Element //用來存儲在內(nèi)存
    list     *list.List               //用來做gc
}

func (pder *Provider) SessionInit(sid string) (session.Session, error) {
    pder.lock.Lock()
    defer pder.lock.Unlock()
    v := make(map[interface{}]interface{}, 0)
    newsess := &SessionStore{sid: sid, timeAccessed: time.Now(), value: v}
    element := pder.list.PushBack(newsess)
    pder.sessions[sid] = element
    return newsess, nil
}

func (pder *Provider) SessionRead(sid string) (session.Session, error) {
    if element, ok := pder.sessions[sid]; ok {
        return element.Value.(*SessionStore), nil
    } else {
        sess, err := pder.SessionInit(sid)
        return sess, err
    }
    return nil, nil
}

func (pder *Provider) SessionDestroy(sid string) error {
    if element, ok := pder.sessions[sid]; ok {
        delete(pder.sessions, sid)
        pder.list.Remove(element)
        return nil
    }
    return nil
}

func (pder *Provider) SessionGC(maxlifetime int64) {
    pder.lock.Lock()
    defer pder.lock.Unlock()

    for {
        element := pder.list.Back()
        if element == nil {
            break
        }
        if (element.Value.(*SessionStore).timeAccessed.Unix() + maxlifetime) < time.Now().Unix() {
            pder.list.Remove(element)
            delete(pder.sessions, element.Value.(*SessionStore).sid)
        } else {
            break
        }
    }
}

func (pder *Provider) SessionUpdate(sid string) error {
    pder.lock.Lock()
    defer pder.lock.Unlock()
    if element, ok := pder.sessions[sid]; ok {
        element.Value.(*SessionStore).timeAccessed = time.Now()
        pder.list.MoveToFront(element)
        return nil
    }
    return nil
}

func init() {
    pder.sessions = make(map[string]*list.Element, 0)
    session.Register("memory", pder)
}

上面這個代碼實現(xiàn)了一個內(nèi)存存儲的session機制。通過init函數(shù)注冊到session管理器中。這樣就可以方便的調(diào)用了。我們?nèi)绾蝸碚{(diào)用該引擎呢？請看下面的代碼

import (
    "github.com/astaxie/session"
    _ "github.com/astaxie/session/providers/memory"
)

當(dāng)import的時候已經(jīng)執(zhí)行了memory函數(shù)里面的init函數(shù)，這樣就已經(jīng)注冊到session管理器中，我們就可以使用了，通過如下方式就可以初始化一個session管理器：

var globalSessions *session.Manager

//然后在init函數(shù)中初始化
func init() {
    globalSessions, _ = session.NewManager("memory", "gosessionid", 3600)
    go globalSessions.GC()
}