2c3e289509
* Working on refactoring jobs service * Making more consistent with the previous existing code * Remove no longer needed functions * Making a base PeridicScheduler to use it in most of the schedulers implementations * Removing accidental complexity from on of the jobs * Removing accidental complexity from expirynotify * Fixing compilation from previous commit * Remove accidental complexity from the export_delete job * Simplifying the workers by making a reusable worker * Using simple worker for export_delete job * Simpliying export process job * Simpliying extract content job * Simpliying import delete job * Simpliying import process job * Simpliying product noticies job * Simpliying fix crt channel unreads job (only removing the uneeded register function) * Simpliying migrations job (only removing the uneeded register function) * fixup * Simpliying plugins job (only removing the uneeded register function) * Simpliying bleve indexing job (only removing the uneeded register function) * Simpliying resend invitation email job (only removing the uneeded register function) * Fixing tests * Simplifying migration tests infrastructure * Adding missed license to files * Adding an empty file to imports package to ensure this package exist even without enterprise repo * Regenerating einterfaces mocks * Adding missed license to files * Updating i18n/en.json file * help fixing enterprise tests compilation * Adding new DailyScheduler * Fixing typo and changing the waitTime type for periodic sechduler * Making the daily scheduler more generic * Adding comments to clarify not used parameters in interface scheduler interface implementations * Using merror to handle multiple errors in jobs workers * Fixing linter errors * Addressing PR review comments * Reverting go.tools.mod changes * Removing the static check for worker type in the model (moving it to the insertion of new jobs * Moving migrations job to the jobs directory * Fixing (and improving a bit) tests * Apply suggestions from code review Co-authored-by: Doug Lauder <wiggin77@warpmail.net> * Fixing enterprise tests * Removing unneeded InitWorkers/InitSchedulers calls * Fix expirenotify job when error happens * Fixing govet errors Co-authored-by: Mattermod <mattermod@users.noreply.github.com> Co-authored-by: Doug Lauder <wiggin77@warpmail.net>
104 строки
2.4 KiB
Go
104 строки
2.4 KiB
Go
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
|
// See LICENSE.txt for license information.
|
|
|
|
package jobs
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"github.com/mattermost/mattermost-server/v6/einterfaces"
|
|
"github.com/mattermost/mattermost-server/v6/model"
|
|
"github.com/mattermost/mattermost-server/v6/services/configservice"
|
|
"github.com/mattermost/mattermost-server/v6/store"
|
|
)
|
|
|
|
type JobServer struct {
|
|
ConfigService configservice.ConfigService
|
|
Store store.Store
|
|
metrics einterfaces.MetricsInterface
|
|
|
|
// mut is used to protect the following fields from concurrent access.
|
|
mut sync.Mutex
|
|
workers *Workers
|
|
schedulers *Schedulers
|
|
}
|
|
|
|
func NewJobServer(configService configservice.ConfigService, store store.Store, metrics einterfaces.MetricsInterface) *JobServer {
|
|
return &JobServer{
|
|
ConfigService: configService,
|
|
Store: store,
|
|
metrics: metrics,
|
|
}
|
|
}
|
|
|
|
func (srv *JobServer) Config() *model.Config {
|
|
return srv.ConfigService.Config()
|
|
}
|
|
|
|
func (srv *JobServer) RegisterJobType(name string, worker model.Worker, scheduler model.Scheduler) {
|
|
srv.mut.Lock()
|
|
defer srv.mut.Unlock()
|
|
if worker != nil {
|
|
srv.workers.AddWorker(name, worker)
|
|
}
|
|
if scheduler != nil {
|
|
srv.schedulers.AddScheduler(name, scheduler)
|
|
}
|
|
}
|
|
|
|
func (srv *JobServer) StartWorkers() error {
|
|
srv.mut.Lock()
|
|
defer srv.mut.Unlock()
|
|
if srv.workers == nil {
|
|
return ErrWorkersUninitialized
|
|
} else if srv.workers.running {
|
|
return ErrWorkersRunning
|
|
}
|
|
srv.workers.Start()
|
|
return nil
|
|
}
|
|
|
|
func (srv *JobServer) StartSchedulers() error {
|
|
srv.mut.Lock()
|
|
defer srv.mut.Unlock()
|
|
if srv.schedulers == nil {
|
|
return ErrSchedulersUninitialized
|
|
} else if srv.schedulers.running {
|
|
return ErrSchedulersRunning
|
|
}
|
|
srv.schedulers.Start()
|
|
return nil
|
|
}
|
|
|
|
func (srv *JobServer) StopWorkers() error {
|
|
srv.mut.Lock()
|
|
defer srv.mut.Unlock()
|
|
if srv.workers == nil {
|
|
return ErrWorkersUninitialized
|
|
} else if !srv.workers.running {
|
|
return ErrWorkersNotRunning
|
|
}
|
|
srv.workers.Stop()
|
|
return nil
|
|
}
|
|
|
|
func (srv *JobServer) StopSchedulers() error {
|
|
srv.mut.Lock()
|
|
defer srv.mut.Unlock()
|
|
if srv.schedulers == nil {
|
|
return ErrSchedulersUninitialized
|
|
} else if !srv.schedulers.running {
|
|
return ErrSchedulersNotRunning
|
|
}
|
|
srv.schedulers.Stop()
|
|
return nil
|
|
}
|
|
|
|
func (srv *JobServer) HandleClusterLeaderChange(isLeader bool) {
|
|
srv.mut.Lock()
|
|
defer srv.mut.Unlock()
|
|
if srv.schedulers != nil {
|
|
srv.schedulers.handleClusterLeaderChange(isLeader)
|
|
}
|
|
}
|