87 строки
2.2 KiB
Go
87 строки
2.2 KiB
Go
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
|
// See LICENSE.txt for license information.
|
|
|
|
package jobs
|
|
|
|
import (
|
|
"math/rand"
|
|
"time"
|
|
|
|
"github.com/mattermost/mattermost/server/public/model"
|
|
"github.com/mattermost/mattermost/server/public/shared/mlog"
|
|
"github.com/mattermost/mattermost/server/public/shared/request"
|
|
)
|
|
|
|
// Default polling interval for jobs termination.
|
|
// (Defining as `var` rather than `const` allows tests to lower the interval.)
|
|
var DefaultWatcherPollingInterval = 15000
|
|
|
|
type Watcher struct {
|
|
srv *JobServer
|
|
workers *Workers
|
|
|
|
stop chan struct{}
|
|
stopped chan struct{}
|
|
pollingInterval int
|
|
}
|
|
|
|
func (srv *JobServer) MakeWatcher(workers *Workers, pollingInterval int) *Watcher {
|
|
return &Watcher{
|
|
stop: make(chan struct{}),
|
|
stopped: make(chan struct{}),
|
|
pollingInterval: pollingInterval,
|
|
workers: workers,
|
|
srv: srv,
|
|
}
|
|
}
|
|
|
|
func (watcher *Watcher) Start() {
|
|
mlog.Debug("Watcher Started")
|
|
// Delay for some random number of milliseconds before starting to ensure that multiple
|
|
// instances of the jobserver don't poll at a time too close to each other.
|
|
rand.Seed(time.Now().UTC().UnixNano())
|
|
<-time.After(time.Duration(rand.Intn(watcher.pollingInterval)) * time.Millisecond)
|
|
|
|
defer func() {
|
|
mlog.Debug("Watcher Finished")
|
|
close(watcher.stopped)
|
|
}()
|
|
|
|
for {
|
|
select {
|
|
case <-watcher.stop:
|
|
mlog.Debug("Watcher: Received stop signal")
|
|
return
|
|
case <-time.After(time.Duration(watcher.pollingInterval) * time.Millisecond):
|
|
watcher.PollAndNotify()
|
|
}
|
|
}
|
|
}
|
|
|
|
func (watcher *Watcher) Stop() {
|
|
mlog.Debug("Watcher Stopping")
|
|
close(watcher.stop)
|
|
<-watcher.stopped
|
|
|
|
watcher.stop = make(chan struct{})
|
|
watcher.stopped = make(chan struct{})
|
|
}
|
|
|
|
func (watcher *Watcher) PollAndNotify() {
|
|
jobs, err := watcher.srv.Store.Job().GetAllByStatus(request.EmptyContext(watcher.srv.logger), model.JobStatusPending)
|
|
if err != nil {
|
|
mlog.Error("Error occurred getting all pending statuses.", mlog.Err(err))
|
|
return
|
|
}
|
|
|
|
for _, job := range jobs {
|
|
worker := watcher.workers.Get(job.Type)
|
|
if worker != nil {
|
|
select {
|
|
case worker.JobChannel() <- *job:
|
|
default:
|
|
}
|
|
}
|
|
}
|
|
}
|