使用超时机制连接 WebSocket：Go 语言实践-Golang-PHP中文网

使用超时机制连接 WebSocket：Go 语言实践

聖光之護

发布： 2025-10-06 11:30:31

原创

198人浏览过

使用超时机制连接 websocket：go 语言实践

本文介绍了在 Go 语言中使用 go .net/websocket 包连接 WebSocket 服务器时，如何设置连接超时。通过结合 net.DialTimeout 和 websocket.NewClient 函数，可以在连接建立之前设置超时时间，避免程序长时间阻塞。本文将提供详细的代码示例和注意事项，帮助开发者更好地控制 WebSocket 连接过程。

在使用 Go 语言的 go.net/websocket 包连接 WebSocket 服务器时，经常需要考虑连接超时的问题。默认情况下，如果服务器无法访问，websocket.Dial 函数可能会阻塞较长时间，影响程序的响应速度。本文将介绍一种常用的方法，结合 net.DialTimeout 和 websocket.NewClient 函数，在连接建立之前设置超时时间。

使用 net.DialTimeout 和 websocket.NewClient

net.DialTimeout 函数允许设置连接超时时间。它尝试连接到指定的网络地址，如果在指定的时间内未能建立连接，则返回错误。websocket.NewClient 函数则可以使用现有的 net.Conn 连接来创建 WebSocket 客户端。

以下是一个示例代码，演示如何使用这两个函数来实现 WebSocket 连接超时：

ViiTor实时翻译

AI实时多语言翻译专家！强大的语音识别、AR翻译功能。

116

查看详情

package main

import (
    "fmt"
    "net"
    "net/url"
    "time"

    "golang.org/x/net/websocket"
)

func main() {
    wsURL := "wss://remote-server" // 替换为你的 WebSocket 服务器地址
    origin := "http://localhost"      // 替换为你的 Origin

    // 设置超时时间
    timeout := 5 * time.Second

    // 使用 net.DialTimeout 建立连接
    conn, err := net.DialTimeout("tcp", getHost(wsURL), timeout)
    if err != nil {
        fmt.Println("DialTimeout error:", err)
        return
    }
    defer conn.Close()

    // 创建 WebSocket 配置
    config := &websocket.Config{
        Origin: &url.URL{Scheme: "http", Host: origin},
        Version: websocket.DraftLatest,
        Location: &url.URL{Scheme: "ws", Host: getHost(wsURL), Path: "/"},
    }

    // 使用 NewClient 函数创建 WebSocket 连接
    ws, err := websocket.NewClient(conn, config, wsURL, "", nil)
    if err != nil {
        fmt.Println("NewClient error:", err)
        return
    }
    defer ws.Close()

    fmt.Println("WebSocket connection established successfully!")

    // 在此处进行 WebSocket 通信
    // 例如：
    // _, err = ws.Write([]byte("Hello, WebSocket!"))
    // if err != nil {
    //  fmt.Println("Write error:", err)
    //  return
    // }
}

// getHost 从 URL 中提取主机名
func getHost(wsURL string) string {
    u, err := url.Parse(wsURL)
    if err != nil {
        panic(err) // 处理错误，这里为了简化直接 panic
    }
    return u.Host
}

登录后复制

代码解释：

net.DialTimeout("tcp", getHost(wsURL), timeout): 使用 net.DialTimeout 函数尝试连接到 WebSocket 服务器的 TCP 地址。timeout 变量指定了超时时间。如果连接在指定时间内未能建立，该函数将返回一个错误。getHost 函数用于从 URL 中提取主机名。
websocket.NewClient(conn, config, wsURL, "", nil): 使用 websocket.NewClient 函数，利用已经建立的 net.Conn 连接创建一个 WebSocket 客户端。config 参数包含了 WebSocket 连接的配置信息，例如 Origin 和 WebSocket 版本。

注意事项：