Perf: use new hostname trie

This commit is contained in:
SukkaW 2024-05-11 00:07:43 +08:00
parent 59b86f706f
commit a486910a26
2 changed files with 6 additions and 6 deletions

View File

@ -3,7 +3,7 @@ import { createTrie } from './trie';
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);
const trie = createTrie(inputDomains, true);
const sets = new Set(inputDomains);
for (let i = 0, len1 = inputDomains.length; i < len1; i++) {

View File

@ -28,10 +28,10 @@ const compare = (a: string, b: string) => {
};
export const sortDomains = (inputs: string[], gorhill: PublicSuffixList) => {
const domains = inputs.reduce<Map<string, string | null>>((acc, cur) => {
const domains = inputs.reduce<Map<string, string>>((acc, cur) => {
if (!acc.has(cur)) {
const topD = gorhill.getDomain(cur[0] === '.' ? cur.slice(1) : cur);
acc.set(cur, topD === cur ? null : topD);
acc.set(cur, topD);
};
return acc;
}, new Map());
@ -39,10 +39,10 @@ export const sortDomains = (inputs: string[], gorhill: PublicSuffixList) => {
const sorter = (a: string, b: string) => {
if (a === b) return 0;
const $a = domains.get(a);
const $b = domains.get(b);
const $a = domains.get(a)!;
const $b = domains.get(b)!;
if ($a == null || $b == null) {
if ($a === a && $b === b) {
return compare(a, b);
}
return compare($a, $b) || compare(a, b);