-
Notifications
You must be signed in to change notification settings - Fork 3.8k
/
compact_sql_stats.go
70 lines (60 loc) · 2.26 KB
/
compact_sql_stats.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
// Copyright 2021 The Cockroach Authors.
//
// Use of this software is governed by the Business Source License
// included in the file licenses/BSL.txt.
//
// As of the Change Date specified in that file, in accordance with
// the Business Source License, use of this software will be governed
// by the Apache License, Version 2.0, included in the file
// licenses/APL.txt.
package sql
import (
"context"
"github.com/cockroachdb/cockroach/pkg/jobs"
"github.com/cockroachdb/cockroach/pkg/jobs/jobspb"
"github.com/cockroachdb/cockroach/pkg/settings/cluster"
"github.com/cockroachdb/cockroach/pkg/sql/sqlstats/persistedsqlstats"
"github.com/cockroachdb/cockroach/pkg/util/log"
"github.com/cockroachdb/errors"
)
type sqlStatsCompactionResumer struct {
*jobs.Job
compactionExecutor *persistedsqlstats.StatsCompactor
}
var _ jobs.Resumer = &sqlStatsCompactionResumer{}
// Resume implements the jobs.Resumer interface.
func (r *sqlStatsCompactionResumer) Resume(ctx context.Context, execCtx interface{}) error {
log.Infof(ctx, "starting sql stats compaction job")
p := execCtx.(JobExecContext)
ie := p.ExecCfg().InternalExecutor
db := p.ExecCfg().DB
// We check for concurrently running SQL Stats compaction jobs. We only allow
// one job to be running at the same time.
if err := persistedsqlstats.CheckExistingCompactionJob(ctx, r.Job, ie, nil /* txn */); err != nil {
if errors.Is(err, persistedsqlstats.ErrConcurrentSQLStatsCompaction) {
log.Infof(ctx, "exiting due to a running sql stats compaction job")
return nil
}
return err
}
jobDetails, ok := r.Job.Details().(jobspb.SQLStatsCompactionDetails)
if !ok {
return errors.AssertionFailedf("invalid job detail payload for sql stats compaction")
}
r.compactionExecutor.Init(ie, db, jobDetails.DisableFollowerRead)
return r.compactionExecutor.DeleteOldestEntries(ctx)
}
// OnFailOrCancel implements the jobs.Resumer interface.
func (r *sqlStatsCompactionResumer) OnFailOrCancel(ctx context.Context, execCtx interface{}) error {
return nil
}
func init() {
jobs.RegisterConstructor(jobspb.TypeSQLStatsCompaction, func(job *jobs.Job, settings *cluster.Settings) jobs.Resumer {
return &sqlStatsCompactionResumer{
Job: job,
compactionExecutor: &persistedsqlstats.StatsCompactor{
Settings: settings,
},
}
})
}