Refactor

Signed-off-by: Oliver Davies <oliver@oliverdavies.uk>

go-api/utils.go

@@ -0,0 +1,33 @@
+package main
+
+import (
+	"log"
+	"net/http"
+	"os"
+	"strconv"
+)
+
+func getPort() string {
+	port := os.Getenv("PORT")
+
+	if port != "" {
+		return port
+	}
+
+	return "8080"
+}
+
+func getResponseCode(r *http.Request) int {
+	// If the `force-fail` header is set, get and return its value.
+	if failCode := r.Header.Get("force-fail"); failCode != "" {
+		log.Println("`force-fail` header set...")
+
+		if code, err := strconv.Atoi(failCode); err == nil {
+			log.Printf("Setting the response code to %d...", code)
+
+			return code
+		}
+	}
+
+	return http.StatusOK
+}