平滑的加权轮询

前言
算法描述
代码实现

前言

如何“优雅的轮询”，是生产环境中很常见一种场景，

所谓“优雅”，即在均衡分配的基础上，要保证连续的请求不能连续分配到某一个后端节点上；

可以理解为时间上的连续请求，要尽量分配到物理上的不同节点，避免某段时间内后端单台节点压力过大。

算法描述

1 2	`On each peer selection we increase current_weight of each eligible peer by its weight,` `Select peer with greatest current_weight and reduce its current_weight by total number of weight points distributed among peers.`

回到顶部

代码实现

type Node struct {
    Name    string
    Current int
    Weight  int
}
 
func SmoothWrr(nodes []*Node) (best *Node) {
    if len(nodes) == 0 {
        return
    }
    total := 0
    for _, node := range nodes {
        if node == nil {
            continue
        }
        total += node.Weight
        node.Current += node.Weight
        if best == nil || node.Current > best.Current {
            best = node
        }
    }
    if best == nil {
        return
    }
    best.Current -= total
    return
}
 
func main() {
    nodes := []*Node{
        &Node{"a", 0, 5},
        &Node{"b", 0, 1},
        &Node{"c", 0, 1},
    }
 
    for i := 0; i < 7; i++ {
        best := SmoothWrr(nodes)
        if best != nil {
            fmt.Println(best.Name)
        }
    }
}