本文介绍了在 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 连接超时：

Golang Uber Dig

使用 uber-go/dig 在 Golang 中实现依赖注入——基于反射容器，提供 Provide/Invoke、dig.In/dig.Out 参数与结果对象，支持命名依赖等

下载

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
}

代码解释：

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

注意事项：

• 错误处理： 代码中包含了错误处理逻辑，可以及时发现连接失败的原因。在实际应用中，应该根据具体情况选择合适的错误处理方式。
• Origin 设置： Origin 字段在 WebSocket 连接中非常重要。确保将其设置为正确的值，否则服务器可能会拒绝连接。
• URL 解析: 使用 url.Parse 解析 URL，提取主机名。
• 连接关闭: 使用 defer conn.Close() 和 defer ws.Close() 确保连接在使用完毕后正确关闭，释放资源。

总结：

通过结合 net.DialTimeout 和 websocket.NewClient 函数，可以在 Go 语言中实现 WebSocket 连接超时功能。这种方法允许在连接建立之前设置超时时间，避免程序长时间阻塞。使用时请注意错误处理和 Origin 设置，确保连接能够成功建立。