如何使用Go语言进行WebSocket通信？

程序员墨松

于 2024-05-27 10:05:54 发布

阅读量374

点赞数 4

分类专栏： Golang 文章标签： golang websocket iphone

本文链接：https://blog.csdn.net/zingsono/article/details/139230892

版权

Golang 专栏收录该内容

75 篇文章 0 订阅

订阅专栏

文章目录

如何使用Go语言进行WebSocket通信？
- 在Go中创建WebSocket服务器
- 在Go中创建WebSocket客户端

如何使用Go语言进行WebSocket通信？

WebSocket是一种网络通信协议，它可以在单个TCP连接上进行全双工通信。这意味着，一旦建立连接，客户端和服务器就可以在任何时刻发送数据，而不需要像HTTP那样通过请求和响应的方式来进行通信。Go语言的标准库net/http中提供了对WebSocket的支持，让你可以轻松地实现WebSocket服务器和客户端。

在Go中创建WebSocket服务器

要在Go中创建一个WebSocket服务器，你需要实现http.Handler接口，并处理http.Upgrade请求。下面是一个简单的示例：

package main

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

// WebSocketHandler 实现http.Handler接口
type WebSocketHandler struct{}

// ServeHTTP 处理WebSocket请求
func (h *WebSocketHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 conn, err := upgrader.Upgrade(w, r, nil)
 if err != nil {
 log.Print("upgrade error: ", err)
 return
 }
 defer conn.Close()

 for {
 // 读取消息
 mt, message, err := conn.ReadMessage()
 if err != nil {
 log.Println("read error: ", err)
 break
 }

 // 打印接收到的消息类型和消息内容
 fmt.Printf("received message: %s\n", message)

 // 发送消息
 err = conn.WriteMessage(mt, message)
 if err != nil {
 log.Println("write error: ", err)
 break
 }
 }
}

var upgrader = &websocket.Upgrader{
 ReadBufferSize:  1024,
 WriteBufferSize: 1024,
}

func main() {
 http.Handle("/", &WebSocketHandler{})
 log.Fatal(http.ListenAndServe(":8080", nil))
}

这个示例中，WebSocketHandler实现了http.Handler接口，它的ServeHTTP方法会处理所有的WebSocket请求。upgrader.Upgrade方法用于将HTTP连接升级为WebSocket连接。一旦连接建立，你就可以使用conn.ReadMessage和conn.WriteMessage方法来读写消息了。

在Go中创建WebSocket客户端

要创建WebSocket客户端，你可以使用net/http包中的Dial函数来建立连接，并使用websocket.Conn接口来进行通信。下面是一个简单的示例：

package main

import (
 "fmt"
 "log"
 "net/url"

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

func main() {
 // 连接到WebSocket服务器
 server := "ws://localhost:8080"
 u, err := url.Parse(server)
 if err != nil {
 log.Fatal("Error parsing URL: ", err)
 }
 conn, _, err := websocket.DefaultDialer.Dial("ws://"+u.Host, nil)
 if err != nil {
 log.Fatal("Error connecting: ", err)
 }
 defer conn.Close()

 // 发送消息
 err = conn.WriteMessage(websocket.TextMessage, []byte("Hello, Server!"))
 if err != nil {
 log.Fatal("Error sending message: ", err)
 }

 // 接收消息
 _, message, err := conn.ReadMessage()
 if err != nil {
 log.Fatal("Error reading message: ", err)
 }
 fmt.Println("Received: ", string(message))
}