Remove more global refs / state (#7723)

* remove more global refs / state

* fix job enterprise initialization

* fix api4 test compilation

* saml api endpoints fix
Этот коммит содержится в:
Chris
2017-10-26 14:21:22 -05:00
коммит произвёл Joram Wilander
родитель a0bfd2885d
Коммит 7ed011745a
23 изменённых файлов: 129 добавлений и 150 удалений

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

@@ -400,7 +400,7 @@ func removeCertificate(c *Context, w http.ResponseWriter, r *http.Request) {
} }
func samlCertificateStatus(c *Context, w http.ResponseWriter, r *http.Request) { func samlCertificateStatus(c *Context, w http.ResponseWriter, r *http.Request) {
status := app.GetSamlCertificateStatus() status := c.App.GetSamlCertificateStatus()
statusMap := map[string]interface{}{} statusMap := map[string]interface{}{}
statusMap["IdpCertificateFile"] = status.IdpCertificateFile statusMap["IdpCertificateFile"] = status.IdpCertificateFile

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

@@ -114,15 +114,6 @@ func Setup() *TestHelper {
return setupTestHelper(false) return setupTestHelper(false)
} }
func ReloadConfigForSetup() {
utils.LoadConfig("config.json")
utils.InitTranslations(utils.Cfg.LocalizationSettings)
*utils.Cfg.TeamSettings.MaxUsersPerTeam = 50
*utils.Cfg.RateLimitSettings.Enable = false
utils.Cfg.EmailSettings.SendEmailNotifications = true
*utils.Cfg.TeamSettings.EnableOpenServer = true
}
func (me *TestHelper) InitBasic() *TestHelper { func (me *TestHelper) InitBasic() *TestHelper {
me.waitForConnectivity() me.waitForConnectivity()

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

@@ -205,7 +205,6 @@ func TestStatuses(t *testing.T) {
} }
func TestGetStatusesByIds(t *testing.T) { func TestGetStatusesByIds(t *testing.T) {
ReloadConfigForSetup()
th := Setup().InitBasic() th := Setup().InitBasic()
defer th.TearDown() defer th.TearDown()

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

@@ -292,7 +292,7 @@ func getInitialLoad(c *Context, w http.ResponseWriter, r *http.Request) {
il.TeamMembers = c.Session.TeamMembers il.TeamMembers = c.Session.TeamMembers
} }
if app.SessionCacheLength() == 0 { if c.App.SessionCacheLength() == 0 {
// Below is a special case when intializating a new server // Below is a special case when intializating a new server
// Lets check to make sure the server is really empty // Lets check to make sure the server is really empty

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

@@ -8,7 +8,6 @@ import (
"net/http" "net/http"
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"
) )
@@ -73,7 +72,7 @@ func addSamlPublicCertificate(c *Context, w http.ResponseWriter, r *http.Request
return return
} }
if err := app.AddSamlPublicCertificate(fileData); err != nil { if err := c.App.AddSamlPublicCertificate(fileData); err != nil {
c.Err = err c.Err = err
return return
} }
@@ -92,7 +91,7 @@ func addSamlPrivateCertificate(c *Context, w http.ResponseWriter, r *http.Reques
return return
} }
if err := app.AddSamlPrivateCertificate(fileData); err != nil { if err := c.App.AddSamlPrivateCertificate(fileData); err != nil {
c.Err = err c.Err = err
return return
} }
@@ -111,7 +110,7 @@ func addSamlIdpCertificate(c *Context, w http.ResponseWriter, r *http.Request) {
return return
} }
if err := app.AddSamlIdpCertificate(fileData); err != nil { if err := c.App.AddSamlIdpCertificate(fileData); err != nil {
c.Err = err c.Err = err
return return
} }
@@ -124,7 +123,7 @@ func removeSamlPublicCertificate(c *Context, w http.ResponseWriter, r *http.Requ
return return
} }
if err := app.RemoveSamlPublicCertificate(); err != nil { if err := c.App.RemoveSamlPublicCertificate(); err != nil {
c.Err = err c.Err = err
return return
} }
@@ -138,7 +137,7 @@ func removeSamlPrivateCertificate(c *Context, w http.ResponseWriter, r *http.Req
return return
} }
if err := app.RemoveSamlPrivateCertificate(); err != nil { if err := c.App.RemoveSamlPrivateCertificate(); err != nil {
c.Err = err c.Err = err
return return
} }
@@ -152,7 +151,7 @@ func removeSamlIdpCertificate(c *Context, w http.ResponseWriter, r *http.Request
return return
} }
if err := app.RemoveSamlIdpCertificate(); err != nil { if err := c.App.RemoveSamlIdpCertificate(); err != nil {
c.Err = err c.Err = err
return return
} }
@@ -166,6 +165,6 @@ func getSamlCertificateStatus(c *Context, w http.ResponseWriter, r *http.Request
return return
} }
status := app.GetSamlCertificateStatus() status := c.App.GetSamlCertificateStatus()
w.Write([]byte(status.ToJson())) w.Write([]byte(status.ToJson()))
} }

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

@@ -10,7 +10,6 @@ import (
"testing" "testing"
"time" "time"
"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"
"github.com/stretchr/testify/assert" "github.com/stretchr/testify/assert"
@@ -961,7 +960,7 @@ func TestUpdateUser(t *testing.T) {
session, _ := th.App.GetSession(Client.AuthToken) session, _ := th.App.GetSession(Client.AuthToken)
session.IsOAuth = true session.IsOAuth = true
app.AddSessionToCache(session) th.App.AddSessionToCache(session)
ruser.Id = user.Id ruser.Id = user.Id
ruser.Email = GenerateTestEmail() ruser.Email = GenerateTestEmail()
@@ -1045,7 +1044,7 @@ func TestPatchUser(t *testing.T) {
session, _ := th.App.GetSession(Client.AuthToken) session, _ := th.App.GetSession(Client.AuthToken)
session.IsOAuth = true session.IsOAuth = true
app.AddSessionToCache(session) th.App.AddSessionToCache(session)
patch.Email = model.NewString(GenerateTestEmail()) patch.Email = model.NewString(GenerateTestEmail())
_, resp = Client.PatchUser(user.Id, patch) _, resp = Client.PatchUser(user.Id, patch)
@@ -1513,7 +1512,7 @@ func TestUpdateUserMfa(t *testing.T) {
session, _ := th.App.GetSession(Client.AuthToken) session, _ := th.App.GetSession(Client.AuthToken)
session.IsOAuth = true session.IsOAuth = true
app.AddSessionToCache(session) th.App.AddSessionToCache(session)
_, resp := Client.UpdateUserMfa(th.BasicUser.Id, "12345", false) _, resp := Client.UpdateUserMfa(th.BasicUser.Id, "12345", false)
CheckForbiddenStatus(t, resp) CheckForbiddenStatus(t, resp)
@@ -1609,7 +1608,7 @@ func TestGenerateMfaSecret(t *testing.T) {
session, _ := th.App.GetSession(Client.AuthToken) session, _ := th.App.GetSession(Client.AuthToken)
session.IsOAuth = true session.IsOAuth = true
app.AddSessionToCache(session) th.App.AddSessionToCache(session)
_, resp = Client.GenerateMfaSecret(th.BasicUser.Id) _, resp = Client.GenerateMfaSecret(th.BasicUser.Id)
CheckForbiddenStatus(t, resp) CheckForbiddenStatus(t, resp)
@@ -2257,7 +2256,7 @@ func TestCreateUserAccessToken(t *testing.T) {
session, _ := th.App.GetSession(Client.AuthToken) session, _ := th.App.GetSession(Client.AuthToken)
session.IsOAuth = true session.IsOAuth = true
app.AddSessionToCache(session) th.App.AddSessionToCache(session)
_, resp = Client.CreateUserAccessToken(th.BasicUser.Id, testDescription) _, resp = Client.CreateUserAccessToken(th.BasicUser.Id, testDescription)
CheckForbiddenStatus(t, resp) CheckForbiddenStatus(t, resp)

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

@@ -114,7 +114,7 @@ func (a *App) InvalidateAllCaches() *model.AppError {
func (a *App) InvalidateAllCachesSkipSend() { func (a *App) InvalidateAllCachesSkipSend() {
l4g.Info(utils.T("api.context.invalidate_all_caches")) l4g.Info(utils.T("api.context.invalidate_all_caches"))
sessionCache.Purge() a.sessionCache.Purge()
ClearStatusCache() ClearStatusCache()
sqlstore.ClearChannelCaches() sqlstore.ClearChannelCaches()
sqlstore.ClearUserCaches() sqlstore.ClearUserCaches()
@@ -133,7 +133,7 @@ func (a *App) GetConfig() *model.Config {
func (a *App) ReloadConfig() { func (a *App) ReloadConfig() {
debug.FreeOSMemory() debug.FreeOSMemory()
utils.LoadConfig(utils.CfgFileName) utils.LoadConfig(a.ConfigFileName())
// start/restart email batching job if necessary // start/restart email batching job if necessary
a.InitEmailBatching() a.InitEmailBatching()
@@ -157,8 +157,8 @@ func (a *App) SaveConfig(cfg *model.Config, sendConfigChangeClusterMessage bool)
} }
utils.DisableConfigWatch() utils.DisableConfigWatch()
utils.SaveConfig(utils.CfgFileName, cfg) utils.SaveConfig(a.ConfigFileName(), cfg)
utils.LoadConfig(utils.CfgFileName) utils.LoadConfig(a.ConfigFileName())
utils.EnableConfigWatch() utils.EnableConfigWatch()
if a.Metrics != nil { if a.Metrics != nil {

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

@@ -48,6 +48,8 @@ type App struct {
Saml einterfaces.SamlInterface Saml einterfaces.SamlInterface
newStore func() store.Store newStore func() store.Store
sessionCache *utils.Cache
} }
var appCount = 0 var appCount = 0
@@ -64,10 +66,10 @@ func New(options ...Option) *App {
app := &App{ app := &App{
goroutineExitSignal: make(chan struct{}, 1), goroutineExitSignal: make(chan struct{}, 1),
Jobs: &jobs.JobServer{},
Srv: &Server{ Srv: &Server{
Router: mux.NewRouter(), Router: mux.NewRouter(),
}, },
sessionCache: utils.NewLru(model.SESSION_CACHE_SIZE),
} }
app.initEnterprise() app.initEnterprise()
@@ -82,7 +84,7 @@ func New(options ...Option) *App {
} }
app.Srv.Store = app.newStore() app.Srv.Store = app.newStore()
app.Jobs.Store = app.Srv.Store app.initJobs()
app.Srv.Router.NotFoundHandler = http.HandlerFunc(app.Handle404) app.Srv.Router.NotFoundHandler = http.HandlerFunc(app.Handle404)
@@ -218,6 +220,10 @@ func (a *App) initEnterprise() {
if dataRetentionInterface != nil { if dataRetentionInterface != nil {
a.DataRetention = dataRetentionInterface(a) a.DataRetention = dataRetentionInterface(a)
} }
}
func (a *App) initJobs() {
a.Jobs = jobs.NewJobServer(a.Config, a.Srv.Store)
if jobsDataRetentionJobInterface != nil { if jobsDataRetentionJobInterface != nil {
a.Jobs.DataRetentionJob = jobsDataRetentionJobInterface(a) a.Jobs.DataRetentionJob = jobsDataRetentionJobInterface(a)
} }
@@ -240,6 +246,10 @@ func (a *App) UpdateConfig(f func(*model.Config)) {
f(utils.Cfg) f(utils.Cfg)
} }
func (a *App) ConfigFileName() string {
return utils.CfgFileName
}
// Go creates a goroutine, but maintains a record of it to ensure that execution completes before // Go creates a goroutine, but maintains a record of it to ensure that execution completes before
// the app is destroyed. // the app is destroyed.
func (a *App) Go(f func()) { func (a *App) Go(f func()) {

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

@@ -25,7 +25,7 @@ func (a *App) CheckPasswordAndAllCriteria(user *model.User, password string, mfa
// This to be used for places we check the users password when they are already logged in // This to be used for places we check the users password when they are already logged in
func (a *App) doubleCheckPassword(user *model.User, password string) *model.AppError { func (a *App) doubleCheckPassword(user *model.User, password string) *model.AppError {
if err := checkUserLoginAttempts(user); err != nil { if err := checkUserLoginAttempts(user, *a.Config().ServiceSettings.MaximumLoginAttempts); err != nil {
return err return err
} }
@@ -83,15 +83,15 @@ func (a *App) CheckUserAdditionalAuthenticationCriteria(user *model.User, mfaTok
return err return err
} }
if err := checkEmailVerified(user); err != nil { if !user.EmailVerified && a.Config().EmailSettings.RequireEmailVerification {
return err return model.NewAppError("Login", "api.user.login.not_verified.app_error", nil, "user_id="+user.Id, http.StatusUnauthorized)
} }
if err := checkUserNotDisabled(user); err != nil { if err := checkUserNotDisabled(user); err != nil {
return err return err
} }
if err := checkUserLoginAttempts(user); err != nil { if err := checkUserLoginAttempts(user, *a.Config().ServiceSettings.MaximumLoginAttempts); err != nil {
return err return err
} }
@@ -116,21 +116,14 @@ func (a *App) CheckUserMfa(user *model.User, token string) *model.AppError {
return nil return nil
} }
func checkUserLoginAttempts(user *model.User) *model.AppError { func checkUserLoginAttempts(user *model.User, max int) *model.AppError {
if user.FailedAttempts >= *utils.Cfg.ServiceSettings.MaximumLoginAttempts { if user.FailedAttempts >= max {
return model.NewAppError("checkUserLoginAttempts", "api.user.check_user_login_attempts.too_many.app_error", nil, "user_id="+user.Id, http.StatusUnauthorized) return model.NewAppError("checkUserLoginAttempts", "api.user.check_user_login_attempts.too_many.app_error", nil, "user_id="+user.Id, http.StatusUnauthorized)
} }
return nil return nil
} }
func checkEmailVerified(user *model.User) *model.AppError {
if !user.EmailVerified && utils.Cfg.EmailSettings.RequireEmailVerification {
return model.NewAppError("Login", "api.user.login.not_verified.app_error", nil, "user_id="+user.Id, http.StatusUnauthorized)
}
return nil
}
func checkUserNotDisabled(user *model.User) *model.AppError { func checkUserNotDisabled(user *model.User) *model.AppError {
if user.DeleteAt > 0 { if user.DeleteAt > 0 {
return model.NewAppError("Login", "api.user.login.inactive.app_error", nil, "user_id="+user.Id, http.StatusUnauthorized) return model.NewAppError("Login", "api.user.login.inactive.app_error", nil, "user_id="+user.Id, http.StatusUnauthorized)

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

@@ -275,7 +275,7 @@ func (a *App) newSession(appName string, user *model.User) (*model.Session, *mod
return nil, model.NewAppError("newSession", "api.oauth.get_access_token.internal_session.app_error", nil, "", http.StatusInternalServerError) return nil, model.NewAppError("newSession", "api.oauth.get_access_token.internal_session.app_error", nil, "", http.StatusInternalServerError)
} else { } else {
session = result.Data.(*model.Session) session = result.Data.(*model.Session)
AddSessionToCache(session) a.AddSessionToCache(session)
} }
return session, nil return session, nil

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

@@ -7,7 +7,6 @@ import (
"testing" "testing"
"github.com/mattermost/mattermost-server/model" "github.com/mattermost/mattermost-server/model"
"github.com/mattermost/mattermost-server/utils"
) )
func TestOAuthRevokeAccessToken(t *testing.T) { func TestOAuthRevokeAccessToken(t *testing.T) {
@@ -50,11 +49,11 @@ func TestOAuthDeleteApp(t *testing.T) {
th := Setup() th := Setup()
defer th.TearDown() defer th.TearDown()
oldSetting := utils.Cfg.ServiceSettings.EnableOAuthServiceProvider oldSetting := th.App.Config().ServiceSettings.EnableOAuthServiceProvider
defer func() { defer th.App.UpdateConfig(func(cfg *model.Config) {
utils.Cfg.ServiceSettings.EnableOAuthServiceProvider = oldSetting cfg.ServiceSettings.EnableOAuthServiceProvider = oldSetting
}() })
utils.Cfg.ServiceSettings.EnableOAuthServiceProvider = true th.App.Config().ServiceSettings.EnableOAuthServiceProvider = true
a1 := &model.OAuthApp{} a1 := &model.OAuthApp{}
a1.CreatorId = model.NewId() a1.CreatorId = model.NewId()

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

@@ -33,7 +33,7 @@ type PluginAPI struct {
} }
func (api *PluginAPI) LoadPluginConfiguration(dest interface{}) error { func (api *PluginAPI) LoadPluginConfiguration(dest interface{}) error {
if b, err := json.Marshal(utils.Cfg.PluginSettings.Plugins[api.id]); err != nil { if b, err := json.Marshal(api.app.Config().PluginSettings.Plugins[api.id]); err != nil {
return err return err
} else { } else {
return json.Unmarshal(b, dest) return json.Unmarshal(b, dest)
@@ -145,7 +145,7 @@ type BuiltInPluginAPI struct {
} }
func (api *BuiltInPluginAPI) LoadPluginConfiguration(dest interface{}) error { func (api *BuiltInPluginAPI) LoadPluginConfiguration(dest interface{}) error {
if b, err := json.Marshal(utils.Cfg.PluginSettings.Plugins[api.id]); err != nil { if b, err := json.Marshal(api.app.Config().PluginSettings.Plugins[api.id]); err != nil {
return err return err
} else { } else {
return json.Unmarshal(b, dest) return json.Unmarshal(b, dest)

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

@@ -52,59 +52,56 @@ func WriteSamlFile(fileData *multipart.FileHeader) *model.AppError {
return nil return nil
} }
func AddSamlPublicCertificate(fileData *multipart.FileHeader) *model.AppError { func (a *App) AddSamlPublicCertificate(fileData *multipart.FileHeader) *model.AppError {
if err := WriteSamlFile(fileData); err != nil { if err := WriteSamlFile(fileData); err != nil {
return err return err
} }
cfg := &model.Config{} cfg := a.Config().Clone()
*cfg = *utils.Cfg
*cfg.SamlSettings.PublicCertificateFile = fileData.Filename *cfg.SamlSettings.PublicCertificateFile = fileData.Filename
if err := cfg.IsValid(); err != nil { if err := cfg.IsValid(); err != nil {
return err return err
} }
utils.SaveConfig(utils.CfgFileName, cfg) a.UpdateConfig(func(dest *model.Config) { *dest = *cfg })
utils.SaveConfig(a.ConfigFileName(), cfg)
return nil return nil
} }
func AddSamlPrivateCertificate(fileData *multipart.FileHeader) *model.AppError { func (a *App) AddSamlPrivateCertificate(fileData *multipart.FileHeader) *model.AppError {
if err := WriteSamlFile(fileData); err != nil { if err := WriteSamlFile(fileData); err != nil {
return err return err
} }
cfg := &model.Config{} cfg := a.Config().Clone()
*cfg = *utils.Cfg
*cfg.SamlSettings.PrivateKeyFile = fileData.Filename *cfg.SamlSettings.PrivateKeyFile = fileData.Filename
if err := cfg.IsValid(); err != nil { if err := cfg.IsValid(); err != nil {
return err return err
} }
utils.SaveConfig(utils.CfgFileName, cfg) a.UpdateConfig(func(dest *model.Config) { *dest = *cfg })
utils.SaveConfig(a.ConfigFileName(), cfg)
return nil return nil
} }
func AddSamlIdpCertificate(fileData *multipart.FileHeader) *model.AppError { func (a *App) AddSamlIdpCertificate(fileData *multipart.FileHeader) *model.AppError {
if err := WriteSamlFile(fileData); err != nil { if err := WriteSamlFile(fileData); err != nil {
return err return err
} }
cfg := &model.Config{} cfg := a.Config().Clone()
*cfg = *utils.Cfg
*cfg.SamlSettings.IdpCertificateFile = fileData.Filename *cfg.SamlSettings.IdpCertificateFile = fileData.Filename
if err := cfg.IsValid(); err != nil { if err := cfg.IsValid(); err != nil {
return err return err
} }
utils.SaveConfig(utils.CfgFileName, cfg) a.UpdateConfig(func(dest *model.Config) { *dest = *cfg })
utils.SaveConfig(a.ConfigFileName(), cfg)
return nil return nil
} }
@@ -123,14 +120,12 @@ func RemoveSamlFile(filename string) *model.AppError {
return nil return nil
} }
func RemoveSamlPublicCertificate() *model.AppError { func (a *App) RemoveSamlPublicCertificate() *model.AppError {
if err := RemoveSamlFile(*utils.Cfg.SamlSettings.PublicCertificateFile); err != nil { if err := RemoveSamlFile(*a.Config().SamlSettings.PublicCertificateFile); err != nil {
return err return err
} }
cfg := &model.Config{} cfg := a.Config().Clone()
*cfg = *utils.Cfg
*cfg.SamlSettings.PublicCertificateFile = "" *cfg.SamlSettings.PublicCertificateFile = ""
*cfg.SamlSettings.Encrypt = false *cfg.SamlSettings.Encrypt = false
@@ -138,19 +133,18 @@ func RemoveSamlPublicCertificate() *model.AppError {
return err return err
} }
utils.SaveConfig(utils.CfgFileName, cfg) a.UpdateConfig(func(dest *model.Config) { *dest = *cfg })
utils.SaveConfig(a.ConfigFileName(), cfg)
return nil return nil
} }
func RemoveSamlPrivateCertificate() *model.AppError { func (a *App) RemoveSamlPrivateCertificate() *model.AppError {
if err := RemoveSamlFile(*utils.Cfg.SamlSettings.PrivateKeyFile); err != nil { if err := RemoveSamlFile(*a.Config().SamlSettings.PrivateKeyFile); err != nil {
return err return err
} }
cfg := &model.Config{} cfg := a.Config().Clone()
*cfg = *utils.Cfg
*cfg.SamlSettings.PrivateKeyFile = "" *cfg.SamlSettings.PrivateKeyFile = ""
*cfg.SamlSettings.Encrypt = false *cfg.SamlSettings.Encrypt = false
@@ -158,19 +152,18 @@ func RemoveSamlPrivateCertificate() *model.AppError {
return err return err
} }
utils.SaveConfig(utils.CfgFileName, cfg) a.UpdateConfig(func(dest *model.Config) { *dest = *cfg })
utils.SaveConfig(a.ConfigFileName(), cfg)
return nil return nil
} }
func RemoveSamlIdpCertificate() *model.AppError { func (a *App) RemoveSamlIdpCertificate() *model.AppError {
if err := RemoveSamlFile(*utils.Cfg.SamlSettings.IdpCertificateFile); err != nil { if err := RemoveSamlFile(*a.Config().SamlSettings.IdpCertificateFile); err != nil {
return err return err
} }
cfg := &model.Config{} cfg := a.Config().Clone()
*cfg = *utils.Cfg
*cfg.SamlSettings.IdpCertificateFile = "" *cfg.SamlSettings.IdpCertificateFile = ""
*cfg.SamlSettings.Enable = false *cfg.SamlSettings.Enable = false
@@ -178,17 +171,18 @@ func RemoveSamlIdpCertificate() *model.AppError {
return err return err
} }
utils.SaveConfig(utils.CfgFileName, cfg) a.UpdateConfig(func(dest *model.Config) { *dest = *cfg })
utils.SaveConfig(a.ConfigFileName(), cfg)
return nil return nil
} }
func GetSamlCertificateStatus() *model.SamlCertificateStatus { func (a *App) GetSamlCertificateStatus() *model.SamlCertificateStatus {
status := &model.SamlCertificateStatus{} status := &model.SamlCertificateStatus{}
status.IdpCertificateFile = utils.FileExistsInConfigFolder(*utils.Cfg.SamlSettings.IdpCertificateFile) status.IdpCertificateFile = utils.FileExistsInConfigFolder(*a.Config().SamlSettings.IdpCertificateFile)
status.PrivateKeyFile = utils.FileExistsInConfigFolder(*utils.Cfg.SamlSettings.PrivateKeyFile) status.PrivateKeyFile = utils.FileExistsInConfigFolder(*a.Config().SamlSettings.PrivateKeyFile)
status.PublicCertificateFile = utils.FileExistsInConfigFolder(*utils.Cfg.SamlSettings.PublicCertificateFile) status.PublicCertificateFile = utils.FileExistsInConfigFolder(*a.Config().SamlSettings.PublicCertificateFile)
return status return status
} }

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

@@ -53,11 +53,12 @@ func (rl *RecoveryLogger) Println(i ...interface{}) {
} }
type CorsWrapper struct { type CorsWrapper struct {
config model.ConfigFunc
router *mux.Router router *mux.Router
} }
func (cw *CorsWrapper) ServeHTTP(w http.ResponseWriter, r *http.Request) { func (cw *CorsWrapper) ServeHTTP(w http.ResponseWriter, r *http.Request) {
if len(*utils.Cfg.ServiceSettings.AllowCorsFrom) > 0 { if len(*cw.config().ServiceSettings.AllowCorsFrom) > 0 {
if utils.OriginChecker(r) { if utils.OriginChecker(r) {
w.Header().Set("Access-Control-Allow-Origin", r.Header.Get("Origin")) w.Header().Set("Access-Control-Allow-Origin", r.Header.Get("Origin"))
@@ -88,25 +89,6 @@ func (m *VaryBy) Key(r *http.Request) string {
return utils.GetIpAddress(r) return utils.GetIpAddress(r)
} }
func initalizeThrottledVaryBy() *throttled.VaryBy {
vary := throttled.VaryBy{}
if utils.Cfg.RateLimitSettings.VaryByRemoteAddr {
vary.RemoteAddr = true
}
if len(utils.Cfg.RateLimitSettings.VaryByHeader) > 0 {
vary.Headers = strings.Fields(utils.Cfg.RateLimitSettings.VaryByHeader)
if utils.Cfg.RateLimitSettings.VaryByRemoteAddr {
l4g.Warn(utils.T("api.server.start_server.rate.warn"))
vary.RemoteAddr = false
}
}
return &vary
}
func redirectHTTPToHTTPS(w http.ResponseWriter, r *http.Request) { func redirectHTTPToHTTPS(w http.ResponseWriter, r *http.Request) {
if r.Host == "" { if r.Host == "" {
http.Error(w, "Not Found", http.StatusNotFound) http.Error(w, "Not Found", http.StatusNotFound)
@@ -121,7 +103,7 @@ func redirectHTTPToHTTPS(w http.ResponseWriter, r *http.Request) {
func (a *App) StartServer() { func (a *App) StartServer() {
l4g.Info(utils.T("api.server.start_server.starting.info")) l4g.Info(utils.T("api.server.start_server.starting.info"))
var handler http.Handler = &CorsWrapper{a.Srv.Router} var handler http.Handler = &CorsWrapper{a.Config, a.Srv.Router}
if *a.Config().RateLimitSettings.Enable { if *a.Config().RateLimitSettings.Enable {
l4g.Info(utils.T("api.server.start_server.rate.info")) l4g.Info(utils.T("api.server.start_server.rate.info"))

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

@@ -12,8 +12,6 @@ import (
l4g "github.com/alecthomas/log4go" l4g "github.com/alecthomas/log4go"
) )
var sessionCache *utils.Cache = utils.NewLru(model.SESSION_CACHE_SIZE)
func (a *App) CreateSession(session *model.Session) (*model.Session, *model.AppError) { func (a *App) CreateSession(session *model.Session) (*model.Session, *model.AppError) {
session.Token = "" session.Token = ""
@@ -22,7 +20,7 @@ func (a *App) CreateSession(session *model.Session) (*model.Session, *model.AppE
} else { } else {
session := result.Data.(*model.Session) session := result.Data.(*model.Session)
AddSessionToCache(session) a.AddSessionToCache(session)
return session, nil return session, nil
} }
@@ -32,7 +30,7 @@ func (a *App) GetSession(token string) (*model.Session, *model.AppError) {
metrics := a.Metrics metrics := a.Metrics
var session *model.Session var session *model.Session
if ts, ok := sessionCache.Get(token); ok { if ts, ok := a.sessionCache.Get(token); ok {
session = ts.(*model.Session) session = ts.(*model.Session)
if metrics != nil { if metrics != nil {
metrics.IncrementMemCacheHitCounterSession() metrics.IncrementMemCacheHitCounterSession()
@@ -53,7 +51,7 @@ func (a *App) GetSession(token string) (*model.Session, *model.AppError) {
} }
if !session.IsExpired() { if !session.IsExpired() {
AddSessionToCache(session) a.AddSessionToCache(session)
} }
} }
} }
@@ -132,13 +130,13 @@ func (a *App) ClearSessionCacheForUser(userId string) {
} }
func (a *App) ClearSessionCacheForUserSkipClusterSend(userId string) { func (a *App) ClearSessionCacheForUserSkipClusterSend(userId string) {
keys := sessionCache.Keys() keys := a.sessionCache.Keys()
for _, key := range keys { for _, key := range keys {
if ts, ok := sessionCache.Get(key); ok { if ts, ok := a.sessionCache.Get(key); ok {
session := ts.(*model.Session) session := ts.(*model.Session)
if session.UserId == userId { if session.UserId == userId {
sessionCache.Remove(key) a.sessionCache.Remove(key)
} }
} }
} }
@@ -146,12 +144,12 @@ func (a *App) ClearSessionCacheForUserSkipClusterSend(userId string) {
a.InvalidateWebConnSessionCacheForUser(userId) a.InvalidateWebConnSessionCacheForUser(userId)
} }
func AddSessionToCache(session *model.Session) { func (a *App) AddSessionToCache(session *model.Session) {
sessionCache.AddWithExpiresInSecs(session.Token, session, int64(*utils.Cfg.ServiceSettings.SessionCacheInMinutes*60)) a.sessionCache.AddWithExpiresInSecs(session.Token, session, int64(*a.Config().ServiceSettings.SessionCacheInMinutes*60))
} }
func SessionCacheLength() int { func (a *App) SessionCacheLength() int {
return sessionCache.Len() return a.sessionCache.Len()
} }
func (a *App) RevokeSessionsForDeviceId(userId string, deviceId string, currentSessionId string) *model.AppError { func (a *App) RevokeSessionsForDeviceId(userId string, deviceId string, currentSessionId string) *model.AppError {
@@ -227,7 +225,7 @@ func (a *App) UpdateLastActivityAtIfNeeded(session model.Session) {
} }
session.LastActivityAt = now session.LastActivityAt = now
AddSessionToCache(&session) a.AddSessionToCache(&session)
} }
func (a *App) CreateUserAccessToken(token *model.UserAccessToken) (*model.UserAccessToken, *model.AppError) { func (a *App) CreateUserAccessToken(token *model.UserAccessToken) (*model.UserAccessToken, *model.AppError) {
@@ -301,7 +299,7 @@ func (a *App) createSessionForUserAccessToken(tokenString string) (*model.Sessio
} else { } else {
session := result.Data.(*model.Session) session := result.Data.(*model.Session)
AddSessionToCache(session) a.AddSessionToCache(session)
return session, nil return session, nil
} }

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

@@ -23,16 +23,16 @@ func TestCache(t *testing.T) {
UserId: model.NewId(), UserId: model.NewId(),
} }
sessionCache.AddWithExpiresInSecs(session.Token, session, 5*60) th.App.sessionCache.AddWithExpiresInSecs(session.Token, session, 5*60)
keys := sessionCache.Keys() keys := th.App.sessionCache.Keys()
if len(keys) <= 0 { if len(keys) <= 0 {
t.Fatal("should have items") t.Fatal("should have items")
} }
th.App.ClearSessionCacheForUser(session.UserId) th.App.ClearSessionCacheForUser(session.UserId)
rkeys := sessionCache.Keys() rkeys := th.App.sessionCache.Keys()
if len(rkeys) != len(keys)-1 { if len(rkeys) != len(keys)-1 {
t.Fatal("should have one less") t.Fatal("should have one less")
} }

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

@@ -159,7 +159,7 @@ func IsUserSignUpAllowed() *model.AppError {
} }
func (a *App) IsFirstUserAccount() bool { func (a *App) IsFirstUserAccount() bool {
if SessionCacheLength() == 0 { if a.SessionCacheLength() == 0 {
if cr := <-a.Srv.Store.User().GetTotalUsersCount(); cr.Err != nil { if cr := <-a.Srv.Store.User().GetTotalUsersCount(); cr.Err != nil {
l4g.Error(cr.Err) l4g.Error(cr.Err)
return false return false

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

@@ -87,11 +87,11 @@ func TestCreateOAuthUser(t *testing.T) {
th.App.PermanentDeleteUser(user) th.App.PermanentDeleteUser(user)
userCreation := utils.Cfg.TeamSettings.EnableUserCreation userCreation := th.App.Config().TeamSettings.EnableUserCreation
defer func() { defer th.App.UpdateConfig(func(cfg *model.Config) {
utils.Cfg.TeamSettings.EnableUserCreation = userCreation cfg.TeamSettings.EnableUserCreation = userCreation
}() })
utils.Cfg.TeamSettings.EnableUserCreation = false th.App.Config().TeamSettings.EnableUserCreation = false
_, err = th.App.CreateOAuthUser(model.USER_AUTH_SERVICE_GITLAB, strings.NewReader(json), th.BasicTeam.Id) _, err = th.App.CreateOAuthUser(model.USER_AUTH_SERVICE_GITLAB, strings.NewReader(json), th.BasicTeam.Id)
if err == nil { if err == nil {

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

@@ -13,6 +13,7 @@ import (
) )
type JobServer struct { type JobServer struct {
Config model.ConfigFunc
Store store.Store Store store.Store
Workers *Workers Workers *Workers
Schedulers *Schedulers Schedulers *Schedulers
@@ -23,6 +24,13 @@ type JobServer struct {
LdapSync ejobs.LdapSyncInterface LdapSync ejobs.LdapSyncInterface
} }
func NewJobServer(config model.ConfigFunc, store store.Store) *JobServer {
return &JobServer{
Config: config,
Store: store,
}
}
func (srv *JobServer) LoadLicense() { func (srv *JobServer) LoadLicense() {
licenseId := "" licenseId := ""
if result := <-srv.Store.System().Get(); result.Err == nil { if result := <-srv.Store.System().Get(); result.Err == nil {

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

@@ -13,6 +13,7 @@ import (
type Workers struct { type Workers struct {
startOnce sync.Once startOnce sync.Once
Config model.ConfigFunc
Watcher *Watcher Watcher *Watcher
DataRetention model.Worker DataRetention model.Worker
@@ -24,7 +25,9 @@ type Workers struct {
} }
func (srv *JobServer) InitWorkers() *Workers { func (srv *JobServer) InitWorkers() *Workers {
workers := &Workers{} workers := &Workers{
Config: srv.Config,
}
workers.Watcher = srv.MakeWatcher(workers, DEFAULT_WATCHER_POLLING_INTERVAL) workers.Watcher = srv.MakeWatcher(workers, DEFAULT_WATCHER_POLLING_INTERVAL)
if srv.DataRetentionJob != nil { if srv.DataRetentionJob != nil {
@@ -50,19 +53,19 @@ func (workers *Workers) Start() *Workers {
l4g.Info("Starting workers") l4g.Info("Starting workers")
workers.startOnce.Do(func() { workers.startOnce.Do(func() {
if workers.DataRetention != nil && (*utils.Cfg.DataRetentionSettings.EnableMessageDeletion || *utils.Cfg.DataRetentionSettings.EnableFileDeletion) { if workers.DataRetention != nil && (*workers.Config().DataRetentionSettings.EnableMessageDeletion || *workers.Config().DataRetentionSettings.EnableFileDeletion) {
go workers.DataRetention.Run() go workers.DataRetention.Run()
} }
if workers.ElasticsearchIndexing != nil && *utils.Cfg.ElasticsearchSettings.EnableIndexing { if workers.ElasticsearchIndexing != nil && *workers.Config().ElasticsearchSettings.EnableIndexing {
go workers.ElasticsearchIndexing.Run() go workers.ElasticsearchIndexing.Run()
} }
if workers.ElasticsearchAggregation != nil && *utils.Cfg.ElasticsearchSettings.EnableIndexing { if workers.ElasticsearchAggregation != nil && *workers.Config().ElasticsearchSettings.EnableIndexing {
go workers.ElasticsearchAggregation.Run() go workers.ElasticsearchAggregation.Run()
} }
if workers.LdapSync != nil && *utils.Cfg.LdapSettings.Enable { if workers.LdapSync != nil && *workers.Config().LdapSettings.Enable {
go workers.LdapSync.Run() go workers.LdapSync.Run()
} }
@@ -113,19 +116,19 @@ func (workers *Workers) Stop() *Workers {
workers.Watcher.Stop() workers.Watcher.Stop()
if workers.DataRetention != nil && (*utils.Cfg.DataRetentionSettings.EnableMessageDeletion || *utils.Cfg.DataRetentionSettings.EnableFileDeletion) { if workers.DataRetention != nil && (*workers.Config().DataRetentionSettings.EnableMessageDeletion || *workers.Config().DataRetentionSettings.EnableFileDeletion) {
workers.DataRetention.Stop() workers.DataRetention.Stop()
} }
if workers.ElasticsearchIndexing != nil && *utils.Cfg.ElasticsearchSettings.EnableIndexing { if workers.ElasticsearchIndexing != nil && *workers.Config().ElasticsearchSettings.EnableIndexing {
workers.ElasticsearchIndexing.Stop() workers.ElasticsearchIndexing.Stop()
} }
if workers.ElasticsearchAggregation != nil && *utils.Cfg.ElasticsearchSettings.EnableIndexing { if workers.ElasticsearchAggregation != nil && *workers.Config().ElasticsearchSettings.EnableIndexing {
workers.ElasticsearchAggregation.Stop() workers.ElasticsearchAggregation.Stop()
} }
if workers.LdapSync != nil && *utils.Cfg.LdapSettings.Enable { if workers.LdapSync != nil && *workers.Config().LdapSettings.Enable {
workers.LdapSync.Stop() workers.LdapSync.Stop()
} }

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

@@ -522,6 +522,8 @@ type PluginSettings struct {
PluginStates map[string]*PluginState PluginStates map[string]*PluginState
} }
type ConfigFunc func() *Config
type Config struct { type Config struct {
ServiceSettings ServiceSettings ServiceSettings ServiceSettings
TeamSettings TeamSettings TeamSettings TeamSettings

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

@@ -285,7 +285,7 @@ func InitAndLoadConfig(filename string) error {
// LoadConfig will try to search around for the corresponding config file. // LoadConfig will try to search around for the corresponding config file.
// It will search /tmp/fileName then attempt ./config/fileName, // It will search /tmp/fileName then attempt ./config/fileName,
// then ../config/fileName and last it will look at fileName // then ../config/fileName and last it will look at fileName
func LoadConfig(fileName string) { func LoadConfig(fileName string) *model.Config {
cfgMutex.Lock() cfgMutex.Lock()
defer cfgMutex.Unlock() defer cfgMutex.Unlock()
@@ -394,6 +394,8 @@ func LoadConfig(fileName string) {
for _, listener := range cfgListeners { for _, listener := range cfgListeners {
listener(&oldConfig, &config) listener(&oldConfig, &config)
} }
return &config
} }
func RegenerateClientConfig() { func RegenerateClientConfig() {

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

@@ -21,14 +21,14 @@ func Init(api3 *api.API) {
mainrouter := api3.BaseRoutes.Root mainrouter := api3.BaseRoutes.Root
if *utils.Cfg.ServiceSettings.WebserverMode != "disabled" { if *api3.App.Config().ServiceSettings.WebserverMode != "disabled" {
staticDir, _ := utils.FindDir(model.CLIENT_DIR) staticDir, _ := utils.FindDir(model.CLIENT_DIR)
l4g.Debug("Using client directory at %v", staticDir) l4g.Debug("Using client directory at %v", staticDir)
staticHandler := staticHandler(http.StripPrefix("/static/", http.FileServer(http.Dir(staticDir)))) staticHandler := staticHandler(http.StripPrefix("/static/", http.FileServer(http.Dir(staticDir))))
pluginHandler := pluginHandler(http.StripPrefix("/static/plugins/", http.FileServer(http.Dir(staticDir+"plugins/")))) pluginHandler := pluginHandler(api3.App.Config, http.StripPrefix("/static/plugins/", http.FileServer(http.Dir(staticDir+"plugins/"))))
if *utils.Cfg.ServiceSettings.WebserverMode == "gzip" { if *api3.App.Config().ServiceSettings.WebserverMode == "gzip" {
staticHandler = gziphandler.GzipHandler(staticHandler) staticHandler = gziphandler.GzipHandler(staticHandler)
pluginHandler = gziphandler.GzipHandler(pluginHandler) pluginHandler = gziphandler.GzipHandler(pluginHandler)
} }
@@ -50,9 +50,9 @@ func staticHandler(handler http.Handler) http.Handler {
}) })
} }
func pluginHandler(handler http.Handler) http.Handler { func pluginHandler(config model.ConfigFunc, handler http.Handler) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
if *utils.Cfg.ServiceSettings.EnableDeveloper { if *config().ServiceSettings.EnableDeveloper {
w.Header().Set("Cache-Control", "no-cache, no-store, must-revalidate") w.Header().Set("Cache-Control", "no-cache, no-store, must-revalidate")
} else { } else {
w.Header().Set("Cache-Control", "max-age=31556926, public") w.Header().Set("Cache-Control", "max-age=31556926, public")