mirror of
https://github.com/hanzoai/team.git
synced 2026-11-01 07:55:31 +00:00
3c1d3a081d
* Standardize request.CTX parameter naming to rctx - Migrate 886 request.CTX parameters across 147 files to use consistent 'rctx' naming - Updated function signatures from 'c', 'ctx', and 'cancelContext' to 'rctx' - Updated function bodies to reference the new parameter names - Preserved underscore parameters unchanged as they are unused - Fixed method receiver context issue in store.go * Use request.CTX interface in batch worker * Manual fixes * Fix parameter naming * Add linter check --------- Co-authored-by: Hanzo Dev <[email protected]>
111 lines
3.8 KiB
Go
111 lines
3.8 KiB
Go
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
|
// See LICENSE.txt for license information.
|
|
|
|
package migrations
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/mattermost/mattermost/server/public/model"
|
|
"github.com/mattermost/mattermost/server/public/shared/mlog"
|
|
"github.com/mattermost/mattermost/server/public/shared/request"
|
|
"github.com/mattermost/mattermost/server/v8/channels/jobs"
|
|
"github.com/mattermost/mattermost/server/v8/channels/store"
|
|
)
|
|
|
|
const (
|
|
MigrationJobWedgedTimeoutMilliseconds = 3600000 // 1 hour
|
|
)
|
|
|
|
type Scheduler struct {
|
|
jobServer *jobs.JobServer
|
|
store store.Store
|
|
allMigrationsCompleted bool
|
|
}
|
|
|
|
var _ jobs.Scheduler = (*Scheduler)(nil)
|
|
|
|
func MakeScheduler(jobServer *jobs.JobServer, store store.Store) *Scheduler {
|
|
return &Scheduler{jobServer, store, false}
|
|
}
|
|
|
|
func (scheduler *Scheduler) Enabled(_ *model.Config) bool {
|
|
return true
|
|
}
|
|
|
|
//nolint:unparam
|
|
func (scheduler *Scheduler) NextScheduleTime(cfg *model.Config, now time.Time, pendingJobs bool, lastSuccessfulJob *model.Job) *time.Time {
|
|
if scheduler.allMigrationsCompleted {
|
|
return nil
|
|
}
|
|
|
|
nextTime := time.Now().Add(60 * time.Second)
|
|
return &nextTime
|
|
}
|
|
|
|
//nolint:unparam
|
|
func (scheduler *Scheduler) ScheduleJob(rctx request.CTX, cfg *model.Config, pendingJobs bool, lastSuccessfulJob *model.Job) (*model.Job, *model.AppError) {
|
|
rctx.Logger().Debug("Scheduling Job", mlog.String("scheduler", model.JobTypeMigrations))
|
|
|
|
// Work through the list of migrations in order. Schedule the first one that isn't done (assuming it isn't in progress already).
|
|
for _, key := range MakeMigrationsList() {
|
|
state, job, err := GetMigrationState(rctx, key, scheduler.store)
|
|
if err != nil {
|
|
rctx.Logger().Error("Failed to determine status of migration: ", mlog.String("scheduler", model.JobTypeMigrations), mlog.String("migration_key", key), mlog.Err(err))
|
|
return nil, nil
|
|
}
|
|
|
|
logger := rctx.Logger().With(jobs.JobLoggerFields(job)...)
|
|
|
|
if state == MigrationStateCompleted {
|
|
// This migration is done. Continue to check the next.
|
|
continue
|
|
}
|
|
|
|
if state == MigrationStateInProgress {
|
|
// Check the migration job isn't wedged.
|
|
if job != nil && job.LastActivityAt < model.GetMillis()-MigrationJobWedgedTimeoutMilliseconds && job.CreateAt < model.GetMillis()-MigrationJobWedgedTimeoutMilliseconds {
|
|
logger.Warn("Job appears to be wedged. Rescheduling another instance.", mlog.String("scheduler", model.JobTypeMigrations), mlog.String("migration_key", key))
|
|
if err := scheduler.jobServer.SetJobError(job, nil); err != nil {
|
|
logger.Error("Worker: Failed to set job error", mlog.String("scheduler", model.JobTypeMigrations), mlog.Err(err))
|
|
}
|
|
return scheduler.createJob(rctx, key, job)
|
|
}
|
|
|
|
return nil, nil
|
|
}
|
|
|
|
if state == MigrationStateUnscheduled {
|
|
logger.Debug("Scheduling a new job for migration.", mlog.String("scheduler", model.JobTypeMigrations), mlog.String("migration_key", key))
|
|
return scheduler.createJob(rctx, key, job)
|
|
}
|
|
|
|
logger.Error("Unknown migration state. Not doing anything.", mlog.String("migration_state", state))
|
|
return nil, nil
|
|
}
|
|
|
|
// If we reached here, then there aren't any migrations left to run.
|
|
scheduler.allMigrationsCompleted = true
|
|
rctx.Logger().Debug("All migrations are complete.", mlog.String("scheduler", model.JobTypeMigrations))
|
|
|
|
return nil, nil
|
|
}
|
|
|
|
func (scheduler *Scheduler) createJob(rctx request.CTX, migrationKey string, lastJob *model.Job) (*model.Job, *model.AppError) {
|
|
var lastDone string
|
|
if lastJob != nil {
|
|
lastDone = lastJob.Data[JobDataKeyMigrationLastDone]
|
|
}
|
|
|
|
data := map[string]string{
|
|
JobDataKeyMigration: migrationKey,
|
|
JobDataKeyMigrationLastDone: lastDone,
|
|
}
|
|
|
|
job, err := scheduler.jobServer.CreateJob(rctx, model.JobTypeMigrations, data)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return job, nil
|
|
}
|