1
0
mirror of https://github.com/labstack/echo.git synced 2024-12-24 20:14:31 +02:00
Signed-off-by: Vishal Rana <vr@labstack.com>
This commit is contained in:
Vishal Rana 2015-04-13 21:57:36 -07:00
parent 604d9411fd
commit 90ff3c14b2
2 changed files with 47 additions and 49 deletions

View File

@ -11,11 +11,11 @@ type (
label byte
prefix string
parent *node
edges edges
children children
handler HandlerFunc
echo *Echo
}
edges []*node
children []*node
param struct {
Name string
Value string
@ -31,7 +31,7 @@ func NewRouter(e *Echo) (r *router) {
for _, m := range methods {
r.trees[m] = &node{
prefix: "",
edges: edges{},
children: children{},
}
}
return
@ -75,8 +75,8 @@ func (r *router) insert(method, path string, h HandlerFunc, echo *Echo) {
}
} else if l < pl {
// Split node
n := newNode(cn.prefix[l:], cn, cn.edges, cn.handler, cn.echo)
cn.edges = edges{n} // Add to parent
n := newNode(cn.prefix[l:], cn, cn.children, cn.handler, cn.echo)
cn.children = children{n} // Add to parent
// Reset parent node
cn.label = cn.prefix[0]
@ -90,20 +90,20 @@ func (r *router) insert(method, path string, h HandlerFunc, echo *Echo) {
cn.echo = echo
} else {
// Create child node
n = newNode(search[l:], cn, edges{}, h, echo)
cn.edges = append(cn.edges, n)
n = newNode(search[l:], cn, children{}, h, echo)
cn.children = append(cn.children, n)
}
} else if l < sl {
search = search[l:]
e := cn.findEdge(search[0])
if e != nil {
c := cn.findChild(search[0])
if c != nil {
// Go deeper
cn = e
cn = c
continue
}
// Create child node
n := newNode(search, cn, edges{}, h, echo)
cn.edges = append(cn.edges, n)
n := newNode(search, cn, children{}, h, echo)
cn.children = append(cn.children, n)
} else {
// Node already exists
if h != nil {
@ -115,22 +115,22 @@ func (r *router) insert(method, path string, h HandlerFunc, echo *Echo) {
}
}
func newNode(pfx string, p *node, e edges, h HandlerFunc, echo *Echo) (n *node) {
func newNode(pfx string, p *node, c children, h HandlerFunc, echo *Echo) (n *node) {
n = &node{
label: pfx[0],
prefix: pfx,
parent: p,
edges: e,
children: c,
handler: h,
echo: echo,
}
return
}
func (n *node) findEdge(l byte) *node {
for _, e := range n.edges {
if e.label == l {
return e
func (n *node) findChild(l byte) *node {
for _, c := range n.children {
if c.label == l {
return c
}
}
return nil
@ -152,6 +152,7 @@ func (r *router) Find(method, path string, params Params) (h HandlerFunc, echo *
cn := r.trees[method] // Current node as root
search := path
n := 0 // Param count
c := new(node) // Child node
// Search order static > param > catch-all
for {
@ -162,30 +163,27 @@ func (r *router) Find(method, path string, params Params) (h HandlerFunc, echo *
return
}
var e *node
pl := len(cn.prefix)
l := lcp(search, cn.prefix)
if l == pl {
search = search[l:]
} else if l < pl {
if cn.label != ':' {
} else if l < pl && cn.label != ':' {
goto Up
}
}
// Static node
e = cn.findEdge(search[0])
if e != nil {
cn = e
c = cn.findChild(search[0])
if c != nil {
cn = c
continue
}
// Param node
Param:
e = cn.findEdge(':')
if e != nil {
cn = e
c = cn.findChild(':')
if c != nil {
cn = c
i, l := 0, len(search)
for ; i < l && search[i] != '/'; i++ {
}
@ -197,9 +195,9 @@ func (r *router) Find(method, path string, params Params) (h HandlerFunc, echo *
}
// Catch-all node
e = cn.findEdge('*')
if e != nil {
cn = e
c = cn.findChild('*')
if c != nil {
cn = c
p := params[:n+1]
p[n].Name = "_name"
p[n].Value = search

View File

@ -531,14 +531,14 @@ func (n *node) printTree(pfx string, tail bool) {
p := prefix(tail, pfx, "└── ", "├── ")
fmt.Printf("%s%s, %p: parent=%p, handler=%v, echo=%v\n", p, n.prefix, n, n.parent, n.handler, n.echo)
nodes := n.edges
l := len(nodes)
children := n.children
l := len(children)
p = prefix(tail, pfx, " ", "│ ")
for i := 0; i < l-1; i++ {
nodes[i].printTree(p, false)
children[i].printTree(p, false)
}
if l > 0 {
nodes[l-1].printTree(p, true)
children[l-1].printTree(p, true)
}
}