2FAS/2fas-server
1
0
Fork 0
mirror of https://github.com/twofas/2fas-server.git synced 2024-12-24 18:10:51 +01:00
Code Issues Packages Projects Releases Wiki Activity

websocket api: fix websocket connection timeout handling

Browse Source
This commit is contained in:
in-void 2023-03-07 23:50:04 +01:00
parent b8dd4c167a
commit 3fb3b1b794
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
internal/websocket/common/handler.go
View File

@ -76,15 +76,15 @@ func (h *ConnectionHandler) serveWs(hub *Hub, w http.ResponseWriter, r *http.Req
go client.readPump() go client.readPump()
go func() { go func() {
disconnectAfter := time.Duration(3) * time.Minute disconnectAfter := 3 * time.Minute
timeout := time.After(disconnectAfter)
<-time.After(disconnectAfter) select {
case <-timeout:
defer func() {
logging.Info("Connection closed after", disconnectAfter) logging.Info("Connection closed after", disconnectAfter)
client.hub.unregister <- client client.hub.unregister <- client
client.conn.Close() client.conn.Close()
}() }
}() }()
} }