当前位置： 技术文章>> 100道Go语言面试题之-在Go中，如何实现WebSocket通信？

文章标题：100道Go语言面试题之-在Go中，如何实现WebSocket通信？
  • 文章分类: 后端
  • 4742 阅读

在Go语言中实现WebSocket通信，通常会使用一些流行的库，如gorilla/websocket。这个库提供了一个简单的API来创建WebSocket客户端和服务器。以下是一个基本的步骤和示例代码，展示如何在Go中使用gorilla/websocket库来实现WebSocket通信。

步骤 1: 安装 gorilla/websocket 库

首先，你需要安装gorilla/websocket库。在你的Go项目目录下，运行以下命令来安装它：

go get github.com/gorilla/websocket

步骤 2: 编写WebSocket服务器

以下是一个简单的WebSocket服务器示例，它监听来自客户端的连接，并在接收到消息时回复一个确认消息。

package main

import (
    "flag"
    "log"
    "net/http"

    "github.com/gorilla/websocket"
)

var upgrader = websocket.Upgrader{} // 使用默认选项

func echo(w http.ResponseWriter, r *http.Request) {
    c, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        log.Print("upgrade:", err)
        return
    }
    defer c.Close()
    for {
        mt, message, err := c.ReadMessage()
        if err != nil {
            log.Println("read:", err)
            break
        }
        log.Printf("recv: %s", message)
        err = c.WriteMessage(mt, message)
        if err != nil {
            log.Println("write:", err)
            break
        }
    }
}

func main() {
    flag.Parse()
    log.SetFlags(0)
    http.HandleFunc("/echo", echo)
    log.Fatal(http.ListenAndServe("localhost:8080", nil))
}

步骤 3: 编写WebSocket客户端

客户端的实现依赖于你使用的环境或框架。但基本上，你会创建一个WebSocket连接，发送消息，并监听回复。以下是一个简单的客户端示例（假设在浏览器中运行）：

<!DOCTYPE html>
<html>
<body>

<h2>WebSocket Test</h2>
<button onclick="sendMessage()">Send Message</button>
<div id="output"></div>

<script>
var ws = new WebSocket("ws://localhost:8080/echo");

ws.onopen = function() {
    console.log("Connection open ...");
    ws.send("Hello Server!");
};

ws.onmessage = function(evt) { 
    var received_msg = evt.data;
    document.getElementById("output").innerHTML += "<br>" + received_msg;
};

ws.onclose = function() { 
    console.log("Connection closed.");
};

function sendMessage() {
    ws.send("Hello again!");
}

</script>

</body>
</html>

注意事项

  • 确保WebSocket服务器和客户端的端口和路径相匹配。
  • WebSocket通信是双向的，服务器和客户端都可以发送和接收消息。
  • 考虑到错误处理和连接重连等更复杂的场景，可能需要添加更多的逻辑来确保系统的健壮性。

通过以上步骤，你可以在Go中成功实现WebSocket通信。gorilla/websocket库提供了丰富的功能和灵活性，支持多种高级特性，如消息分片、压缩等。

推荐文章