aim-oscar-server/session_manager.go

43 lines
779 B
Go
Raw Normal View History

2021-12-24 12:41:16 -05:00
package main
import (
"aim-oscar/oscar"
"sync"
)
type SessionManager struct {
sessions map[string]*oscar.Session
mutex *sync.RWMutex
}
func NewSessionManager() *SessionManager {
sm := &SessionManager{
sessions: make(map[string]*oscar.Session),
mutex: &sync.RWMutex{},
}
return sm
}
func (sm *SessionManager) SetSession(screen_name string, session *oscar.Session) {
2021-12-24 12:41:16 -05:00
sm.mutex.Lock()
sm.sessions[screen_name] = session
2021-12-24 12:41:16 -05:00
sm.mutex.Unlock()
}
func (sm *SessionManager) GetSession(screen_name string) *oscar.Session {
2021-12-24 12:41:16 -05:00
sm.mutex.RLock()
s, ok := sm.sessions[screen_name]
2021-12-24 12:41:16 -05:00
sm.mutex.RUnlock()
if ok {
return s
}
return nil
}
func (sm *SessionManager) RemoveSession(screen_name string) {
2021-12-24 12:41:16 -05:00
sm.mutex.Lock()
sm.sessions[screen_name] = nil
2021-12-24 12:41:16 -05:00
sm.mutex.Unlock()
}