43 lines
1.2 KiB
Go
43 lines
1.2 KiB
Go
package go_sensitive_word
|
||
|
||
import (
|
||
"errors"
|
||
"github.com/zmexing/go-sensitive-word/filter"
|
||
"github.com/zmexing/go-sensitive-word/store"
|
||
)
|
||
|
||
// Manager 是敏感词过滤系统的核心结构，整合了词库存储和过滤算法
|
||
type Manager struct {
|
||
store.Store // // 词库存储接口（支持内存、本地文件、远程等）
|
||
filter.Filter // // 敏感词匹配算法接口（如 DFA）
|
||
}
|
||
|
||
// NewFilter 初始化过滤器和词库存储
|
||
// 参数：storeOption 指定存储方式，filterOption 指定过滤算法
|
||
func NewFilter(storeOption StoreOption, filterOption FilterOption) (*Manager, error) {
|
||
var filterStore store.Store
|
||
var myFilter filter.Filter
|
||
|
||
switch storeOption.Type {
|
||
case StoreMemory: // 使用内存词库
|
||
filterStore = store.NewMemoryModel()
|
||
default:
|
||
return nil, errors.New("invalid store type")
|
||
}
|
||
|
||
switch filterOption.Type {
|
||
case FilterDfa: // 使用 DFA 算法
|
||
dfaModel := filter.NewDfaModel()
|
||
// 启动监听协程，实时接收新增/删除词的通知
|
||
go dfaModel.Listen(filterStore.GetAddChan(), filterStore.GetDelChan())
|
||
myFilter = dfaModel
|
||
default:
|
||
return nil, errors.New("invalid filter type")
|
||
}
|
||
|
||
return &Manager{
|
||
Store: filterStore,
|
||
Filter: myFilter,
|
||
}, nil
|
||
}
|