From a0fa5a94b6e05dfe7232706a1e691ead5a3ecf49 Mon Sep 17 00:00:00 2001 From: Dane Everitt Date: Sun, 27 Sep 2020 09:46:00 -0700 Subject: [PATCH] Support websocket event to send server stats; ref pterodactyl/panel#2414 --- router/websocket/message.go | 1 + router/websocket/websocket.go | 10 ++++++++++ 2 files changed, 11 insertions(+) diff --git a/router/websocket/message.go b/router/websocket/message.go index 5186826..85fb77f 100644 --- a/router/websocket/message.go +++ b/router/websocket/message.go @@ -8,6 +8,7 @@ const ( SetStateEvent = "set state" SendServerLogsEvent = "send logs" SendCommandEvent = "send command" + SendStatsEvent = "send stats" ErrorEvent = "daemon error" JwtErrorEvent = "jwt error" ) diff --git a/router/websocket/websocket.go b/router/websocket/websocket.go index 45cc575..45ddf9f 100644 --- a/router/websocket/websocket.go +++ b/router/websocket/websocket.go @@ -360,6 +360,16 @@ func (h *Handler) HandleInbound(m Message) error { }) } + return nil + } + case SendStatsEvent: + { + b, _ := json.Marshal(h.server.Proc()) + h.SendJson(&Message{ + Event: server.StatsEvent, + Args: []string{string(b)}, + }) + return nil } case SendCommandEvent: