mirror of
https://github.com/amigan/aim-oscar-server.git
synced 2024-10-18 05:33:32 -04:00
43 lines
761 B
Go
43 lines
761 B
Go
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(username string, session *oscar.Session) {
|
|
sm.mutex.Lock()
|
|
sm.sessions[username] = session
|
|
sm.mutex.Unlock()
|
|
}
|
|
|
|
func (sm *SessionManager) GetSession(username string) *oscar.Session {
|
|
sm.mutex.RLock()
|
|
s, ok := sm.sessions[username]
|
|
sm.mutex.RUnlock()
|
|
|
|
if ok {
|
|
return s
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (sm *SessionManager) RemoveSession(username string) {
|
|
sm.mutex.Lock()
|
|
sm.sessions[username] = nil
|
|
sm.mutex.Unlock()
|
|
}
|