CL-Softwares
/
telegraf
8
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feat(inputs.prometheus): Add support for custom header (#12050)

This commit is contained in:
Charly Nanfack 2022-11-21 15:55:21 -05:00 committed by GitHub
parent 6f407c5949
commit d4eda21742
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 46 additions and 32 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
plugins/inputs/prometheus/README.md
View File

@ -98,6 +98,9 @@ See the [CONFIGURATION.md][CONFIGURATION.md] for more details.
# username = "" # username = ""
# password = "" # password = ""
## Optional custom HTTP headers
# headers = {"X-Special-Header" = "Special-Value"}
## Specify timeout duration for slower prometheus clients (default is 3s) ## Specify timeout duration for slower prometheus clients (default is 3s)
# response_timeout = "3s" # response_timeout = "3s"

8
plugins/inputs/prometheus/prometheus.go
View File

@ -58,6 +58,8 @@ type Prometheus struct {
Username string `toml:"username"` Username string `toml:"username"`
Password string `toml:"password"` Password string `toml:"password"`
HTTPHeaders map[string]string `toml:"http_headers"`
ResponseTimeout config.Duration `toml:"response_timeout"` ResponseTimeout config.Duration `toml:"response_timeout"`
MetricVersion int `toml:"metric_version"` MetricVersion int `toml:"metric_version"`
@ -298,6 +300,12 @@ func (p *Prometheus) gatherURL(u URLAndAddress, acc telegraf.Accumulator) error
req.SetBasicAuth(p.Username, p.Password) req.SetBasicAuth(p.Username, p.Password)
} }
if p.HTTPHeaders != nil {
for key, value := range p.HTTPHeaders {
req.Header.Add(key, value)
}
}
var resp *http.Response var resp *http.Response
if u.URL.Scheme != "unix" { if u.URL.Scheme != "unix" {
//nolint:bodyclose // False positive (because of if-else) - body will be closed in `defer` //nolint:bodyclose // False positive (because of if-else) - body will be closed in `defer`

3
plugins/inputs/prometheus/sample.conf
View File

@ -81,6 +81,9 @@
# username = "" # username = ""
# password = "" # password = ""
## Optional custom HTTP headers
# headers = {"X-Special-Header" = "Special-Value"}
## Specify timeout duration for slower prometheus clients (default is 3s) ## Specify timeout duration for slower prometheus clients (default is 3s)
# response_timeout = "3s" # response_timeout = "3s"