local
/
router
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

custom not found

master
Bel LaPointe 2021-04-20 05:49:09 -05:00
parent 5deb0e5f0d
commit b77f3987c9
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
router.go
View File

@ -6,12 +6,14 @@ import (
) )
type Router struct { type Router struct {
t *tree t *tree
NotFound http.HandlerFunc
} }
func New() *Router { func New() *Router {
return &Router{ return &Router{
t: newTree(), t: newTree(),
NotFound: http.NotFound,
} }
} }
@ -22,7 +24,7 @@ func (rt *Router) Add(path string, foo http.HandlerFunc) error {
func (rt *Router) ServeHTTP(w http.ResponseWriter, r *http.Request) { func (rt *Router) ServeHTTP(w http.ResponseWriter, r *http.Request) {
foo := rt.t.Lookup(r.URL.Path) foo := rt.t.Lookup(r.URL.Path)
if foo == nil { if foo == nil {
http.NotFound(w, r) rt.NotFound(w, r)
return return
} }
foo(w, r) foo(w, r)