yanic/websocketserver/client.go

package websocketserver

import (
	"fmt"

	"golang.org/x/net/websocket"
)

const channelBufSize = 100

var maxID = 0

//Client struct
type Client struct {
	id     int
	ws     *websocket.Conn
	server *Server
	ch     chan interface{}
	doneCh chan bool
}

//NewClient creates a new Client
func NewClient(ws *websocket.Conn, server *Server) *Client {

	if ws == nil {
		panic("ws cannot be nil")
	}

	if server == nil {
		panic("server cannot be nil")
	}

	maxID++
	ch := make(chan interface{}, channelBufSize)
	doneCh := make(chan bool)

	return &Client{maxID, ws, server, ch, doneCh}
}

//GetConnection the websocket connection of a listen client
func (c *Client) GetConnection() *websocket.Conn {
	return c.ws
}

//Write send the msg informations to the clients
func (c *Client) Write(msg interface{}) {
	select {
	case c.ch <- msg:
	default:
		c.server.Del(c)
		err := fmt.Errorf("Client %d is disconnected.", c.id)
		c.server.Err(err)
	}
}

// Listen Write and Read request via chanel
func (c *Client) Listen() {
	c.listen()
}

// listen for new msg informations
func (c *Client) listen() {
	for {
		select {
		case msg := <-c.ch:
			err := websocket.JSON.Send(c.ws, msg)
			if err != nil {
				c.doneCh <- true
			}
		case gone := <-c.doneCh:
			if gone {
				c.server.Del(c)
				err := fmt.Errorf("Client %d is disconnected.", c.id)
				c.server.Err(err)
			}
		}
	}
}