Fixes performance regression

This commit is contained in:
Manu Mtz-Almeida 2015-06-04 01:54:36 +02:00
parent 21979d6a99
commit 0b043d0886
2 changed files with 22 additions and 1 deletions

3
gin.go
View File

@ -158,7 +158,7 @@ func (engine *Engine) addRoute(method, path string, handlers HandlersChain) {
panic("there must be at least one handler")
}
root := engine.trees.get("method")
root := engine.trees.get(method)
if root == nil {
root = new(node)
engine.trees = append(engine.trees, methodTree{
@ -247,6 +247,7 @@ func (engine *Engine) handleHTTPRequest(context *Context) {
return
}
}
break
}
}

View File

@ -78,6 +78,26 @@ func testRouteNotOK2(method string, t *testing.T) {
assert.Equal(t, w.Code, http.StatusMethodNotAllowed)
}
func TestRouterMethod(t *testing.T) {
router := New()
router.PUT("/hey2", func(c *Context) {
c.String(200, "sup2")
})
router.PUT("/hey", func(c *Context) {
c.String(200, "called")
})
router.PUT("/hey3", func(c *Context) {
c.String(200, "sup3")
})
w := performRequest(router, "PUT", "/hey")
assert.Equal(t, w.Code, 200)
assert.Equal(t, w.Body.String(), "called")
}
func TestRouterGroupRouteOK(t *testing.T) {
testRouteOK("GET", t)
testRouteOK("POST", t)