golang的websocket服务端识别客户端

在Golang实战开发的过程中，我们经常会遇到一些这样那样的问题，然后要卡好半天，等问题解决了才发现原来一些细节知识点还是没有掌握好。今天golang学习网就整理分享《golang的websocket服务端识别客户端》，聊聊golang、websocket，希望可以帮助到正在努力赚钱的你。

问题内容

老师，golang的websocket服务端，同时接受多个客户端的连接，当多个客户端同时向服务端发送消息，如何识别到具体是哪个客户端发的消息？

正确答案

在Go语言中，可以通过使用 github.com/gorilla/websocket 包来实现 WebSocket 服务端，并且支持同时接收多个客户端的连接。

要区分不同的客户端，可以在处理每个客户端的连接时，将连接实例保存在一个集合中，比如使用 map[*websocket.Conn]bool 的形式，其中键是 *websocket.Conn 类型的指针，值是一个布尔值，用于表示该连接是否处于活动状态。这样就可以根据连接实例来识别具体是哪个客户端发送的消息。

下面是一个简单的示例代码：

package main

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

    "github.com/gorilla/websocket"
)

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

var upgrader = websocket.Upgrader{
    ReadBufferSize:  1024,
    WriteBufferSize: 1024,
}

func main() {
    http.HandleFunc("/ws", handleWebSocket)
    go handleMessages()
    log.Fatal(http.ListenAndServe(":8080", nil))
}

func handleWebSocket(w http.ResponseWriter, r *http.Request) {
    conn, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        log.Println(err)
        return
    }
    connections[conn] = true

    go func() {
        for {
            _, message, err := conn.ReadMessage()
            if err != nil {
                log.Println(err)
                delete(connections, conn)
                return
            }
            message = []byte(fmt.Sprintf("%s: %s", conn.RemoteAddr(), message))
            broadcast 

在上面的示例中，connections 是一个 map[*websocket.Conn]bool 类型的变量，用于保存连接实例和连接状态。当一个客户端连接到服务端时，将该连接实例添加到 connections 中。

在 handleWebSocket 函数中，针对每个连接实例启动一个协程，用于接收该客户端发送的消息，并将其转发给其他客户端。在每个消息的前面添加了客户端的 IP 地址，以便区分不同的客户端。

在 handleMessages 函数中，使用一个无限循环来接收 broadcast 通道中的消息，并将其发送给所有活动的客户端。如果发送失败，则从 connections 中删除该连接实例。

通过这种方式，可以轻松地实现一个支持多个客户端连接的 WebSocket 服务端，并且能够区分不同的客户端发送的消息。

文中关于golang的知识介绍，希望对你的学习有所帮助！若是受益匪浅，那就动动鼠标收藏这篇《golang的websocket服务端识别客户端》文章吧，也可关注golang学习网公众号了解相关技术文章。