This commit adds new metrics to know the number of registered users
we have in every WebSocket broadcasts hub
Этот коммит содержится в:
Mario de Frutos Dieguez
2020-03-10 10:15:56 +01:00
коммит произвёл GitHub
родитель cd382412fd
Коммит ee86413cdd
3 изменённых файлов: 18 добавлений и 0 удалений

Просмотреть файл

@@ -161,6 +161,9 @@ func (a *App) GetHubForUserId(userId string) *Hub {
func (a *App) HubRegister(webConn *WebConn) {
hub := a.GetHubForUserId(webConn.UserId)
if hub != nil {
if metrics := a.Metrics(); metrics != nil {
metrics.IncrementWebSocketBroadcastUsersRegistered(strconv.Itoa(hub.connectionIndex), 1)
}
hub.Register(webConn)
}
}
@@ -168,6 +171,9 @@ func (a *App) HubRegister(webConn *WebConn) {
func (a *App) HubUnregister(webConn *WebConn) {
hub := a.GetHubForUserId(webConn.UserId)
if hub != nil {
if metrics := a.Metrics(); metrics != nil {
metrics.DecrementWebSocketBroadcastUsersRegistered(strconv.Itoa(hub.connectionIndex), 1)
}
hub.Unregister(webConn)
}
}