mirror of
https://github.com/hay-kot/homebox.git
synced 2025-08-03 08:10:28 +00:00
backend: Periodically send client aliveness pings following event websocket connection
This commit is contained in:
parent
aba853d598
commit
55e5cf3f1a
1 changed files with 39 additions and 3 deletions
|
@ -2,8 +2,9 @@
|
||||||
package v1
|
package v1
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"encoding/json"
|
||||||
"net/http"
|
"net/http"
|
||||||
|
"time"
|
||||||
|
|
||||||
"github.com/google/uuid"
|
"github.com/google/uuid"
|
||||||
"github.com/hay-kot/homebox/backend/internal/core/services"
|
"github.com/hay-kot/homebox/backend/internal/core/services"
|
||||||
|
@ -144,11 +145,40 @@ func (ctrl *V1Controller) HandleCurrency() errchain.HandlerFunc {
|
||||||
}
|
}
|
||||||
|
|
||||||
func (ctrl *V1Controller) HandleCacheWS() errchain.HandlerFunc {
|
func (ctrl *V1Controller) HandleCacheWS() errchain.HandlerFunc {
|
||||||
|
type eventMsg struct {
|
||||||
|
Event string `json:"event"`
|
||||||
|
}
|
||||||
|
|
||||||
m := melody.New()
|
m := melody.New()
|
||||||
|
|
||||||
m.HandleConnect(func(s *melody.Session) {
|
m.HandleConnect(func(s *melody.Session) {
|
||||||
auth := services.NewContext(s.Request.Context())
|
auth := services.NewContext(s.Request.Context())
|
||||||
s.Set("gid", auth.GID)
|
s.Set("gid", auth.GID)
|
||||||
|
|
||||||
|
// Asynchronous ticker that keeps the websocket connection alive with periodic pings.
|
||||||
|
go func() {
|
||||||
|
const interval = 10 * time.Second
|
||||||
|
|
||||||
|
ping := time.NewTicker(interval)
|
||||||
|
defer ping.Stop()
|
||||||
|
|
||||||
|
for {
|
||||||
|
select {
|
||||||
|
case <-s.Request.Context().Done():
|
||||||
|
return
|
||||||
|
|
||||||
|
case <-ping.C:
|
||||||
|
msg := &eventMsg{Event: "ping"}
|
||||||
|
|
||||||
|
pingBytes, err := json.Marshal(msg)
|
||||||
|
if err != nil {
|
||||||
|
log.Log().Msgf("error marshaling ping: %v", err)
|
||||||
|
} else {
|
||||||
|
_ = m.BroadcastMultiple(pingBytes, []*melody.Session{s})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}()
|
||||||
})
|
})
|
||||||
|
|
||||||
factory := func(e string) func(data any) {
|
factory := func(e string) func(data any) {
|
||||||
|
@ -159,9 +189,15 @@ func (ctrl *V1Controller) HandleCacheWS() errchain.HandlerFunc {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
jsonStr := fmt.Sprintf(`{"event": "%s"}`, e)
|
msg := &eventMsg{Event: e}
|
||||||
|
|
||||||
_ = m.BroadcastFilter([]byte(jsonStr), func(s *melody.Session) bool {
|
jsonBytes, err := json.Marshal(msg)
|
||||||
|
if err != nil {
|
||||||
|
log.Log().Msgf("error marshling event data %v: %v", data, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
_ = m.BroadcastFilter(jsonBytes, func(s *melody.Session) bool {
|
||||||
groupIDStr, ok := s.Get("gid")
|
groupIDStr, ok := s.Get("gid")
|
||||||
if !ok {
|
if !ok {
|
||||||
return false
|
return false
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue