rune

bench_test.go

@@ -23,7 +23,8 @@ func benchRequest(b *testing.B, router http.Handler, r *http.Request) {
 	}
 }
 
-func BenchmarkRouterStatic(b *testing.B) {
+// func BenchmarkRouterStatic(b *testing.B) {
+func BenchmarkXX(b *testing.B) {
 	router := New()
 	router.HandleFunc("/hello", func(w http.ResponseWriter, r *http.Request) {}, "GET,HEAD")
 	r, _ := http.NewRequest("GET", "/hello", nil)

trie.go

@@ -1,6 +1,7 @@
 package violetear
 
 import (
+	"bytes"
 	"errors"
 	"net/http"
 	"strings"
@@ -97,3 +98,24 @@ func (t *Trie) Get(path []string, version string) (trie *Trie, p []string, leaf
 
 	return t, path, false, nil
 }
+
+// Split path by "/"
+func (t *Trie) Split(path string) []string {
+	if path == "" {
+		return []string{"/"}
+	}
+	var key bytes.Buffer
+	for i, rune := range path {
+		if rune == '/' && i > 0 {
+			return []string{key.String(), path[i:]}
+		} else if rune == '*' {
+			return []string{"*"}
+		} else if rune != '/' {
+			key.WriteRune(rune)
+		}
+	}
+	if key.Len() > 0 {
+		return []string{key.String()}
+	}
+	return nil
+}

violetear.go

@@ -250,7 +250,7 @@ func (r *Router) ServeHTTP(w http.ResponseWriter, req *http.Request) {
 	}
 
 	// _ path never empty, defaults to ("/")
-	node, path, leaf, _ := r.routes.Get(r.splitPath(req.URL.Path), version)
+	node, path, leaf, _ := r.routes.Get(r.routes.Split(req.URL.Path), version)
 
 	// h http.Handler
 	h, p := r.match(node, path, leaf, nil, req.Method, version)