Add server-side error handler

camdram · Feb 14, 2021 · 2ef4dfc · 2ef4dfc
1 parent 5bf7d94
commit 2ef4dfc
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/internal/server/controller.go b/internal/server/controller.go
@@ -35,6 +35,7 @@ func newController(driver *SQLDriver, token string, addr string) *Controller {
 }
 
 func (c *Controller) ListenAndServe() error {
+	c.router.Use(recoveryMiddleware)
 	c.router.Use(logMiddleware)
 	c.router.Use(headerMiddleware)
 	c.router.Use(c.authMiddleware)

diff --git a/internal/server/middleware.go b/internal/server/middleware.go
@@ -41,7 +41,7 @@ func recoveryMiddleware(next http.Handler) http.Handler {
 	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
 		defer func() {
 			if err := recover(); err != nil {
-				http.Error(w, "500 Internal Server Error", http.StatusInternalServerError)
+				http.Error(w, "500 internal server error", http.StatusInternalServerError)
 				log.Println("An internal server error occurred:", err)
 				log.Println(string(debug.Stack()))
 			}