add members & host to stats.

2021-03-19 22:06:40 +01:00
parent bbae073104
commit b96ba47224
4 changed files with 23 additions and 9 deletions
--- a/server/internal/websocket/websocket.go
+++ b/server/internal/websocket/websocket.go
@ -198,8 +198,18 @@ func (ws *WebSocketHandler) Upgrade(w http.ResponseWriter, r *http.Request) erro
 	return nil
 }

-func (ws *WebSocketHandler) TotalConns() uint32 {
-	return atomic.LoadUint32(&ws.conns)
+func (ws *WebSocketHandler) Stats() types.Stats {
+	host := ""
+	session, ok := ws.sessions.GetHost()
+	if ok {
+		host = session.ID()
+	}
+
+	return types.Stats{
+		Connections: atomic.LoadUint32(&ws.conns),
+		Host:        host,
+		Members:     ws.sessions.Members(),
+	}
 }

 func (ws *WebSocketHandler) IsAdmin(password string) (bool, error) {