|
|
package core
|
|
|
|
|
|
import (
|
|
|
"git.noahlan.cn/noahlan/nnet/entity"
|
|
|
"sync"
|
|
|
"sync/atomic"
|
|
|
)
|
|
|
|
|
|
type session struct {
|
|
|
sync.RWMutex // 数据锁
|
|
|
|
|
|
// 网络单元
|
|
|
entity entity.NetworkEntity
|
|
|
|
|
|
id int64 // Session全局唯一ID
|
|
|
uid string // 用户ID，不绑定的情况下与sid一致
|
|
|
data map[string]interface{} // session数据存储（内存）
|
|
|
}
|
|
|
|
|
|
func newSession(entity entity.NetworkEntity, id int64) *session {
|
|
|
return &session{
|
|
|
id: id,
|
|
|
entity: entity,
|
|
|
uid: "",
|
|
|
data: make(map[string]interface{}),
|
|
|
}
|
|
|
}
|
|
|
|
|
|
// ID 获取 session ID
|
|
|
func (s *session) ID() int64 {
|
|
|
return s.id
|
|
|
}
|
|
|
|
|
|
// UID 获取UID
|
|
|
func (s *session) UID() string {
|
|
|
return s.uid
|
|
|
}
|
|
|
|
|
|
// Bind 绑定uid
|
|
|
func (s *session) Bind(uid string) {
|
|
|
s.uid = uid
|
|
|
}
|
|
|
|
|
|
// Attribute 获取指定key对应参数
|
|
|
func (s *session) Attribute(key string) interface{} {
|
|
|
s.RLock()
|
|
|
defer s.RUnlock()
|
|
|
|
|
|
return s.data[key]
|
|
|
}
|
|
|
|
|
|
// Keys 获取所有参数key
|
|
|
func (s *session) Keys() []string {
|
|
|
s.RLock()
|
|
|
defer s.RUnlock()
|
|
|
|
|
|
keys := make([]string, 0, len(s.data))
|
|
|
for k := range s.data {
|
|
|
keys = append(keys, k)
|
|
|
}
|
|
|
return keys
|
|
|
}
|
|
|
|
|
|
// Exists 指定key是否存在
|
|
|
func (s *session) Exists(key string) bool {
|
|
|
s.RLock()
|
|
|
defer s.RUnlock()
|
|
|
|
|
|
_, has := s.data[key]
|
|
|
return has
|
|
|
}
|
|
|
|
|
|
// Attributes 获取所有参数
|
|
|
func (s *session) Attributes() map[string]interface{} {
|
|
|
s.RLock()
|
|
|
defer s.RUnlock()
|
|
|
|
|
|
return s.data
|
|
|
}
|
|
|
|
|
|
// RemoveAttribute 移除指定key对应参数
|
|
|
func (s *session) RemoveAttribute(key string) {
|
|
|
s.Lock()
|
|
|
defer s.Unlock()
|
|
|
|
|
|
delete(s.data, key)
|
|
|
}
|
|
|
|
|
|
// SetAttribute 设置参数
|
|
|
func (s *session) SetAttribute(key string, value interface{}) {
|
|
|
s.Lock()
|
|
|
defer s.Unlock()
|
|
|
|
|
|
s.data[key] = value
|
|
|
}
|
|
|
|
|
|
// Invalidate 清理
|
|
|
func (s *session) Invalidate() {
|
|
|
s.Lock()
|
|
|
defer s.Unlock()
|
|
|
|
|
|
s.id = 0
|
|
|
s.uid = ""
|
|
|
s.data = make(map[string]interface{})
|
|
|
}
|
|
|
|
|
|
// Close 关闭
|
|
|
func (s *session) Close() {
|
|
|
//s.entity.Close()
|
|
|
s.Invalidate()
|
|
|
}
|
|
|
|
|
|
type sessionIDMgr struct {
|
|
|
count int64
|
|
|
sid int64
|
|
|
}
|
|
|
|
|
|
func newSessionIDMgr() *sessionIDMgr {
|
|
|
return &sessionIDMgr{}
|
|
|
}
|
|
|
|
|
|
// Increment the connection count
|
|
|
func (c *sessionIDMgr) Increment() {
|
|
|
atomic.AddInt64(&c.count, 1)
|
|
|
}
|
|
|
|
|
|
// Decrement the connection count
|
|
|
func (c *sessionIDMgr) Decrement() {
|
|
|
atomic.AddInt64(&c.count, -1)
|
|
|
}
|
|
|
|
|
|
// Count returns the connection numbers in current
|
|
|
func (c *sessionIDMgr) Count() int64 {
|
|
|
return atomic.LoadInt64(&c.count)
|
|
|
}
|
|
|
|
|
|
// Reset the connection service status
|
|
|
func (c *sessionIDMgr) Reset() {
|
|
|
atomic.StoreInt64(&c.count, 0)
|
|
|
atomic.StoreInt64(&c.sid, 0)
|
|
|
}
|
|
|
|
|
|
// SessionID returns the session id
|
|
|
func (c *sessionIDMgr) SessionID() int64 {
|
|
|
return atomic.AddInt64(&c.sid, 1)
|
|
|
}
|