parent
e2d7f82c96
commit
8d2e458d61
@ -0,0 +1,93 @@
|
|||||||
|
package ncmd
|
||||||
|
|
||||||
|
import "sync"
|
||||||
|
|
||||||
|
type Session struct {
|
||||||
|
id int64 // 唯一ID
|
||||||
|
data map[string]any // 数据存储
|
||||||
|
|
||||||
|
*sync.RWMutex
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewSession(id int64) *Session {
|
||||||
|
return &Session{
|
||||||
|
id: id,
|
||||||
|
data: make(map[string]any),
|
||||||
|
RWMutex: &sync.RWMutex{},
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *Session) Id() int64 {
|
||||||
|
return s.id
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *Session) SetId(id int64) {
|
||||||
|
s.id = id
|
||||||
|
}
|
||||||
|
|
||||||
|
// Attribute 获取指定key对应参数
|
||||||
|
func (s *Session) Attribute(key string) any {
|
||||||
|
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]any {
|
||||||
|
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 any) {
|
||||||
|
s.Lock()
|
||||||
|
defer s.Unlock()
|
||||||
|
|
||||||
|
s.data[key] = value
|
||||||
|
}
|
||||||
|
|
||||||
|
// Invalidate 清理
|
||||||
|
func (s *Session) Invalidate() {
|
||||||
|
s.Lock()
|
||||||
|
defer s.Unlock()
|
||||||
|
|
||||||
|
s.id = 0
|
||||||
|
s.data = make(map[string]any)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Close 关闭
|
||||||
|
func (s *Session) Close() {
|
||||||
|
s.Invalidate()
|
||||||
|
}
|
Loading…
Reference in New Issue