120 lines
2.1 KiB
Go
120 lines
2.1 KiB
Go
package server
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"github.com/khlieng/dispatch/irc"
|
|
"github.com/khlieng/dispatch/storage"
|
|
)
|
|
|
|
type Session struct {
|
|
irc map[string]*irc.Client
|
|
connectionState map[string]bool
|
|
ircLock sync.Mutex
|
|
|
|
ws map[string]*wsConn
|
|
wsLock sync.Mutex
|
|
out chan WSResponse
|
|
|
|
user *storage.User
|
|
}
|
|
|
|
func NewSession(user *storage.User) *Session {
|
|
return &Session{
|
|
irc: make(map[string]*irc.Client),
|
|
connectionState: make(map[string]bool),
|
|
ws: make(map[string]*wsConn),
|
|
out: make(chan WSResponse, 32),
|
|
user: user,
|
|
}
|
|
}
|
|
|
|
func (s *Session) getIRC(server string) (*irc.Client, bool) {
|
|
s.ircLock.Lock()
|
|
i, ok := s.irc[server]
|
|
s.ircLock.Unlock()
|
|
|
|
return i, ok
|
|
}
|
|
|
|
func (s *Session) setIRC(server string, i *irc.Client) {
|
|
s.ircLock.Lock()
|
|
s.irc[server] = i
|
|
s.connectionState[server] = false
|
|
s.ircLock.Unlock()
|
|
}
|
|
|
|
func (s *Session) deleteIRC(server string) {
|
|
s.ircLock.Lock()
|
|
delete(s.irc, server)
|
|
delete(s.connectionState, server)
|
|
s.ircLock.Unlock()
|
|
}
|
|
|
|
func (s *Session) numIRC() int {
|
|
s.ircLock.Lock()
|
|
n := len(s.irc)
|
|
s.ircLock.Unlock()
|
|
|
|
return n
|
|
}
|
|
|
|
func (s *Session) getConnectionStates() map[string]bool {
|
|
s.ircLock.Lock()
|
|
state := make(map[string]bool, len(s.connectionState))
|
|
|
|
for k, v := range s.connectionState {
|
|
state[k] = v
|
|
}
|
|
s.ircLock.Unlock()
|
|
|
|
return state
|
|
}
|
|
|
|
func (s *Session) setConnectionState(server string, connected bool) {
|
|
s.ircLock.Lock()
|
|
s.connectionState[server] = connected
|
|
s.ircLock.Unlock()
|
|
}
|
|
|
|
func (s *Session) setWS(addr string, w *wsConn) {
|
|
s.wsLock.Lock()
|
|
s.ws[addr] = w
|
|
s.wsLock.Unlock()
|
|
}
|
|
|
|
func (s *Session) deleteWS(addr string) {
|
|
s.wsLock.Lock()
|
|
delete(s.ws, addr)
|
|
s.wsLock.Unlock()
|
|
}
|
|
|
|
func (s *Session) numWS() int {
|
|
s.ircLock.Lock()
|
|
n := len(s.ws)
|
|
s.ircLock.Unlock()
|
|
|
|
return n
|
|
}
|
|
|
|
func (s *Session) sendJSON(t string, v interface{}) {
|
|
s.out <- WSResponse{t, v}
|
|
}
|
|
|
|
func (s *Session) sendError(err error, server string) {
|
|
s.sendJSON("error", Error{
|
|
Server: server,
|
|
Message: err.Error(),
|
|
})
|
|
}
|
|
|
|
func (s *Session) write() {
|
|
for res := range s.out {
|
|
s.wsLock.Lock()
|
|
for _, ws := range s.ws {
|
|
ws.out <- res
|
|
}
|
|
s.wsLock.Unlock()
|
|
}
|
|
}
|