Merge pull request #8 from golobby/any-method

implement any method

Author: miladrahimi

radix.go

@@ -81,7 +81,9 @@ func (t *tree) searchByParts(parent *node, parts []string, position int, paramet
 				return child
 			}
 
-			return t.searchByParts(child, parts, position+1, parameters)
+			if node := t.searchByParts(child, parts, position+1, parameters); node != nil {
+				return node
+			}
 		}
 	}
 
@@ -106,16 +108,16 @@ func (t *tree) searchByName(node *node, name string) *node {
 }
 
 // insert adds a new route to the radix tree by recursive traversing.
-func (t *tree) insert(parent, node *node, parts []string, index int) {
-	isLeaf := index == len(parts)-1
+func (t *tree) insert(parent, node *node, parts []string, position int) {
+	isLeaf := position == len(parts)-1
 
 	for i, child := range parent.Children {
-		if child.content == parts[index] {
+		if child.content == parts[position] {
 			if isLeaf {
 				node.Children = parent.Children[i].Children
 				parent.Children[i] = node
 			} else {
-				t.insert(child, node, parts, index+1)
+				t.insert(child, node, parts, position+1)
 			}
 			return
 		}
@@ -124,9 +126,9 @@ func (t *tree) insert(parent, node *node, parts []string, index int) {
 	if isLeaf {
 		parent.Children = append(parent.Children, node)
 	} else {
-		newNode := newNode(nil, parts[index])
+		newNode := newNode(nil, parts[position])
 		parent.Children = append(parent.Children, newNode)
-		t.insert(newNode, node, parts, index+1)
+		t.insert(newNode, node, parts, position+1)
 	}
 }
 

router.go

@@ -86,6 +86,11 @@ func (r Router) Serve(rw http.ResponseWriter, request *http.Request) {
 	r.director.ServeHTTP(rw, request)
 }
 
+// Any maps a method-agnostic Route.
+func (r Router) Any(path string, handler Handler) *Route {
+	return r.Map(":__METHOD__", path, handler)
+}
+
 // GET maps a GET Route.
 func (r Router) GET(path string, handler Handler) *Route {
 	return r.Map("GET", path, handler)

router_test.go

@@ -91,13 +91,19 @@ func TestRouter_With_Different_HTTP_Methods(t *testing.T) {
 	r.HEAD("/", handler)
 	r.OPTIONS("/", handler)
 	r.Map("CUSTOM", "/", handler)
+	r.Any("/any", handler)
 
 	methods := []string{"GET", "POST", "PUT", "PATCH", "DELETE", "HEAD", "OPTIONS", "CUSTOM"}
 	for _, m := range methods {
 		rw := newResponse()
 		r.Serve(rw, newRequest(m, "/"))
 		assert.Equal(t, 200, rw.status)
 		assert.Equal(t, m, rw.stringBody())
+
+		rw = newResponse()
+		r.Serve(rw, newRequest(m, "/any"))
+		assert.Equal(t, 200, rw.status)
+		assert.Equal(t, m, rw.stringBody())
 	}
 
 	rw := newResponse()