Trie 树,是一种树形结构,是一种哈希树的变种。典型应用是用于统计,排序和保存大量的字符串(但不仅限于字符串),所以经常被搜索引擎系统用于文本词频统计。它的优点是:利用字符串的公共前缀来减少查询时间,最大限度地减少无谓的字符串比较,查询效率比哈希树高。
type Trie func New() *Trie func (trie *Trie) GetNode(v string) (ok bool, node *Node) func (trie *Trie) Has(pattern string) bool func (trie *Trie) Match(v string) (bool, *Result) func (trie *Trie) Put(pattern string, value interface{}) error func (trie *Trie) SetDelimeter(delimeter string)
基于该 Trie 数实现的 HTTP router
type Context func NewContent(rw http.ResponseWriter, r *http.Request) *Context func (context *Context) ParamInt(key string, d ...int) (int, error) func (context *Context) ParamString(key string, d ...string) string func (context *Context) WriteString(v string) type Handler func NewHanlder() *Handler func (handler *Handler) Delete(handleFunc func(*Context)) func (handler *Handler) DoDelete(context *Context) func (handler *Handler) DoGet(context *Context) func (handler *Handler) DoPatch(context *Context) func (handler *Handler) DoPost(context *Context) func (handler *Handler) DoPut(context *Context) func (handler *Handler) Get(handleFunc func(*Context)) func (handler *Handler) Patch(handleFunc func(*Context)) func (handler *Handler) Post(handleFunc func(*Context)) func (handler *Handler) Put(handleFunc func(*Context)) type HandlerInterface type Router func New() *Router func (router *Router) After(pattern string, midwares ...func(context *Context)) func (router *Router) Before(pattern string, midwares ...func(*Context)) func (router *Router) Delete(pattern string, handlefunc func(*Context)) func (router *Router) Get(pattern string, handlefunc func(*Context)) func (router *Router) Patch(pattern string, handlefunc func(*Context)) func (router *Router) Post(pattern string, handlefunc func(*Context)) func (router *Router) Put(pattern string, handlefunc func(*Context)) func (router *Router) Router(pattern string, handler HandlerInterface) func (router *Router) ServeHTTP(rw http.ResponseWriter, r *http.Request)
示例
package main import ( "fmt" "net/http" "os" "github.com/importcjj/trie.go/router" ) func Helloworld(ctx *router.Context) { ctx.WriteString("hello, world!") } func ParamHandler(ctx *router.Context) { username := ctx.ParamString("username") text := fmt.Sprintf("hi, %s", username) ctx.WriteString(text) } var PageResource = &router.Handler{ OnGet: func(ctx *router.Context) { filepath := ctx.ParamString("filepath") text := fmt.Sprintf("Get page %s", filepath) ctx.WriteString(text) }, OnPost: func(ctx *router.Context) { filepath := ctx.ParamString("filepath") text := fmt.Sprintf("Post page %s", filepath) ctx.WriteString(text) }, } // BasicAuth is a Midwares func BasicAuth(ctx *router.Context) { fmt.Fprintln(os.Stderr, ctx.Request.URL, "Call Basic Auth.") } // BeforeMetric mark a time point when the request start. func BeforeMetric(ctx *router.Context) { // just a example, so use the params map to // record the time. ctx.Params["time"] = time.Now().Format("Mon Jan 2 15:04:05 -0700 MST 2006") } // AfterMetric log the time spent to handle the requeset. func AfterMetric(ctx *router.Context) { start, _ := time.Parse("Mon Jan 2 15:04:05 -0700 MST 2006", ctx.Params["time"]) dur := time.Since(start) fmt.Fprintf(os.Stderr, "%s spent", dur.String()) } var r = router.New() func init() { r.Get("/hello/world", Helloworld) r.Get("/hi/<username:str>", ParamHandler) // restful api style, this pattern can match such as // "/page/hi.html" "/page/static/inde.html" eta. r.Router("/page/<filepath:*>", PageResource) r.Before("/", BasicAuth, BeforeMetric) r.After("/", AfterMetric) } func main() { server := &http.Server{ Addr: ":8080", Handler: r, } server.ListenAndServe() }
项目地址: https://github.com/importcjj/trie.go
没有花很多时间,没有进行彻底测试。不过简单测试了一下,没有发现什么问题,性能也还过的去。如果可以的话,希望大家去试试,提一点意见。如果觉得写得可以的话,欢迎 star 和 fork 哦!然后,基于该 HTTP router 可以实现一个简单的 web 框架(懒得搞了)。有兴趣的同学可以试试。