/
analysis.go
68 lines (56 loc) · 1.27 KB
/
analysis.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
package video_analysis
import (
"time"
"go.uber.org/fx"
"go.uber.org/zap"
)
type Score interface {
GetKeyType() string
GetScore(value string) int
Reload() error
}
const (
Tag = "tag"
Mid = "mid"
)
type Analysis struct {
logger *zap.Logger
scoreMap map[string]Score
}
func NewAnalysis(logger *zap.Logger, scores ...Score) *Analysis {
analysis := &Analysis{
logger: logger,
scoreMap: map[string]Score{},
}
for _, score := range scores {
analysis.scoreMap[score.GetKeyType()] = score
}
go analysis.tick(time.NewTicker(10 * time.Minute))
return analysis
}
func NewFxAnalysis(logger *zap.Logger, blacklist *Blacklist, tagScore *TagScore) *Analysis {
return NewAnalysis(logger, blacklist, tagScore)
}
func (als *Analysis) Calculate(key, value string) int {
if score, ok := als.scoreMap[key]; ok {
return score.GetScore(value)
}
return 0
}
func (als *Analysis) tick(tk *time.Ticker) {
for {
select {
case <-tk.C:
for k := range als.scoreMap {
als.logger.Info("reload score start", zap.String("key", k))
err := als.scoreMap[k].Reload()
if err != nil {
als.logger.Error("reload score error", zap.String("key", k), zap.Error(err))
}
}
}
}
}
func Provide() fx.Option {
return fx.Provide(NewFxAnalysis, NewTagScore, NewBlacklist)
}