From 05dc6e83eded4bb6fe8500c4432659c80d26ae9e Mon Sep 17 00:00:00 2001
From: Tom Andrade <wolvie@gmail.com>
Date: Thu, 26 Jan 2023 11:53:00 +0100
Subject: [PATCH] feat: switch to http.Error()

---
 main.go | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/main.go b/main.go
index 5a31527..b8daae1 100644
--- a/main.go
+++ b/main.go
@@ -102,8 +102,7 @@ func handler(l int) http.HandlerFunc {
 func httpHealth(l int) http.HandlerFunc {
 	return func(w http.ResponseWriter, r *http.Request) {
 		if !serve(l) {
-			w.WriteHeader(http.StatusInternalServerError)
-			fmt.Fprintf(w, `{"status":"FAIL","requests":"%v"}`, C.Get())
+			http.Error(w, fmt.Sprintf(`{"status":"FAIL","requests":"%v"}`, C.Get()), http.StatusInternalServerError)
 			return
 		}
 		fmt.Fprintf(w, `{"status":"ok","requests":"%v"}`, C.Get())
@@ -116,8 +115,7 @@ func reset(rt string) http.HandlerFunc {
 		r.ParseForm()
 		t := r.FormValue("TOKEN")
 		if len(t) == 0 || t != rt {
-			w.WriteHeader(http.StatusBadRequest)
-			fmt.Fprintf(w, "Bad request, invalid token")
+			http.Error(w, "Bad request, invalid token", http.StatusBadRequest)
 			return
 		}
 		C.Reset()