mirror of
https://github.com/hanzoai/team.git
synced 2026-09-30 01:57:35 +00:00
zeekay
8feb24ea1f
The server module named github.com/hanzoai/team/server/v8 — a path that resolves to the retired workspace's repository, not this one — while server/public still published github.com/mattermost/mattermost/server/public. Both are github.com/hanzoteam/server/... now, which is where this code actually lives. That second half is what made the binary unbootable. Upstream's public path stayed claimed by upstream, mattermost-plugin-ai imported it, and the build linked public/model twice; public/plugin's gob.Register then ran twice under one name and panicked at init, before main. Reverting the rename traded the brand for a working binary. Rebranding the plugin alongside it needs no such trade: the agents fork is github.com/hanzoteam/agents/v2 against our public module, so no module named mattermost/mattermost is in the graph at all. Measured: `go list -deps ./cmd/mattermost` names exactly one public/model, and it is ours.
136 lines
3.3 KiB
Go
136 lines
3.3 KiB
Go
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
|
// See LICENSE.txt for license information.
|
|
|
|
package jobs
|
|
|
|
import (
|
|
"sync"
|
|
"time"
|
|
|
|
"github.com/hanzoteam/server/server/public/model"
|
|
"github.com/hanzoteam/server/server/public/shared/configservice"
|
|
"github.com/hanzoteam/server/server/public/shared/mlog"
|
|
"github.com/hanzoteam/server/server/v8/channels/store"
|
|
"github.com/hanzoteam/server/server/v8/einterfaces"
|
|
)
|
|
|
|
type JobServer struct {
|
|
ConfigService configservice.ConfigService
|
|
Store store.Store
|
|
metrics einterfaces.MetricsInterface
|
|
logger mlog.LoggerIFace
|
|
publish func(*model.WebSocketEvent)
|
|
|
|
// 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, logger mlog.LoggerIFace, publish func(*model.WebSocketEvent)) *JobServer {
|
|
srv := &JobServer{
|
|
ConfigService: configService,
|
|
Store: store,
|
|
metrics: metrics,
|
|
logger: logger,
|
|
publish: publish,
|
|
}
|
|
srv.initWorkers()
|
|
srv.initSchedulers()
|
|
return srv
|
|
}
|
|
|
|
func (srv *JobServer) initWorkers() {
|
|
workers := NewWorkers(srv.ConfigService)
|
|
workers.Watcher = srv.MakeWatcher(workers, DefaultWatcherPollingInterval)
|
|
srv.workers = workers
|
|
}
|
|
|
|
func (srv *JobServer) initSchedulers() {
|
|
schedulers := &Schedulers{
|
|
configChanged: make(chan *model.Config),
|
|
clusterLeaderChanged: make(chan bool, 1),
|
|
jobs: srv,
|
|
isLeader: true,
|
|
schedulers: make(map[string]Scheduler),
|
|
nextRunTimes: make(map[string]*time.Time),
|
|
}
|
|
|
|
srv.schedulers = schedulers
|
|
}
|
|
|
|
func (srv *JobServer) Config() *model.Config {
|
|
return srv.ConfigService.Config()
|
|
}
|
|
|
|
func (srv *JobServer) Logger() mlog.LoggerIFace {
|
|
return srv.logger
|
|
}
|
|
|
|
func (srv *JobServer) RegisterJobType(name string, worker model.Worker, scheduler 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)
|
|
}
|
|
}
|