mirror of
https://github.com/prometheus/node_exporter.git
synced 2024-12-23 00:46:08 +01:00
46 lines
889 B
Go
46 lines
889 B
Go
|
package kingpin
|
||
|
|
||
|
import (
|
||
|
"os"
|
||
|
"regexp"
|
||
|
)
|
||
|
|
||
|
var (
|
||
|
envVarValuesSeparator = "\r?\n"
|
||
|
envVarValuesTrimmer = regexp.MustCompile(envVarValuesSeparator + "$")
|
||
|
envVarValuesSplitter = regexp.MustCompile(envVarValuesSeparator)
|
||
|
)
|
||
|
|
||
|
type envarMixin struct {
|
||
|
envar string
|
||
|
noEnvar bool
|
||
|
}
|
||
|
|
||
|
func (e *envarMixin) HasEnvarValue() bool {
|
||
|
return e.GetEnvarValue() != ""
|
||
|
}
|
||
|
|
||
|
func (e *envarMixin) GetEnvarValue() string {
|
||
|
if e.noEnvar || e.envar == "" {
|
||
|
return ""
|
||
|
}
|
||
|
return os.Getenv(e.envar)
|
||
|
}
|
||
|
|
||
|
func (e *envarMixin) GetSplitEnvarValue() []string {
|
||
|
values := make([]string, 0)
|
||
|
|
||
|
envarValue := e.GetEnvarValue()
|
||
|
if envarValue == "" {
|
||
|
return values
|
||
|
}
|
||
|
|
||
|
// Split by new line to extract multiple values, if any.
|
||
|
trimmed := envVarValuesTrimmer.ReplaceAllString(envarValue, "")
|
||
|
for _, value := range envVarValuesSplitter.Split(trimmed, -1) {
|
||
|
values = append(values, value)
|
||
|
}
|
||
|
|
||
|
return values
|
||
|
}
|