当前位置：首页 > news >正文

golang-websocket

news 2025/7/9 20:04:03

WebSocket 是一种新型的网络通信协议，可以在 Web 应用程序中实现双向通信。

WebSocket与HTTP协议的主要区别是：

HTTP 和 WebSocket 协议的区别 HTTP 是单向的，而 WebSocket 是双向的。

在客户端和服务器之间的通信中，每个来自客户端的 HTTP 请求都会建立一个新的连接，并在从服务器收到响应后关闭，而WebSocket连接只建立一次，并在服务器或客户端终止连接之前重复使用。

HTTP 的空闲连接的超时时间为 60 秒。可用于响应时间非常短的调用。
WebSocket 空闲连接的超时时间为 3600 秒（1 小时），可应用于响应时间长于 60 秒的调用，最大连接数为 250。

WebSocket服务器

首先，创建一个 HTTP 服务器。在这个服务器中，创建一个用于处理 WebSocket 连接的函数 handleWebSocket。

其实就是开启一个http服务器。

func HandleWebSocket(w http.ResponseWriter, r *http.Request) {}
func main() {http.HandleFunc("/ws", HandleWebSocket)http.ListenAndServe(":8080", nil)
}

然后在该函数中，使用 gorilla/websocket 包来处理 WebSocket 连接。

1.建立upgrade变量。

websocket协议建立连接的握手过程就是http协议的头部加入了upgrade字段（upgrade:websocket）。
首先要导入"github.com/gorilla/websocket"包

var upgrade = websocket.Upgrader{ReadBufferSize:  1024,WriteBufferSize: 1024,CheckOrigin: func(r *http.Request) bool {return true},
}

2.升级http连接成websocket连接，处理websocket连接

将http连接头部加入upgrade字段，成为websocket协议。

func HandleHome(w http.ResponseWriter, r *http.Request) {//升级成websocketconn, err := upgrade.Upgrade(w, r, nil)if err != nil {log.Println(err)}//websocket读写逻辑for {messageType, p, err := conn.ReadMessage()if err != nil {log.Println(err)return}log.Println("receive:", string(p))conn.WriteMessage(messageType, []byte("hello, i am server"))}
}

WebSocket客户端

客户端主要是用来测试websocket服务器是否正确工作的，下面提供两种websocket服务器。

html+js实现

<!DOCTYPE html>
<html>
<head><title>WebSocket Test</title>
</head>
<body><script>var ws = new WebSocket("ws://localhost:8080/ws");ws.onopen = function() {console.log("WebSocket connection opened.");ws.send("Hi From the Client!")};ws.onmessage = function(event) {console.log("Received message: " + event.data);};ws.onclose = function() {console.log("WebSocket connection closed.");};ws.onerror = function(event) {console.log("WebSocket error: " + event.data);};</script>
</body>
</html>

利用golang websocket包实现

func WebSocketClient() {conn, _, err := websocket.DefaultDialer.Dial("ws://127.0.0.1:8080/ws", nil)if err != nil {panic(err)}defer conn.Close()for {time.Sleep(time.Second)conn.WriteMessage(websocket.TextMessage, []byte("hi i am client"))_, p, err := conn.ReadMessage()if err != nil {log.Println(err)}println("client recv:", string(p))}}