mirror of
https://github.com/hanzoai/team.git
synced 2026-11-01 07:55:31 +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.
90 lines
2.7 KiB
Go
90 lines
2.7 KiB
Go
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
|
// See LICENSE.txt for license information.
|
|
|
|
package jobs
|
|
|
|
import (
|
|
"crypto/rand"
|
|
"math/big"
|
|
"time"
|
|
|
|
"github.com/hanzoteam/server/server/public/model"
|
|
"github.com/hanzoteam/server/server/public/shared/request"
|
|
)
|
|
|
|
type PeriodicScheduler struct {
|
|
jobs *JobServer
|
|
period time.Duration
|
|
jobType string
|
|
enabledFunc func(cfg *model.Config) bool
|
|
}
|
|
|
|
var _ Scheduler = (*DailyScheduler)(nil)
|
|
|
|
func NewPeriodicScheduler(jobs *JobServer, jobType string, period time.Duration, enabledFunc func(cfg *model.Config) bool) *PeriodicScheduler {
|
|
return &PeriodicScheduler{
|
|
period: period,
|
|
jobType: jobType,
|
|
enabledFunc: enabledFunc,
|
|
jobs: jobs,
|
|
}
|
|
}
|
|
|
|
func (scheduler *PeriodicScheduler) Enabled(cfg *model.Config) bool {
|
|
return scheduler.enabledFunc(cfg)
|
|
}
|
|
|
|
func (scheduler *PeriodicScheduler) NextScheduleTime(_ *model.Config, _ time.Time /* pendingJobs */, _ bool /* lastSuccessfulJob */, _ *model.Job) *time.Time {
|
|
nextTime := time.Now().Add(getRandomDelay(jitterRange)).Add(scheduler.period)
|
|
return &nextTime
|
|
}
|
|
|
|
func (scheduler *PeriodicScheduler) ScheduleJob(rctx request.CTX, _ *model.Config /* pendingJobs */, _ bool /* lastSuccessfulJob */, _ *model.Job) (*model.Job, *model.AppError) {
|
|
return scheduler.jobs.CreateJob(rctx, scheduler.jobType, nil)
|
|
}
|
|
|
|
type DailyScheduler struct {
|
|
jobs *JobServer
|
|
startTimeFunc func(cfg *model.Config) *time.Time
|
|
jobType string
|
|
enabledFunc func(cfg *model.Config) bool
|
|
}
|
|
|
|
var _ Scheduler = (*DailyScheduler)(nil)
|
|
|
|
func NewDailyScheduler(jobs *JobServer, jobType string, startTimeFunc func(cfg *model.Config) *time.Time, enabledFunc func(cfg *model.Config) bool) *DailyScheduler {
|
|
return &DailyScheduler{
|
|
startTimeFunc: startTimeFunc,
|
|
jobType: jobType,
|
|
enabledFunc: enabledFunc,
|
|
jobs: jobs,
|
|
}
|
|
}
|
|
|
|
func (scheduler *DailyScheduler) Enabled(cfg *model.Config) bool {
|
|
return scheduler.enabledFunc(cfg)
|
|
}
|
|
|
|
func (scheduler *DailyScheduler) NextScheduleTime(cfg *model.Config, now time.Time /* pendingJobs */, _ bool /* lastSuccessfulJob */, _ *model.Job) *time.Time {
|
|
scheduledTime := scheduler.startTimeFunc(cfg)
|
|
if scheduledTime == nil {
|
|
return nil
|
|
}
|
|
|
|
return GenerateNextStartDateTime(now, *scheduledTime)
|
|
}
|
|
|
|
func (scheduler *DailyScheduler) ScheduleJob(rctx request.CTX, _ *model.Config /* pendingJobs */, _ bool /* lastSuccessfulJob */, _ *model.Job) (*model.Job, *model.AppError) {
|
|
return scheduler.jobs.CreateJob(rctx, scheduler.jobType, nil)
|
|
}
|
|
|
|
const jitterRange = 2000 // milliseconds
|
|
|
|
func getRandomDelay(limit int64) time.Duration {
|
|
num, err := rand.Int(rand.Reader, big.NewInt(limit))
|
|
if err != nil {
|
|
return time.Millisecond
|
|
}
|
|
return time.Millisecond * time.Duration(num.Int64())
|
|
}
|