forked from st3v/go-plugins
/
blacklist.go
111 lines (89 loc) · 1.93 KB
/
blacklist.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
98
99
100
101
102
103
104
105
106
107
108
109
110
111
package blacklist
import (
"math/rand"
"time"
"github.com/micro/go-micro/registry"
"github.com/micro/go-micro/selector"
)
type blacklistSelector struct {
so selector.Options
exit chan bool
bl *blacklist
}
func init() {
rand.Seed(time.Now().Unix())
}
func (r *blacklistSelector) Init(opts ...selector.Option) error {
for _, o := range opts {
o(&r.so)
}
return nil
}
func (r *blacklistSelector) Options() selector.Options {
return r.so
}
func (r *blacklistSelector) Select(service string, opts ...selector.SelectOption) (selector.Next, error) {
sopts := selector.SelectOptions{
Strategy: r.so.Strategy,
}
for _, opt := range opts {
opt(&sopts)
}
// get the service
services, err := r.so.Registry.GetService(service)
if err != nil {
return nil, err
}
// apply the filters
for _, filter := range sopts.Filters {
services = filter(services)
}
// apply the blacklist
services, err = r.bl.Filter(services)
if err != nil {
return nil, err
}
// if there's nothing left, return
if len(services) == 0 {
return nil, selector.ErrNoneAvailable
}
return sopts.Strategy(services), nil
}
func (r *blacklistSelector) Mark(service string, node *registry.Node, err error) {
r.bl.Mark(service, node, err)
}
func (r *blacklistSelector) Reset(service string) {
r.bl.Reset(service)
}
func (r *blacklistSelector) Close() error {
select {
case <-r.exit:
return nil
default:
close(r.exit)
r.bl.Close()
}
return nil
}
func (r *blacklistSelector) String() string {
return "blacklist"
}
func newSelector(opts ...selector.Option) selector.Selector {
sopts := selector.Options{
Strategy: selector.Random,
}
for _, opt := range opts {
opt(&sopts)
}
if sopts.Registry == nil {
sopts.Registry = registry.DefaultRegistry
}
return &blacklistSelector{
so: sopts,
exit: make(chan bool),
bl: newBlacklist(),
}
}
func NewSelector(opts ...selector.Option) selector.Selector {
return newSelector(opts...)
}