如何使用golang创建WebSocket服务器

2023-12-17 22:33

短信预约 -IT技能 免费直播动态提醒

如何使用golang创建WebSocket服务器

WebSocket是一种在客户端和服务器之间实现双向通信的协议。它允许我们通过一个持久的连接在客户端和服务器之间传输数据。在这篇文章中，我将向你介绍如何使用golang创建一个简单的WebSocket服务器，并提供具体的代码示例。

准备工作
在开始之前，你需要确保你已经正确安装了golang。你可以从官方网站下载安装包并按照指示进行安装。
导入依赖
我们需要导入net/http和github.com/gorilla/websocket这两个依赖包。net/http用于创建HTTP服务器，gorilla/websocket用于处理WebSocket连接。

import (
    "net/http"
    "github.com/gorilla/websocket"
)

创建WebSocket服务器
我们使用http.HandleFunc()函数来处理WebSocket连接请求，并使用websocket.Upgrade()函数将HTTP协议升级为WebSocket协议。

func wsHandler(w http.ResponseWriter, r *http.Request) {
    upgrader := websocket.Upgrader{
        CheckOrigin: func(r *http.Request) bool { return true },
    }

    conn, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        log.Println("Upgrade error:", err)
        return
    }

    // 在这里处理WebSocket连接
}

在wsHandler()函数中，我们创建了一个Upgrader对象，并设置了CheckOrigin属性为一个匿名函数，返回true表示接受所有的跨域请求。

接下来，我们使用Upgrade()函数将HTTP协议升级为WebSocket协议，如果升级失败则打印错误信息并返回。如果升级成功，我们可以使用conn对象来处理WebSocket连接。

处理WebSocket连接
在WebSocket连接建立后，我们可以使用conn对象进行读写操作。下面是一个简单的处理示例：

for {
    messageType, message, err := conn.ReadMessage()
    if err != nil {
        log.Println("Read error:", err)
        break
    }

    log.Println("received message:", string(message))

    err = conn.WriteMessage(messageType, message)
    if err != nil {
        log.Println("Write error:", err)
        break
    }
}

在上面的示例中，我们使用ReadMessage()函数读取客户端发送的消息，并使用WriteMessage()函数将消息回传给客户端。如果读取或写入过程中出现错误，我们打印错误信息并跳出循环。

启动WebSocket服务器
最后，我们使用http.HandleFunc()函数将wsHandler()函数注册为处理WebSocket连接的处理器，并使用http.ListenAndServe()函数在指定端口上启动WebSocket服务器。

func main() {
    http.HandleFunc("/ws", wsHandler)
    log.Println("WebSocket server started.")

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

在上面的示例中，我们将wsHandler()函数注册为"/ws"路径的处理器。你可以根据需要自定义路径。