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
113
114
115
116
117
118
119
120
121
122
123
124
125
|
package config
import (
"fmt"
"log/slog"
"os"
"path/filepath"
"sync"
"github.com/goccy/go-yaml"
"github.com/prometheus/client_golang/prometheus"
"github.com/prometheus/client_golang/prometheus/promauto"
)
type Config struct {
Scripts []Script `yaml:"scripts"`
}
func (c *Config) GetScript(name string) *Script {
for _, script := range c.Scripts {
if name == script.Name {
return &script
}
}
return nil
}
type Script struct {
Name string `yaml:"name"`
Command []string `yaml:"command"`
Args []string `yaml:"args"`
Env map[string]string `yaml:"env"`
AllowEnvOverwrite bool `yaml:"allow_env_overwrite"`
Sudo bool `yaml:"sudo"`
Output Output `yaml:"output"`
Timeout Timeout `yaml:"timeout"`
Cache Cache `yaml:"cache"`
Discovery Discovery `yaml:"discovery"`
}
type Output struct {
Ignore bool `yaml:"ignore"`
IgnoreOnError bool `yaml:"ignore_on_error"`
}
type Timeout struct {
MaxTimeout float64 `yaml:"max_timeout"`
Enforced bool `yaml:"enforced"`
WaitDelay float64 `yaml:"wait_delay"`
}
type Cache struct {
Duration *float64 `yaml:"duration"`
CacheOnError bool `yaml:"cache_on_error"`
UseExpiredCacheOnError bool `yaml:"use_expired_cache_on_error"`
}
type Discovery struct {
Params map[string]string `yaml:"params"`
ScrapeInterval string `yaml:"scrape_interval"`
ScrapeTimeout string `yaml:"scrape_timeout"`
}
type SafeConfig struct {
sync.RWMutex
C *Config
configReloadSuccess prometheus.Gauge
configReloadSeconds prometheus.Gauge
}
func NewSafeConfig(reg prometheus.Registerer) *SafeConfig {
configReloadSuccess := promauto.With(reg).NewGauge(prometheus.GaugeOpts{
Namespace: "script_exporter",
Name: "config_last_reload_successful",
Help: "Script Exporter config loaded successfully.",
})
configReloadSeconds := promauto.With(reg).NewGauge(prometheus.GaugeOpts{
Namespace: "script_exporter",
Name: "config_last_reload_success_timestamp_seconds",
Help: "Timestamp of the last successful configuration reload.",
})
return &SafeConfig{C: &Config{}, configReloadSuccess: configReloadSuccess, configReloadSeconds: configReloadSeconds}
}
func (sc *SafeConfig) ReloadConfig(configFiles string, logger *slog.Logger) (err error) {
var c = &Config{}
defer func() {
if err != nil {
sc.configReloadSuccess.Set(0)
} else {
sc.configReloadSuccess.Set(1)
sc.configReloadSeconds.SetToCurrentTime()
}
}()
files, err := filepath.Glob(configFiles)
if err != nil {
return err
}
for _, file := range files {
var fc = &Config{}
yamlReader, err := os.Open(file)
if err != nil {
return fmt.Errorf("error reading config file: %s", err)
}
defer yamlReader.Close()
decoder := yaml.NewDecoder(yamlReader, yaml.DisallowUnknownField())
if err = decoder.Decode(fc); err != nil {
return fmt.Errorf("error parsing config file: %s", err)
}
c.Scripts = append(c.Scripts, fc.Scripts...)
}
sc.Lock()
sc.C = c
sc.Unlock()
return nil
}
|
