69 lines
1.4 KiB
Go
69 lines
1.4 KiB
Go
package main
|
|
//import "fmt"
|
|
import "strings"
|
|
import "sync"
|
|
import "net/http"
|
|
|
|
type Router struct {
|
|
mu sync.RWMutex
|
|
routes map[string]func(http.ResponseWriter, *http.Request)
|
|
}
|
|
|
|
func NewRouter() *Router {
|
|
return &Router{
|
|
routes: make(map[string]func(http.ResponseWriter, *http.Request)),
|
|
}
|
|
}
|
|
|
|
func (router *Router) Handle(pattern string, handle http.Handler) {
|
|
router.mu.Lock()
|
|
router.routes[pattern] = handle.ServeHTTP
|
|
router.mu.Unlock()
|
|
}
|
|
|
|
func (router *Router) HandleFunc(pattern string, handle func(http.ResponseWriter, *http.Request)) {
|
|
router.mu.Lock()
|
|
router.routes[pattern] = handle
|
|
router.mu.Unlock()
|
|
}
|
|
|
|
func (router *Router) ServeHTTP(w http.ResponseWriter, req *http.Request) {
|
|
if req.URL.Path[0] != '/' {
|
|
w.WriteHeader(405)
|
|
w.Write([]byte(""))
|
|
return
|
|
}
|
|
|
|
router.mu.RLock()
|
|
handle, ok := router.routes[req.URL.Path]
|
|
if ok {
|
|
router.mu.RUnlock()
|
|
handle(w,req)
|
|
return
|
|
}
|
|
|
|
if req.URL.Path[len(req.URL.Path) - 1] == '/' {
|
|
router.mu.RUnlock()
|
|
NotFound(w,req)
|
|
return
|
|
}
|
|
|
|
handle, ok = router.routes[req.URL.Path[:strings.LastIndexByte(req.URL.Path,'/') + 1]]
|
|
if ok {
|
|
router.mu.RUnlock()
|
|
handle(w,req)
|
|
return
|
|
}
|
|
//fmt.Println(req.URL.Path[:strings.LastIndexByte(req.URL.Path,'/')])
|
|
|
|
handle, ok = router.routes[req.URL.Path + "/"]
|
|
if ok {
|
|
router.mu.RUnlock()
|
|
handle(w,req)
|
|
return
|
|
}
|
|
|
|
router.mu.RUnlock()
|
|
NotFound(w,req)
|
|
return
|
|
} |