/go-domain-suffix-trie

安全开发基础组件：域名后缀树（Golang实现）

Primary LanguageGoMIT LicenseMIT

域名后缀树（Golang）

一、什么是域名后缀树

类似于字典后缀树，不同的是域名后缀树是以.切分域名的各个部分， 对域名中的每个部分作为一个Node建立后缀树以便高效进行后缀匹配查询。

比如：

www.google.com

会以.分割域名为三个部分，每个部分建立一个节点：

再增加一个：

baidu.com

此时后缀树是这样子的：

二、业务场景举例

比如现在有n个域名后缀，称之为集合A：

google.com
api.baidu.com
007.qq.com

然后有m个域名，称之为集合B：

a.google.com
b.google.com
c.google.com
google.com
google3.com
a.api.baidu.com
b.api.baidu.com
003.qq.com
a.007.qq.com

现在要为这集合B中的每个域名从集合A中做后缀匹配，这个工具类就是用来解决这个问题的，尤其是在海量子域名关联到根域名上时效率极高。

三、Example

添加此项目作为依赖：

go get -u github.com/golang-infrastructure/go-domain-suffix-trie

代码示例（DomainSuffixTree是线程安全的）：

package main

import (
	"fmt"
	domain_suffix_trie "github.com/golang-infrastructure/go-domain-suffix-trie"
)

func main() {

	// 调用 #NewDomainSuffixTrie 创建一颗后缀树
	tree := domain_suffix_trie.NewDomainSuffixTrie[string]()

	// 将需要匹配的域名后缀依次调用 #AddDomainSuffix 添加到树上，添加的时候可以为后缀指定一个payload（使用集合A构建树）
	_ = tree.AddDomainSuffix("google.com", "谷歌主站子域名")
	_ = tree.AddDomainSuffix("map.google.com", "谷歌地图子域名")
	_ = tree.AddDomainSuffix("baidu.com", "百度主站子域名")
	_ = tree.AddDomainSuffix("jd.com", "京东子域名")

	// 需要查询的时候调用 #FindMatchDomainSuffixPayload 或者 #FindMatchDomainSuffixNode 查询，
	// 参数是一个完整的域名，会返回此域名匹配到的后缀在之前指定的payload（将集合B的每个元素依次在树上查询）
	fmt.Println(tree.FindMatchDomainSuffixPayload("test.google.com"))               // output: 谷歌主站子域名
	fmt.Println(tree.FindMatchDomainSuffixPayload("test.map.google.com"))           // output: 谷歌地图子域名
	fmt.Println(tree.FindMatchDomainSuffixNode("test.baidu.com").GetNodeTriePath()) // output: baidu.com
	fmt.Println(tree.FindMatchDomainSuffixNode("test.jd.com").GetNodeTrieValue())   // output: jd
}