summaryrefslogtreecommitdiffstats
path: root/src/go/collectors/go.d.plugin/modules/hdfs/client.go
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--src/go/collectors/go.d.plugin/modules/hdfs/client.go69
1 files changed, 69 insertions, 0 deletions
diff --git a/src/go/collectors/go.d.plugin/modules/hdfs/client.go b/src/go/collectors/go.d.plugin/modules/hdfs/client.go
new file mode 100644
index 000000000..bdeced146
--- /dev/null
+++ b/src/go/collectors/go.d.plugin/modules/hdfs/client.go
@@ -0,0 +1,69 @@
+// SPDX-License-Identifier: GPL-3.0-or-later
+
+package hdfs
+
+import (
+ "encoding/json"
+ "fmt"
+ "io"
+ "net/http"
+
+ "github.com/netdata/netdata/go/go.d.plugin/pkg/web"
+)
+
+func newClient(httpClient *http.Client, request web.Request) *client {
+ return &client{
+ httpClient: httpClient,
+ request: request,
+ }
+}
+
+type client struct {
+ httpClient *http.Client
+ request web.Request
+}
+
+func (c *client) do() (*http.Response, error) {
+ req, err := web.NewHTTPRequest(c.request)
+ if err != nil {
+ return nil, fmt.Errorf("error on creating http request to %s : %v", c.request.URL, err)
+ }
+
+ // req.Header.Add("Accept-Encoding", "gzip")
+ // req.Header.Set("User-Agent", "netdata/go.d.plugin")
+
+ return c.httpClient.Do(req)
+}
+
+func (c *client) doOK() (*http.Response, error) {
+ resp, err := c.do()
+ if err != nil {
+ return nil, err
+ }
+
+ if resp.StatusCode != http.StatusOK {
+ return resp, fmt.Errorf("%s returned %d", c.request.URL, resp.StatusCode)
+ }
+ return resp, nil
+}
+
+func (c *client) doOKWithDecodeJSON(dst interface{}) error {
+ resp, err := c.doOK()
+ defer closeBody(resp)
+ if err != nil {
+ return err
+ }
+
+ err = json.NewDecoder(resp.Body).Decode(dst)
+ if err != nil {
+ return fmt.Errorf("error on decoding response from %s : %v", c.request.URL, err)
+ }
+ return nil
+}
+
+func closeBody(resp *http.Response) {
+ if resp != nil && resp.Body != nil {
+ _, _ = io.Copy(io.Discard, resp.Body)
+ _ = resp.Body.Close()
+ }
+}