-
Notifications
You must be signed in to change notification settings - Fork 1.2k
/
configs_agent_task.go
59 lines (47 loc) · 1.58 KB
/
configs_agent_task.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
// Unless explicitly stated otherwise all files in this repository are licensed
// under the Apache License Version 2.0.
// This product includes software developed at Datadog (https://www.datadoghq.com/).
// Copyright 2023-present Datadog, Inc.
package state
import (
"encoding/json"
"fmt"
)
// AgentTaskConfig is a deserialized agent task configuration file
// along with the associated metadata
type AgentTaskConfig struct {
Config AgentTaskData
Metadata Metadata
}
// AgentTaskData is the content of a agent task configuration file
type AgentTaskData struct {
TaskType string `json:"task_type"`
UUID string `json:"uuid"`
TaskArgs map[string]string `json:"args"`
}
// ParseConfigAgentTask parses an agent task config
func ParseConfigAgentTask(data []byte, metadata Metadata) (AgentTaskConfig, error) {
var d AgentTaskData
err := json.Unmarshal(data, &d)
if err != nil {
return AgentTaskConfig{}, fmt.Errorf("Unexpected AGENT_TASK received through remote-config: %s", err)
}
return AgentTaskConfig{
Config: d,
Metadata: metadata,
}, nil
}
// AgentTaskConfigs returns the currently active AGENT_TASK configs
func (r *Repository) AgentTaskConfigs() map[string]AgentTaskConfig {
typedConfigs := make(map[string]AgentTaskConfig)
configs := r.getConfigs(ProductAgentTask)
for path, conf := range configs {
// We control this, so if this has gone wrong something has gone horribly wrong
typed, ok := conf.(AgentTaskConfig)
if !ok {
panic("unexpected config stored as AgentTaskConfigs")
}
typedConfigs[path] = typed
}
return typedConfigs
}