2016-05-03 23:06:19 +02:00
|
|
|
package selector
|
|
|
|
|
|
|
|
import (
|
|
|
|
"math/rand"
|
|
|
|
"sync"
|
|
|
|
"time"
|
|
|
|
|
2021-10-12 13:55:53 +02:00
|
|
|
"go-micro.dev/v4/registry"
|
2016-05-03 23:06:19 +02:00
|
|
|
)
|
|
|
|
|
|
|
|
func init() {
|
|
|
|
rand.Seed(time.Now().UnixNano())
|
|
|
|
}
|
|
|
|
|
|
|
|
// Random is a random strategy algorithm for node selection
|
|
|
|
func Random(services []*registry.Service) Next {
|
2019-11-05 19:43:12 +02:00
|
|
|
nodes := make([]*registry.Node, 0, len(services))
|
2016-05-03 23:06:19 +02:00
|
|
|
|
|
|
|
for _, service := range services {
|
|
|
|
nodes = append(nodes, service.Nodes...)
|
|
|
|
}
|
|
|
|
|
|
|
|
return func() (*registry.Node, error) {
|
|
|
|
if len(nodes) == 0 {
|
2016-05-07 01:04:08 +02:00
|
|
|
return nil, ErrNoneAvailable
|
2016-05-03 23:06:19 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
i := rand.Int() % len(nodes)
|
|
|
|
return nodes[i], nil
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// RoundRobin is a roundrobin strategy algorithm for node selection
|
|
|
|
func RoundRobin(services []*registry.Service) Next {
|
2019-12-03 21:59:44 +02:00
|
|
|
nodes := make([]*registry.Node, 0, len(services))
|
2016-05-03 23:06:19 +02:00
|
|
|
|
|
|
|
for _, service := range services {
|
|
|
|
nodes = append(nodes, service.Nodes...)
|
|
|
|
}
|
|
|
|
|
2016-10-29 10:31:32 +02:00
|
|
|
var i = rand.Int()
|
2016-05-03 23:06:19 +02:00
|
|
|
var mtx sync.Mutex
|
|
|
|
|
|
|
|
return func() (*registry.Node, error) {
|
|
|
|
if len(nodes) == 0 {
|
2016-05-07 01:04:08 +02:00
|
|
|
return nil, ErrNoneAvailable
|
2016-05-03 23:06:19 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
mtx.Lock()
|
|
|
|
node := nodes[i%len(nodes)]
|
|
|
|
i++
|
|
|
|
mtx.Unlock()
|
|
|
|
|
|
|
|
return node, nil
|
|
|
|
}
|
|
|
|
}
|