Support for multiple status codes

Author: digiserg

README.md

@@ -119,4 +119,4 @@ This application can be used to verify whether a server is healthy before perfor
 
 ## License
 
-This project is licensed under the Apache License.
+This project is licensed under the Apache License.

config.yaml

@@ -30,3 +30,9 @@ endpoints:
   - name: external-api
     url: http://api.example.com
     status: 404
+  - name: other
+    url: http://something.example.com
+    statuses:
+      - 404
+      - 301
+      - 302

main.go

@@ -53,9 +53,10 @@ type Port struct {
 }
 
 type Endpoint struct {
-	Name   string `yaml:"name"`
-	URL    string `yaml:"url"`
-	Status int    `yaml:"status"`
+	Name     string `yaml:"name"`
+	URL      string `yaml:"url"`
+	Status   int    `yaml:"status"`
+	Statuses []int  `yaml:"statuses"`
 }
 
 var outputMessages []string
@@ -187,7 +188,7 @@ func checkEndpoints(endpoints []Endpoint) bool {
 		}
 		defer resp.Body.Close()
 
-		if resp.StatusCode == endpoint.Status {
+		if resp.StatusCode == endpoint.Status || contains(endpoint.Statuses, resp.StatusCode) {
 			addToOutputMessages("Endpoint Name: %s, URL: %s, Status: %d is as expected", endpoint.Name, endpoint.URL, endpoint.Status)
 		} else {
 			addToOutputMessages("Endpoint Name: %s, URL: %s, Status: %d is not as expected, got: %d", endpoint.Name, endpoint.URL, endpoint.Status, resp.StatusCode)
@@ -219,3 +220,12 @@ func GetEnvInt(key string, fallback int) int {
 	}
 	return fallback
 }
+
+func contains(numbers []int, target int) bool {
+	for _, num := range numbers {
+		if num == target {
+			return true
+		}
+	}
+	return false
+}

resources/post_install.sh

@@ -2,4 +2,4 @@
 
 getent passwd health > /dev/null || useradd -s /bin/false --no-create-home -r health
 
-[ -x /usr/bin/systemctl ] && systemctl daemon-reload
+[ -x /usr/bin/systemctl ] && systemctl daemon-reload