«

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

时间:2024-3-26 09:13     作者:韩俊     分类: Go语言


        <p style="text-indent:2em;">如何使用Go和http.Transport进行WebSocket通信?</p><p style="text-indent:2em;">WebSocket 是一种在网络应用中,提供双向通信的协议。它允许服务器主动向客户端推送消息,而不需要客户端先发起请求。在Go语言中,可以使用http.Transport来创建和管理WebSocket连接。</p><p style="text-indent:2em;">下面是一个使用Go和http.Transport进行WebSocket通信的示例代码:</p><pre>package main

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

&quot;golang.org/x/net/websocket&quot;

)

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(&quot;send failed:&quot;, err)
        break
    }
    log.Println(&quot;sent:&quot;, 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"

&quot;golang.org/x/net/websocket&quot;

)

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

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

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

fmt.Println(&quot;received:&quot;, response)

}

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

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

标签: golang

热门推荐