-
Notifications
You must be signed in to change notification settings - Fork 6
/
filter.go
97 lines (80 loc) · 2.14 KB
/
filter.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
package berecall
import (
"encoding/json"
"fmt"
"github.com/alibaba/pairec/v2/context"
"github.com/alibaba/pairec/v2/module"
"github.com/alibaba/pairec/v2/recconf"
"github.com/alibaba/pairec/v2/utils"
)
var filterMapping = make(map[string]IBeFilter)
var filterSigns = make(map[string]string)
var filterService *FilterService
func init() {
filterService = &FilterService{}
filterService.Filters = make(map[string][]IBeFilter)
}
type IBeFilter interface {
BuildQueryParams(user *module.User, context *context.RecommendContext) map[string]string
}
type FilterService struct {
Filters map[string][]IBeFilter
}
func (fs *FilterService) AddFilter(scene string, filter IBeFilter) {
// if exist in Filters, direct return
filters, ok := fs.Filters[scene]
if ok {
for _, f := range filters {
if f == filter {
return
}
}
filters = append(filters, filter)
fs.Filters[scene] = filters
} else {
filters := []IBeFilter{filter}
fs.Filters[scene] = filters
}
}
func (fs *FilterService) AddFilters(scene string, filters []IBeFilter) {
fs.Filters[scene] = filters
}
func RegisterFilterWithConfig(config *recconf.RecommendConfig) {
for _, conf := range config.BeFilterConfs {
if _, ok := filterMapping[conf.Name]; ok {
sign, _ := json.Marshal(&conf)
if utils.Md5(string(sign)) == filterSigns[conf.Name] {
continue
}
}
var f IBeFilter
if conf.FilterType == "User2ItemExposureFilter" {
f = NewUser2ItemExposureFilter(conf)
}
if f == nil {
panic("Filter is nil, name:" + conf.Name)
}
sign, _ := json.Marshal(&conf)
registerFilterWithSign(conf.Name, f, utils.Md5(string(sign)))
}
}
func RegisterFilter(name string, filter IBeFilter) {
if filter == nil {
panic("Filter is nil, name:" + name)
}
if _, ok := filterMapping[name]; !ok {
filterMapping[name] = filter
}
}
func registerFilterWithSign(name string, filter IBeFilter, sign string) {
filterMapping[name] = filter
filterSigns[name] = sign
}
// GetFilter get filter by the name
func GetFilter(name string) (IBeFilter, error) {
filter, ok := filterMapping[name]
if !ok {
return nil, fmt.Errorf("befilter not found, name:%s", name)
}
return filter, nil
}