79 lines
2.0 KiB
Go
79 lines
2.0 KiB
Go
package balance
|
||
|
||
import (
|
||
"hpds_control_center/model"
|
||
)
|
||
|
||
// WeightRoundRobinBalance 加权轮询负载
|
||
type WeightRoundRobinBalance struct {
|
||
curIndex int
|
||
rss []*WeightNode
|
||
rsw []int
|
||
}
|
||
|
||
type WeightNode struct {
|
||
node model.NodeLastStateItem
|
||
Weight int //初始化时对节点约定的权重
|
||
currentWeight int //节点临时权重，每轮都会变化
|
||
effectiveWeight int //有效权重, 默认与weight相同 , totalWeight = sum(effectiveWeight) //出现故障就-1
|
||
}
|
||
|
||
//1, currentWeight = currentWeight + effectiveWeight
|
||
//2, 选中最大的currentWeight节点为选中节点
|
||
//3, currentWeight = currentWeight - totalWeight
|
||
|
||
func (r *WeightRoundRobinBalance) Add(params model.NodeLastStateItem) error {
|
||
weightCpu := params.CpuUsed
|
||
weightMem := float64(params.MemUsed*100) / float64(params.MemUsed)
|
||
weightSwap := float64(params.SwapTotal*100) / float64(params.SwapTotal)
|
||
weightTask := 0.0
|
||
if len(params.ExecTask) > 0 {
|
||
weightTask = 0
|
||
} else {
|
||
weightTask = 100
|
||
}
|
||
node := &WeightNode{
|
||
node: params,
|
||
Weight: int(400 - weightCpu - weightMem - weightSwap - weightTask),
|
||
}
|
||
node.effectiveWeight = node.Weight
|
||
r.rss = append(r.rss, node)
|
||
return nil
|
||
}
|
||
|
||
func (r *WeightRoundRobinBalance) Next() int64 {
|
||
var best *WeightNode
|
||
total := 0
|
||
for i := 0; i < len(r.rss); i++ {
|
||
w := r.rss[i]
|
||
//1 计算所有有效权重
|
||
total += w.effectiveWeight
|
||
//2 修改当前节点临时权重
|
||
w.currentWeight += w.effectiveWeight
|
||
//3 有效权重默认与权重相同，通讯异常时-1, 通讯成功+1，直到恢复到weight大小
|
||
if w.effectiveWeight < w.Weight {
|
||
w.effectiveWeight++
|
||
}
|
||
|
||
//4 选中最大临时权重节点
|
||
if best == nil || w.currentWeight > best.currentWeight {
|
||
best = w
|
||
}
|
||
}
|
||
|
||
if best == nil {
|
||
return 0
|
||
}
|
||
//5 变更临时权重为 临时权重-有效权重之和
|
||
best.currentWeight -= total
|
||
return best.node.NodeId
|
||
}
|
||
|
||
func (r *WeightRoundRobinBalance) Get(int64) (int64, error) {
|
||
return r.Next(), nil
|
||
}
|
||
|
||
func (r *WeightRoundRobinBalance) Update() {
|
||
|
||
}
|