https://github.com/golang-infrastructure/go-domain-suffix-trie

安全开发基础组件：域名后缀树（Golang实现）
https://github.com/golang-infrastructure/go-domain-suffix-trie

go golang security-development suffix-tree utils-library

Last synced: about 2 months ago
JSON representation

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

• Host: GitHub
• URL: https://github.com/golang-infrastructure/go-domain-suffix-trie
• Owner: golang-infrastructure
• License: mit
• Created: 2022-07-07T15:10:51.000Z (over 2 years ago)
• Default Branch: main
• Last Pushed: 2022-11-21T06:40:21.000Z (about 2 years ago)
• Last Synced: 2024-06-21T14:40:12.593Z (7 months ago)
• Topics: go, golang, security-development, suffix-tree, utils-library
• Language: Go
• Homepage:
• Size: 78.1 KB
• Stars: 35
• Watchers: 2
• Forks: 10
• Open Issues: 2
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

        
# 域名后缀树（Golang）

## 一、什么是域名后缀树

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

比如：

```text

www.google.com

```

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

![](.README_images/5f9cf648.png)

再增加一个：

```text

baidu.com

```

此时后缀树是这样子的：

![](.README_images/b7910457.png)

## 二、业务场景举例

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

```text

google.com

api.baidu.com

007.qq.com

```

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

```text

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

添加此项目作为依赖：

```text

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

```

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

```go

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

}

```