pomodoro/GoTomato
3
0
Fork 0
Code Issues Pull requests Releases 15 Activity

drop sendPing()

Browse source
This commit is contained in:
Sebastian Mark 2024-11-21 20:11:09 +01:00
parent a98b4bc154
commit 994526a30d
1 changed files with 4 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
internal/websocket/staleClients.go
View file

@ -16,7 +16,10 @@ func RemoveStaleClients() {
for range ticker.C { for range ticker.C {
mu.Lock() mu.Lock()
for _, client := range Clients { for _, client := range Clients {
if !sendPing(client) || isStale(client) { client.Conn.SetWriteDeadline(time.Now().Add(SEND_TIMEOUT * time.Second))
client.Conn.WriteMessage(websocket.PingMessage, nil)
if isStale(client) {
client.Conn.Close() client.Conn.Close()
delete(Clients, client.Conn.LocalAddr()) delete(Clients, client.Conn.LocalAddr())
helper.Logger.Info("Removed stale client", "host", client.RealIP, "lastPong", client.LastPong) helper.Logger.Info("Removed stale client", "host", client.RealIP, "lastPong", client.LastPong)
@ -26,17 +29,6 @@ func RemoveStaleClients() {
} }
} }
func sendPing(client *models.WebsocketClient) bool {
client.Conn.SetWriteDeadline(time.Now().Add(SEND_TIMEOUT * time.Second))
err := client.Conn.WriteMessage(websocket.PingMessage, nil)
if err != nil {
return false
}
return true
}
func isStale(client *models.WebsocketClient) bool { func isStale(client *models.WebsocketClient) bool {
return time.Since(client.LastPong) > (STALE_CLIENT_TIMEOUT * time.Second) return time.Since(client.LastPong) > (STALE_CLIENT_TIMEOUT * time.Second)
} }