Overriding NotFound & MethodNotAllowed errors
No One
Actions

Authored By

	Eevans
	Oct 13 2021, 8:12 PM

Size

833 B

Referenced Files

None

Subscribers

None

Overriding NotFound & MethodNotAllowed errors
View Options

	package main

	import (
	"fmt"
	"log"
	"net/http"

	"github.com/julienschmidt/httprouter"
	"schneider.vip/problem"
	)

	func problemHandler(status int) http.Handler {
	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
	w.Header().Set("Content-Type", "text/plain; charset=utf-8")
	w.Header().Set("X-Content-Type-Options", "nosniff")
	problem.Of(status).WriteTo(w)
	})
	}

	func main() {
	router := httprouter.New()

	// Assign http.Handlers that return a JSON-encoded problem object
	router.NotFound = problemHandler(http.StatusNotFound)
	router.MethodNotAllowed = problemHandler(http.StatusMethodNotAllowed)

	router.GET("/hello/:name", func(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
	fmt.Fprintf(w, "hello, %s!\n", ps.ByName("name"))
	})

	log.Fatal(http.ListenAndServe(":8080", router))
	}