update

app/echo.go

@@ -0,0 +1,89 @@
+package main
+
+import (
+	"github.com/gorilla/websocket"
+	"net/http"
+	//"fmt"
+)
+
+func newHub() *Hub {
+	return &Hub{
+		broadcast:  make(chan []byte),
+		register:   make(chan *Client),
+		unregister: make(chan *Client),
+		clients:    make(map[*Client]bool),
+	}
+}
+
+type Hub struct {
+	clients    map[*Client]bool
+	broadcast  chan []byte
+	register   chan *Client
+	unregister chan *Client
+}
+
+type Client struct {
+	hub  *Hub
+	conn *websocket.Conn
+	send chan []byte
+}
+
+func (h *Hub) run() {
+	for {
+		select {
+		case client := <-h.register:
+			h.clients[client] = true
+		case client := <-h.unregister:
+			if _, ok := h.clients[client]; ok {
+				delete(h.clients, client)
+				close(client.send)
+			}
+		case message := <-h.broadcast:
+			//fmt.Println("map channel:", len(h.broadcast), cap(h.broadcast))
+			for client := range h.clients {
+				select {
+				case client.send <- message:
+				default:
+					close(client.send)
+					delete(h.clients, client)
+				}
+			}
+		}
+	}
+}
+
+func (c *Client) writePump() {
+	for {
+		message, ok := <-c.send
+		if !ok {
+			// The hub closed the channel.
+			c.conn.WriteMessage(websocket.CloseMessage, []byte{})
+			return
+		}
+		c.conn.WriteMessage(1, message)
+	}
+	c.conn.Close()
+}
+
+func (c *Client) readPump() {
+	for {
+		// Client close connection
+		_, _, err := c.conn.ReadMessage()
+		if err != nil {
+			break
+		}
+	}
+	c.hub.unregister <- c
+	c.conn.Close()
+}
+
+func (hub *Hub) wsHandler(w http.ResponseWriter, r *http.Request) {
+	conn, err := websocket.Upgrade(w, r, w.Header(), 1024, 1024)
+	if err != nil {
+		return
+	}
+	client := &Client{hub: hub, conn: conn, send: make(chan []byte)}
+	client.hub.register <- client
+	go client.readPump()
+	go client.writePump()
+}