109 lines
2.5 KiB
Go
109 lines
2.5 KiB
Go
|
// 会话操作.
|
||
|
|
// Wide 服务器端需要维护两种会话：
|
||
|
|
// 1. HTTP 会话：主要用于验证登录
|
||
|
|
// 2. Wide 会话：浏览器 tab 打开/刷新会创建一个，并和 HTTP 会话进行关联
|
||
|
|
//
|
||
|
|
// TODO: 当 HTTP 会话失效时，关联的 Wide 会话也会做失效处理：释放所有和该会话相关的资源，例如运行中的程序进程、事件队列等
|
||
|
|
package session
|
||
|
|
|
||
|
|
import (
|
||
|
|
"math/rand"
|
||
|
|
"strconv"
|
||
|
|
"sync"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/golang/glog"
|
||
|
|
"github.com/gorilla/sessions"
|
||
|
|
)
|
||
|
|
|
||
|
|
const (
|
||
|
|
SessionStateActive = iota // 会话状态：活的
|
||
|
|
)
|
||
|
|
|
||
|
|
// 用户 HTTP 会话，用于验证登录.
|
||
|
|
var HTTPSession = sessions.NewCookieStore([]byte("BEYOND"))
|
||
|
|
|
||
|
|
// Wide 会话，对应一个浏览器 tab.
|
||
|
|
type WideSession struct {
|
||
|
|
Id string // 唯一标识
|
||
|
|
HTTPSession *sessions.Session // 关联的 HTTP 会话
|
||
|
|
State int // 状态
|
||
|
|
Created time.Time // 创建时间
|
||
|
|
Updated time.Time // 最近一次使用时间
|
||
|
|
}
|
||
|
|
|
||
|
|
type Sessions []*WideSession
|
||
|
|
|
||
|
|
// 所有 Wide 会话集.
|
||
|
|
var WideSessions Sessions
|
||
|
|
|
||
|
|
// 排它锁，防止并发问题.
|
||
|
|
var mutex sync.Mutex
|
||
|
|
|
||
|
|
// 创建一个 Wide 会话.
|
||
|
|
func (sessions *Sessions) New(httpSession *sessions.Session) *WideSession {
|
||
|
|
mutex.Lock()
|
||
|
|
defer mutex.Unlock()
|
||
|
|
|
||
|
|
rand.Seed(time.Now().UnixNano())
|
||
|
|
|
||
|
|
id := strconv.Itoa(rand.Int())
|
||
|
|
now := time.Now()
|
||
|
|
|
||
|
|
ret := &WideSession{
|
||
|
|
Id: id,
|
||
|
|
HTTPSession: httpSession,
|
||
|
|
State: SessionStateActive,
|
||
|
|
Created: now,
|
||
|
|
Updated: now,
|
||
|
|
}
|
||
|
|
|
||
|
|
*sessions = append(*sessions, ret)
|
||
|
|
|
||
|
|
return ret
|
||
|
|
}
|
||
|
|
|
||
|
|
// 移除 Wide 会话.
|
||
|
|
func (sessions *Sessions) Remove(sid string) {
|
||
|
|
mutex.Lock()
|
||
|
|
defer mutex.Unlock()
|
||
|
|
|
||
|
|
for i, s := range *sessions {
|
||
|
|
if s.Id == sid {
|
||
|
|
*sessions = append((*sessions)[:i], (*sessions)[i+1:]...)
|
||
|
|
|
||
|
|
glog.V(3).Infof("Has [%d] wide sessions currently", len(*sessions))
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// 获取 HTTP 会话关联的所有 Wide 会话.
|
||
|
|
func (sessions *Sessions) GetByHTTPSession(httpSession *sessions.Session) []*WideSession {
|
||
|
|
mutex.Lock()
|
||
|
|
defer mutex.Unlock()
|
||
|
|
|
||
|
|
ret := []*WideSession{}
|
||
|
|
|
||
|
|
for _, s := range *sessions {
|
||
|
|
if s.HTTPSession.ID == httpSession.ID {
|
||
|
|
ret = append(ret, s)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return ret
|
||
|
|
}
|
||
|
|
|
||
|
|
// 移除 HTTP 会话关联的所有 Wide 会话.
|
||
|
|
func (sessions *Sessions) RemoveByHTTPSession(httpSession *sessions.Session) {
|
||
|
|
mutex.Lock()
|
||
|
|
defer mutex.Unlock()
|
||
|
|
|
||
|
|
for i, s := range *sessions {
|
||
|
|
if s.HTTPSession.ID == httpSession.ID {
|
||
|
|
*sessions = append((*sessions)[:i], (*sessions)[i+1:]...)
|
||
|
|
|
||
|
|
glog.V(3).Infof("Has [%d] wide sessions currently", len(*sessions))
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|