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

Perf: Aho-Corasick node uses extends

Browse Source
This commit is contained in:
SukkaW 2024-11-05 20:31:00 +08:00
parent 5c3b86bce0
commit 7b6b14b1dc
1 changed files with 15 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
Build/lib/aho-corasick.ts
View File

@ -1,20 +1,14 @@
const WORDEND = Symbol('wordEnd');
const FAIL = Symbol('fail');
type Node = Map<string, Node> & {
[WORDEND]: boolean,
[FAIL]: Node | undefined
};
function createNode(): Node {
const node = new Map<string, Node | undefined>() as Node;
node[WORDEND] = false;
node[FAIL] = undefined;
return node;
class Node extends Map<string, Node> {
constructor(
public wordEnd: boolean,
public fail: Node | undefined
) {
super();
}
}
function createKeywordFilter(keys: string[] | Set<string>) {
const root = createNode();
const root = new Node(false, undefined);
// Create a trie with extra fields and information
const put = (key: string) => {
@ -28,7 +22,7 @@ function createKeywordFilter(keys: string[] | Set<string>) {
if (node.has(char)) {
node = node.get(char)!;
} else {
const newNode = createNode();
const newNode = new Node(false, undefined);
node.set(char, newNode);
node = newNode;
}
@ -36,7 +30,7 @@ function createKeywordFilter(keys: string[] | Set<string>) {
// If a new node is created, mark it as a word end when loop finish
if (node !== root) {
node[WORDEND] = true;
node.wordEnd = true;
}
};
@ -49,13 +43,13 @@ function createKeywordFilter(keys: string[] | Set<string>) {
const beginNode = queue.pop()!;
beginNode.forEach((node, char) => {
let failNode = beginNode[FAIL];
let failNode = beginNode.fail;
while (failNode && !failNode.has(char)) {
failNode = failNode[FAIL];
failNode = failNode.fail;
}
node[FAIL] = failNode ? failNode.get(char) : root;
node.fail = failNode ? failNode.get(char) : root;
queue.push(node);
});
@ -70,12 +64,12 @@ function createKeywordFilter(keys: string[] | Set<string>) {
const char = text[i];
while (node && !node.has(char)) {
node = node[FAIL];
node = node.fail;
}
node = node ? node.get(char)! : root;
if (node[WORDEND]) {
if (node.wordEnd) {
return true;
}
}