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

Perf: avoid mutate string in Trie

Browse Source
This commit is contained in:
SukkaW 2024-12-14 20:24:52 +08:00
parent e53e6b9375
commit 5c85a0c504
1 changed files with 38 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

64
Build/lib/trie.ts
View File

@ -38,12 +38,12 @@ function deepTrieNodeToJSON(node: TrieNode,
const createNode = <Meta = any>(allSubdomain = false, parent: TrieNode | null = null): TrieNode => [false, allSubdomain, parent, new Map<string, TrieNode>(), null] as TrieNode<Meta>; const createNode = <Meta = any>(allSubdomain = false, parent: TrieNode | null = null): TrieNode => [false, allSubdomain, parent, new Map<string, TrieNode>(), null] as TrieNode<Meta>;
export function hostnameToTokens(hostname: string): string[] { export function hostnameToTokens(hostname: string, hostnameFromIndex: number): string[] {
const tokens = hostname.split('.'); const tokens = hostname.split('.');
const results: string[] = []; const results: string[] = [];
let token = ''; let token = '';
for (let i = 0, l = tokens.length; i < l; i++) { for (let i = hostnameFromIndex, l = tokens.length; i < l; i++) {
token = tokens[i]; token = tokens[i];
if (token.length > 0) { if (token.length > 0) {
results.push(token); results.push(token);
@ -53,7 +53,7 @@ export function hostnameToTokens(hostname: string): string[] {
return results; return results;
} }
function walkHostnameTokens(hostname: string, onToken: (token: string) => boolean | null): boolean | null { function walkHostnameTokens(hostname: string, onToken: (token: string) => boolean | null, hostnameFromIndex: number): boolean | null {
const tokens = hostname.split('.'); const tokens = hostname.split('.');
const l = tokens.length - 1; const l = tokens.length - 1;
@ -61,7 +61,7 @@ function walkHostnameTokens(hostname: string, onToken: (token: string) => boolea
// we are at the first of hostname, no splitor there // we are at the first of hostname, no splitor there
let token = ''; let token = '';
for (let i = l; i >= 0; i--) { for (let i = l; i >= hostnameFromIndex; i--) {
token = tokens[i]; token = tokens[i];
if (token.length > 0) { if (token.length > 0) {
const t = onToken(token); const t = onToken(token);
@ -104,7 +104,7 @@ abstract class Triebase<Meta = any> {
} }
} }
public abstract add(suffix: string, includeAllSubdomain?: boolean, meta?: Meta): void; public abstract add(suffix: string, includeAllSubdomain?: boolean, meta?: Meta, hostnameFromIndex?: number): void;
protected walkIntoLeafWithTokens( protected walkIntoLeafWithTokens(
tokens: string[], tokens: string[],
@ -138,6 +138,7 @@ abstract class Triebase<Meta = any> {
protected walkIntoLeafWithSuffix( protected walkIntoLeafWithSuffix(
suffix: string, suffix: string,
hostnameFromIndex: number,
onLoop: (node: TrieNode, parent: TrieNode, token: string) => void = noop onLoop: (node: TrieNode, parent: TrieNode, token: string) => void = noop
) { ) {
let node: TrieNode = this.$root; let node: TrieNode = this.$root;
@ -161,7 +162,7 @@ abstract class Triebase<Meta = any> {
return false; return false;
}; };
if (walkHostnameTokens(suffix, onToken) === null) { if (walkHostnameTokens(suffix, onToken, hostnameFromIndex) === null) {
return null; return null;
} }
@ -169,10 +170,11 @@ abstract class Triebase<Meta = any> {
}; };
public contains(suffix: string, includeAllSubdomain = suffix[0] === '.'): boolean { public contains(suffix: string, includeAllSubdomain = suffix[0] === '.'): boolean {
let hostnameFromIndex = 0;
if (suffix[0] === '.') { if (suffix[0] === '.') {
suffix = suffix.slice(1); hostnameFromIndex = 1;
} }
const res = this.walkIntoLeafWithSuffix(suffix); const res = this.walkIntoLeafWithSuffix(suffix, hostnameFromIndex);
if (!res) return false; if (!res) return false;
if (includeAllSubdomain) return res.node[1]; if (includeAllSubdomain) return res.node[1];
return true; return true;
@ -330,14 +332,15 @@ abstract class Triebase<Meta = any> {
*/ */
public find( public find(
inputSuffix: string, inputSuffix: string,
subdomainOnly = inputSuffix[0] === '.' subdomainOnly = inputSuffix[0] === '.',
hostnameFromIndex = 0
// /** @default true */ includeEqualWithSuffix = true // /** @default true */ includeEqualWithSuffix = true
): string[] { ): string[] {
if (inputSuffix[0] === '.') { if (inputSuffix[0] === '.') {
inputSuffix = inputSuffix.slice(1); hostnameFromIndex = 1;
} }
const inputTokens = hostnameToTokens(inputSuffix); const inputTokens = hostnameToTokens(inputSuffix, hostnameFromIndex);
const res = this.walkIntoLeafWithTokens(inputTokens); const res = this.walkIntoLeafWithTokens(inputTokens);
if (res === null) return []; if (res === null) return [];
@ -346,7 +349,7 @@ abstract class Triebase<Meta = any> {
const onMatches = subdomainOnly const onMatches = subdomainOnly
? (suffix: string[], subdomain: boolean) => { // fast path (default option) ? (suffix: string[], subdomain: boolean) => { // fast path (default option)
const d = fastStringArrayJoin(suffix, '.'); const d = fastStringArrayJoin(suffix, '.');
if (!subdomain && d === inputSuffix) return; if (!subdomain && subStringEqual(inputSuffix, d, 1)) return;
results.push(subdomain ? '.' + d : d); results.push(subdomain ? '.' + d : d);
} }
@ -368,7 +371,7 @@ abstract class Triebase<Meta = any> {
* Method used to delete a prefix from the trie. * Method used to delete a prefix from the trie.
*/ */
public remove(suffix: string): boolean { public remove(suffix: string): boolean {
const res = this.getSingleChildLeaf(hostnameToTokens(suffix)); const res = this.getSingleChildLeaf(hostnameToTokens(suffix, 0));
if (res === null) return false; if (res === null) return false;
if (!res.node[0]) return false; if (!res.node[0]) return false;
@ -392,11 +395,13 @@ abstract class Triebase<Meta = any> {
* Method used to assert whether the given prefix exists in the Trie. * Method used to assert whether the given prefix exists in the Trie.
*/ */
public has(suffix: string, includeAllSubdomain = suffix[0] === '.'): boolean { public has(suffix: string, includeAllSubdomain = suffix[0] === '.'): boolean {
let hostnameFromIndex = 0;
if (suffix[0] === '.') { if (suffix[0] === '.') {
suffix = suffix.slice(1); hostnameFromIndex = 1;
} }
const res = this.walkIntoLeafWithSuffix(suffix); const res = this.walkIntoLeafWithSuffix(suffix, hostnameFromIndex);
if (res === null) return false; if (res === null) return false;
if (!res.node[0]) return false; if (!res.node[0]) return false;
@ -485,12 +490,12 @@ abstract class Triebase<Meta = any> {
export class HostnameSmolTrie<Meta = any> extends Triebase<Meta> { export class HostnameSmolTrie<Meta = any> extends Triebase<Meta> {
public smolTree = true; public smolTree = true;
add(suffix: string, includeAllSubdomain = suffix[0] === '.', meta?: Meta): void { add(suffix: string, includeAllSubdomain = suffix[0] === '.', meta?: Meta, hostnameFromIndex = 0): void {
let node: TrieNode<Meta> = this.$root; let node: TrieNode<Meta> = this.$root;
let curNodeChildren: Map<string, TrieNode<Meta>> = node[3]; let curNodeChildren: Map<string, TrieNode<Meta>> = node[3];
if (suffix[0] === '.') { if (hostnameFromIndex === 0 && suffix[0] === '.') {
suffix = suffix.slice(1); hostnameFromIndex = 1;
} }
const onToken = (token: string) => { const onToken = (token: string) => {
@ -512,7 +517,7 @@ export class HostnameSmolTrie<Meta = any> extends Triebase<Meta> {
}; };
// When walkHostnameTokens returns true, we should skip the rest // When walkHostnameTokens returns true, we should skip the rest
if (walkHostnameTokens(suffix, onToken)) { if (walkHostnameTokens(suffix, onToken, hostnameFromIndex)) {
return; return;
} }
@ -539,12 +544,12 @@ export class HostnameSmolTrie<Meta = any> extends Triebase<Meta> {
node[4] = meta!; node[4] = meta!;
} }
public whitelist(suffix: string, includeAllSubdomain = suffix[0] === '.') { public whitelist(suffix: string, includeAllSubdomain = suffix[0] === '.', hostnameFromIndex = 0) {
if (suffix[0] === '.') { if (suffix[0] === '.') {
suffix = suffix.slice(1); hostnameFromIndex = 1;
} }
const tokens = hostnameToTokens(suffix); const tokens = hostnameToTokens(suffix, hostnameFromIndex);
const res = this.getSingleChildLeaf(tokens); const res = this.getSingleChildLeaf(tokens);
if (res === null) return; if (res === null) return;
@ -579,7 +584,7 @@ export class HostnameTrie<Meta = any> extends Triebase<Meta> {
return this.$size; return this.$size;
} }
add(suffix: string, includeAllSubdomain = suffix[0] === '.', meta?: Meta): void { add(suffix: string, includeAllSubdomain = suffix[0] === '.', meta?: Meta, hostnameFromIndex = 0): void {
let node: TrieNode<Meta> = this.$root; let node: TrieNode<Meta> = this.$root;
const onToken = (token: string) => { const onToken = (token: string) => {
@ -594,12 +599,12 @@ export class HostnameTrie<Meta = any> extends Triebase<Meta> {
return false; return false;
}; };
if (suffix[0] === '.') { if (hostnameFromIndex === 0 && suffix[0] === '.') {
suffix = suffix.slice(1); hostnameFromIndex = 1;
} }
// When walkHostnameTokens returns true, we should skip the rest // When walkHostnameTokens returns true, we should skip the rest
if (walkHostnameTokens(suffix, onToken)) { if (walkHostnameTokens(suffix, onToken, hostnameFromIndex)) {
return; return;
} }
@ -634,3 +639,10 @@ export type Trie = ReturnType<typeof createTrie>;
// } // }
// return true; // return true;
// }; // };
function subStringEqual(needle: string, haystack: string, needleIndex = 0) {
for (let i = 0, l = haystack.length; i < l; i++) {
if (needle[i + needleIndex] !== haystack[i]) return false;
}
return true;
}