forked from evergreen-ci/evergreen
-
Notifications
You must be signed in to change notification settings - Fork 0
/
config_commitqueue.go
64 lines (53 loc) · 2.44 KB
/
config_commitqueue.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
package evergreen
import (
"context"
"github.com/mongodb/anser/bsonutil"
"github.com/pkg/errors"
"go.mongodb.org/mongo-driver/bson"
"go.mongodb.org/mongo-driver/mongo"
"go.mongodb.org/mongo-driver/mongo/options"
)
type CommitQueueConfig struct {
MergeTaskDistro string `yaml:"merge_task_distro" bson:"merge_task_distro" json:"merge_task_distro"`
CommitterName string `yaml:"committer_name" bson:"committer_name" json:"committer_name"`
CommitterEmail string `yaml:"committer_email" bson:"committer_email" json:"committer_email"`
BatchSize int `yaml:"batch_size" bson:"batch_size" json:"batch_size"`
MaxSystemFailedTaskRetries int `yaml:"max_system_failed_task_retries" bson:"max_system_failed_task_retries" json:"max_system_failed_task_retries"`
}
var (
mergeTaskDistroKey = bsonutil.MustHaveTag(CommitQueueConfig{}, "MergeTaskDistro")
committerNameKey = bsonutil.MustHaveTag(CommitQueueConfig{}, "CommitterName")
committerEmailKey = bsonutil.MustHaveTag(CommitQueueConfig{}, "CommitterEmail")
commitQueueBatchSizeKey = bsonutil.MustHaveTag(CommitQueueConfig{}, "BatchSize")
maxSystemFailedTaskRetriesKey = bsonutil.MustHaveTag(CommitQueueConfig{}, "MaxSystemFailedTaskRetries")
)
func (c *CommitQueueConfig) SectionId() string { return "commit_queue" }
func (c *CommitQueueConfig) Get(ctx context.Context) error {
res := GetEnvironment().DB().Collection(ConfigCollection).FindOne(ctx, byId(c.SectionId()))
if err := res.Err(); err != nil {
if err == mongo.ErrNoDocuments {
*c = CommitQueueConfig{}
return nil
}
return errors.Wrapf(err, "getting config section '%s'", c.SectionId())
}
if err := res.Decode(&c); err != nil {
return errors.Wrapf(err, "decoding config section '%s'", c.SectionId())
}
return nil
}
func (c *CommitQueueConfig) Set(ctx context.Context) error {
_, err := GetEnvironment().DB().Collection(ConfigCollection).UpdateOne(ctx, byId(c.SectionId()), bson.M{
"$set": bson.M{
mergeTaskDistroKey: c.MergeTaskDistro,
committerNameKey: c.CommitterName,
committerEmailKey: c.CommitterEmail,
commitQueueBatchSizeKey: c.BatchSize,
maxSystemFailedTaskRetriesKey: c.MaxSystemFailedTaskRetries,
},
}, options.Update().SetUpsert(true))
return errors.Wrapf(err, "updating config section '%s'", c.SectionId())
}
func (c *CommitQueueConfig) ValidateAndDefault() error {
return nil
}