From 91c998156336e34ab4b8979db77cc65c97a65782 Mon Sep 17 00:00:00 2001 From: Joram Wilander Date: Fri, 11 May 2018 11:56:02 -0400 Subject: More potential panic fixes (#8776) --- app/web_hub.go | 15 +++++++++------ 1 file changed, 9 insertions(+), 6 deletions(-) (limited to 'app/web_hub.go') diff --git a/app/web_hub.go b/app/web_hub.go index f69645f50..2ce78b5ef 100644 --- a/app/web_hub.go +++ b/app/web_hub.go @@ -177,8 +177,9 @@ func (a *App) Publish(message *model.WebSocketEvent) { func (a *App) PublishSkipClusterSend(message *model.WebSocketEvent) { if message.Broadcast.UserId != "" { - if len(a.Hubs) != 0 { - a.GetHubForUserId(message.Broadcast.UserId).Broadcast(message) + hub := a.GetHubForUserId(message.Broadcast.UserId) + if hub != nil { + hub.Broadcast(message) } } else { for _, hub := range a.Hubs { @@ -299,8 +300,9 @@ func (a *App) InvalidateCacheForUserSkipClusterSend(userId string) { a.Srv.Store.User().InvalidateProfilesInChannelCacheByUser(userId) a.Srv.Store.User().InvalidatProfileCacheForUser(userId) - if len(a.Hubs) != 0 { - a.GetHubForUserId(userId).InvalidateUser(userId) + hub := a.GetHubForUserId(userId) + if hub != nil { + hub.InvalidateUser(userId) } } @@ -322,8 +324,9 @@ func (a *App) InvalidateCacheForWebhookSkipClusterSend(webhookId string) { } func (a *App) InvalidateWebConnSessionCacheForUser(userId string) { - if len(a.Hubs) != 0 { - a.GetHubForUserId(userId).InvalidateUser(userId) + hub := a.GetHubForUserId(userId) + if hub != nil { + hub.InvalidateUser(userId) } } -- cgit v1.2.3-1-g7c22