yanic/websocketserver/server.go

126 lines
2.1 KiB
Go

package websocketserver
import (
"fmt"
"log"
"net/http"
"golang.org/x/net/websocket"
)
//Server struct
type Server struct {
pattern string
clients map[int]*Client
addCh chan *Client
delCh chan *Client
sendAllCh chan interface{}
closeCh chan bool
errCh chan error
}
//NewServer creates a new server
func NewServer(pattern string) *Server {
clients := make(map[int]*Client)
addCh := make(chan *Client)
delCh := make(chan *Client)
sendAllCh := make(chan interface{})
closeCh := make(chan bool)
errCh := make(chan error)
return &Server{
pattern,
clients,
addCh,
delCh,
sendAllCh,
closeCh,
errCh,
}
}
//Add a listen client
func (s *Server) Add(c *Client) {
s.addCh <- c
}
//Del a listen client
func (s *Server) Del(c *Client) {
s.delCh <- c
}
//SendAll to all listen clients a msg
func (s *Server) SendAll(msg interface{}) {
s.sendAllCh <- msg
}
//Close stops server
func (s *Server) Close() {
s.closeCh <- true
}
//Err send to server
func (s *Server) Err(err error) {
s.errCh <- err
}
func (s *Server) sendAll(msg interface{}) {
for _, c := range s.clients {
c.Write(msg)
}
}
// Listen and serve.
// It serves client connection and broadcast request.
func (s *Server) Listen() {
log.Println("Listening Server...")
// websocket handler
onConnected := func(ws *websocket.Conn) {
defer func() {
err := ws.Close()
if err != nil {
s.errCh <- err
}
}()
client := NewClient(ws, s)
s.Add(client)
defer func() {
s.Del(client)
err := fmt.Errorf("Client %d is disconnected.", client.id)
s.Err(err)
}()
client.Listen()
}
http.Handle(s.pattern, websocket.Handler(onConnected))
log.Println("Created handler")
for {
select {
// Add new a client
case c := <-s.addCh:
log.Println("Added new client")
s.clients[c.id] = c
log.Println("Now", len(s.clients), "clients connected.")
// del a client
case c := <-s.delCh:
log.Println("Delete client")
delete(s.clients, c.id)
// broadcast message for all clients
case msg := <-s.sendAllCh:
s.sendAll(msg)
case err := <-s.errCh:
log.Println("Error:", err.Error())
case <-s.closeCh:
return
}
}
}