GithubFollow/Surge_by_SukkaW
1
0
Fork 0
mirror of https://github.com/SukkaW/Surge.git synced 2025-12-12 09:10:35 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Perf: micro optmize trie

Browse Source
This commit is contained in:
SukkaW 2024-09-23 17:08:33 +08:00
parent af8086f25a
commit 428f30574c
1 changed files with 13 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
Build/lib/trie.ts
View File

@ -92,19 +92,20 @@ export const createTrie = <Meta = any>(from?: string[] | Set<string> | null, smo
const add = smolTree
? (suffix: string, meta?: Meta): void => {
let node: TrieNode<Meta> = root;
let curNodeChildren: Map<string, TrieNode<Meta>> = node[2];
const onToken = (token: string) => {
if (node[2].has(token)) {
node = node[2].get(token)!;
curNodeChildren = node[2];
if (curNodeChildren.has(token)) {
node = curNodeChildren.get(token)!;
// During the adding of `[start]blog|.skk.moe` and find out that there is a `[start].skk.moe` in the trie
// Dedupe the covered subdomain by skipping
if (token === '.' && node[0]) {
// During the adding of `[start]blog|.skk.moe` and find out that there is a `[start].skk.moe` in the trie, skip adding the rest of the node
if (node[0] && token === '.') {
return true;
}
} else {
const newNode = createNode(node);
node[2].set(token, newNode);
curNodeChildren.set(token, newNode);
node = newNode;
}
@ -176,9 +177,9 @@ export const createTrie = <Meta = any>(from?: string[] | Set<string> | null, smo
for (let i = tokens.length - 1; i >= 0; i--) {
token = tokens[i];
if (token === '') {
break;
}
// if (token === '') {
// break;
// }
parent = node;
@ -416,8 +417,9 @@ export const createTrie = <Meta = any>(from?: string[] | Set<string> | null, smo
if (tokens[0] === '.') {
// If there is a `[start]sub.example.com` here, remove it
parent[0] = false;
// Removing all the child nodes by disconnecting ".", which removes "blog.sub.example.com"
parent[2].delete('.');
// Removing all the child nodes by empty the children
// This removes the only child ".", which removes "blog.sub.example.com"
parent[2].clear();
}
// Trying to whitelist `example.com` when there is already a `.example.com` in the trie
@ -425,9 +427,6 @@ export const createTrie = <Meta = any>(from?: string[] | Set<string> | null, smo
if (dotNode) {
dotNode[0] = false;
}
// if (dotNode?.s === true) {
// dotnode[0] = false;
// }
// return early if not found
if (!node[0]) return;