120 lines
2.3 KiB
Go
120 lines
2.3 KiB
Go
package router
|
|
|
|
import (
|
|
"errors"
|
|
"net/http"
|
|
"strings"
|
|
)
|
|
|
|
type route struct {
|
|
method string
|
|
path string
|
|
callback http.HandlerFunc
|
|
}
|
|
|
|
type segment struct {
|
|
path string
|
|
methods map[string]http.HandlerFunc
|
|
children map[string]*segment
|
|
}
|
|
|
|
// Router is the main router object that keeps track of an looks up routes.
|
|
type Router struct {
|
|
routes []route
|
|
lookup *segment
|
|
}
|
|
|
|
// NewRouter is a constructor for Router.
|
|
func NewRouter() (r Router) {
|
|
return
|
|
}
|
|
|
|
// AddRoute adds a new route with a corresponding callback to the router.
|
|
func (r *Router) AddRoute(method string, path string, callback http.HandlerFunc) (err error) {
|
|
keys := setupKeys(strings.Split(path, "/"))
|
|
if r.lookup == nil {
|
|
r.lookup = &segment{}
|
|
r.lookup.children = map[string]*segment{}
|
|
r.lookup.methods = map[string]http.HandlerFunc{}
|
|
r.lookup.path = "/"
|
|
}
|
|
|
|
curr := r.lookup
|
|
|
|
for i, key := range keys {
|
|
if i == 0 {
|
|
continue
|
|
}
|
|
|
|
var seg segment
|
|
|
|
if child, ok := curr.children[key]; !ok {
|
|
seg = *newSegment(curr.path, key)
|
|
curr.children[key] = &seg
|
|
curr = &seg
|
|
} else {
|
|
curr = child
|
|
}
|
|
}
|
|
|
|
if _, ok := curr.methods[method]; ok {
|
|
err = errors.New("path already exists")
|
|
}
|
|
|
|
if err == nil {
|
|
curr.methods[method] = callback
|
|
r.routes = append(r.routes, route{method, path, callback})
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
func addSegment(curr *segment, keys []string) (seg *segment) {
|
|
for _, key := range keys {
|
|
if child, ok := curr.children[key]; !ok {
|
|
seg = newSegment(curr.path, key)
|
|
curr.children[key] = seg
|
|
curr = seg
|
|
} else {
|
|
curr = child
|
|
}
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
func setupKeys(slice []string) (clean []string) {
|
|
clean = append(clean, "/")
|
|
for _, v := range slice {
|
|
if v != "" {
|
|
clean = append(clean, "/"+v)
|
|
}
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
// Get is a convinience method which calls Router.AddRoute with the "GET" method.
|
|
func (r *Router) Get(path string, callback http.HandlerFunc) {
|
|
r.AddRoute(http.MethodGet, path, callback)
|
|
}
|
|
|
|
func (r Router) ServeHTTP(w http.ResponseWriter, req *http.Request) {
|
|
return
|
|
}
|
|
|
|
func newSegment(parentPath string, key string) (seg *segment) {
|
|
var path string
|
|
if parentPath == "/" {
|
|
path = key
|
|
} else {
|
|
path = parentPath + key
|
|
}
|
|
seg = &segment{}
|
|
seg.children = map[string]*segment{}
|
|
seg.methods = map[string]http.HandlerFunc{}
|
|
seg.path = path
|
|
|
|
return
|
|
}
|