client.go

func (c *Client) listenWrite() {
	log.Println("Listening write to client")
	for {
		select {
		case msg := <-c.ch:
			log.Println("Send: ", msg)
			websocket.JSON.Send(c.ws, msg)
		case <-c.doneCh:
			c.server.Del(c)
			c.doneCh <- true
			return
		}
	}
}

func (c *Client) listenRead() {
	log.Println("Listening read from client")
	for {
		select {
		case <-c.doneCh:
			c.server.Del(c)
			c.doneCh <- true
			return
		default:
			var msg Message
			err := websocket.JSON.Receive(c.ws, &msg)
			if (err == io.EOF) {
				c.doneCh <- true
			} else if err != nil {
				c.server.Err(err)
			} else {
				c.server.SendAll(&msg)
			}
		}
	}
}

client2.go

func (c *Client) listenWrite() {
	log.Println("Listening write to client")
  defer func(){
  			c.server.Del(c)
        c.doneCh <- true
  }()
	for {
		select {
		case msg := <-c.ch:
			log.Println("Send: ", msg)
			websocket.JSON.Send(c.ws, msg)
		case <-c.doneCh:
			return
		}
	}
}

func (c *Client) listenRead() {
	log.Println("Listening read from client")
	for {
		select {
		case <-c.doneCh:
			return
		default:
			var msg Message
			err := websocket.JSON.Receive(c.ws, &msg)
			if (err == io.EOF) {
				c.doneCh <- true
			} else if err != nil {
				c.server.Err(err)
			} else {
				c.server.SendAll(&msg)
			}
		}
	}
}