99 lines
1.7 KiB
Go
99 lines
1.7 KiB
Go
package server
|
|
|
|
import (
|
|
"encoding/json"
|
|
"sync"
|
|
|
|
"github.com/khlieng/name_pending/Godeps/_workspace/src/github.com/gorilla/websocket"
|
|
|
|
"github.com/khlieng/name_pending/storage"
|
|
)
|
|
|
|
type Session struct {
|
|
irc map[string]*IRC
|
|
ircLock sync.Mutex
|
|
|
|
ws map[string]*WebSocket
|
|
wsLock sync.Mutex
|
|
out chan []byte
|
|
|
|
user *storage.User
|
|
}
|
|
|
|
func NewSession() *Session {
|
|
return &Session{
|
|
irc: make(map[string]*IRC),
|
|
ws: make(map[string]*WebSocket),
|
|
out: make(chan []byte, 32),
|
|
}
|
|
}
|
|
|
|
func (s *Session) getIRC(server string) (*IRC, bool) {
|
|
s.ircLock.Lock()
|
|
irc, ok := s.irc[server]
|
|
s.ircLock.Unlock()
|
|
|
|
return irc, ok
|
|
}
|
|
|
|
func (s *Session) setIRC(server string, irc *IRC) {
|
|
s.ircLock.Lock()
|
|
s.irc[server] = irc
|
|
s.ircLock.Unlock()
|
|
}
|
|
|
|
func (s *Session) deleteIRC(server string) {
|
|
s.ircLock.Lock()
|
|
delete(s.irc, server)
|
|
s.ircLock.Unlock()
|
|
}
|
|
|
|
func (s *Session) numIRC() int {
|
|
s.ircLock.Lock()
|
|
n := len(s.irc)
|
|
s.ircLock.Unlock()
|
|
|
|
return n
|
|
}
|
|
|
|
func (s *Session) setWS(addr string, ws *websocket.Conn) {
|
|
socket := NewWebSocket(ws)
|
|
go socket.write()
|
|
|
|
s.wsLock.Lock()
|
|
s.ws[addr] = socket
|
|
s.wsLock.Unlock()
|
|
}
|
|
|
|
func (s *Session) deleteWS(addr string) {
|
|
s.wsLock.Lock()
|
|
s.ws[addr].close()
|
|
delete(s.ws, addr)
|
|
s.wsLock.Unlock()
|
|
}
|
|
|
|
func (s *Session) sendJSON(t string, v interface{}) {
|
|
data, _ := json.Marshal(v)
|
|
raw := json.RawMessage(data)
|
|
res, _ := json.Marshal(WSResponse{Type: t, Response: &raw})
|
|
|
|
s.out <- res
|
|
}
|
|
|
|
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()
|
|
}
|
|
}
|