forked from evergreen-ci/evergreen
-
Notifications
You must be signed in to change notification settings - Fork 0
/
config_serviceflags.go
112 lines (101 loc) · 6.03 KB
/
config_serviceflags.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
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
package evergreen
import (
"github.com/pkg/errors"
"go.mongodb.org/mongo-driver/bson"
"go.mongodb.org/mongo-driver/mongo"
"go.mongodb.org/mongo-driver/mongo/options"
)
// ServiceFlags holds the state of each of the runner/API processes
type ServiceFlags struct {
PluginAdminPageDisabled bool `bson:"plugin_admin_page_disabled" json:"plugin_admin_page_disabled"`
TaskDispatchDisabled bool `bson:"task_dispatch_disabled" json:"task_dispatch_disabled"`
HostInitDisabled bool `bson:"host_init_disabled" json:"host_init_disabled"`
PodInitDisabled bool `bson:"pod_init_disabled" json:"pod_init_disabled"`
S3BinaryDownloadsDisabled bool `bson:"s3_binary_downloads_disabled" json:"s3_binary_downloads_disabled"`
MonitorDisabled bool `bson:"monitor_disabled" json:"monitor_disabled"`
AlertsDisabled bool `bson:"alerts_disabled" json:"alerts_disabled"`
AgentStartDisabled bool `bson:"agent_start_disabled" json:"agent_start_disabled"`
RepotrackerDisabled bool `bson:"repotracker_disabled" json:"repotracker_disabled"`
SchedulerDisabled bool `bson:"scheduler_disabled" json:"scheduler_disabled"`
CheckBlockedTasksDisabled bool `bson:"check_blocked_tasks_disabled" json:"check_blocked_tasks_disabled"`
GithubPRTestingDisabled bool `bson:"github_pr_testing_disabled" json:"github_pr_testing_disabled"`
CLIUpdatesDisabled bool `bson:"cli_updates_disabled" json:"cli_updates_disabled"`
BackgroundStatsDisabled bool `bson:"background_stats_disabled" json:"background_stats_disabled"`
TaskLoggingDisabled bool `bson:"task_logging_disabled" json:"task_logging_disabled"`
CacheStatsJobDisabled bool `bson:"cache_stats_job_disabled" json:"cache_stats_job_disabled"`
CacheStatsEndpointDisabled bool `bson:"cache_stats_endpoint_disabled" json:"cache_stats_endpoint_disabled"`
TaskReliabilityDisabled bool `bson:"task_reliability_disabled" json:"task_reliability_disabled"`
CommitQueueDisabled bool `bson:"commit_queue_disabled" json:"commit_queue_disabled"`
PlannerDisabled bool `bson:"planner_disabled" json:"planner_disabled"`
HostAllocatorDisabled bool `bson:"host_allocator_disabled" json:"host_allocator_disabled"`
BackgroundReauthDisabled bool `bson:"background_reauth_disabled" json:"background_reauth_disabled"`
BackgroundCleanupDisabled bool `bson:"background_cleanup_disabled" json:"background_cleanup_disabled"`
AmboyRemoteManagementDisabled bool `bson:"amboy_remote_management_disabled" json:"amboy_remote_management_disabled"`
// Notification Flags
EventProcessingDisabled bool `bson:"event_processing_disabled" json:"event_processing_disabled"`
JIRANotificationsDisabled bool `bson:"jira_notifications_disabled" json:"jira_notifications_disabled"`
SlackNotificationsDisabled bool `bson:"slack_notifications_disabled" json:"slack_notifications_disabled"`
EmailNotificationsDisabled bool `bson:"email_notifications_disabled" json:"email_notifications_disabled"`
WebhookNotificationsDisabled bool `bson:"webhook_notifications_disabled" json:"webhook_notifications_disabled"`
GithubStatusAPIDisabled bool `bson:"github_status_api_disabled" json:"github_status_api_disabled"`
}
func (c *ServiceFlags) SectionId() string { return "service_flags" }
func (c *ServiceFlags) Get(env Environment) error {
ctx, cancel := env.Context()
defer cancel()
coll := env.DB().Collection(ConfigCollection)
res := coll.FindOne(ctx, byId(c.SectionId()))
if err := res.Err(); err != nil {
if err == mongo.ErrNoDocuments {
*c = ServiceFlags{}
return nil
}
return errors.Wrapf(err, "error retrieving section %s", c.SectionId())
}
if err := res.Decode(c); err != nil {
return errors.Wrap(err, "problem decoding result")
}
return nil
}
func (c *ServiceFlags) Set() error {
env := GetEnvironment()
ctx, cancel := env.Context()
defer cancel()
coll := env.DB().Collection(ConfigCollection)
_, err := coll.UpdateOne(ctx, byId(c.SectionId()), bson.M{
"$set": bson.M{
pluginAdminPageDisabledKey: c.PluginAdminPageDisabled,
taskDispatchKey: c.TaskDispatchDisabled,
hostInitKey: c.HostInitDisabled,
podInitDisabledKey: c.PodInitDisabled,
s3BinaryDownloadsDisabledKey: c.S3BinaryDownloadsDisabled,
monitorKey: c.MonitorDisabled,
alertsKey: c.AlertsDisabled,
agentStartKey: c.AgentStartDisabled,
repotrackerKey: c.RepotrackerDisabled,
schedulerKey: c.SchedulerDisabled,
checkBlockedTasksKey: c.CheckBlockedTasksDisabled,
githubPRTestingDisabledKey: c.GithubPRTestingDisabled,
cliUpdatesDisabledKey: c.CLIUpdatesDisabled,
backgroundStatsDisabledKey: c.BackgroundStatsDisabled,
eventProcessingDisabledKey: c.EventProcessingDisabled,
jiraNotificationsDisabledKey: c.JIRANotificationsDisabled,
slackNotificationsDisabledKey: c.SlackNotificationsDisabled,
emailNotificationsDisabledKey: c.EmailNotificationsDisabled,
webhookNotificationsDisabledKey: c.WebhookNotificationsDisabled,
githubStatusAPIDisabledKey: c.GithubStatusAPIDisabled,
taskLoggingDisabledKey: c.TaskLoggingDisabled,
cacheStatsJobDisabledKey: c.CacheStatsJobDisabled,
cacheStatsEndpointDisabledKey: c.CacheStatsEndpointDisabled,
taskReliabilityDisabledKey: c.TaskReliabilityDisabled,
commitQueueDisabledKey: c.CommitQueueDisabled,
plannerDisabledKey: c.PlannerDisabled,
hostAllocatorDisabledKey: c.HostAllocatorDisabled,
backgroundCleanupDisabledKey: c.BackgroundCleanupDisabled,
backgroundReauthDisabledKey: c.BackgroundReauthDisabled,
amboyRemoteManagementDisabledKey: c.AmboyRemoteManagementDisabled,
},
}, options.Update().SetUpsert(true))
return errors.Wrapf(err, "error updating section %s", c.SectionId())
}
func (c *ServiceFlags) ValidateAndDefault() error { return nil }