如何使用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:", respo
.........................................................