Perf: re-use trie when building reject domainsets

This commit is contained in:
SukkaW
2024-05-26 01:20:45 +08:00
parent 02bff12245
commit 5b725192e3
3 changed files with 31 additions and 25 deletions

View File

@@ -1,9 +1,17 @@
import { createTrie } from './trie';
import { createTrie, type Trie } from './trie';
export function domainDeduper(inputDomains: string[] | Trie, toArray?: true): string[];
export function domainDeduper(inputDomains: string[] | Trie, toArray: false): Set<string>;
export function domainDeduper(inputDomains: string[] | Trie, toArray = true): string[] | Set<string> {
let trie: Trie;
if (Array.isArray(inputDomains)) {
trie = createTrie(inputDomains, true, true);
} else if (!inputDomains.hostnameMode || !inputDomains.smolTree) {
throw new Error('Invalid trie');
} else {
trie = inputDomains;
}
export function domainDeduper(inputDomains: string[], toArray?: true): string[];
export function domainDeduper(inputDomains: string[], toArray: false): Set<string>;
export function domainDeduper(inputDomains: string[], toArray = true): string[] | Set<string> {
const trie = createTrie(inputDomains, true, true);
const dumped = trie.dump();
if (toArray) {
return dumped;