-
Notifications
You must be signed in to change notification settings - Fork 1.2k
/
providers.go
92 lines (75 loc) · 2.76 KB
/
providers.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
// 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 2016-present Datadog, Inc.
package flare
import (
"path/filepath"
"regexp"
"strings"
"github.com/DataDog/datadog-agent/comp/core/flare/types"
)
var (
// Match .yaml and .yml to ship configuration files in the flare.
cnfFileExtRx = regexp.MustCompile(`(?i)\.ya?ml`)
)
func getFirstSuffix(s string) string {
return filepath.Ext(strings.TrimSuffix(s, filepath.Ext(s)))
}
func (f *flare) collectLogsFiles(fb types.FlareBuilder) error {
logFile := f.config.GetString("log_file")
if logFile == "" {
logFile = f.params.defaultLogFile
}
jmxLogFile := f.config.GetString("jmx_log_file")
if jmxLogFile == "" {
jmxLogFile = f.params.defaultJMXLogFile
}
dogstatsdLogFile := f.config.GetString("dogstatsd_log_file")
if dogstatsdLogFile == "" {
dogstatsdLogFile = f.params.defaultDogstatsdLogFile
}
shouldIncludeFunc := func(path string) bool {
if filepath.Ext(path) == ".log" || getFirstSuffix(path) == ".log" {
return true
}
return false
}
f.log.Flush()
fb.CopyDirToWithoutScrubbing(filepath.Dir(logFile), "logs", shouldIncludeFunc)
fb.CopyDirToWithoutScrubbing(filepath.Dir(jmxLogFile), "logs", shouldIncludeFunc)
fb.CopyDirToWithoutScrubbing(filepath.Dir(dogstatsdLogFile), "logs", shouldIncludeFunc)
return nil
}
func (f *flare) collectConfigFiles(fb types.FlareBuilder) error {
confSearchPaths := map[string]string{
"": f.config.GetString("confd_path"),
"dist": filepath.Join(f.params.distPath, "conf.d"),
"checksd": f.params.pythonChecksPath,
}
for prefix, filePath := range confSearchPaths {
fb.CopyDirTo(filePath, filepath.Join("etc", "confd", prefix), func(path string) bool {
// ignore .example file
if filepath.Ext(path) == ".example" {
return false
}
firstSuffix := []byte(getFirstSuffix(path))
ext := []byte(filepath.Ext(path))
if cnfFileExtRx.Match(firstSuffix) || cnfFileExtRx.Match(ext) {
return true
}
return false
})
}
if mainConfpath := f.config.ConfigFileUsed(); mainConfpath != "" {
confDir := filepath.Dir(mainConfpath)
// zip up the config file that was actually used, if one exists
fb.CopyFileTo(mainConfpath, filepath.Join("etc", "datadog.yaml"))
// figure out system-probe file path based on main config path, and use best effort to include
// system-probe.yaml to the flare
fb.CopyFileTo(filepath.Join(confDir, "system-probe.yaml"), filepath.Join("etc", "system-probe.yaml"))
// use best effort to include security-agent.yaml to the flare
fb.CopyFileTo(filepath.Join(confDir, "security-agent.yaml"), filepath.Join("etc", "security-agent.yaml"))
}
return nil
}