web socket server in golang

socket_server.go

package main

import (
	"fmt"
	"io"
	"net/http"

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

type Server struct {
	conns map[*websocket.Conn]bool
}

func NewServer() *Server {
	return &Server{
		conns: make(map[*websocket.Conn]bool),
	}
}

func (s *Server) handleWS(ws *websocket.Conn){
	fmt.Println("new incoming connection from client", ws.RemoteAddr().Network())

	s.conns[ws] = true

	s.readLoop(ws)
}

func (s *Server) readLoop(ws *websocket.Conn)  {
	buf := make([]byte, 1024)
	for{
		n, err := ws.Read(buf)
		if err != nil{
			if err == io.EOF{
				fmt.Println("EOF err")
				break
			}
			fmt.Println("read error:", err)
			continue
		}
		msg := buf[:n]
		
		s.broadCast(msg)
	}
}

func (s *Server) broadCast(b []byte)  {
	for ws := range s.conns{
		go func(ws *websocket.Conn) {
			if _, err := ws.Write(b); err != nil {
				fmt.Println("write error", err)
			}
		}(ws)
	}
}

func main() {
	server := NewServer()
	http.Handle("/ws", websocket.Handler(server.handleWS))
	http.ListenAndServe(":3000", nil)
}