go语言实现WebSocket服务

WebSocket 是基于 TCP 的一个双向传输数据的协议，和HTTP协议同属应用层。它解决了http请求无法进行网页与后端持久双向传输数据的问题。
Go 原生没有提供对 WebSocket 的支持，只能选择第三方库。使用比较多的是 gorilla/websocket 开源库，下面以 gorilla/websocket 来写一个简单的通信示例。

go get -u github.com/gorilla/websocket

package websocket

// 将http请求升级为websocket
type Upgrader struct
// websocket 长连接用于双工i/o
type Conn struct

// 升级方法
func (u *Upgrader) Upgrade(w http.ResponseWriter, r *http.Request, responseHeader http.Header) (*Conn, error)
// 或不经过Upgrader自定义参数进行升级
func Upgrade(w http.ResponseWriter, r *http.Request, responseHeader http.Header, readBufSize, writeBufSize int) (*Conn, error)

后端以go代码写一个示例，使用postman作接口测试。

使用gin + websocket 框架

func NewConnection(c *gin.Context) {
    // Upgrader.CheckOrigin() 定义同源检查，这里只作简单试验不校验
    // upGrader := websocket.Upgrader{
    //     CheckOrigin: func(r *http.Request) bool {
    //         return true
    //     },
    // }
    // ws, err := upGrader.Upgrade(c.Writer, c.Request, nil)

    ws, err := websocket.Upgrade(c.Writer, c.Request, nil, 1024, 1024)
    if err != nil {
        c.JSON(http.StatusInternalServerError, gin.H{
            "msg": "服务端错误",
        })
        return
    }

    defer ws.Close()

    // 开始通信
    for {
        fmt.Println("start transfer message")
        msgType, msg, err := ws.ReadMessage()
        if err != nil {
            fmt.Println("webSocket read error")
            return
        }
        err = ws.WriteMessage(msgType, msg)
        if err != nil {
            fmt.Println("webSocket write error")
            return
        }
    }
}

func main() {
    r := gin.Default()

    r.GET("/wsTest", NewConnection)

    r.Run("127.0.0.1:9090")
}

• 点击new按钮，新建一个websocket请求

• 输入url： ws://127.0.0.1:9090/wsTest 发送请求

效果：

后端简单地给该websocket连接回显文本，完成双工io通信。实际应用中应该以json或其他数据格式写入前端所需要的对象信息。
注：这里go后端代码socket读写未分离，ReadMessage()为空时阻塞下方的WriteMessage()操作。可分离到2个goroutine中分别读写，不占用同一buffer。