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]>
prometheus-community · Jul 11, 2022 · f8ddc2f · f8ddc2f
1 parent a03b913
commit f8ddc2f
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 1 deletion.
diff --git a/config/config.go b/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 {

diff --git a/examples/config.yml b/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 ]
diff --git a/exporter/util.go b/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)
 	}