Golang Gorilla WebSocket服务器 - 如何处理客户端断开连接

Golang Gorilla WebSocket服务器 - 如何处理客户端断开连接 大家好。我使用 Gorilla 实现了以下 WebSocket 服务器代码,该 WebSocket 将检查数据库中的用户消息,如果发现未读消息,则将这些消息传递给该用户。所有操作都应在 msgTicker.C 内完成,问题是当我关闭客户端页面时,定时器仍会执行几次才停止。如何立即知道客户端已断开连接或关闭了页面?以下是我的代码:

package main

import (
	"log"
	"net/http"
	"time"

	"github.com/gorilla/websocket"
)

const (
	// 允许向客户端写入文件的时间
	writeWait = 10 * time.Second

	// 允许从客户端读取下一个 pong 消息的时间
	pongWait = 60 * time.Second

	// 向客户端发送 ping 的周期,必须小于 pongWait
	pingPeriod = (pongWait * 9) / 10

	// 检查文件更改的周期
	msgPeriod = 10 * time.Second
)

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

func reader(ws *websocket.Conn) {
	defer ws.Close()
	ws.SetReadLimit(512)
	ws.SetReadDeadline(time.Now().Add(pongWait))
	ws.SetPongHandler(func(string) error { ws.SetReadDeadline(time.Now().Add(pongWait)); return nil })
	for {
		_, _, err := ws.ReadMessage()
		if err != nil {
			break
		}
	}
}

func writer(ws *websocket.Conn, idUsuario string) {
	pingTicker := time.NewTicker(pingPeriod)
	msgTicker := time.NewTicker(msgPeriod)
	defer func() {
		pingTicker.Stop()
		msgTicker.Stop()
		ws.Close()
	}()
	for {
		select {
		case <-msgTicker.C:
			var p []byte
			//var err error

			//通过 idUsuario 在此处检查数据库中的用户消息
			//变量 "p" 包含要发送给客户端的包

			p = nil //仅用于测试,之后移除
			//p = []byte("测试 WebSocket 连接!!用户ID:" + idUsuario)

			println("测试连接。")

			if p != nil {
				ws.SetWriteDeadline(time.Now().Add(writeWait))
				if err := ws.WriteMessage(websocket.TextMessage, p); err == nil {
					//包已发送,在此处更新数据库并将消息标记为已接收 = 1
				} else {
					return
				}
			}
		case <-pingTicker.C:
			ws.SetWriteDeadline(time.Now().Add(writeWait))
			if err := ws.WriteMessage(websocket.PingMessage, []byte{}); err != nil {
				return
			}
		}
	}
}

func wswebverificamensagensusuario(w http.ResponseWriter, r *http.Request) {
	ws, err := upgrader.Upgrade(w, r, nil)
	if err != nil {
		if _, ok := err.(websocket.HandshakeError); !ok {
			log.Println(err)
		}
		println(err.Error())
		return
	}

	go writer(ws, r.FormValue("idusuario"))
	reader(ws)
}

更多关于Golang Gorilla WebSocket服务器 - 如何处理客户端断开连接的实战教程也可以访问 https://www.itying.com/category-94-b0.html

5 回复

不 😊 但你已经将 r 和 w 传递给了套接字(连接),所以值得一试。

更多关于Golang Gorilla WebSocket服务器 - 如何处理客户端断开连接的实战系列教程也可以访问 https://www.itying.com/category-94-b0.html


不,那样行不通,因为这个函数很快就结束了,我的代码是在一个goroutine中执行的。这是一个WebSocket。你确定我可以用这种方式处理WebSocket吗?

wswebverificamensagensusuarior 中,您应该拥有 Context。或许您可以将其进一步传播(作为写入器的参数)并进行检查。

在 Conn 结构体(ws)中,有诸如 PingHandler、PongHandler、CloseHandler 等方法,您或许可以利用它们来进行连接检查。

Package websocket implements the WebSocket protocol defined in RFC 6455.

Package websocket

但您可能是对的,这两个连接是相互独立的。也许您可以像这样检查心跳:

GitHub图标 GitHub

头像

websockets/ws

ws - 简单易用、速度极快且经过全面测试的Node.js WebSocket客户端和服务器

回到顶部