如何使用Go和http.Transport进行WebSocket通信？

如何使用Go和http.Transport进行WebSocket通信？

WebSocket 是一种在网络应用中，提供双向通信的协议。它允许服务器主动向客户端推送消息，而不需要客户端先发起请求。在Go语言中，可以使用http.Transport来创建和管理WebSocket连接。

下面是一个使用Go和http.Transport进行WebSocket通信的示例代码：

package main

import (
    "fmt"
    "log"
    "net/http"

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

func EchoServer(ws *websocket.Conn) {
    for {
        var message string
        err := websocket.Message.Receive(ws, &message)
        if err != nil {
            log.Println("receive failed:", err)
            break
        }
        log.Println("received:", message)

        err = websocket.Message.Send(ws, message)
        if err != nil {
            log.Println("send failed:", err)
            break
        }
        log.Println("sent:", message)
    }
}

func main() {
    http.Handle("/", websocket.Handler(EchoServer))
    err := http.ListenAndServe(":8080", nil)
    if err != nil {
        log.Fatal("ListenAndServe: ", err)
    }
}

上述代码创建了一个WebSocket服务器，它将收到的消息原样发送回客户端。在main函数中，我们使用http.Handle和websocket.Handler来将EchoServer函数注册为处理器。然后使用http.ListenAndServe来启动服务器监听端口8080。

在客户端，我们可以使用http.Transport来创建WebSocket连接，并通过Send和Receive方法来发送和接收数据。下面是一个简单的客户端示例代码：

package main

import (
    "fmt"
    "log"
    "net/http"

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

func main() {
    origin := "http://localhost/"
    url := "ws://localhost:8080/"
    ws, err := websocket.Dial(url, "", origin)
    if err != nil {
        log.Fatal("dial failed:", err)
    }

    message := "Hello, WebSocket!"
    err = websocket.Message.Send(ws, message)
    if err != nil {
        log.Fatal("send failed:", err)
    }

    var response string
    err = websocket.Message.Receive(ws, &response)
    if err != nil {
        log.Fatal("receive failed:", err)
    }

    fmt.Println("received:", response)
}

上述代码中，我们使用websocket.Dial来创建WebSocket连接。然后使用websocket.Message.Send方法发送消息，通过websocket.Message.Receive来接收服务器返回的消息。

通过上述代码示例，我们可以使用Go和http.Transport轻松地创建和管理WebSocket连接。WebSocket 为应用程序提供了高效且实时的双向通信能力，并且适用于许多不同的应用场景。希望本文对你有所帮助！