记录一下动态路由,思路是用Trie树来记录节点,然后用户输入URL的时候,根据Trie树来查询到对应的节点,对应的节点再返回出一个准确的string,由此string找到处理函数。
Trie树的实现和设计
type node struct {
pattern string // 这其实算是一个准确的字符串,只有叶子节点会有
part string // 这就是节点的值,查询URL中的“/”的一部分
children []*node // 子节点
isWild bool // 是否精确匹配,part 含有 : 或 * 时为true
}
路由主要的功能是完成注册和查询,所以需要两个API来提供插入和查找,查找具体可以分为查找所有的子节点和只需要查找到第一个子节点就足够了。
代码的理解上,就是用递归来完成节点的插入,实际上,会有一个根节点来进行最开始的插入,所以height比起层数,其实用长度来length表达会更合适,不然确实有点迷糊。
所以理解起来就是将URL字符串进行拆解后,插入到Trie树中
func (n *node) insert(pattern string, parts []string, height int) {
if len(parts) == height {
n.pattern = pattern
return
}
part := parts[height]
child := n.matchChild(part)
if child == nil {
child = &node{part: part, isWild: part[0] == ':' || part[0] == '*'}
n.children = append(n.children, child)
}
child.insert(pattern, parts, height+1)
}
func (n *node) search(parts []string, height int) *node {
if len(parts) == height || strings.HasPrefix(n.part, "*") {
if n.pattern == "" {
return nil
}
return n
}
part := parts[height]
children := n.matchChildren(part)
for _, child := range children {
result := child.search(parts, height+1)
if result != nil {
return result
}
}
return nil
}
最后两个查找匹配子节点的API
// 第一个匹配成功的节点,用于插入
func (n *node) matchChild(part string) *node {
for _, child := range n.children {
if child.part == part || child.isWild {
return child
}
}
return nil
}
// 所有匹配成功的节点,用于查找
func (n *node) matchChildren(part string) []*node {
nodes := make([]*node, 0)
for _, child := range n.children {
if child.part == part || child.isWild {
nodes = append(nodes, child)
}
}
return nodes
}