code migration

This commit is contained in:
landaiqing
2024-11-15 02:02:19 +08:00
parent b2d753e832
commit 34c4690f80
85 changed files with 4349 additions and 421 deletions

View File

@@ -0,0 +1,34 @@
package sensitivex
import (
"os"
"path/filepath"
sensitive "github.com/zmexing/go-sensitive-word"
)
func NewSensitive() *sensitive.Manager {
filter, err := sensitive.NewFilter(
sensitive.StoreOption{Type: sensitive.StoreMemory},
sensitive.FilterOption{Type: sensitive.FilterDfa},
)
if err != nil {
panic(err)
}
cwd, err := os.Getwd()
if err != nil {
panic(err)
}
// 加载敏感词库
err = filter.Store.LoadDictPath(
filepath.Join(cwd, "resources/sensitive/", "反动词库.txt"),
filepath.Join(cwd, "resources/sensitive/", "暴恐词库.txt"),
filepath.Join(cwd, "resources/sensitive/", "色情词库.txt"),
filepath.Join(cwd, "resources/sensitive/", "贪腐词库.txt"),
filepath.Join(cwd, "resources/sensitive/", "民生词库.txt"),
)
if err != nil {
panic(err)
}
return filter
}