golang WebSocket示例：构建一个简单的聊天应用

Golang WebSocket示例：构建一个简单的聊天应用

随着互联网的发展，实时通信越来越受人们的关注。WebSocket作为一种实时通信协议，能够在客户端和服务器之间建立持久性的连接，使得双向实时数据传输成为可能。本文将介绍如何使用Golang构建一个简单的聊天应用，通过WebSocket实现实时消息的发送和接收。

在开始之前，确保你已经在计算机上正确安装了Golang开发环境。

首先，我们需要导入"Gorilla Websocket"包，这是Golang中一个常用的WebSocket库。在终端中使用以下命令安装：

go get github.com/gorilla/websocket

接下来，我们创建一个main.go文件并导入所需的包：

package main

import (
    "log"
    "net/http"

    "github.com/gorilla/websocket"
)

在main函数中，我们定义一个WebSocket处理器函数，用于处理来自客户端的WebSocket连接请求：

func wsHandler(w http.ResponseWriter, r *http.Request) {
    // 升级HTTP连接为WebSocket连接
    conn, err := websocket.Upgrade(w, r, nil, 1024, 1024)
    if err != nil {
        log.Println(err)
        return
    }

    // 处理WebSocket连接
    for {
        // 读取客户端发送的消息
        _, msg, err := conn.ReadMessage()
        if err != nil {
            log.Println(err)
            break
        }

        // 将消息发送给所有连接的客户端
        for client := range clients {
            err := client.WriteMessage(websocket.TextMessage, msg)
            if err != nil {
                log.Println(err)
                client.Close()
                delete(clients, client)
            }
        }
    }

    // 关闭WebSocket连接
    conn.Close()
}

在上面的代码中，我们首先将HTTP连接升级为WebSocket连接，然后使用for循环不断地接收客户端发送的消息，并将这些消息发送给所有连接的客户端。当有错误发生或客户端断开连接时，我们会关闭相应的连接。

接下来，我们需要定义一个全局变量clients来保存所有连接的客户端：

var clients = make(map[*websocket.Conn]bool)

然后，我们修改主函数中的代码，将WebSocket处理器函数与"/ws"路径关联起来，并启动一个HTTP服务器：

func main() {
    http.HandleFunc("/ws", wsHandler)

    err := http.ListenAndServe(":8000", nil)
    if err != nil {
        log.Fatal("ListenAndServe: ", err)
    }
}

最后，我们可以使用浏览器或者WebSocket客户端工具来测试我们的聊天应用了。在浏览器中输入"http://localhost:8000"，打开控制台，并执行以下JavaScript代码：

var ws = new WebSocket("ws://localhost:8000/ws");

// 连接建立时触发的事件
ws.onopen = function() {
    console.log("连接成功");
};

// 收到消息时触发的事件
ws.onmessage = function(evt) {
    console.log("收到消息：" + evt.data);
};

// 发送消息
ws.send("Hello, World!");

通过上述代码，我们将会收到来自服务器的“Hello, World!”消息，并在控制台中输出。

至此，我们已经成功地构建了一个简单的聊天应用，使用Golang和WebSocket实现了实时消息的发送和接收。当然，这只是一个基础的示例，你可以根据自己的需求进行扩展和优化。希望这篇文章对你有所帮助，祝你在使用Golang和WebSocket开发实时通信应用时，取得更多的成功！