remove remaining Global() calls (outside of tests) (#7521)

Этот коммит содержится в:
Chris
2017-09-27 11:52:34 -05:00
коммит произвёл Saturnino Abril
родитель 1bd66589a2
Коммит 8c80cdde38
45 изменённых файлов: 207 добавлений и 222 удалений

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

@@ -121,7 +121,7 @@ func Init(a *app.App, root *mux.Router) *API {
utils.InitHTML() utils.InitHTML()
app.InitEmailBatching() a.InitEmailBatching()
if *utils.Cfg.ServiceSettings.EnableAPIv3 { if *utils.Cfg.ServiceSettings.EnableAPIv3 {
l4g.Info("API version 3 is scheduled for deprecation. Please see https://api.mattermost.com for details.") l4g.Info("API version 3 is scheduled for deprecation. Please see https://api.mattermost.com for details.")

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

@@ -49,11 +49,11 @@ func setupTestHelper(enterprise bool) *TestHelper {
th.App.NewServer() th.App.NewServer()
th.App.InitStores() th.App.InitStores()
th.App.Srv.Router = NewRouter() th.App.Srv.Router = NewRouter()
wsapi.InitRouter() th.App.Srv.WebSocketRouter = th.App.NewWebSocketRouter()
th.App.StartServer() th.App.StartServer()
api4.Init(th.App, th.App.Srv.Router, false) api4.Init(th.App, th.App.Srv.Router, false)
Init(th.App, th.App.Srv.Router) Init(th.App, th.App.Srv.Router)
wsapi.InitApi() wsapi.Init(th.App, th.App.Srv.WebSocketRouter)
utils.EnableDebugLogForTest() utils.EnableDebugLogForTest()
th.App.Srv.Store.MarkSystemRanUnitTests() th.App.Srv.Store.MarkSystemRanUnitTests()

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

@@ -8,7 +8,6 @@ import (
"strings" "strings"
"testing" "testing"
"github.com/mattermost/mattermost-server/app"
"github.com/mattermost/mattermost-server/model" "github.com/mattermost/mattermost-server/model"
) )
@@ -96,7 +95,7 @@ func TestCliCreateUserWithoutTeam(t *testing.T) {
t.SkipNow() t.SkipNow()
} }
Setup() th := Setup()
id := model.NewId() id := model.NewId()
email := "success+" + id + "@simulator.amazonses.com" email := "success+" + id + "@simulator.amazonses.com"
username := "name" + id username := "name" + id
@@ -108,7 +107,7 @@ func TestCliCreateUserWithoutTeam(t *testing.T) {
t.Fatal(err) t.Fatal(err)
} }
if result := <-app.Global().Srv.Store.User().GetByEmail(email); result.Err != nil { if result := <-th.App.Srv.Store.User().GetByEmail(email); result.Err != nil {
t.Fatal() t.Fatal()
} else { } else {
user := result.Data.(*model.User) user := result.Data.(*model.User)

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

@@ -118,7 +118,7 @@ func createEmoji(c *Context, w http.ResponseWriter, r *http.Request) {
message := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_EMOJI_ADDED, "", "", "", nil) message := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_EMOJI_ADDED, "", "", "", nil)
message.Add("emoji", result.Data.(*model.Emoji).ToJson()) message.Add("emoji", result.Data.(*model.Emoji).ToJson())
app.Publish(message) c.App.Publish(message)
w.Write([]byte(result.Data.(*model.Emoji).ToJson())) w.Write([]byte(result.Data.(*model.Emoji).ToJson()))
} }
} }

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

@@ -141,7 +141,7 @@ func saveIsPinnedPost(c *Context, w http.ResponseWriter, r *http.Request, isPinn
message := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_POST_EDITED, "", rpost.ChannelId, "", nil) message := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_POST_EDITED, "", rpost.ChannelId, "", nil)
message.Add("post", rpost.ToJson()) message.Add("post", rpost.ToJson())
go app.Publish(message) go c.App.Publish(message)
c.App.InvalidateCacheForChannelPosts(rpost.ChannelId) c.App.InvalidateCacheForChannelPosts(rpost.ChannelId)

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

@@ -8,7 +8,6 @@ import (
l4g "github.com/alecthomas/log4go" l4g "github.com/alecthomas/log4go"
"github.com/gorilla/mux" "github.com/gorilla/mux"
"github.com/mattermost/mattermost-server/app"
"github.com/mattermost/mattermost-server/model" "github.com/mattermost/mattermost-server/model"
"github.com/mattermost/mattermost-server/utils" "github.com/mattermost/mattermost-server/utils"
) )
@@ -112,23 +111,6 @@ func deleteReaction(c *Context, w http.ResponseWriter, r *http.Request) {
ReturnStatusOK(w) ReturnStatusOK(w)
} }
func sendReactionEvent(event string, channelId string, reaction *model.Reaction, post *model.Post) {
// send out that a reaction has been added/removed
message := model.NewWebSocketEvent(event, "", channelId, "", nil)
message.Add("reaction", reaction.ToJson())
app.Publish(message)
// THe post is always modified since the UpdateAt always changes
app.Global().InvalidateCacheForChannelPosts(post.ChannelId)
post.HasReactions = true
post.UpdateAt = model.GetMillis()
umessage := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_POST_EDITED, "", channelId, "", nil)
umessage.Add("post", post.ToJson())
app.Publish(umessage)
}
func listReactions(c *Context, w http.ResponseWriter, r *http.Request) { func listReactions(c *Context, w http.ResponseWriter, r *http.Request) {
params := mux.Vars(r) params := mux.Vars(r)

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

@@ -8,7 +8,6 @@ import (
l4g "github.com/alecthomas/log4go" l4g "github.com/alecthomas/log4go"
"github.com/gorilla/websocket" "github.com/gorilla/websocket"
"github.com/mattermost/mattermost-server/app"
"github.com/mattermost/mattermost-server/model" "github.com/mattermost/mattermost-server/model"
"github.com/mattermost/mattermost-server/utils" "github.com/mattermost/mattermost-server/utils"
) )
@@ -37,7 +36,7 @@ func connect(c *Context, w http.ResponseWriter, r *http.Request) {
wc := c.App.NewWebConn(ws, c.Session, c.T, c.Locale) wc := c.App.NewWebConn(ws, c.Session, c.T, c.Locale)
if len(c.Session.UserId) > 0 { if len(c.Session.UserId) > 0 {
app.HubRegister(wc) c.App.HubRegister(wc)
} }
go wc.WritePump() go wc.WritePump()

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

@@ -11,7 +11,6 @@ import (
"time" "time"
"github.com/gorilla/websocket" "github.com/gorilla/websocket"
"github.com/mattermost/mattermost-server/app"
"github.com/mattermost/mattermost-server/model" "github.com/mattermost/mattermost-server/model"
"github.com/mattermost/mattermost-server/utils" "github.com/mattermost/mattermost-server/utils"
) )
@@ -195,7 +194,7 @@ func TestWebSocketEvent(t *testing.T) {
omitUser["somerandomid"] = true omitUser["somerandomid"] = true
evt1 := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_TYPING, "", th.BasicChannel.Id, "", omitUser) evt1 := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_TYPING, "", th.BasicChannel.Id, "", omitUser)
evt1.Add("user_id", "somerandomid") evt1.Add("user_id", "somerandomid")
app.Publish(evt1) th.App.Publish(evt1)
time.Sleep(300 * time.Millisecond) time.Sleep(300 * time.Millisecond)
@@ -224,7 +223,7 @@ func TestWebSocketEvent(t *testing.T) {
} }
evt2 := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_TYPING, "", "somerandomid", "", nil) evt2 := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_TYPING, "", "somerandomid", "", nil)
go app.Publish(evt2) go th.App.Publish(evt2)
time.Sleep(300 * time.Millisecond) time.Sleep(300 * time.Millisecond)
eventHit = false eventHit = false

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

@@ -226,7 +226,7 @@ func Init(a *app.App, root *mux.Router, full bool) *API {
if full { if full {
utils.InitHTML() utils.InitHTML()
app.InitEmailBatching() a.InitEmailBatching()
} }
return api return api

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

@@ -61,10 +61,10 @@ func setupTestHelper(enterprise bool) *TestHelper {
th.App.NewServer() th.App.NewServer()
th.App.InitStores() th.App.InitStores()
th.App.Srv.Router = NewRouter() th.App.Srv.Router = NewRouter()
wsapi.InitRouter() th.App.Srv.WebSocketRouter = th.App.NewWebSocketRouter()
th.App.StartServer() th.App.StartServer()
Init(th.App, th.App.Srv.Router, true) Init(th.App, th.App.Srv.Router, true)
wsapi.InitApi() wsapi.Init(th.App, th.App.Srv.WebSocketRouter)
utils.EnableDebugLogForTest() utils.EnableDebugLogForTest()
th.App.Srv.Store.MarkSystemRanUnitTests() th.App.Srv.Store.MarkSystemRanUnitTests()

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

@@ -8,7 +8,6 @@ import (
l4g "github.com/alecthomas/log4go" l4g "github.com/alecthomas/log4go"
"github.com/gorilla/websocket" "github.com/gorilla/websocket"
"github.com/mattermost/mattermost-server/app"
"github.com/mattermost/mattermost-server/model" "github.com/mattermost/mattermost-server/model"
"github.com/mattermost/mattermost-server/utils" "github.com/mattermost/mattermost-server/utils"
) )
@@ -38,7 +37,7 @@ func connectWebSocket(c *Context, w http.ResponseWriter, r *http.Request) {
wc := c.App.NewWebConn(ws, c.Session, c.T, "") wc := c.App.NewWebConn(ws, c.Session, c.T, "")
if len(c.Session.UserId) > 0 { if len(c.Session.UserId) > 0 {
app.HubRegister(wc) c.App.HubRegister(wc)
} }
go wc.WritePump() go wc.WritePump()

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

@@ -138,7 +138,7 @@ func (a *App) ReloadConfig() {
utils.LoadConfig(utils.CfgFileName) utils.LoadConfig(utils.CfgFileName)
// start/restart email batching job if necessary // start/restart email batching job if necessary
InitEmailBatching() a.InitEmailBatching()
} }
func (a *App) SaveConfig(cfg *model.Config, sendConfigChangeClusterMessage bool) *model.AppError { func (a *App) SaveConfig(cfg *model.Config, sendConfigChangeClusterMessage bool) *model.AppError {
@@ -179,7 +179,7 @@ func (a *App) SaveConfig(cfg *model.Config, sendConfigChangeClusterMessage bool)
} }
// start/restart email batching job if necessary // start/restart email batching job if necessary
InitEmailBatching() a.InitEmailBatching()
return nil return nil
} }

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

@@ -103,7 +103,7 @@ func (a *App) GetAnalytics(name string, teamId string) (model.AnalyticsRows, *mo
return nil, err return nil, err
} }
totalSockets := TotalWebsocketConnections() totalSockets := a.TotalWebsocketConnections()
totalMasterDb := a.Srv.Store.TotalMasterDbConnections() totalMasterDb := a.Srv.Store.TotalMasterDbConnections()
totalReadDb := a.Srv.Store.TotalReadDbConnections() totalReadDb := a.Srv.Store.TotalReadDbConnections()
@@ -118,7 +118,7 @@ func (a *App) GetAnalytics(name string, teamId string) (model.AnalyticsRows, *mo
rows[7].Value = float64(totalReadDb) rows[7].Value = float64(totalReadDb)
} else { } else {
rows[5].Value = float64(TotalWebsocketConnections()) rows[5].Value = float64(a.TotalWebsocketConnections())
rows[6].Value = float64(a.Srv.Store.TotalMasterDbConnections()) rows[6].Value = float64(a.Srv.Store.TotalMasterDbConnections())
rows[7].Value = float64(a.Srv.Store.TotalReadDbConnections()) rows[7].Value = float64(a.Srv.Store.TotalReadDbConnections())
} }

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

@@ -20,6 +20,11 @@ type App struct {
PluginEnv *pluginenv.Environment PluginEnv *pluginenv.Environment
PluginConfigListenerId string PluginConfigListenerId string
EmailBatching *EmailBatchingJob
Hubs []*Hub
HubsStopCheckingForDeadlock chan bool
AccountMigration einterfaces.AccountMigrationInterface AccountMigration einterfaces.AccountMigrationInterface
Brand einterfaces.BrandInterface Brand einterfaces.BrandInterface
Cluster einterfaces.ClusterInterface Cluster einterfaces.ClusterInterface

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

@@ -16,7 +16,7 @@ type TestEnvironment struct {
Environments []TeamEnvironment Environments []TeamEnvironment
} }
func CreateTestEnvironmentWithTeams(client *model.Client, rangeTeams utils.Range, rangeChannels utils.Range, rangeUsers utils.Range, rangePosts utils.Range, fuzzy bool) (TestEnvironment, bool) { func CreateTestEnvironmentWithTeams(a *App, client *model.Client, rangeTeams utils.Range, rangeChannels utils.Range, rangeUsers utils.Range, rangePosts utils.Range, fuzzy bool) (TestEnvironment, bool) {
rand.Seed(time.Now().UTC().UnixNano()) rand.Seed(time.Now().UTC().UnixNano())
teamCreator := NewAutoTeamCreator(client) teamCreator := NewAutoTeamCreator(client)
@@ -29,7 +29,7 @@ func CreateTestEnvironmentWithTeams(client *model.Client, rangeTeams utils.Range
environment := TestEnvironment{teams, make([]TeamEnvironment, len(teams))} environment := TestEnvironment{teams, make([]TeamEnvironment, len(teams))}
for i, team := range teams { for i, team := range teams {
userCreator := NewAutoUserCreator(client, team) userCreator := NewAutoUserCreator(a, client, team)
userCreator.Fuzzy = fuzzy userCreator.Fuzzy = fuzzy
randomUser, err := userCreator.createRandomUser() randomUser, err := userCreator.createRandomUser()
if err != true { if err != true {
@@ -37,7 +37,7 @@ func CreateTestEnvironmentWithTeams(client *model.Client, rangeTeams utils.Range
} }
client.LoginById(randomUser.Id, USER_PASSWORD) client.LoginById(randomUser.Id, USER_PASSWORD)
client.SetTeamId(team.Id) client.SetTeamId(team.Id)
teamEnvironment, err := CreateTestEnvironmentInTeam(client, team, rangeChannels, rangeUsers, rangePosts, fuzzy) teamEnvironment, err := CreateTestEnvironmentInTeam(a, client, team, rangeChannels, rangeUsers, rangePosts, fuzzy)
if err != true { if err != true {
return TestEnvironment{}, false return TestEnvironment{}, false
} }
@@ -47,7 +47,7 @@ func CreateTestEnvironmentWithTeams(client *model.Client, rangeTeams utils.Range
return environment, true return environment, true
} }
func CreateTestEnvironmentInTeam(client *model.Client, team *model.Team, rangeChannels utils.Range, rangeUsers utils.Range, rangePosts utils.Range, fuzzy bool) (TeamEnvironment, bool) { func CreateTestEnvironmentInTeam(a *App, client *model.Client, team *model.Team, rangeChannels utils.Range, rangeUsers utils.Range, rangePosts utils.Range, fuzzy bool) (TeamEnvironment, bool) {
rand.Seed(time.Now().UTC().UnixNano()) rand.Seed(time.Now().UTC().UnixNano())
// We need to create at least one user // We need to create at least one user
@@ -55,7 +55,7 @@ func CreateTestEnvironmentInTeam(client *model.Client, team *model.Team, rangeCh
rangeUsers.Begin = 1 rangeUsers.Begin = 1
} }
userCreator := NewAutoUserCreator(client, team) userCreator := NewAutoUserCreator(a, client, team)
userCreator.Fuzzy = fuzzy userCreator.Fuzzy = fuzzy
users, err := userCreator.CreateTestUsers(rangeUsers) users, err := userCreator.CreateTestUsers(rangeUsers)
if err != true { if err != true {

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

@@ -12,6 +12,7 @@ import (
) )
type AutoUserCreator struct { type AutoUserCreator struct {
app *App
client *model.Client client *model.Client
team *model.Team team *model.Team
EmailLength utils.Range EmailLength utils.Range
@@ -21,8 +22,9 @@ type AutoUserCreator struct {
Fuzzy bool Fuzzy bool
} }
func NewAutoUserCreator(client *model.Client, team *model.Team) *AutoUserCreator { func NewAutoUserCreator(a *App, client *model.Client, team *model.Team) *AutoUserCreator {
return &AutoUserCreator{ return &AutoUserCreator{
app: a,
client: client, client: client,
team: team, team: team,
EmailLength: USER_EMAIL_LEN, EmailLength: USER_EMAIL_LEN,
@@ -81,14 +83,14 @@ func (cfg *AutoUserCreator) createRandomUser() (*model.User, bool) {
ruser := result.Data.(*model.User) ruser := result.Data.(*model.User)
status := &model.Status{UserId: ruser.Id, Status: model.STATUS_ONLINE, Manual: false, LastActivityAt: model.GetMillis(), ActiveChannel: ""} status := &model.Status{UserId: ruser.Id, Status: model.STATUS_ONLINE, Manual: false, LastActivityAt: model.GetMillis(), ActiveChannel: ""}
if result := <-Global().Srv.Store.Status().SaveOrUpdate(status); result.Err != nil { if result := <-cfg.app.Srv.Store.Status().SaveOrUpdate(status); result.Err != nil {
result.Err.Translate(utils.T) result.Err.Translate(utils.T)
l4g.Error(result.Err.Error()) l4g.Error(result.Err.Error())
return nil, false return nil, false
} }
// We need to cheat to verify the user's email // We need to cheat to verify the user's email
store.Must(Global().Srv.Store.User().VerifyEmail(ruser.Id)) store.Must(cfg.app.Srv.Store.User().VerifyEmail(ruser.Id))
return result.Data.(*model.User), true return result.Data.(*model.User), true
} }

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

@@ -136,7 +136,7 @@ func (a *App) CreateChannelWithUser(channel *model.Channel, userId string) (*mod
message := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_CHANNEL_CREATED, "", "", userId, nil) message := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_CHANNEL_CREATED, "", "", userId, nil)
message.Add("channel_id", channel.Id) message.Add("channel_id", channel.Id)
message.Add("team_id", channel.TeamId) message.Add("team_id", channel.TeamId)
Publish(message) a.Publish(message)
return rchannel, nil return rchannel, nil
} }
@@ -181,7 +181,7 @@ func (a *App) CreateDirectChannel(userId string, otherUserId string) (*model.Cha
message := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_DIRECT_ADDED, "", channel.Id, "", nil) message := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_DIRECT_ADDED, "", channel.Id, "", nil)
message.Add("teammate_id", otherUserId) message.Add("teammate_id", otherUserId)
Publish(message) a.Publish(message)
return channel, nil return channel, nil
} }
@@ -254,7 +254,7 @@ func (a *App) CreateGroupChannel(userIds []string, creatorId string) (*model.Cha
message := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_GROUP_ADDED, "", channel.Id, "", nil) message := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_GROUP_ADDED, "", channel.Id, "", nil)
message.Add("teammate_ids", model.ArrayToJson(userIds)) message.Add("teammate_ids", model.ArrayToJson(userIds))
Publish(message) a.Publish(message)
return channel, nil return channel, nil
} }
@@ -316,7 +316,7 @@ func (a *App) UpdateChannel(channel *model.Channel) (*model.Channel, *model.AppE
messageWs := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_CHANNEL_UPDATED, "", channel.Id, "", nil) messageWs := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_CHANNEL_UPDATED, "", channel.Id, "", nil)
messageWs.Add("channel", channel.ToJson()) messageWs.Add("channel", channel.ToJson())
Publish(messageWs) a.Publish(messageWs)
return channel, nil return channel, nil
} }
@@ -484,7 +484,7 @@ func (a *App) DeleteChannel(channel *model.Channel, userId string) *model.AppErr
message := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_CHANNEL_DELETED, channel.TeamId, "", "", nil) message := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_CHANNEL_DELETED, channel.TeamId, "", "", nil)
message.Add("channel_id", channel.Id) message.Add("channel_id", channel.Id)
Publish(message) a.Publish(message)
} }
return nil return nil
@@ -550,7 +550,7 @@ func (a *App) AddUserToChannel(user *model.User, channel *model.Channel) (*model
message := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_USER_ADDED, "", channel.Id, "", nil) message := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_USER_ADDED, "", channel.Id, "", nil)
message.Add("user_id", user.Id) message.Add("user_id", user.Id)
message.Add("team_id", channel.TeamId) message.Add("team_id", channel.TeamId)
Publish(message) a.Publish(message)
return newMember, nil return newMember, nil
} }
@@ -1039,13 +1039,13 @@ func (a *App) removeUserFromChannel(userIdToRemove string, removerUserId string,
message := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_USER_REMOVED, "", channel.Id, "", nil) message := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_USER_REMOVED, "", channel.Id, "", nil)
message.Add("user_id", userIdToRemove) message.Add("user_id", userIdToRemove)
message.Add("remover_id", removerUserId) message.Add("remover_id", removerUserId)
go Publish(message) go a.Publish(message)
// because the removed user no longer belongs to the channel we need to send a separate websocket event // because the removed user no longer belongs to the channel we need to send a separate websocket event
userMsg := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_USER_REMOVED, "", "", userIdToRemove, nil) userMsg := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_USER_REMOVED, "", "", userIdToRemove, nil)
userMsg.Add("channel_id", channel.Id) userMsg.Add("channel_id", channel.Id)
userMsg.Add("remover_id", removerUserId) userMsg.Add("remover_id", removerUserId)
go Publish(userMsg) go a.Publish(userMsg)
return nil return nil
} }
@@ -1098,7 +1098,7 @@ func (a *App) SetActiveChannel(userId string, channelId string) *model.AppError
a.AddStatusCache(status) a.AddStatusCache(status)
if status.Status != oldStatus { if status.Status != oldStatus {
BroadcastStatus(status) a.BroadcastStatus(status)
} }
return nil return nil
@@ -1113,7 +1113,7 @@ func (a *App) UpdateChannelLastViewedAt(channelIds []string, userId string) *mod
for _, channelId := range channelIds { for _, channelId := range channelIds {
message := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_CHANNEL_VIEWED, "", "", userId, nil) message := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_CHANNEL_VIEWED, "", "", userId, nil)
message.Add("channel_id", channelId) message.Add("channel_id", channelId)
go Publish(message) go a.Publish(message)
} }
} }
@@ -1179,7 +1179,7 @@ func (a *App) ViewChannel(view *model.ChannelView, userId string, clearPushNotif
if *utils.Cfg.ServiceSettings.EnableChannelViewedMessages && model.IsValidId(view.ChannelId) { if *utils.Cfg.ServiceSettings.EnableChannelViewedMessages && model.IsValidId(view.ChannelId) {
message := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_CHANNEL_VIEWED, "", "", userId, nil) message := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_CHANNEL_VIEWED, "", "", userId, nil)
message.Add("channel_id", view.ChannelId) message.Add("channel_id", view.ChannelId)
go Publish(message) go a.Publish(message)
} }
return nil return nil

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

@@ -20,12 +20,12 @@ func (a *App) RegisterAllClusterMessageHandlers() {
a.Cluster.RegisterClusterMessageHandler(model.CLUSTER_EVENT_INVALIDATE_CACHE_FOR_CHANNEL_BY_NAME, a.ClusterInvalidateCacheForChannelByNameHandler) a.Cluster.RegisterClusterMessageHandler(model.CLUSTER_EVENT_INVALIDATE_CACHE_FOR_CHANNEL_BY_NAME, a.ClusterInvalidateCacheForChannelByNameHandler)
a.Cluster.RegisterClusterMessageHandler(model.CLUSTER_EVENT_INVALIDATE_CACHE_FOR_CHANNEL, a.ClusterInvalidateCacheForChannelHandler) a.Cluster.RegisterClusterMessageHandler(model.CLUSTER_EVENT_INVALIDATE_CACHE_FOR_CHANNEL, a.ClusterInvalidateCacheForChannelHandler)
a.Cluster.RegisterClusterMessageHandler(model.CLUSTER_EVENT_INVALIDATE_CACHE_FOR_USER, a.ClusterInvalidateCacheForUserHandler) a.Cluster.RegisterClusterMessageHandler(model.CLUSTER_EVENT_INVALIDATE_CACHE_FOR_USER, a.ClusterInvalidateCacheForUserHandler)
a.Cluster.RegisterClusterMessageHandler(model.CLUSTER_EVENT_CLEAR_SESSION_CACHE_FOR_USER, ClusterClearSessionCacheForUserHandler) a.Cluster.RegisterClusterMessageHandler(model.CLUSTER_EVENT_CLEAR_SESSION_CACHE_FOR_USER, a.ClusterClearSessionCacheForUserHandler)
} }
func (a *App) ClusterPublishHandler(msg *model.ClusterMessage) { func (a *App) ClusterPublishHandler(msg *model.ClusterMessage) {
event := model.WebSocketEventFromJson(strings.NewReader(msg.Data)) event := model.WebSocketEventFromJson(strings.NewReader(msg.Data))
PublishSkipClusterSend(event) a.PublishSkipClusterSend(event)
} }
func (a *App) ClusterUpdateStatusHandler(msg *model.ClusterMessage) { func (a *App) ClusterUpdateStatusHandler(msg *model.ClusterMessage) {
@@ -65,6 +65,6 @@ func (a *App) ClusterInvalidateCacheForUserHandler(msg *model.ClusterMessage) {
a.InvalidateCacheForUserSkipClusterSend(msg.Data) a.InvalidateCacheForUserSkipClusterSend(msg.Data)
} }
func ClusterClearSessionCacheForUserHandler(msg *model.ClusterMessage) { func (a *App) ClusterClearSessionCacheForUserHandler(msg *model.ClusterMessage) {
ClearSessionCacheForUserSkipClusterSend(msg.Data) a.ClearSessionCacheForUserSkipClusterSend(msg.Data)
} }

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

@@ -49,7 +49,7 @@ func (a *App) CreateCommandPost(post *model.Post, teamId string, response *model
return a.CreatePostMissingChannel(post, true) return a.CreatePostMissingChannel(post, true)
} else if response.ResponseType == "" || response.ResponseType == model.COMMAND_RESPONSE_TYPE_EPHEMERAL { } else if response.ResponseType == "" || response.ResponseType == model.COMMAND_RESPONSE_TYPE_EPHEMERAL {
post.ParentId = "" post.ParentId = ""
SendEphemeralPost(post.UserId, post) a.SendEphemeralPost(post.UserId, post)
} }
return post, nil return post, nil

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

@@ -74,7 +74,7 @@ func (a *App) setCollapsePreference(args *model.CommandArgs, isCollapse bool) *m
socketMessage := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_PREFERENCE_CHANGED, "", "", args.UserId, nil) socketMessage := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_PREFERENCE_CHANGED, "", "", args.UserId, nil)
socketMessage.Add("preference", pref.ToJson()) socketMessage.Add("preference", pref.ToJson())
go Publish(socketMessage) go a.Publish(socketMessage)
var rmsg string var rmsg string

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

@@ -166,6 +166,7 @@ func (me *LoadTestProvider) SetupCommand(a *App, args *model.CommandArgs, messag
} }
client.Login(BTEST_USER_EMAIL, BTEST_USER_PASSWORD) client.Login(BTEST_USER_EMAIL, BTEST_USER_PASSWORD)
environment, err := CreateTestEnvironmentWithTeams( environment, err := CreateTestEnvironmentWithTeams(
a,
client, client,
utils.Range{Begin: numTeams, End: numTeams}, utils.Range{Begin: numTeams, End: numTeams},
utils.Range{Begin: numChannels, End: numChannels}, utils.Range{Begin: numChannels, End: numChannels},
@@ -193,6 +194,7 @@ func (me *LoadTestProvider) SetupCommand(a *App, args *model.CommandArgs, messag
client.MockSession(args.Session.Token) client.MockSession(args.Session.Token)
client.SetTeamId(args.TeamId) client.SetTeamId(args.TeamId)
CreateTestEnvironmentInTeam( CreateTestEnvironmentInTeam(
a,
client, client,
team, team,
utils.Range{Begin: numChannels, End: numChannels}, utils.Range{Begin: numChannels, End: numChannels},
@@ -227,7 +229,7 @@ func (me *LoadTestProvider) UsersCommand(a *App, args *model.CommandArgs, messag
client := model.NewClient(args.SiteURL) client := model.NewClient(args.SiteURL)
client.SetTeamId(team.Id) client.SetTeamId(team.Id)
userCreator := NewAutoUserCreator(client, team) userCreator := NewAutoUserCreator(a, client, team)
userCreator.Fuzzy = doFuzz userCreator.Fuzzy = doFuzz
userCreator.CreateTestUsers(usersr) userCreator.CreateTestUsers(usersr)

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

@@ -22,26 +22,24 @@ const (
EMAIL_BATCHING_TASK_NAME = "Email Batching" EMAIL_BATCHING_TASK_NAME = "Email Batching"
) )
var emailBatchingJob *EmailBatchingJob func (a *App) InitEmailBatching() {
func InitEmailBatching() {
if *utils.Cfg.EmailSettings.EnableEmailBatching { if *utils.Cfg.EmailSettings.EnableEmailBatching {
if emailBatchingJob == nil { if a.EmailBatching == nil {
emailBatchingJob = MakeEmailBatchingJob(*utils.Cfg.EmailSettings.EmailBatchingBufferSize) a.EmailBatching = NewEmailBatchingJob(a, *utils.Cfg.EmailSettings.EmailBatchingBufferSize)
} }
// note that we don't support changing EmailBatchingBufferSize without restarting the server // note that we don't support changing EmailBatchingBufferSize without restarting the server
emailBatchingJob.Start() a.EmailBatching.Start()
} }
} }
func AddNotificationEmailToBatch(user *model.User, post *model.Post, team *model.Team) *model.AppError { func (a *App) AddNotificationEmailToBatch(user *model.User, post *model.Post, team *model.Team) *model.AppError {
if !*utils.Cfg.EmailSettings.EnableEmailBatching { if !*utils.Cfg.EmailSettings.EnableEmailBatching {
return model.NewAppError("AddNotificationEmailToBatch", "api.email_batching.add_notification_email_to_batch.disabled.app_error", nil, "", http.StatusNotImplemented) return model.NewAppError("AddNotificationEmailToBatch", "api.email_batching.add_notification_email_to_batch.disabled.app_error", nil, "", http.StatusNotImplemented)
} }
if !emailBatchingJob.Add(user, post, team) { if !a.EmailBatching.Add(user, post, team) {
l4g.Error(utils.T("api.email_batching.add_notification_email_to_batch.channel_full.app_error")) l4g.Error(utils.T("api.email_batching.add_notification_email_to_batch.channel_full.app_error"))
return model.NewAppError("AddNotificationEmailToBatch", "api.email_batching.add_notification_email_to_batch.channel_full.app_error", nil, "", http.StatusInternalServerError) return model.NewAppError("AddNotificationEmailToBatch", "api.email_batching.add_notification_email_to_batch.channel_full.app_error", nil, "", http.StatusInternalServerError)
} }
@@ -56,12 +54,14 @@ type batchedNotification struct {
} }
type EmailBatchingJob struct { type EmailBatchingJob struct {
app *App
newNotifications chan *batchedNotification newNotifications chan *batchedNotification
pendingNotifications map[string][]*batchedNotification pendingNotifications map[string][]*batchedNotification
} }
func MakeEmailBatchingJob(bufferSize int) *EmailBatchingJob { func NewEmailBatchingJob(a *App, bufferSize int) *EmailBatchingJob {
return &EmailBatchingJob{ return &EmailBatchingJob{
app: a,
newNotifications: make(chan *batchedNotification, bufferSize), newNotifications: make(chan *batchedNotification, bufferSize),
pendingNotifications: make(map[string][]*batchedNotification), pendingNotifications: make(map[string][]*batchedNotification),
} }
@@ -97,7 +97,7 @@ func (job *EmailBatchingJob) CheckPendingEmails() {
// it's a bit weird to pass the send email function through here, but it makes it so that we can test // it's a bit weird to pass the send email function through here, but it makes it so that we can test
// without actually sending emails // without actually sending emails
job.checkPendingNotifications(time.Now(), Global().sendBatchedEmailNotification) job.checkPendingNotifications(time.Now(), job.app.sendBatchedEmailNotification)
l4g.Debug(utils.T("api.email_batching.check_pending_emails.finished_running"), len(job.pendingNotifications)) l4g.Debug(utils.T("api.email_batching.check_pending_emails.finished_running"), len(job.pendingNotifications))
} }
@@ -131,7 +131,7 @@ func (job *EmailBatchingJob) checkPendingNotifications(now time.Time, handler fu
if inspectedTeamNames[notification.teamName] != "" { if inspectedTeamNames[notification.teamName] != "" {
continue continue
} }
tchan := Global().Srv.Store.Team().GetByName(notifications[0].teamName) tchan := job.app.Srv.Store.Team().GetByName(notifications[0].teamName)
if result := <-tchan; result.Err != nil { if result := <-tchan; result.Err != nil {
l4g.Error("Unable to find Team id for notification", result.Err) l4g.Error("Unable to find Team id for notification", result.Err)
continue continue
@@ -141,7 +141,7 @@ func (job *EmailBatchingJob) checkPendingNotifications(now time.Time, handler fu
// if the user has viewed any channels in this team since the notification was queued, delete // if the user has viewed any channels in this team since the notification was queued, delete
// all queued notifications // all queued notifications
mchan := Global().Srv.Store.Channel().GetMembersForUser(inspectedTeamNames[notification.teamName], userId) mchan := job.app.Srv.Store.Channel().GetMembersForUser(inspectedTeamNames[notification.teamName], userId)
if result := <-mchan; result.Err != nil { if result := <-mchan; result.Err != nil {
l4g.Error("Unable to find ChannelMembers for user", result.Err) l4g.Error("Unable to find ChannelMembers for user", result.Err)
continue continue
@@ -158,7 +158,7 @@ func (job *EmailBatchingJob) checkPendingNotifications(now time.Time, handler fu
// get how long we need to wait to send notifications to the user // get how long we need to wait to send notifications to the user
var interval int64 var interval int64
pchan := Global().Srv.Store.Preference().Get(userId, model.PREFERENCE_CATEGORY_NOTIFICATIONS, model.PREFERENCE_NAME_EMAIL_INTERVAL) pchan := job.app.Srv.Store.Preference().Get(userId, model.PREFERENCE_CATEGORY_NOTIFICATIONS, model.PREFERENCE_NAME_EMAIL_INTERVAL)
if result := <-pchan; result.Err != nil { if result := <-pchan; result.Err != nil {
// use the default batching interval if an error ocurrs while fetching user preferences // use the default batching interval if an error ocurrs while fetching user preferences
interval, _ = strconv.ParseInt(model.PREFERENCE_EMAIL_INTERVAL_BATCHING_SECONDS, 10, 64) interval, _ = strconv.ParseInt(model.PREFERENCE_EMAIL_INTERVAL_BATCHING_SECONDS, 10, 64)

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

@@ -13,14 +13,14 @@ import (
) )
func TestHandleNewNotifications(t *testing.T) { func TestHandleNewNotifications(t *testing.T) {
Setup() th := Setup()
id1 := model.NewId() id1 := model.NewId()
id2 := model.NewId() id2 := model.NewId()
id3 := model.NewId() id3 := model.NewId()
// test queueing of received posts by user // test queueing of received posts by user
job := MakeEmailBatchingJob(128) job := NewEmailBatchingJob(th.App, 128)
job.handleNewNotifications() job.handleNewNotifications()
@@ -74,7 +74,7 @@ func TestHandleNewNotifications(t *testing.T) {
} }
// test ordering of received posts // test ordering of received posts
job = MakeEmailBatchingJob(128) job = NewEmailBatchingJob(th.App, 128)
job.Add(&model.User{Id: id1}, &model.Post{UserId: id1, Message: "test1"}, &model.Team{Name: "team"}) job.Add(&model.User{Id: id1}, &model.Post{UserId: id1, Message: "test1"}, &model.Team{Name: "team"})
job.Add(&model.User{Id: id1}, &model.Post{UserId: id1, Message: "test2"}, &model.Team{Name: "team"}) job.Add(&model.User{Id: id1}, &model.Post{UserId: id1, Message: "test2"}, &model.Team{Name: "team"})
@@ -95,7 +95,7 @@ func TestHandleNewNotifications(t *testing.T) {
func TestCheckPendingNotifications(t *testing.T) { func TestCheckPendingNotifications(t *testing.T) {
th := Setup().InitBasic() th := Setup().InitBasic()
job := MakeEmailBatchingJob(128) job := NewEmailBatchingJob(th.App, 128)
job.pendingNotifications[th.BasicUser.Id] = []*batchedNotification{ job.pendingNotifications[th.BasicUser.Id] = []*batchedNotification{
{ {
post: &model.Post{ post: &model.Post{
@@ -201,7 +201,7 @@ func TestCheckPendingNotifications(t *testing.T) {
*/ */
func TestCheckPendingNotificationsDefaultInterval(t *testing.T) { func TestCheckPendingNotificationsDefaultInterval(t *testing.T) {
th := Setup().InitBasic() th := Setup().InitBasic()
job := MakeEmailBatchingJob(128) job := NewEmailBatchingJob(th.App, 128)
// bypasses recent user activity check // bypasses recent user activity check
channelMember := store.Must(th.App.Srv.Store.Channel().GetMember(th.BasicChannel.Id, th.BasicUser.Id)).(*model.ChannelMember) channelMember := store.Must(th.App.Srv.Store.Channel().GetMember(th.BasicChannel.Id, th.BasicUser.Id)).(*model.ChannelMember)
@@ -237,7 +237,7 @@ func TestCheckPendingNotificationsDefaultInterval(t *testing.T) {
*/ */
func TestCheckPendingNotificationsCantParseInterval(t *testing.T) { func TestCheckPendingNotificationsCantParseInterval(t *testing.T) {
th := Setup().InitBasic() th := Setup().InitBasic()
job := MakeEmailBatchingJob(128) job := NewEmailBatchingJob(th.App, 128)
// bypasses recent user activity check // bypasses recent user activity check
channelMember := store.Must(th.App.Srv.Store.Channel().GetMember(th.BasicChannel.Id, th.BasicUser.Id)).(*model.ChannelMember) channelMember := store.Must(th.App.Srv.Store.Channel().GetMember(th.BasicChannel.Id, th.BasicUser.Id)).(*model.ChannelMember)

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

@@ -61,7 +61,7 @@ func (a *App) CreateEmoji(sessionUserId string, emoji *model.Emoji, multiPartIma
message := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_EMOJI_ADDED, "", "", "", nil) message := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_EMOJI_ADDED, "", "", "", nil)
message.Add("emoji", emoji.ToJson()) message.Add("emoji", emoji.ToJson())
Publish(message) a.Publish(message)
return result.Data.(*model.Emoji), nil return result.Data.(*model.Emoji), nil
} }
} }

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

@@ -94,7 +94,7 @@ func (a *App) SendNotifications(post *model.Post, team *model.Team, channel *mod
if result := <-a.Srv.Store.User().GetProfilesByUsernames(potentialOtherMentions, team.Id); result.Err == nil { if result := <-a.Srv.Store.User().GetProfilesByUsernames(potentialOtherMentions, team.Id); result.Err == nil {
outOfChannelMentions := result.Data.([]*model.User) outOfChannelMentions := result.Data.([]*model.User)
if channel.Type != model.CHANNEL_GROUP { if channel.Type != model.CHANNEL_GROUP {
go sendOutOfChannelMentions(sender, post, team.Id, outOfChannelMentions) go a.sendOutOfChannelMentions(sender, post, team.Id, outOfChannelMentions)
} }
} }
} }
@@ -186,7 +186,7 @@ func (a *App) SendNotifications(post *model.Post, team *model.Team, channel *mod
// If the channel has more than 1K users then @here is disabled // If the channel has more than 1K users then @here is disabled
if hereNotification && int64(len(profileMap)) > *utils.Cfg.TeamSettings.MaxNotificationsPerChannel { if hereNotification && int64(len(profileMap)) > *utils.Cfg.TeamSettings.MaxNotificationsPerChannel {
hereNotification = false hereNotification = false
SendEphemeralPost( a.SendEphemeralPost(
post.UserId, post.UserId,
&model.Post{ &model.Post{
ChannelId: post.ChannelId, ChannelId: post.ChannelId,
@@ -198,7 +198,7 @@ func (a *App) SendNotifications(post *model.Post, team *model.Team, channel *mod
// If the channel has more than 1K users then @channel is disabled // If the channel has more than 1K users then @channel is disabled
if channelNotification && int64(len(profileMap)) > *utils.Cfg.TeamSettings.MaxNotificationsPerChannel { if channelNotification && int64(len(profileMap)) > *utils.Cfg.TeamSettings.MaxNotificationsPerChannel {
SendEphemeralPost( a.SendEphemeralPost(
post.UserId, post.UserId,
&model.Post{ &model.Post{
ChannelId: post.ChannelId, ChannelId: post.ChannelId,
@@ -210,7 +210,7 @@ func (a *App) SendNotifications(post *model.Post, team *model.Team, channel *mod
// If the channel has more than 1K users then @all is disabled // If the channel has more than 1K users then @all is disabled
if allNotification && int64(len(profileMap)) > *utils.Cfg.TeamSettings.MaxNotificationsPerChannel { if allNotification && int64(len(profileMap)) > *utils.Cfg.TeamSettings.MaxNotificationsPerChannel {
SendEphemeralPost( a.SendEphemeralPost(
post.UserId, post.UserId,
&model.Post{ &model.Post{
ChannelId: post.ChannelId, ChannelId: post.ChannelId,
@@ -298,7 +298,7 @@ func (a *App) SendNotifications(post *model.Post, team *model.Team, channel *mod
message.Add("mentions", model.ArrayToJson(mentionedUsersList)) message.Add("mentions", model.ArrayToJson(mentionedUsersList))
} }
Publish(message) a.Publish(message)
return mentionedUsersList, nil return mentionedUsersList, nil
} }
@@ -337,7 +337,7 @@ func (a *App) sendNotificationEmail(post *model.Post, user *model.User, channel
} }
if sendBatched { if sendBatched {
if err := AddNotificationEmailToBatch(user, post, team); err == nil { if err := a.AddNotificationEmailToBatch(user, post, team); err == nil {
return nil return nil
} }
} }
@@ -717,7 +717,7 @@ func (a *App) getMobileAppSessions(userId string) ([]*model.Session, *model.AppE
} }
} }
func sendOutOfChannelMentions(sender *model.User, post *model.Post, teamId string, users []*model.User) *model.AppError { func (a *App) sendOutOfChannelMentions(sender *model.User, post *model.Post, teamId string, users []*model.User) *model.AppError {
if len(users) == 0 { if len(users) == 0 {
return nil return nil
} }
@@ -742,7 +742,7 @@ func sendOutOfChannelMentions(sender *model.User, post *model.Post, teamId strin
}) })
} }
SendEphemeralPost( a.SendEphemeralPost(
post.UserId, post.UserId,
&model.Post{ &model.Post{
ChannelId: post.ChannelId, ChannelId: post.ChannelId,

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

@@ -335,7 +335,7 @@ func (a *App) UnpackAndActivatePlugin(pluginFile io.Reader) (*model.Manifest, *m
if manifest.HasClient() { if manifest.HasClient() {
message := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_PLUGIN_ACTIVATED, "", "", "", nil) message := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_PLUGIN_ACTIVATED, "", "", "", nil)
message.Add("manifest", manifest.ClientManifest()) message.Add("manifest", manifest.ClientManifest())
Publish(message) a.Publish(message)
} }
return manifest, nil return manifest, nil
@@ -383,7 +383,7 @@ func (a *App) RemovePlugin(id string) *model.AppError {
if manifest.HasClient() { if manifest.HasClient() {
message := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_PLUGIN_DEACTIVATED, "", "", "", nil) message := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_PLUGIN_DEACTIVATED, "", "", "", nil)
message.Add("manifest", manifest.ClientManifest()) message.Add("manifest", manifest.ClientManifest())
Publish(message) a.Publish(message)
} }
return nil return nil

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

@@ -51,7 +51,7 @@ func (a *App) CreatePostAsUser(post *model.Post) (*model.Post, *model.AppError)
} }
T := utils.GetUserTranslations(user.Locale) T := utils.GetUserTranslations(user.Locale)
SendEphemeralPost( a.SendEphemeralPost(
post.UserId, post.UserId,
&model.Post{ &model.Post{
ChannelId: channel.Id, ChannelId: channel.Id,
@@ -75,7 +75,7 @@ func (a *App) CreatePostAsUser(post *model.Post) (*model.Post, *model.AppError)
if *utils.Cfg.ServiceSettings.EnableChannelViewedMessages { if *utils.Cfg.ServiceSettings.EnableChannelViewedMessages {
message := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_CHANNEL_VIEWED, "", "", post.UserId, nil) message := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_CHANNEL_VIEWED, "", "", post.UserId, nil)
message.Add("channel_id", post.ChannelId) message.Add("channel_id", post.ChannelId)
go Publish(message) go a.Publish(message)
} }
} }
@@ -239,7 +239,7 @@ func parseSlackLinksToMarkdown(text string) string {
return linkWithTextRegex.ReplaceAllString(text, "[${2}](${1})") return linkWithTextRegex.ReplaceAllString(text, "[${2}](${1})")
} }
func SendEphemeralPost(userId string, post *model.Post) *model.Post { func (a *App) SendEphemeralPost(userId string, post *model.Post) *model.Post {
post.Type = model.POST_EPHEMERAL post.Type = model.POST_EPHEMERAL
// fill in fields which haven't been specified which have sensible defaults // fill in fields which haven't been specified which have sensible defaults
@@ -256,7 +256,7 @@ func SendEphemeralPost(userId string, post *model.Post) *model.Post {
message := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_EPHEMERAL_MESSAGE, "", post.ChannelId, userId, nil) message := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_EPHEMERAL_MESSAGE, "", post.ChannelId, userId, nil)
message.Add("post", post.ToJson()) message.Add("post", post.ToJson())
go Publish(message) go a.Publish(message)
return post return post
} }
@@ -330,7 +330,7 @@ func (a *App) UpdatePost(post *model.Post, safeUpdate bool) (*model.Post, *model
}() }()
} }
sendUpdatedPostEvent(rpost) a.sendUpdatedPostEvent(rpost)
a.InvalidateCacheForChannelPosts(rpost.ChannelId) a.InvalidateCacheForChannelPosts(rpost.ChannelId)
@@ -351,17 +351,17 @@ func (a *App) PatchPost(postId string, patch *model.PostPatch) (*model.Post, *mo
return nil, err return nil, err
} }
sendUpdatedPostEvent(updatedPost) a.sendUpdatedPostEvent(updatedPost)
a.InvalidateCacheForChannelPosts(updatedPost.ChannelId) a.InvalidateCacheForChannelPosts(updatedPost.ChannelId)
return updatedPost, nil return updatedPost, nil
} }
func sendUpdatedPostEvent(post *model.Post) { func (a *App) sendUpdatedPostEvent(post *model.Post) {
message := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_POST_EDITED, "", post.ChannelId, "", nil) message := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_POST_EDITED, "", post.ChannelId, "", nil)
message.Add("post", post.ToJson()) message.Add("post", post.ToJson())
go Publish(message) go a.Publish(message)
} }
func (a *App) GetPostsPage(channelId string, page int, perPage int) (*model.PostList, *model.AppError) { func (a *App) GetPostsPage(channelId string, page int, perPage int) (*model.PostList, *model.AppError) {
@@ -502,7 +502,7 @@ func (a *App) DeletePost(postId string) (*model.Post, *model.AppError) {
message := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_POST_DELETED, "", post.ChannelId, "", nil) message := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_POST_DELETED, "", post.ChannelId, "", nil)
message.Add("post", post.ToJson()) message.Add("post", post.ToJson())
go Publish(message) go a.Publish(message)
go a.DeletePostFiles(post) go a.DeletePostFiles(post)
go a.DeleteFlaggedPosts(post.Id) go a.DeleteFlaggedPosts(post.Id)
@@ -724,7 +724,7 @@ func (a *App) DoPostAction(postId string, actionId string, userId string) *model
} }
ephemeralPost.UserId = userId ephemeralPost.UserId = userId
ephemeralPost.AddProp("from_webhook", "true") ephemeralPost.AddProp("from_webhook", "true")
SendEphemeralPost(userId, ephemeralPost) a.SendEphemeralPost(userId, ephemeralPost)
} }
return nil return nil

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

@@ -55,7 +55,7 @@ func (a *App) UpdatePreferences(userId string, preferences model.Preferences) *m
message := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_PREFERENCES_CHANGED, "", "", userId, nil) message := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_PREFERENCES_CHANGED, "", "", userId, nil)
message.Add("preferences", preferences.ToJson()) message.Add("preferences", preferences.ToJson())
go Publish(message) go a.Publish(message)
return nil return nil
} }
@@ -78,7 +78,7 @@ func (a *App) DeletePreferences(userId string, preferences model.Preferences) *m
message := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_PREFERENCES_DELETED, "", "", userId, nil) message := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_PREFERENCES_DELETED, "", "", userId, nil)
message.Add("preferences", preferences.ToJson()) message.Add("preferences", preferences.ToJson())
go Publish(message) go a.Publish(message)
return nil return nil
} }

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

@@ -51,7 +51,7 @@ func (a *App) sendReactionEvent(event string, reaction *model.Reaction, post *mo
// send out that a reaction has been added/removed // send out that a reaction has been added/removed
message := model.NewWebSocketEvent(event, "", post.ChannelId, "", nil) message := model.NewWebSocketEvent(event, "", post.ChannelId, "", nil)
message.Add("reaction", reaction.ToJson()) message.Add("reaction", reaction.ToJson())
Publish(message) a.Publish(message)
// The post is always modified since the UpdateAt always changes // The post is always modified since the UpdateAt always changes
a.InvalidateCacheForChannelPosts(post.ChannelId) a.InvalidateCacheForChannelPosts(post.ChannelId)
@@ -59,5 +59,5 @@ func (a *App) sendReactionEvent(event string, reaction *model.Reaction, post *mo
post.UpdateAt = model.GetMillis() post.UpdateAt = model.GetMillis()
umessage := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_POST_EDITED, "", post.ChannelId, "", nil) umessage := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_POST_EDITED, "", post.ChannelId, "", nil)
umessage.Add("post", post.ToJson()) umessage.Add("post", post.ToJson())
Publish(umessage) a.Publish(umessage)
} }

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

@@ -218,7 +218,7 @@ func (a *App) StopServer() {
a.Srv.GracefulServer.Stop(TIME_TO_WAIT_FOR_CONNECTIONS_TO_CLOSE_ON_SERVER_SHUTDOWN) a.Srv.GracefulServer.Stop(TIME_TO_WAIT_FOR_CONNECTIONS_TO_CLOSE_ON_SERVER_SHUTDOWN)
a.Srv.Store.Close() a.Srv.Store.Close()
HubStop() a.HubStop()
a.ShutDownPlugins() a.ShutDownPlugins()

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

@@ -107,8 +107,7 @@ func (a *App) RevokeAllSessions(userId string) *model.AppError {
} }
func (a *App) ClearSessionCacheForUser(userId string) { func (a *App) ClearSessionCacheForUser(userId string) {
a.ClearSessionCacheForUserSkipClusterSend(userId)
ClearSessionCacheForUserSkipClusterSend(userId)
if a.Cluster != nil { if a.Cluster != nil {
msg := &model.ClusterMessage{ msg := &model.ClusterMessage{
@@ -120,7 +119,7 @@ func (a *App) ClearSessionCacheForUser(userId string) {
} }
} }
func ClearSessionCacheForUserSkipClusterSend(userId string) { func (a *App) ClearSessionCacheForUserSkipClusterSend(userId string) {
keys := sessionCache.Keys() keys := sessionCache.Keys()
for _, key := range keys { for _, key := range keys {
@@ -132,8 +131,7 @@ func ClearSessionCacheForUserSkipClusterSend(userId string) {
} }
} }
InvalidateWebConnSessionCacheForUser(userId) a.InvalidateWebConnSessionCacheForUser(userId)
} }
func AddSessionToCache(session *model.Session) { func AddSessionToCache(session *model.Session) {

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

@@ -213,15 +213,15 @@ func (a *App) SetStatusOnline(userId string, sessionId string, manual bool) {
} }
if broadcast { if broadcast {
BroadcastStatus(status) a.BroadcastStatus(status)
} }
} }
func BroadcastStatus(status *model.Status) { func (a *App) BroadcastStatus(status *model.Status) {
event := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_STATUS_CHANGE, "", "", status.UserId, nil) event := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_STATUS_CHANGE, "", "", status.UserId, nil)
event.Add("status", status.Status) event.Add("status", status.Status)
event.Add("user_id", status.UserId) event.Add("user_id", status.UserId)
go Publish(event) go a.Publish(event)
} }
func (a *App) SetStatusOffline(userId string, manual bool) { func (a *App) SetStatusOffline(userId string, manual bool) {
@@ -245,7 +245,7 @@ func (a *App) SetStatusOffline(userId string, manual bool) {
event := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_STATUS_CHANGE, "", "", status.UserId, nil) event := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_STATUS_CHANGE, "", "", status.UserId, nil)
event.Add("status", model.STATUS_OFFLINE) event.Add("status", model.STATUS_OFFLINE)
event.Add("user_id", status.UserId) event.Add("user_id", status.UserId)
go Publish(event) go a.Publish(event)
} }
func (a *App) SetStatusAwayIfNeeded(userId string, manual bool) { func (a *App) SetStatusAwayIfNeeded(userId string, manual bool) {
@@ -286,7 +286,7 @@ func (a *App) SetStatusAwayIfNeeded(userId string, manual bool) {
event := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_STATUS_CHANGE, "", "", status.UserId, nil) event := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_STATUS_CHANGE, "", "", status.UserId, nil)
event.Add("status", model.STATUS_AWAY) event.Add("status", model.STATUS_AWAY)
event.Add("user_id", status.UserId) event.Add("user_id", status.UserId)
go Publish(event) go a.Publish(event)
} }
func GetStatusFromCache(userId string) *model.Status { func GetStatusFromCache(userId string) *model.Status {

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

@@ -106,7 +106,7 @@ func (a *App) UpdateTeam(team *model.Team) (*model.Team, *model.AppError) {
oldTeam.Sanitize() oldTeam.Sanitize()
sendUpdatedTeamEvent(oldTeam) a.sendUpdatedTeamEvent(oldTeam)
return oldTeam, nil return oldTeam, nil
} }
@@ -126,15 +126,15 @@ func (a *App) PatchTeam(teamId string, patch *model.TeamPatch) (*model.Team, *mo
updatedTeam.Sanitize() updatedTeam.Sanitize()
sendUpdatedTeamEvent(updatedTeam) a.sendUpdatedTeamEvent(updatedTeam)
return updatedTeam, nil return updatedTeam, nil
} }
func sendUpdatedTeamEvent(team *model.Team) { func (a *App) sendUpdatedTeamEvent(team *model.Team) {
message := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_UPDATE_TEAM, "", "", "", nil) message := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_UPDATE_TEAM, "", "", "", nil)
message.Add("team", team.ToJson()) message.Add("team", team.ToJson())
go Publish(message) go a.Publish(message)
} }
func (a *App) UpdateTeamMemberRoles(teamId string, userId string, newRoles string) (*model.TeamMember, *model.AppError) { func (a *App) UpdateTeamMemberRoles(teamId string, userId string, newRoles string) (*model.TeamMember, *model.AppError) {
@@ -163,16 +163,16 @@ func (a *App) UpdateTeamMemberRoles(teamId string, userId string, newRoles strin
a.ClearSessionCacheForUser(userId) a.ClearSessionCacheForUser(userId)
sendUpdatedMemberRoleEvent(userId, member) a.sendUpdatedMemberRoleEvent(userId, member)
return member, nil return member, nil
} }
func sendUpdatedMemberRoleEvent(userId string, member *model.TeamMember) { func (a *App) sendUpdatedMemberRoleEvent(userId string, member *model.TeamMember) {
message := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_MEMBERROLE_UPDATED, "", "", userId, nil) message := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_MEMBERROLE_UPDATED, "", "", userId, nil)
message.Add("member", member.ToJson()) message.Add("member", member.ToJson())
go Publish(message) go a.Publish(message)
} }
func (a *App) AddUserToTeam(teamId string, userId string, userRequestorId string) (*model.Team, *model.AppError) { func (a *App) AddUserToTeam(teamId string, userId string, userRequestorId string) (*model.Team, *model.AppError) {
@@ -330,7 +330,7 @@ func (a *App) JoinUserToTeam(team *model.Team, user *model.User, userRequestorId
message := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_ADDED_TO_TEAM, "", "", user.Id, nil) message := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_ADDED_TO_TEAM, "", "", user.Id, nil)
message.Add("team_id", team.Id) message.Add("team_id", team.Id)
message.Add("user_id", user.Id) message.Add("user_id", user.Id)
Publish(message) a.Publish(message)
return nil return nil
} }
@@ -462,7 +462,7 @@ func (a *App) AddTeamMember(teamId, userId string) (*model.TeamMember, *model.Ap
message := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_ADDED_TO_TEAM, "", "", userId, nil) message := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_ADDED_TO_TEAM, "", "", userId, nil)
message.Add("team_id", teamId) message.Add("team_id", teamId)
message.Add("user_id", userId) message.Add("user_id", userId)
Publish(message) a.Publish(message)
return teamMember, nil return teamMember, nil
} }
@@ -484,7 +484,7 @@ func (a *App) AddTeamMembers(teamId string, userIds []string, userRequestorId st
message := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_ADDED_TO_TEAM, "", "", userId, nil) message := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_ADDED_TO_TEAM, "", "", userId, nil)
message.Add("team_id", teamId) message.Add("team_id", teamId)
message.Add("user_id", userId) message.Add("user_id", userId)
Publish(message) a.Publish(message)
} }
return members, nil return members, nil
@@ -603,7 +603,7 @@ func (a *App) LeaveTeam(team *model.Team, user *model.User) *model.AppError {
message := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_LEAVE_TEAM, team.Id, "", "", nil) message := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_LEAVE_TEAM, team.Id, "", "", nil)
message.Add("user_id", user.Id) message.Add("user_id", user.Id)
message.Add("team_id", team.Id) message.Add("team_id", team.Id)
Publish(message) a.Publish(message)
teamMember.Roles = "" teamMember.Roles = ""
teamMember.DeleteAt = model.GetMillis() teamMember.DeleteAt = model.GetMillis()

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

@@ -202,7 +202,7 @@ func (a *App) CreateUser(user *model.User) (*model.User, *model.AppError) {
// This message goes to everyone, so the teamId, channelId and userId are irrelevant // This message goes to everyone, so the teamId, channelId and userId are irrelevant
message := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_NEW_USER, "", "", "", nil) message := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_NEW_USER, "", "", "", nil)
message.Add("user_id", ruser.Id) message.Add("user_id", ruser.Id)
go Publish(message) go a.Publish(message)
return ruser, nil return ruser, nil
} }
@@ -829,7 +829,7 @@ func (a *App) SetProfileImage(userId string, imageData *multipart.FileHeader) *m
message := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_USER_UPDATED, "", "", "", omitUsers) message := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_USER_UPDATED, "", "", "", omitUsers)
message.Add("user", user) message.Add("user", user)
Publish(message) a.Publish(message)
} }
return nil return nil
@@ -950,7 +950,7 @@ func (a *App) UpdateUserAsUser(user *model.User, asAdmin bool) (*model.User, *mo
return nil, err return nil, err
} }
sendUpdatedUserEvent(*updatedUser, asAdmin) a.sendUpdatedUserEvent(*updatedUser, asAdmin)
return updatedUser, nil return updatedUser, nil
} }
@@ -968,19 +968,19 @@ func (a *App) PatchUser(userId string, patch *model.UserPatch, asAdmin bool) (*m
return nil, err return nil, err
} }
sendUpdatedUserEvent(*updatedUser, asAdmin) a.sendUpdatedUserEvent(*updatedUser, asAdmin)
return updatedUser, nil return updatedUser, nil
} }
func sendUpdatedUserEvent(user model.User, asAdmin bool) { func (a *App) sendUpdatedUserEvent(user model.User, asAdmin bool) {
SanitizeProfile(&user, asAdmin) SanitizeProfile(&user, asAdmin)
omitUsers := make(map[string]bool, 1) omitUsers := make(map[string]bool, 1)
omitUsers[user.Id] = true omitUsers[user.Id] = true
message := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_USER_UPDATED, "", "", "", omitUsers) message := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_USER_UPDATED, "", "", "", omitUsers)
message.Add("user", user) message.Add("user", user)
go Publish(message) go a.Publish(message)
} }
func (a *App) UpdateUser(user *model.User, sendNotifications bool) (*model.User, *model.AppError) { func (a *App) UpdateUser(user *model.User, sendNotifications bool) (*model.User, *model.AppError) {

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

@@ -96,7 +96,7 @@ func (c *WebConn) SetSession(v *model.Session) {
func (c *WebConn) ReadPump() { func (c *WebConn) ReadPump() {
defer func() { defer func() {
HubUnregister(c) c.App.HubUnregister(c)
c.WebSocket.Close() c.WebSocket.Close()
}() }()
c.WebSocket.SetReadLimit(model.SOCKET_MAX_MESSAGE_SIZE_KB) c.WebSocket.SetReadLimit(model.SOCKET_MAX_MESSAGE_SIZE_KB)

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

@@ -29,6 +29,7 @@ type Hub struct {
// connectionCount should be kept first. // connectionCount should be kept first.
// See https://github.com/mattermost/mattermost-server/pull/7281 // See https://github.com/mattermost/mattermost-server/pull/7281
connectionCount int64 connectionCount int64
app *App
connections []*WebConn connections []*WebConn
connectionIndex int connectionIndex int
register chan *WebConn register chan *WebConn
@@ -40,11 +41,9 @@ type Hub struct {
goroutineId int goroutineId int
} }
var hubs []*Hub = make([]*Hub, 0) func (a *App) NewWebHub() *Hub {
var stopCheckingForDeadlock chan bool
func NewWebHub() *Hub {
return &Hub{ return &Hub{
app: a,
register: make(chan *WebConn), register: make(chan *WebConn),
unregister: make(chan *WebConn), unregister: make(chan *WebConn),
connections: make([]*WebConn, 0, model.SESSION_CACHE_SIZE), connections: make([]*WebConn, 0, model.SESSION_CACHE_SIZE),
@@ -55,26 +54,27 @@ func NewWebHub() *Hub {
} }
} }
func TotalWebsocketConnections() int { func (a *App) TotalWebsocketConnections() int {
count := int64(0) count := int64(0)
for _, hub := range hubs { for _, hub := range a.Hubs {
count = count + atomic.LoadInt64(&hub.connectionCount) count = count + atomic.LoadInt64(&hub.connectionCount)
} }
return int(count) return int(count)
} }
func HubStart() { func (a *App) HubStart() {
// Total number of hubs is twice the number of CPUs. // Total number of hubs is twice the number of CPUs.
numberOfHubs := runtime.NumCPU() * 2 numberOfHubs := runtime.NumCPU() * 2
l4g.Info(utils.T("api.web_hub.start.starting.debug"), numberOfHubs) l4g.Info(utils.T("api.web_hub.start.starting.debug"), numberOfHubs)
hubs = make([]*Hub, numberOfHubs) a.Hubs = make([]*Hub, numberOfHubs)
a.HubsStopCheckingForDeadlock = make(chan bool, 1)
for i := 0; i < len(hubs); i++ { for i := 0; i < len(a.Hubs); i++ {
hubs[i] = NewWebHub() a.Hubs[i] = a.NewWebHub()
hubs[i].connectionIndex = i a.Hubs[i].connectionIndex = i
hubs[i].Start() a.Hubs[i].Start()
} }
go func() { go func() {
@@ -84,12 +84,10 @@ func HubStart() {
ticker.Stop() ticker.Stop()
}() }()
stopCheckingForDeadlock = make(chan bool, 1)
for { for {
select { select {
case <-ticker.C: case <-ticker.C:
for _, hub := range hubs { for _, hub := range a.Hubs {
if len(hub.broadcast) >= DEADLOCK_WARN { if len(hub.broadcast) >= DEADLOCK_WARN {
l4g.Error("Hub processing might be deadlock on hub %v goroutine %v with %v events in the buffer", hub.connectionIndex, hub.goroutineId, len(hub.broadcast)) l4g.Error("Hub processing might be deadlock on hub %v goroutine %v with %v events in the buffer", hub.connectionIndex, hub.goroutineId, len(hub.broadcast))
buf := make([]byte, 1<<16) buf := make([]byte, 1<<16)
@@ -105,46 +103,42 @@ func HubStart() {
} }
} }
case <-stopCheckingForDeadlock: case <-a.HubsStopCheckingForDeadlock:
return return
} }
} }
}() }()
} }
func HubStop() { func (a *App) HubStop() {
l4g.Info(utils.T("api.web_hub.start.stopping.debug")) l4g.Info(utils.T("api.web_hub.start.stopping.debug"))
select { select {
case stopCheckingForDeadlock <- true: case a.HubsStopCheckingForDeadlock <- true:
default: default:
l4g.Warn("We appear to have already sent the stop checking for deadlocks command") l4g.Warn("We appear to have already sent the stop checking for deadlocks command")
} }
for _, hub := range hubs { for _, hub := range a.Hubs {
hub.Stop() hub.Stop()
} }
hubs = make([]*Hub, 0) a.Hubs = []*Hub{}
} }
func GetHubForUserId(userId string) *Hub { func (a *App) GetHubForUserId(userId string) *Hub {
hash := fnv.New32a() hash := fnv.New32a()
hash.Write([]byte(userId)) hash.Write([]byte(userId))
index := hash.Sum32() % uint32(len(hubs)) index := hash.Sum32() % uint32(len(a.Hubs))
return hubs[index] return a.Hubs[index]
} }
func HubRegister(webConn *WebConn) { func (a *App) HubRegister(webConn *WebConn) {
GetHubForUserId(webConn.UserId).Register(webConn) a.GetHubForUserId(webConn.UserId).Register(webConn)
} }
func HubUnregister(webConn *WebConn) { func (a *App) HubUnregister(webConn *WebConn) {
GetHubForUserId(webConn.UserId).Unregister(webConn) a.GetHubForUserId(webConn.UserId).Unregister(webConn)
}
func Publish(message *model.WebSocketEvent) {
Global().Publish(message)
} }
func (a *App) Publish(message *model.WebSocketEvent) { func (a *App) Publish(message *model.WebSocketEvent) {
@@ -152,7 +146,7 @@ func (a *App) Publish(message *model.WebSocketEvent) {
metrics.IncrementWebsocketEvent(message.Event) metrics.IncrementWebsocketEvent(message.Event)
} }
PublishSkipClusterSend(message) a.PublishSkipClusterSend(message)
if a.Cluster != nil { if a.Cluster != nil {
cm := &model.ClusterMessage{ cm := &model.ClusterMessage{
@@ -173,8 +167,8 @@ func (a *App) Publish(message *model.WebSocketEvent) {
} }
} }
func PublishSkipClusterSend(message *model.WebSocketEvent) { func (a *App) PublishSkipClusterSend(message *model.WebSocketEvent) {
for _, hub := range hubs { for _, hub := range a.Hubs {
hub.Broadcast(message) hub.Broadcast(message)
} }
} }
@@ -291,8 +285,8 @@ func (a *App) InvalidateCacheForUserSkipClusterSend(userId string) {
a.Srv.Store.User().InvalidateProfilesInChannelCacheByUser(userId) a.Srv.Store.User().InvalidateProfilesInChannelCacheByUser(userId)
a.Srv.Store.User().InvalidatProfileCacheForUser(userId) a.Srv.Store.User().InvalidatProfileCacheForUser(userId)
if len(hubs) != 0 { if len(a.Hubs) != 0 {
GetHubForUserId(userId).InvalidateUser(userId) a.GetHubForUserId(userId).InvalidateUser(userId)
} }
} }
@@ -313,9 +307,9 @@ func (a *App) InvalidateCacheForWebhookSkipClusterSend(webhookId string) {
a.Srv.Store.Webhook().InvalidateWebhookCache(webhookId) a.Srv.Store.Webhook().InvalidateWebhookCache(webhookId)
} }
func InvalidateWebConnSessionCacheForUser(userId string) { func (a *App) InvalidateWebConnSessionCacheForUser(userId string) {
if len(hubs) != 0 { if len(a.Hubs) != 0 {
GetHubForUserId(userId).InvalidateUser(userId) a.GetHubForUserId(userId).InvalidateUser(userId)
} }
} }
@@ -401,7 +395,7 @@ func (h *Hub) Start() {
} }
if !found { if !found {
go Global().SetStatusOffline(userId, false) go h.app.SetStatusOffline(userId, false)
} }
case userId := <-h.invalidateUser: case userId := <-h.invalidateUser:

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

@@ -17,13 +17,15 @@ type webSocketHandler interface {
} }
type WebSocketRouter struct { type WebSocketRouter struct {
app *App
handlers map[string]webSocketHandler handlers map[string]webSocketHandler
} }
func NewWebSocketRouter() *WebSocketRouter { func (a *App) NewWebSocketRouter() *WebSocketRouter {
router := &WebSocketRouter{} return &WebSocketRouter{
router.handlers = make(map[string]webSocketHandler) app: a,
return router handlers: make(map[string]webSocketHandler),
}
} }
func (wr *WebSocketRouter) Handle(action string, handler webSocketHandler) { func (wr *WebSocketRouter) Handle(action string, handler webSocketHandler) {
@@ -54,21 +56,21 @@ func (wr *WebSocketRouter) ServeWebSocket(conn *WebConn, r *model.WebSocketReque
return return
} }
session, err := Global().GetSession(token) session, err := wr.app.GetSession(token)
if err != nil { if err != nil {
conn.WebSocket.Close() conn.WebSocket.Close()
} else { } else {
go func() { go func() {
Global().SetStatusOnline(session.UserId, session.Id, false) wr.app.SetStatusOnline(session.UserId, session.Id, false)
Global().UpdateLastActivityAtIfNeeded(*session) wr.app.UpdateLastActivityAtIfNeeded(*session)
}() }()
conn.SetSession(session) conn.SetSession(session)
conn.SetSessionToken(session.Token) conn.SetSessionToken(session.Token)
conn.UserId = session.UserId conn.UserId = session.UserId
HubRegister(conn) wr.app.HubRegister(conn)
resp := model.NewWebSocketResponse(model.STATUS_OK, r.Seq, nil) resp := model.NewWebSocketResponse(model.STATUS_OK, r.Seq, nil)
conn.Send <- resp conn.Send <- resp

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

@@ -69,6 +69,7 @@ func runServer(configFileLocation string) {
a.NewServer() a.NewServer()
a.InitStores() a.InitStores()
a.Srv.Router = api.NewRouter() a.Srv.Router = api.NewRouter()
a.Srv.WebSocketRouter = a.NewWebSocketRouter()
if model.BuildEnterpriseReady == "true" { if model.BuildEnterpriseReady == "true" {
a.LoadLicense() a.LoadLicense()
@@ -80,10 +81,9 @@ func runServer(configFileLocation string) {
l4g.Error("Unable to find webapp directory, could not initialize plugins") l4g.Error("Unable to find webapp directory, could not initialize plugins")
} }
wsapi.InitRouter()
api4.Init(a, a.Srv.Router, false) api4.Init(a, a.Srv.Router, false)
api3 := api.Init(a, a.Srv.Router) api3 := api.Init(a, a.Srv.Router)
wsapi.InitApi() wsapi.Init(a, a.Srv.WebSocketRouter)
web.Init(api3) web.Init(api3)
if !utils.IsLicensed() && len(utils.Cfg.SqlSettings.DataSourceReplicas) > 1 { if !utils.IsLicensed() && len(utils.Cfg.SqlSettings.DataSourceReplicas) > 1 {

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

@@ -52,10 +52,10 @@ func webClientTestsCmdF(cmd *cobra.Command, args []string) error {
utils.InitTranslations(utils.Cfg.LocalizationSettings) utils.InitTranslations(utils.Cfg.LocalizationSettings)
a.Srv.Router = api.NewRouter() a.Srv.Router = api.NewRouter()
wsapi.InitRouter() a.Srv.WebSocketRouter = a.NewWebSocketRouter()
api4.Init(a, a.Srv.Router, false) api4.Init(a, a.Srv.Router, false)
api.Init(a, a.Srv.Router) api.Init(a, a.Srv.Router)
wsapi.InitApi() wsapi.Init(a, a.Srv.WebSocketRouter)
setupClientTests() setupClientTests()
a.StartServer() a.StartServer()
runWebClientTests() runWebClientTests()
@@ -72,10 +72,10 @@ func serverForWebClientTestsCmdF(cmd *cobra.Command, args []string) error {
utils.InitTranslations(utils.Cfg.LocalizationSettings) utils.InitTranslations(utils.Cfg.LocalizationSettings)
a.Srv.Router = api.NewRouter() a.Srv.Router = api.NewRouter()
wsapi.InitRouter() a.Srv.WebSocketRouter = a.NewWebSocketRouter()
api4.Init(a, a.Srv.Router, false) api4.Init(a, a.Srv.Router, false)
api.Init(a, a.Srv.Router) api.Init(a, a.Srv.Router)
wsapi.InitApi() wsapi.Init(a, a.Srv.WebSocketRouter)
setupClientTests() setupClientTests()
a.StartServer() a.StartServer()

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

@@ -7,15 +7,21 @@ import (
"github.com/mattermost/mattermost-server/app" "github.com/mattermost/mattermost-server/app"
) )
func InitRouter() { type API struct {
app.Global().Srv.WebSocketRouter = app.NewWebSocketRouter() App *app.App
Router *app.WebSocketRouter
} }
func InitApi() { func Init(a *app.App, router *app.WebSocketRouter) {
InitUser() api := &API{
InitSystem() App: a,
InitStatus() Router: router,
InitWebrtc() }
app.HubStart() api.InitUser()
api.InitSystem()
api.InitStatus()
api.InitWebrtc()
a.HubStart()
} }

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

@@ -10,11 +10,11 @@ import (
"github.com/mattermost/mattermost-server/utils" "github.com/mattermost/mattermost-server/utils"
) )
func InitStatus() { func (api *API) InitStatus() {
l4g.Debug(utils.T("wsapi.status.init.debug")) l4g.Debug(utils.T("wsapi.status.init.debug"))
app.Global().Srv.WebSocketRouter.Handle("get_statuses", ApiWebSocketHandler(getStatuses)) api.Router.Handle("get_statuses", api.ApiWebSocketHandler(getStatuses))
app.Global().Srv.WebSocketRouter.Handle("get_statuses_by_ids", ApiWebSocketHandler(getStatusesByIds)) api.Router.Handle("get_statuses_by_ids", api.ApiWebSocketHandler(api.getStatusesByIds))
} }
func getStatuses(req *model.WebSocketRequest) (map[string]interface{}, *model.AppError) { func getStatuses(req *model.WebSocketRequest) (map[string]interface{}, *model.AppError) {
@@ -22,14 +22,14 @@ func getStatuses(req *model.WebSocketRequest) (map[string]interface{}, *model.Ap
return model.StatusMapToInterfaceMap(statusMap), nil return model.StatusMapToInterfaceMap(statusMap), nil
} }
func getStatusesByIds(req *model.WebSocketRequest) (map[string]interface{}, *model.AppError) { func (api *API) getStatusesByIds(req *model.WebSocketRequest) (map[string]interface{}, *model.AppError) {
var userIds []string var userIds []string
if userIds = model.ArrayFromInterface(req.Data["user_ids"]); len(userIds) == 0 { if userIds = model.ArrayFromInterface(req.Data["user_ids"]); len(userIds) == 0 {
l4g.Error(model.StringInterfaceToJson(req.Data)) l4g.Error(model.StringInterfaceToJson(req.Data))
return nil, NewInvalidWebSocketParamError(req.Action, "user_ids") return nil, NewInvalidWebSocketParamError(req.Action, "user_ids")
} }
statusMap, err := app.Global().GetStatusesByIds(userIds) statusMap, err := api.App.GetStatusesByIds(userIds)
if err != nil { if err != nil {
return nil, err return nil, err
} }

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

@@ -5,15 +5,14 @@ package wsapi
import ( import (
l4g "github.com/alecthomas/log4go" l4g "github.com/alecthomas/log4go"
"github.com/mattermost/mattermost-server/app"
"github.com/mattermost/mattermost-server/model" "github.com/mattermost/mattermost-server/model"
"github.com/mattermost/mattermost-server/utils" "github.com/mattermost/mattermost-server/utils"
) )
func InitSystem() { func (api *API) InitSystem() {
l4g.Debug(utils.T("wsapi.system.init.debug")) l4g.Debug(utils.T("wsapi.system.init.debug"))
app.Global().Srv.WebSocketRouter.Handle("ping", ApiWebSocketHandler(ping)) api.Router.Handle("ping", api.ApiWebSocketHandler(ping))
} }
func ping(req *model.WebSocketRequest) (map[string]interface{}, *model.AppError) { func ping(req *model.WebSocketRequest) (map[string]interface{}, *model.AppError) {

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

@@ -5,18 +5,17 @@ package wsapi
import ( import (
l4g "github.com/alecthomas/log4go" l4g "github.com/alecthomas/log4go"
"github.com/mattermost/mattermost-server/app"
"github.com/mattermost/mattermost-server/model" "github.com/mattermost/mattermost-server/model"
"github.com/mattermost/mattermost-server/utils" "github.com/mattermost/mattermost-server/utils"
) )
func InitUser() { func (api *API) InitUser() {
l4g.Debug(utils.T("wsapi.user.init.debug")) l4g.Debug(utils.T("wsapi.user.init.debug"))
app.Global().Srv.WebSocketRouter.Handle("user_typing", ApiWebSocketHandler(userTyping)) api.Router.Handle("user_typing", api.ApiWebSocketHandler(api.userTyping))
} }
func userTyping(req *model.WebSocketRequest) (map[string]interface{}, *model.AppError) { func (api *API) userTyping(req *model.WebSocketRequest) (map[string]interface{}, *model.AppError) {
var ok bool var ok bool
var channelId string var channelId string
if channelId, ok = req.Data["channel_id"].(string); !ok || len(channelId) != 26 { if channelId, ok = req.Data["channel_id"].(string); !ok || len(channelId) != 26 {
@@ -34,7 +33,7 @@ func userTyping(req *model.WebSocketRequest) (map[string]interface{}, *model.App
event := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_TYPING, "", channelId, "", omitUsers) event := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_TYPING, "", channelId, "", omitUsers)
event.Add("parent_id", parentId) event.Add("parent_id", parentId)
event.Add("user_id", req.Session.UserId) event.Add("user_id", req.Session.UserId)
go app.Publish(event) go api.App.Publish(event)
return nil, nil return nil, nil
} }

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

@@ -5,18 +5,17 @@ package wsapi
import ( import (
l4g "github.com/alecthomas/log4go" l4g "github.com/alecthomas/log4go"
"github.com/mattermost/mattermost-server/app"
"github.com/mattermost/mattermost-server/model" "github.com/mattermost/mattermost-server/model"
"github.com/mattermost/mattermost-server/utils" "github.com/mattermost/mattermost-server/utils"
) )
func InitWebrtc() { func (api *API) InitWebrtc() {
l4g.Debug(utils.T("wsapi.webtrc.init.debug")) l4g.Debug(utils.T("wsapi.webtrc.init.debug"))
app.Global().Srv.WebSocketRouter.Handle("webrtc", ApiWebSocketHandler(webrtcMessage)) api.Router.Handle("webrtc", api.ApiWebSocketHandler(api.webrtcMessage))
} }
func webrtcMessage(req *model.WebSocketRequest) (map[string]interface{}, *model.AppError) { func (api *API) webrtcMessage(req *model.WebSocketRequest) (map[string]interface{}, *model.AppError) {
var ok bool var ok bool
var toUserId string var toUserId string
if toUserId, ok = req.Data["to_user_id"].(string); !ok || len(toUserId) != 26 { if toUserId, ok = req.Data["to_user_id"].(string); !ok || len(toUserId) != 26 {
@@ -25,7 +24,7 @@ func webrtcMessage(req *model.WebSocketRequest) (map[string]interface{}, *model.
event := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_WEBRTC, "", "", toUserId, nil) event := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_WEBRTC, "", "", toUserId, nil)
event.Data = req.Data event.Data = req.Data
go app.Publish(event) go api.App.Publish(event)
return nil, nil return nil, nil
} }

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

@@ -13,18 +13,19 @@ import (
"github.com/mattermost/mattermost-server/utils" "github.com/mattermost/mattermost-server/utils"
) )
func ApiWebSocketHandler(wh func(*model.WebSocketRequest) (map[string]interface{}, *model.AppError)) webSocketHandler { func (api *API) ApiWebSocketHandler(wh func(*model.WebSocketRequest) (map[string]interface{}, *model.AppError)) webSocketHandler {
return webSocketHandler{wh} return webSocketHandler{api.App, wh}
} }
type webSocketHandler struct { type webSocketHandler struct {
app *app.App
handlerFunc func(*model.WebSocketRequest) (map[string]interface{}, *model.AppError) handlerFunc func(*model.WebSocketRequest) (map[string]interface{}, *model.AppError)
} }
func (wh webSocketHandler) ServeWebSocket(conn *app.WebConn, r *model.WebSocketRequest) { func (wh webSocketHandler) ServeWebSocket(conn *app.WebConn, r *model.WebSocketRequest) {
l4g.Debug("/api/v3/users/websocket:%s", r.Action) l4g.Debug("/api/v3/users/websocket:%s", r.Action)
session, sessionErr := app.Global().GetSession(conn.GetSessionToken()) session, sessionErr := wh.app.GetSession(conn.GetSessionToken())
if sessionErr != nil { if sessionErr != nil {
l4g.Error(utils.T("api.web_socket_handler.log.error"), "/api/v3/users/websocket", r.Action, r.Seq, conn.UserId, sessionErr.SystemMessage(utils.T), sessionErr.Error()) l4g.Error(utils.T("api.web_socket_handler.log.error"), "/api/v3/users/websocket", r.Action, r.Seq, conn.UserId, sessionErr.SystemMessage(utils.T), sessionErr.Error())
sessionErr.DetailedError = "" sessionErr.DetailedError = ""