實現功能: websocket+chan 通信

使用包:

github.com/gin-gonic/gin
github.com/gorilla/websocket

代碼

package websocket

import (
    "encoding/json"
    "log"
    "net/http"

    "github.com/gin-gonic/gin"
    "github.com/gorilla/websocket"
)

// ClientManager is a websocket manager
type ClientManager struct {
    Clients    map[*Client]bool
    Broadcast  chan []byte
    Register   chan *Client
    Unregister chan *Client
}

// Client is a websocket client
type Client struct {
    ID     int
    Socket *websocket.Conn
    Send   chan []byte
}

// Message is return msg
type Message struct {
    Sender    string `json:"sender,omitempty"`
    Recipient string `json:"recipient,omitempty"`
    Content   string `json:"content,omitempty"`
}

// Manager define a ws server manager
var Manager = ClientManager{
    Broadcast:  make(chan []byte),
    Register:   make(chan *Client),
    Unregister: make(chan *Client),
    Clients:    make(map[*Client]bool),
}

// Start is  項目運行前, 協(xié)程開啟start -> go Manager.Start()
func (manager *ClientManager) Start() {
    for {
        log.Println("<---管道通信--->")

        select {
        case conn := <-Manager.Register:
            log.Printf("新用戶加入:%v", conn.ID)
            Manager.Clients[conn] = true
            jsonMessage, _ := json.Marshal(&Message{Content: "Successful connection to socket service"})
            Manager.Send(jsonMessage, conn)
        case conn := <-Manager.Unregister:
            log.Printf("用戶離開:%v", conn.ID)
            if _, ok := Manager.Clients[conn]; ok {
                close(conn.Send)
                delete(Manager.Clients, conn)
                jsonMessage, _ := json.Marshal(&Message{Content: "A socket has disconnected"})
                Manager.Send(jsonMessage, conn)
            }

        case message := <-Manager.Broadcast:

            jsonMessage, _ := json.Marshal(&Message{Content: string(message)})
            for conn := range Manager.Clients {
                select {
                case conn.Send <- jsonMessage:
                default:
                    close(conn.Send)
                    delete(Manager.Clients, conn)
                }
            }
        }
    }
}

// Send is to send ws message to ws client
func (manager *ClientManager) Send(message []byte, ignore *Client) {

    for conn := range manager.Clients {
        // if conn != ignore { //向除了自己的socket 用戶發(fā)送
        conn.Send <- message
        // }
    }
}

func (c *Client) Read() {
    defer func() {
        Manager.Unregister <- c
        c.Socket.Close()
    }()

    for {

        _, message, err := c.Socket.ReadMessage()
        if err != nil {
            Manager.Unregister <- c
            c.Socket.Close()
            break
        }
        log.Printf("讀取到客戶端的信息:%s", string(message))
        Manager.Broadcast <- message
    }
}

func (c *Client) Write() {
    defer func() {
        c.Socket.Close()
    }()

    for {
        select {
        case message, ok := <-c.Send:
            if !ok {
                c.Socket.WriteMessage(websocket.CloseMessage, []byte{})
                return
            }
            log.Printf("發(fā)送到到客戶端的信息:%s", string(message))

            c.Socket.WriteMessage(websocket.TextMessage, message)
        }
    }

}

//TestHandler socket 連接 中間件 作用:升級協(xié)議,用戶驗證,自定義信息等
func TestHandler(c *gin.Context) {
    conn, err := websocket.Upgrade(c.Writer, c.Request, nil, 1024, 1024)
    if err != nil {
        http.NotFound(c.Writer, c.Request)
        return
    }
    //可以添加用戶信息驗證
    userID := 1
    client := &Client{
        ID:     userID,
        Socket: conn,
        Send:   make(chan []byte),
    }
    Manager.Register <- client
    go client.Read()
    go client.Write()
}

使用

定義個ws請求路徑路由 , 路由驗證 使用TestHandler 中間件, 用于升級服務,驗證用戶信息
例如 :

 route.Get("/ws",TestHandler)

==main 入口 寫入 go Manager.Start() 協(xié)程開啟服務==
如果文件不在main包里, 需要寫上包名 例如: 包名為websocket
go websocket.Manager.Start()

每個方法的作用

Start():啟動websocket服務
Send():向連接websocket的管道chan寫入數據
Read():讀取在websocket管道中的數據
Write():通過websocket協(xié)議向連接到ws的客戶端發(fā)送數據
TestHandler(): ws鏈接交互的中間件, 用于協(xié)議升級, 用戶信息驗證等

前端測試 我使用的是 http://easyswoole.com/wstool.html 這個測試工具

代碼原處:http://www.reibang.com/p/f058fdbdea58 只是稍微修改下代碼