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>
105 строки
2.9 KiB
Go
105 строки
2.9 KiB
Go
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
|
// See LICENSE.txt for license information.
|
|
|
|
package scheduler
|
|
|
|
import (
|
|
"github.com/mattermost/mattermost-server/v6/jobs"
|
|
"github.com/mattermost/mattermost-server/v6/model"
|
|
"github.com/mattermost/mattermost-server/v6/shared/mlog"
|
|
)
|
|
|
|
type AppIface interface {
|
|
DeleteAllExpiredPluginKeys() *model.AppError
|
|
}
|
|
|
|
type Worker struct {
|
|
name string
|
|
stop chan bool
|
|
stopped chan bool
|
|
jobs chan model.Job
|
|
jobServer *jobs.JobServer
|
|
app AppIface
|
|
}
|
|
|
|
func MakeWorker(jobServer *jobs.JobServer, app AppIface) model.Worker {
|
|
worker := Worker{
|
|
name: "Plugins",
|
|
stop: make(chan bool, 1),
|
|
stopped: make(chan bool, 1),
|
|
jobs: make(chan model.Job),
|
|
jobServer: jobServer,
|
|
app: app,
|
|
}
|
|
|
|
return &worker
|
|
}
|
|
|
|
func (worker *Worker) Run() {
|
|
mlog.Debug("Worker started", mlog.String("worker", worker.name))
|
|
|
|
defer func() {
|
|
mlog.Debug("Worker finished", mlog.String("worker", worker.name))
|
|
worker.stopped <- true
|
|
}()
|
|
|
|
for {
|
|
select {
|
|
case <-worker.stop:
|
|
mlog.Debug("Worker received stop signal", mlog.String("worker", worker.name))
|
|
return
|
|
case job := <-worker.jobs:
|
|
mlog.Debug("Worker received a new candidate job.", mlog.String("worker", worker.name))
|
|
worker.DoJob(&job)
|
|
}
|
|
}
|
|
}
|
|
|
|
func (worker *Worker) Stop() {
|
|
mlog.Debug("Worker stopping", mlog.String("worker", worker.name))
|
|
worker.stop <- true
|
|
<-worker.stopped
|
|
}
|
|
|
|
func (worker *Worker) JobChannel() chan<- model.Job {
|
|
return worker.jobs
|
|
}
|
|
|
|
func (worker *Worker) IsEnabled(cfg *model.Config) bool {
|
|
return true
|
|
}
|
|
|
|
func (worker *Worker) DoJob(job *model.Job) {
|
|
if claimed, err := worker.jobServer.ClaimJob(job); err != nil {
|
|
mlog.Info("Worker experienced an error while trying to claim job",
|
|
mlog.String("worker", worker.name),
|
|
mlog.String("job_id", job.Id),
|
|
mlog.String("error", err.Error()))
|
|
return
|
|
} else if !claimed {
|
|
return
|
|
}
|
|
|
|
if err := worker.app.DeleteAllExpiredPluginKeys(); err != nil {
|
|
mlog.Error("Worker: Failed to delete expired keys", mlog.String("worker", worker.name), mlog.String("job_id", job.Id), mlog.String("error", err.Error()))
|
|
worker.setJobError(job, err)
|
|
return
|
|
}
|
|
|
|
mlog.Info("Worker: Job is complete", mlog.String("worker", worker.name), mlog.String("job_id", job.Id))
|
|
worker.setJobSuccess(job)
|
|
}
|
|
|
|
func (worker *Worker) setJobSuccess(job *model.Job) {
|
|
if err := worker.jobServer.SetJobSuccess(job); err != nil {
|
|
mlog.Error("Worker: Failed to set success for job", mlog.String("worker", worker.name), mlog.String("job_id", job.Id), mlog.String("error", err.Error()))
|
|
worker.setJobError(job, err)
|
|
}
|
|
}
|
|
|
|
func (worker *Worker) setJobError(job *model.Job, appError *model.AppError) {
|
|
if err := worker.jobServer.SetJobError(job, appError); err != nil {
|
|
mlog.Error("Worker: Failed to set job error", mlog.String("worker", worker.name), mlog.String("job_id", job.Id), mlog.String("error", err.Error()))
|
|
}
|
|
}
|