负载均衡 轮询

负载均衡轮询：在一组固定的服务器节点中，按序轮流选择服务节点。

package com.tech.da.loadbalance;

import java.util.concurrent.atomic.AtomicInteger;

/**
 * 负载均衡 轮询
 */
public class Code1_RoundRobin {
    private static final AtomicInteger nextIndex = new AtomicInteger(0);

    public static void main(String[] args) {
        for (int i = 0; i < 10; i++) {
            System.out.println(new Code1_RoundRobin().roundRobin(3));
        }
    }

    /*
    轮询
    serverSize 服务节点数量
    for 用于自旋，当不满足返回条件时进行自旋
    current < serverSize 避免nextIndex初始化设置的值大于服务器节点数量，或在程序运行一段时间serverSize设置成一个小于nextIndex的值，此时需要循环，下次再返回
     */
    int roundRobin(int serverSize) {
        for (; ; ) {
            int current = nextIndex.get();
            int next = (current + 1) % serverSize;
            if (nextIndex.compareAndSet(current, next) && current < serverSize) {
                return current;
            }
        }

    }

}