feat: recover middleware

miyataka · Jan 1, 2024 · 332ba5f · 332ba5f
1 parent 9b54d70
commit 332ba5f
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/recover_middleware.go b/recover_middleware.go
@@ -0,0 +1,18 @@
+package minimalmux
+
+import (
+	"fmt"
+	"net/http"
+)
+
+func RecoverMiddleware(next http.Handler) http.Handler {
+	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		defer func() {
+			if r := recover(); r != nil {
+				w.WriteHeader(http.StatusInternalServerError)
+				fmt.Fprintf(w, "Internal Server Error")
+			}
+		}()
+		next.ServeHTTP(w, r)
+	})
+}