ec1f89217a
Full Mattermost server source with integrated Community Enterprise features. Includes vendor directory for offline/air-gapped builds. Structure: - enterprise-impl/: Enterprise feature implementations - enterprise-community/: Init files that register implementations - enterprise/: Bridge imports (community_imports.go) - vendor/: All dependencies for offline builds Build (online): go build ./cmd/mattermost Build (offline/air-gapped): go build -mod=vendor ./cmd/mattermost 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
92 lines
2.2 KiB
Go
92 lines
2.2 KiB
Go
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
|
// See LICENSE.txt for license information.
|
|
|
|
package jobs
|
|
|
|
import (
|
|
"errors"
|
|
|
|
"github.com/mattermost/mattermost/server/public/model"
|
|
"github.com/mattermost/mattermost/server/public/shared/configservice"
|
|
"github.com/mattermost/mattermost/server/public/shared/mlog"
|
|
)
|
|
|
|
type Workers struct {
|
|
ConfigService configservice.ConfigService
|
|
Watcher *Watcher
|
|
|
|
workers map[string]model.Worker
|
|
|
|
listenerId string
|
|
running bool
|
|
}
|
|
|
|
var (
|
|
ErrWorkersNotRunning = errors.New("job workers are not running")
|
|
ErrWorkersRunning = errors.New("job workers are running")
|
|
ErrWorkersUninitialized = errors.New("job workers are not initialized")
|
|
)
|
|
|
|
func NewWorkers(configService configservice.ConfigService) *Workers {
|
|
return &Workers{
|
|
ConfigService: configService,
|
|
workers: make(map[string]model.Worker),
|
|
}
|
|
}
|
|
|
|
func (workers *Workers) AddWorker(name string, worker model.Worker) {
|
|
workers.workers[name] = worker
|
|
}
|
|
|
|
func (workers *Workers) Get(name string) model.Worker {
|
|
return workers.workers[name]
|
|
}
|
|
|
|
// Start starts the workers. This call is not safe for concurrent use.
|
|
// Synchronization should be implemented by the caller.
|
|
func (workers *Workers) Start() {
|
|
mlog.Info("Starting workers")
|
|
|
|
for _, w := range workers.workers {
|
|
if w.IsEnabled(workers.ConfigService.Config()) {
|
|
go w.Run()
|
|
}
|
|
}
|
|
|
|
go workers.Watcher.Start()
|
|
|
|
workers.listenerId = workers.ConfigService.AddConfigListener(workers.handleConfigChange)
|
|
workers.running = true
|
|
}
|
|
|
|
func (workers *Workers) handleConfigChange(oldConfig *model.Config, newConfig *model.Config) {
|
|
mlog.Debug("Workers received config change.")
|
|
|
|
for _, w := range workers.workers {
|
|
if w.IsEnabled(oldConfig) && !w.IsEnabled(newConfig) {
|
|
w.Stop()
|
|
}
|
|
if !w.IsEnabled(oldConfig) && w.IsEnabled(newConfig) {
|
|
go w.Run()
|
|
}
|
|
}
|
|
}
|
|
|
|
// Stop stops the workers. This call is not safe for concurrent use.
|
|
// Synchronization should be implemented by the caller.
|
|
func (workers *Workers) Stop() {
|
|
workers.ConfigService.RemoveConfigListener(workers.listenerId)
|
|
|
|
workers.Watcher.Stop()
|
|
|
|
for _, w := range workers.workers {
|
|
if w.IsEnabled(workers.ConfigService.Config()) {
|
|
w.Stop()
|
|
}
|
|
}
|
|
|
|
workers.running = false
|
|
|
|
mlog.Info("Stopped workers")
|
|
}
|