Adding session cache directly to web-conn (#4861)

api/context.go

@@ -566,6 +566,8 @@ func RemoveAllSessionsForUserIdSkipClusterSend(userId string) {
 		}
 	}
 
+	InvalidateWebConnSessionCacheForUser(userId)
+
 }
 
 func AddSessionToCache(session *model.Session) {

api/post.go

@@ -705,7 +705,7 @@ func sendNotifications(c *Context, post *model.Post, team *model.Team, channel *
 			}
 
 			if userAllowsEmails && status.Status != model.STATUS_ONLINE {
-				go sendNotificationEmail(c, post, profileMap[id], channel, team, senderName[id], sender)
+				sendNotificationEmail(c, post, profileMap[id], channel, team, senderName[id], sender)
 			}
 		}
 	}
@@ -801,7 +801,7 @@ func sendNotifications(c *Context, post *model.Post, team *model.Team, channel *
 			}
 
 			if DoesStatusAllowPushNotification(profileMap[id], status, post.ChannelId) {
-				go sendPushNotification(post, profileMap[id], channel, senderName[id], true)
+				sendPushNotification(post, profileMap[id], channel, senderName[id], true)
 			}
 		}
 
@@ -814,7 +814,7 @@ func sendNotifications(c *Context, post *model.Post, team *model.Team, channel *
 				}
 
 				if DoesStatusAllowPushNotification(profileMap[id], status, post.ChannelId) {
-					go sendPushNotification(post, profileMap[id], channel, senderName[id], false)
+					sendPushNotification(post, profileMap[id], channel, senderName[id], false)
 				}
 			}
 		}

api/web_conn.go

@@ -27,6 +27,7 @@ type WebConn struct {
 	WebSocket                 *websocket.Conn
 	Send                      chan model.WebSocketMessage
 	SessionToken              string
+	SessionExpiresAt          int64
 	UserId                    string
 	T                         goi18n.TranslateFunc
 	Locale                    string
@@ -44,6 +45,7 @@ func NewWebConn(c *Context, ws *websocket.Conn) *WebConn {
 		WebSocket:        ws,
 		UserId:           c.Session.UserId,
 		SessionToken:     c.Session.Token,
+		SessionExpiresAt: c.Session.ExpiresAt,
 		T:                c.T,
 		Locale:           c.Locale,
 	}
@@ -144,18 +146,27 @@ func (c *WebConn) writePump() {
 func (webCon *WebConn) InvalidateCache() {
 	webCon.AllChannelMembers = nil
 	webCon.LastAllChannelMembersTime = 0
+	webCon.SessionExpiresAt = 0
 }
 
 func (webCon *WebConn) isAuthenticated() bool {
+	// Check the expiry to see if we need to check for a new session
+	if webCon.SessionExpiresAt < model.GetMillis() {
 		if webCon.SessionToken == "" {
 			return false
 		}
 
 		session := GetSession(webCon.SessionToken)
 		if session == nil || session.IsExpired() {
+			webCon.SessionToken = ""
+			webCon.SessionExpiresAt = 0
 			return false
 		}
 
+		webCon.SessionToken = session.Token
+		webCon.SessionExpiresAt = session.ExpiresAt
+	}
+
 	return true
 }
 

api/web_hub.go

@@ -145,6 +145,12 @@ func InvalidateCacheForUserSkipClusterSend(userId string) {
 	}
 }
 
+func InvalidateWebConnSessionCacheForUser(userId string) {
+	if len(hubs) != 0 {
+		GetHubForUserId(userId).InvalidateUser(userId)
+	}
+}
+
 func (h *Hub) Register(webConn *WebConn) {
 	h.register <- webConn