Add ignore_status (#161)

* Add ignore_status

Signed-off-by: Alexander Ryabov <[email protected]>

* Use valid_status_codes slice

Signed-off-by: Alexander Ryabov <[email protected]>

Co-authored-by: Alexander Ryabov <[email protected]>

Author: sepich

config/config.go

@@ -57,6 +57,7 @@ type Module struct {
 	Metrics          []Metric                 `yaml:"metrics"`
 	HTTPClientConfig pconfig.HTTPClientConfig `yaml:"http_client_config,omitempty"`
 	Body             Body                     `yaml:"body,omitempty"`
+	ValidStatusCodes []int                    `yaml:"valid_status_codes,omitempty"`
 }
 
 type Body struct {

examples/config.yml

@@ -43,3 +43,6 @@ modules:
     #     username: myuser
     #     #password: veryverysecret
     #     password_file: /tmp/mysecret.txt
+
+    # Accepted status codes for this probe. Defaults to 2xx.
+    # valid_status_codes: [ <int>, ... | default = 2xx ]

exporter/util.go

@@ -179,7 +179,18 @@ func (f *JSONFetcher) FetchJSON(endpoint string) ([]byte, error) {
 		resp.Body.Close()
 	}()
 
-	if resp.StatusCode/100 != 2 {
+	if len(f.module.ValidStatusCodes) != 0 {
+		success := false
+		for _, code := range f.module.ValidStatusCodes {
+			if resp.StatusCode == code {
+				success = true
+				break
+			}
+		}
+		if !success {
+			return nil, errors.New(resp.Status)
+		}
+	} else if resp.StatusCode/100 != 2 {
 		return nil, errors.New(resp.Status)
 	}