feat(controllers): add custom not found handler

course-go · Aug 26, 2024 · 85aab00 · 85aab00
1 parent 71bb8d9
commit 85aab00
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/internal/controllers/api.go b/internal/controllers/api.go
@@ -54,6 +54,7 @@ func NewAPIRouter(
 
 func (a API) mountCommonControllers(mux *http.ServeMux) {
 	mux.Handle("/metrics", promhttp.Handler())
+	mux.HandleFunc("/", NotFound)
 }
 
 func (a API) mountTodoControllers(mux *http.ServeMux) {

diff --git a/internal/controllers/common.go b/internal/controllers/common.go
@@ -0,0 +1,9 @@
+package controllers
+
+import "net/http"
+
+func NotFound(w http.ResponseWriter, r *http.Request) {
+	code := http.StatusNotFound
+	w.WriteHeader(code)
+	w.Write(responseErrorBytes(code))
+}