-
Notifications
You must be signed in to change notification settings - Fork 1
/
selector_byid.go
108 lines (91 loc) · 2.02 KB
/
selector_byid.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
/*
@Time : 2018/8/24 14:22
@Author : kenny zhu
@File : selector_byid.go
@Software: GoLand
@Others: Implements multi-database for route services...
*/
package router
import (
"github.com/micro/go-micro/selector"
"math/rand"
"time"
"github.com/micro/go-micro/registry"
"sync"
)
// Built in random hashed node selector
// One datasource must be named...
type idSelector struct {
opts selector.Options
}
var (
// All server the same tag in the same set..
datacenter = "dbrouter"
)
func init() {
rand.Seed(time.Now().Unix())
}
func (n *idSelector) Init(opts ...selector.Option) error {
for _, o := range opts {
o(&n.opts)
}
return nil
}
func (n *idSelector) Options() selector.Options {
return n.opts
}
// Select directly not use other Strategy
func (n *idSelector) Select(service string, opts ...selector.SelectOption) (selector.Next, error) {
// Default is ConsulRegistry.
services, err := n.opts.Registry.GetService(service)
if err != nil {
return nil, err
}
if len(services) == 0 {
return nil, selector.ErrNotFound
}
var nodes []*registry.Node
// Filter the nodes for datacenter marked by the server..
for _, service := range services {
for _, node := range service.Nodes {
if node.Metadata["datacenter"] == datacenter {
nodes = append(nodes, node)
}
}
}
if len(nodes) == 0 {
return nil, selector.ErrNotFound
}
var i int
var mtx sync.Mutex
// Round bin..
return func() (*registry.Node, error) {
mtx.Lock()
defer mtx.Unlock()
i++
return nodes[i%len(nodes)], nil
}, nil
}
func (n *idSelector) Mark(service string, node *registry.Node, err error) {
return
}
func (n *idSelector) Reset(service string) {
return
}
func (n *idSelector) Close() error {
return nil
}
func (n *idSelector) String() string {
return datacenter
}
// Return a new first node selector
func IDSelector(opts ...selector.Option) selector.Selector {
var sopts selector.Options
for _, opt := range opts {
opt(&sopts)
}
if sopts.Registry == nil {
sopts.Registry = registry.DefaultRegistry
}
return &idSelector{sopts}
}