Andreas/siyuan
1
0
Fork 0
mirror of https://github.com/siyuan-note/siyuan.git synced 2025-12-29 12:58:48 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

⚡ 使用 AC 算法优化虚拟引用匹配性能

Browse source
This commit is contained in:
Liang Ding 2023-02-17 15:25:59 +08:00
parent 5f72a7b2cd
commit 72631ad0e9
No known key found for this signature in database
GPG key ID: 136F30F901A2231D
3 changed files with 8 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

9
kernel/model/virutalref.go
View file

@ -27,7 +27,7 @@ import (
"github.com/88250/lute"
"github.com/88250/lute/ast"
"github.com/88250/lute/parse"
ahocorasick "github.com/BobuSumisu/aho-corasick"
ahocorasick "github.com/ClarkThan/ahocorasick"
"github.com/dgraph-io/ristretto"
"github.com/siyuan-note/siyuan/kernel/search"
"github.com/siyuan-note/siyuan/kernel/sql"
@ -82,10 +82,11 @@ func putBlockVirtualRefKeywords(blockContent, blockID, docTitle string) (ret []s
}
}
trie := ahocorasick.NewTrieBuilder().AddStrings(keywordsTmp).Build()
hits := trie.MatchString(contentTmp)
m := ahocorasick.NewMatcher()
m.BuildWithPatterns(keywordsTmp)
hits := m.Search(contentTmp)
for _, hit := range hits {
ret = append(ret, hit.MatchString())
ret = append(ret, hit)
}
if 1 > len(ret) {