Go 实现 WebSockets和什么是 WebSockets

1. 什么是 WebSockets

WebSockets 是一种在单个 TCP 连接上进行全双工通信的协议。传统上，标准的 HTTP 通信通过客户端发出请求，服务器响应请求，然后终止连接。但是，在 WebSockets 中，连接保持开放状态，使双方能够通过 WebSockets 连接交换数据。

1. Go 实现 WebSockets

Go 语言中可以使用内置的 net/http 包实现 WebSockets。以下是 Go 实现 WebSockets 的步骤：

• 创建一个 HTTP 服务器来运行 WebSockets。
• 创建一个函数或处理器，该处理器将指定要在 WebSockets 连接上执行的操作。
• 在 HTTP 服务器中注册处理器。
• 在处理器中设置与 WebSockets 版本的握手交互，以建立连接并开始通信。

下面是一个具体的实现示例：

package main

import (
    "fmt"
    "log"
    "net/http"

    "github.com/gorilla/websocket"
)

var connections = make(map[*websocket.Conn]bool)
var broadcast = make(chan []byte)

// WebSockets 接收器类型
type SocketReceiver struct{}

func (s SocketReceiver) ServeHTTP(w http.ResponseWriter, r *http.Request) {
    fmt.Println("开启 WebSockets 连接...")
    upgrader := websocket.Upgrader{}
    conn, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        log.Println(err)
    }
    connections[conn] = true
    for {
        // 读取 WebSockets 信息
        _, msg, err := conn.ReadMessage()
        if err != nil {
            log.Println(err)
            return
        }
        //将消息广播给所有 WebSockets 客户端
        broadcast <- msg
    }
}

func handleMessages() {
    for {
        //取出广播信道中的消息
        msg := <-broadcast
        //将消息发送给所有 WebSockets 客户端
        for conn := range connections {
            err := conn.WriteMessage(websocket.TextMessage, msg)
            if err != nil {
                log.Println(err)
                conn.Close()
                delete(connections, conn)
            }
        }
    }
}

func main() {
    fmt.Println("起始运行 WebSockets!")
    //初始化 WebSockets 接收器
    receiver := SocketReceiver{}
    http.Handle("/", receiver)
    //开启广播器
    go handleMessages()
    //开启 HTTP 服务器
    log.Fatal(http.ListenAndServe(":8080", nil))
}

在这个例子中，我们使用 Gorilla 的 WebSockets 库来处理 Go 中的 WebSockets。handleMessages 函数从 broadcast 通道中获取消息并将消息发送到所有 WebSockets 连接。

除了这个示例，Go 语言还有支持 WebSockets 的其他库，比如 websocket，gobwas 等等。具体使用可以查阅相关的官方文档。