gosora/router.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) {
	router.mu.RLock()
	
	if req.URL.Path[0] != '/' {
		router.mu.RUnlock()
		w.WriteHeader(405)
		w.Write([]byte(""))
		return
	}
	
	// Do something on the path to turn slashes facing the wrong way "\" into "/" slashes. Like what? Wouldn't that be slow? Might need to move to fasthttp for this
	
	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
}