From 2fe2be0b9a67621ebc4f9b0a97b6f0fe7f256183 Mon Sep 17 00:00:00 2001 From: Erik Brakkee Date: Sat, 10 Aug 2024 13:23:59 +0200 Subject: [PATCH] with the previous fix connection loss was not detected anymore, now using the ping method based on writing data which is more robust but may take a bit longer for connection loss to be detected. --- pkg/server/converge/websessions.go | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/pkg/server/converge/websessions.go b/pkg/server/converge/websessions.go index 745de7a..a868204 100644 --- a/pkg/server/converge/websessions.go +++ b/pkg/server/converge/websessions.go @@ -71,6 +71,8 @@ func GetUserLocation(r *http.Request) (*time.Location, error) { } func (session *WebSession) WriteNotifications(location *time.Location) { + timer := time.NewTicker(10 * time.Second) + defer timer.Stop() for { select { case notification, ok := <-session.notifications: @@ -83,6 +85,12 @@ func (session *WebSession) WriteNotifications(location *time.Location) { log.Printf("WS connection closed: %v", err) return } + case <-timer.C: + _, err := session.conn.Write(make([]byte, 0, 0)) + if err != nil { + log.Printf("WS connection closed: %v", err) + return + } } } }