代码拉取完成,页面将自动刷新
/*
普通轮询
*/
package round_robin
import (
"gitee.com/mosache/YFrame/micro/loadbalance"
"google.golang.org/grpc/balancer"
"google.golang.org/grpc/balancer/base"
"sync/atomic"
)
type Builder struct {
*loadbalance.Builder
}
func (b *Builder) Build(info base.PickerBuildInfo) balancer.Picker {
p := b.Builder.Build(info).(*Balancer)
return p
}
type Balancer struct {
index uint64
length uint64
loadbalance.Picker
}
func NewBalancer(nodes []*loadbalance.NodeInfo) *Balancer {
return &Balancer{
index: 0,
length: uint64(len(nodes)),
Picker: loadbalance.Picker{Nodes: nodes},
}
}
func (b *Balancer) Pick(info balancer.PickInfo) (balancer.PickResult, error) {
if b.length == 0 {
return balancer.PickResult{}, balancer.ErrNoSubConnAvailable
}
idx := atomic.AddUint64(&b.index, 1)
return balancer.PickResult{
SubConn: b.Nodes[idx%b.length].SubConn,
}, nil
}
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。