diff options
Diffstat (limited to '')
-rw-r--r-- | src/go/plugin/go.d/modules/puppet/collect.go | 75 |
1 files changed, 75 insertions, 0 deletions
diff --git a/src/go/plugin/go.d/modules/puppet/collect.go b/src/go/plugin/go.d/modules/puppet/collect.go new file mode 100644 index 000000000..a1b95e09c --- /dev/null +++ b/src/go/plugin/go.d/modules/puppet/collect.go @@ -0,0 +1,75 @@ +// SPDX-License-Identifier: GPL-3.0-or-later + +package puppet + +import ( + "encoding/json" + "fmt" + "io" + "net/http" + "net/url" + + "github.com/netdata/netdata/go/plugins/plugin/go.d/pkg/stm" + "github.com/netdata/netdata/go/plugins/plugin/go.d/pkg/web" +) + +var ( + //https://puppet.com/docs/puppet/8/server/status-api/v1/services + urlPathStatusService = "/status/v1/services" + urlQueryStatusService = url.Values{"level": {"debug"}}.Encode() +) + +func (p *Puppet) collect() (map[string]int64, error) { + stats, err := p.queryStatsService() + if err != nil { + return nil, err + } + + mx := stm.ToMap(stats) + + return mx, nil +} + +func (p *Puppet) queryStatsService() (*statusServiceResponse, error) { + req, err := web.NewHTTPRequestWithPath(p.Request, urlPathStatusService) + if err != nil { + return nil, err + } + + req.URL.RawQuery = urlQueryStatusService + + var stats statusServiceResponse + if err := p.doOKDecode(req, &stats); err != nil { + return nil, err + } + + if stats.StatusService == nil { + return nil, fmt.Errorf("unexpected response: not puppet service status data") + } + + return &stats, nil +} + +func (p *Puppet) doOKDecode(req *http.Request, in interface{}) error { + resp, err := p.httpClient.Do(req) + if err != nil { + return fmt.Errorf("error on HTTP request '%s': %v", req.URL, err) + } + defer closeBody(resp) + + if resp.StatusCode != http.StatusOK { + return fmt.Errorf("'%s' returned HTTP status code: %d", req.URL, resp.StatusCode) + } + + if err := json.NewDecoder(resp.Body).Decode(in); err != nil { + return fmt.Errorf("error on decoding response from '%s': %v", req.URL, err) + } + return nil +} + +func closeBody(resp *http.Response) { + if resp != nil && resp.Body != nil { + _, _ = io.Copy(io.Discard, resp.Body) + _ = resp.Body.Close() + } +} |