package server import ( "git.solsynth.dev/hydrogen/identity/pkg/models" "git.solsynth.dev/hydrogen/identity/pkg/services" "github.com/gofiber/contrib/websocket" "github.com/samber/lo" ) func listenNotifications(c *websocket.Conn) { user := c.Locals("principal").(models.Account) // Push connection services.WsConn[user.ID] = append(services.WsConn[user.ID], c) // Event loop var err error for { if _, _, err = c.ReadMessage(); err != nil { break } } // Pop connection services.WsConn[user.ID] = lo.Filter(services.WsConn[user.ID], func(item *websocket.Conn, idx int) bool { return item != c }) }