47bfa2b66b
* pass a logger instead of embedding on job * leverage mlog.Millis * use worker logger with HandleJobPanic * rely on existing LogClone instead * guard Job.LogClone against nil Job * s/workername/worker_name * Revert "rely on existing LogClone instead" This reverts commit 17303cbac90d4b01815abca1309b78b97de368fb. * Revert "guard Job.LogClone against nil Job" This reverts commit f1ae22dee58d76f084582857830ffe8d4c546d7e.
75 строки
2.1 KiB
Go
75 строки
2.1 KiB
Go
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
|
// See LICENSE.txt for license information.
|
|
|
|
package notify_admin
|
|
|
|
import (
|
|
"github.com/mattermost/mattermost/server/public/model"
|
|
"github.com/mattermost/mattermost/server/public/shared/mlog"
|
|
"github.com/mattermost/mattermost/server/v8/channels/jobs"
|
|
)
|
|
|
|
const (
|
|
UpgradeNotifyJobName = "UpgradeNotifyAdmin"
|
|
TrialNotifyJobName = "TrialNotifyAdmin"
|
|
InstallNotifyJobName = "InstallNotifyAdmin"
|
|
)
|
|
|
|
type AppIface interface {
|
|
DoCheckForAdminNotifications(trial bool) *model.AppError
|
|
}
|
|
|
|
func MakeUpgradeNotifyWorker(jobServer *jobs.JobServer, license *model.License, app AppIface) *jobs.SimpleWorker {
|
|
isEnabled := func(_ *model.Config) bool {
|
|
return license != nil && license.Features != nil && *license.Features.Cloud
|
|
}
|
|
execute := func(logger mlog.LoggerIFace, job *model.Job) error {
|
|
defer jobServer.HandleJobPanic(logger, job)
|
|
|
|
appErr := app.DoCheckForAdminNotifications(false)
|
|
if appErr != nil {
|
|
return appErr
|
|
}
|
|
|
|
return nil
|
|
}
|
|
worker := jobs.NewSimpleWorker(UpgradeNotifyJobName, jobServer, execute, isEnabled)
|
|
return worker
|
|
}
|
|
|
|
func MakeTrialNotifyWorker(jobServer *jobs.JobServer, license *model.License, app AppIface) *jobs.SimpleWorker {
|
|
isEnabled := func(_ *model.Config) bool {
|
|
return license != nil && license.Features != nil && *license.Features.Cloud
|
|
}
|
|
execute := func(logger mlog.LoggerIFace, job *model.Job) error {
|
|
defer jobServer.HandleJobPanic(logger, job)
|
|
|
|
appErr := app.DoCheckForAdminNotifications(true)
|
|
if appErr != nil {
|
|
return appErr
|
|
}
|
|
|
|
return nil
|
|
}
|
|
worker := jobs.NewSimpleWorker(TrialNotifyJobName, jobServer, execute, isEnabled)
|
|
return worker
|
|
}
|
|
|
|
func MakeInstallPluginNotifyWorker(jobServer *jobs.JobServer, app AppIface) *jobs.SimpleWorker {
|
|
isEnabled := func(_ *model.Config) bool {
|
|
return true
|
|
}
|
|
execute := func(logger mlog.LoggerIFace, job *model.Job) error {
|
|
defer jobServer.HandleJobPanic(logger, job)
|
|
|
|
appErr := app.DoCheckForAdminNotifications(false)
|
|
if appErr != nil {
|
|
return appErr
|
|
}
|
|
|
|
return nil
|
|
}
|
|
worker := jobs.NewSimpleWorker(InstallNotifyJobName, jobServer, execute, isEnabled)
|
|
return worker
|
|
}
|