웹 애플리케이션의 인기와 함께 가장 일반적으로 사용되는 통신 프로토콜 중 하나인 WebSocket은 클라이언트와 서버 간의 실시간 통신에 자주 사용됩니다. Go 언어에서 gorilla/websocket은 WebSocket을 사용하는 데 가장 널리 사용되는 타사 라이브러리 중 하나입니다. 하지만 WebSocket을 사용하다 보면 연결이 자주 끊어지는 문제에 직면하는 경우가 많습니다. 이 문제를 해결하기 위한 몇 가지 팁과 관련 Go 코드 예제는 다음과 같습니다. gorilla/websocket 是使用WebSocket的最受欢迎的第三方库之一。但是，通过使用WebSocket时经常会遇到连接频繁断开的问题。解决这个问题的一些技巧，与相关的Go代码示例如下。
1.连接心跳
在WebSocket的使用中，一些因素可能导致连接断开，例如浏览器/客户端网络连接的丢失或服务器端假死。为了解决这些问题，可以使用心跳操作来保持连接。心跳操作专门发送ping消息，以保持WebSocket连接活动状态，并启动goroutine以检测响应。
下面是一个示例代码：
package main
import (
"log"
"net/http"
"time"
"github.com/gorilla/websocket"
)
var (
upgrader = websocket.Upgrader{
ReadBufferSize: 1024,
WriteBufferSize: 1024,
CheckOrigin: func(r *http.Request) bool { return true },
}
)
func main() {
http.HandleFunc("/ws", wsHandler)
log.Fatal(http.ListenAndServe(":8080", nil))
}
func wsHandler(w http.ResponseWriter, r *http.Request) {
conn, err := upgrader.Upgrade(w, r, nil)
if err != nil {
log.Println("upgrade error:", err)
return
}
defer conn.Close()
heartbeat(conn)
}
func heartbeat(conn *websocket.Conn) {
ticker := time.NewTicker(10 * time.Second)
defer ticker.Stop()
for {
select {
case <-ticker.C:
if err := conn.WriteMessage(websocket.PingMessage, []byte{}); err != nil {
log.Println("write ping error:", err)
return
}
}
}
}在上面的代码中，heartbeat函数启动了一个goroutine，该goroutine每10秒发送ping消息。当一个ping消息发送时，select语句等待500毫秒来接收pong消息。如果未收到pong消息，则断开连接或返回错误。这个例子展示了如何利用心跳来维持WebSocket连接。
2.降低客户端ping请求频率
在某些情况下，频繁发送ping请求也可能会中断WebSocket连接。例如，当WebSocket与公司网络代理一起使用时，代理可能会阻止连续的ping请求。
为了避免这种情况，可以降低ping消息的频率。例如，在下面的代码段中，心跳以10秒为间隔发送ping请求。控制ping频率并适当地调整它可以改善WebSocket连接的稳定性。
func heartbeat(conn *websocket.Conn) {
ticker := time.NewTicker(10 * time.Second)
defer ticker.Stop()
for {
select {
case <-ticker.C:
if err := conn.WriteMessage(websocket.PingMessage, []byte{}); err != nil {
log.Println("write ping error:", err)
return
}
}
}
}3.处理WebSocket关闭
定期检查WebSocket连接是否关闭，可以避免连接断开的情况。可以添加一个defer函数来关闭WebSocket连接，以确保在函数退出之前 WebSocket连接已经被完全关闭。
func wsHandler(w http.ResponseWriter, r *http.Request) {
conn, err := upgrader.Upgrade(w, r, nil)
if err != nil {
log.Println("upgrade error:", err)
return
}
defer conn.Close()
ticker := time.NewTicker(10 * time.Second)
defer ticker.Stop()
for {
_, _, err := conn.ReadMessage()
if err != nil {
log.Println("read error:", err)
return
}
}
}在上面的代码中，错误检查与conn.ReadMessage()的调用并置。在WebSocket连接中，ReadMessage()会阻塞等待收到的消息，所以我们可以利用无限循环来实现定期检查WebSocket是否已经关闭，以防止进入 错误检查-调用-错误检查-调用
goroutine을 실행하여 응답을 감지합니다. 샘플 코드는 다음과 같습니다. 🎜rrreee🎜위 코드에서 heartbeat 함수는 10초마다 goroutine을 시작하고 ping 메시지를 보냅니다. 핑 메시지가 전송되면 select 문은 퐁 메시지를 수신하기 위해 500밀리초를 기다립니다. Pong 메시지가 수신되지 않으면 연결이 끊어지거나 오류가 반환됩니다. 이 예에서는 하트비트를 사용하여 WebSocket 연결을 유지하는 방법을 보여줍니다. 🎜🎜2. 클라이언트 핑 요청 빈도를 줄이세요🎜🎜어떤 경우에는 핑 요청을 자주 보내면 WebSocket 연결이 중단될 수도 있습니다. 예를 들어 WebSocket을 회사 네트워크 프록시와 함께 사용하면 프록시가 지속적인 ping 요청을 차단할 수 있습니다. 🎜🎜이 상황을 방지하려면 핑 메시지 빈도를 줄일 수 있습니다. 예를 들어 아래 코드 조각에서 Heartbeat는 10초 간격으로 ping 요청을 보냅니다. 핑 빈도를 제어하고 적절하게 조정하면 WebSocket 연결의 안정성을 향상시킬 수 있습니다. 🎜rrreee🎜3. WebSocket 폐쇄 처리🎜🎜 연결이 끊어지지 않도록 WebSocket 연결이 닫혀 있는지 정기적으로 확인하세요. WebSocket 연결을 닫는 defer 함수를 추가하면 함수가 종료되기 전에 WebSocket 연결이 완전히 닫혔는지 확인할 수 있습니다. 🎜rrreee🎜위 코드에서 오류 검사는 conn.ReadMessage() 호출과 함께 배치됩니다. WebSocket 연결에서 ReadMessage()는 수신된 메시지에 대한 대기를 차단하므로 무한 루프를 사용하여 WebSocket이 닫혔는지 정기적으로 확인하여 오류 확인 호출을 방지할 수 있습니다. error Check-call의 무한 루프입니다. 🎜🎜결론🎜🎜위의 팁은 WebSocket 연결이 자주 끊어지는 문제를 해결하는 데 도움이 될 수 있습니다. 하트비트 메커니즘을 활용하고, 핑 빈도를 줄이고, WebSocket 종료를 처리함으로써 WebSocket 연결의 안정성이 크게 향상되어 클라이언트와의 보다 원활한 실시간 통신이 보장됩니다. 🎜위 내용은 Go 언어에서 Websocket 연결이 자주 끊어지는 문제를 해결하는 팁의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!
![[웹 프런트엔드] Node.js 빠른 시작](https://img.php.cn/upload/course/000/000/067/662b5d34ba7c0227.png)
