🐞 fix: fix route matching issue, support path with trailing slash

sohaha · Nov 21, 2024 · 3abc61a · 3abc61a
1 parent d647887
commit 3abc61a
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 1 deletion.
diff --git a/znet/utils.go b/znet/utils.go
@@ -205,7 +205,7 @@ func (u utils) TreeFind(t *Tree, path string) (*Engine, handlerFn, []handlerFn,
 		}
 	}
 
-	if len(nodes) == 0 {
+	if len(nodes) == 0 || strings.HasSuffix(path, "/") {
 		res := strings.Split(path, "/")
 		p := ""
 		if len(res) == 1 {

diff --git a/znet/web_test.go b/znet/web_test.go
@@ -159,6 +159,22 @@ func TestWeb(t *testing.T) {
 	})
 	tt.Equal(200, w.Code)
 	tt.Equal(expected, w.Body.String())
+
+	r.GET("/web-get/*", func(c *Context) string {
+		return "web-get:" + c.GetParam("*")
+	})
+
+	w = request(r, "GET", "/web-get/", nil)
+	tt.Equal(200, w.Code)
+	tt.Equal("web-get:", w.Body.String())
+
+	w = request(r, "GET", "/web-get/1", nil)
+	tt.Equal(200, w.Code)
+	tt.Equal("web-get:1", w.Body.String())
+
+	w = request(r, "GET", "/web-get/a", nil)
+	tt.Equal(200, w.Code)
+	tt.Equal("web-get:a", w.Body.String())
 }
 
 func TestMoreMethod(t *testing.T) {