Java實現負載均衡算法--輪詢和加權輪詢


1.普通輪詢算法

輪詢(Round Robin,RR)是依次將用戶的訪問請求,按循環順序分配到web服務節點上,從1開始到最后一台服務器節點結束,然后再開始新一輪的循環。這種算法簡單,但是沒有考慮到每台節點服務器的具體性能,請求分發往往不均衡。

代碼實現:

/**
 * 普通輪詢算法
 */
public class RoundRobin {
    private static Integer index = 0;
    private static List<String> nodes = new ArrayList<>();
    // 記錄輪詢輸出結果
    private static StringBuffer stringBuffer = new StringBuffer();
    // 准備模擬數據
    static {
        nodes.add("192.168.1.101");
        nodes.add("192.168.1.103");
        nodes.add("192.168.1.102");
        System.out.println("普通輪詢算法的所有節點:"+nodes);//打印所有節點
    }

    // 關鍵代碼
    public String selectNode(){
        String ip = null;
//      之前寫錯的代碼
//      synchronized (index){
        synchronized (RoundRobin.class){
            // 下標復位
            if(index>=nodes.size()) index = 0;
            ip = nodes.get(index);
            stringBuffer.append(Thread.currentThread().getName()+"==獲取節點:"+ ip +"\n");
            index++;
        }
        return ip;
    }

    // 並發測試:兩個線程循環獲取節點
    public static void main(String[] args) throws InterruptedException {
        new Thread(() -> {
            RoundRobin roundRobin1 = new RoundRobin();
            for (int i=1;i<=21;i++){
                roundRobin1.selectNode();
                try {
                    // 模擬業務處理耗時
                    Thread.sleep(10);
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }
        }).start();
        new Thread(() -> {
            RoundRobin roundRobin1 = new RoundRobin();
            for (int i=1;i<=21;i++){
                roundRobin1.selectNode();
                try {
                    // 模擬業務處理耗時
                    Thread.sleep(10);
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }
        }).start();
        // 等待輪詢完成,統一輸出
        Thread.sleep(3000);
        System.out.println(stringBuffer.toString());
    }
}

執行結果:不同線程訪問,結果依舊是按順序循環分配節點

普通輪詢算法的所有節點:[192.168.1.101, 192.168.1.103, 192.168.1.102]

Thread-0==獲取節點:192.168.1.101

Thread-1==獲取節點:192.168.1.103

Thread-1==獲取節點:192.168.1.102

Thread-0==獲取節點:192.168.1.101

Thread-1==獲取節點:192.168.1.103

Thread-0==獲取節點:192.168.1.102

2.加權輪詢算法

加權輪詢(Weighted Round Robin,WRR)是根據設定的權重值來分配訪問請求,權重值越大的,被分到的請求數也就越多。一般根據每台節點服務器的具體性能來分配權重。

2.1.實現方式一

將需要輪詢的所有節點按權重數循環生成一個List 集合,然后就跟普通輪詢算法一樣,來一個、分配一個、進1位。

例如:

所有節點信息:{{“192.168.1.100“,5},{“192.168.1.101“,1},{“192.168.1.102“,3}}

那么生成的List 集合為:

{“192.168.1.100“,

“192.168.1.100“,

“192.168.1.100“,

“192.168.1.100“,

“192.168.1.100“,

“192.168.1.101“,

“192.168.1.102“,

“192.168.1.102“,

“192.168.1.102“}

后面就是普通輪詢算法的邏輯

代碼實現:

類似於二維數組 降維成 一維數組,然后使用普通輪詢

/**
 *  簡單版的加權輪詢
 */public class WeightedRoundRobinSimple {
    private static Integer index = 0;
    private static Map<String,Integer> mapNodes = new HashMap<>();
    // 記錄輪詢輸出結果
    private static StringBuffer stringBuffer = new StringBuffer();

    // 准備模擬數據
    static {
        mapNodes.put("192.168.1.101",1);
        mapNodes.put("192.168.1.102",3);
        mapNodes.put("192.168.1.103",2);
        /* -- 以下代碼只為了方便查看所有節點,刪除不影響 -- S */
        List<String> nodes = new ArrayList<>();
        Iterator<Map.Entry<String, Integer>> iterator = mapNodes.entrySet().iterator();
        while (iterator.hasNext()){
            Map.Entry<String, Integer> entry = iterator.next();
            String key = entry.getKey();
            for (int i=0;i<entry.getValue();i++){
                nodes.add(key);
            }
        }
        System.out.println("簡單版的加權輪詢:"+nodes);//打印所有節點
        /* -- 以上代碼只為了方便查看所有節點,刪除不影響-- E */
    }

    // 關鍵代碼:類似於二維數組 降維成 一維數組,然后使用普通輪詢
    public String selectNode(){
        List<String> nodes = new ArrayList<>();
        Iterator<Map.Entry<String, Integer>> iterator = mapNodes.entrySet().iterator();
        while (iterator.hasNext()){
            Map.Entry<String, Integer> entry = iterator.next();
            String key = entry.getKey();
            for (int i=0;i<entry.getValue();i++){
                nodes.add(key);
            }
        }
        String ip = null;
//      之前寫錯的代碼
//      synchronized (index){
        synchronized (WeightedRoundRobinSimple.class){
            // 下標復位
            if(index>=nodes.size()) index = 0;
            ip = nodes.get(index);
            stringBuffer.append(Thread.currentThread().getName()+"==獲取節點:"+ ip +"\n");
            index++;
        }
        return ip;
    }

    // 並發測試:兩個線程循環獲取節點
    public static void main(String[] args) throws InterruptedException {
        new Thread(() -> {
            WeightedRoundRobinSimple roundRobin1 = new WeightedRoundRobinSimple();
            for (int i=1;i<=6;i++){
                roundRobin1.selectNode();
                try {
                    // 模擬業務處理耗時
                    Thread.sleep(10);
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }
        }).start();
        new Thread(() -> {
            WeightedRoundRobinSimple roundRobin1 = new WeightedRoundRobinSimple();
            for (int i=1;i<=6;i++){
                roundRobin1.selectNode();
                try {
                    // 模擬業務處理耗時
                    Thread.sleep(10);
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }
        }).start();

        // 等待輪詢完成,統一輸出
        Thread.sleep(3000);
        System.out.println(stringBuffer.toString());
    }
}

執行結果:兩個線程循環測試,輸出結果會出現交替分配到不同的IP,但最終的效果都是一個個按順序分配,類似於普通輪詢算法。

簡單版的加權輪詢:[192.168.1.103, 192.168.1.103, 192.168.1.101, 192.168.1.102, 192.168.1.102, 192.168.1.102]

Thread-0==獲取節點:192.168.1.103

Thread-1==獲取節點:192.168.1.103

Thread-1==獲取節點:192.168.1.101

Thread-0==獲取節點:192.168.1.102

Thread-0==獲取節點:192.168.1.102

Thread-1==獲取節點:192.168.1.102

Thread-0==獲取節點:192.168.1.103

Thread-1==獲取節點:192.168.1.103

Thread-1==獲取節點:192.168.1.101

Thread-0==獲取節點:192.168.1.102

Thread-0==獲取節點:192.168.1.102

Thread-1==獲取節點:192.168.1.102

2.2.實現方式二(重點難點)

本文的重點難點。

在實現方式一的算法中可以很明顯的看到,同權重的IP會被連續分配,也就是說同一個IP在短時間內收到不同的請求,過了這個連續點,就要等到下一輪才會被分配到,並沒有做到均勻分配節點。

實現方式二將盡可能地均勻分配每個節點,節點分配不再是連續的,但最終的權重比和上一個方式一樣,這種加權輪詢又被稱為平滑加權輪詢。

理解關鍵的幾個參數和算法邏輯,方便理解代碼的實現。

2.2.1.概述

關鍵參數

  • ip:負載IP
  • weight:權重,保存配置的權重
  • effectiveWeight:有效權重,輪詢的過程權重可能變化
  • currentWeight:當前權重,比對該值大小獲取節點

注意幾個點:

weight 權重,在整個過程不會對它做修改,只用來保存配置時的權重參數值。如果直接拿weight 運算而不保存配置的最原始權重參數,那么將會丟失最關鍵的用戶配置的權重參數。

effectiveWeight 有效權重,在整個過程可能會變化,初始值等於weight,主要用於當節點出現分配失敗時降低權重值,成功時提高權重值(但不能大於weight值),本案例為了簡化算法,並未加入這功能,因此本案例中effectiveWeight始終等於weight。

currentWeight 當前權重,通過循環所有節點比對該值大小來分配權重最大的節點,初始值等於weight。

三個權重參數的變化情況

僅僅針對本案例,因為本案例為了簡化算法,並未加入[節點出現分配失敗時降低權重值,成功時提高權重值(但不能大於weight值)的功能],所以有效權重effectiveWeight 不會發生變化。

  • 第一次加權輪詢時:currentWeight = weight = effectiveWeight;
  • 后面每次加權輪詢時:currentWeight 的值都會不斷變化,weight 和effectiveWeight 的值不變;
  • 被分配的節點的currentWeight = currentWeight - 權重之和
  • 所有節點的currentWeight = currentWeight + effectiveWeight

2.2.2.舉個例子理解算法

你面前有三個瓶子A、B、C,分別裝有1L、3L、2L水。

第一輪分配情況:B多,所以把B瓶子的3L水,分1L給A,分2L給C(按權重分),分完之后:A、B、C分別為:2L、0L、4L

第二輪分配情況:C多,所以把C瓶子的4L水,分1L給A,分3L給B(按權重分),分完之后:A、B、C分別為:3L、3L、0L

第三輪分配情況:A和B一樣多,那么拿誰去分呢?拿誰其實都一樣(算法中寫了A大於B才選A,現在等於,所以不選A),所以把B瓶子的3L水,分1L給A,分2L給C(按權重分),分完之后:A、B、C分別為:4L、0L、2L

然后不斷的進行下去……

簡化成數學邏輯(代碼實現)的關鍵兩步

  • 被分配的節點的currentWeight = currentWeight - 權重之和
  • 所有節點的currentWeight = currentWeight + effectiveWeight

下面通過閱讀代碼來理解

2.2.3.代碼實現

節點對象

/**
 * String ip:負載IP
 * final Integer weight:權重,保存配置的權重
 * Integer effectiveWeight:有效權重,輪詢的過程權重可能變化
 * Integer currentWeight:當前權重,比對該值大小獲取節點
 *   第一次加權輪詢時:currentWeight = weight = effectiveWeight
 *   后面每次加權輪詢時:currentWeight 的值都會不斷變化,其他權重不變
 */public class Node implements Comparable<Node>{
    private String ip;
    private final Integer weight;
    private Integer effectiveWeight;
    private Integer currentWeight;

    public Node(String ip,Integer weight){
        this.ip = ip;
        this.weight = weight;
        this.effectiveWeight = weight;
        this.currentWeight = weight;
    }

    public Node(String ip, Integer weight, Integer effectiveWeight, Integer currentWeight) {
        this.ip = ip;
        this.weight = weight;
        this.effectiveWeight = effectiveWeight;
        this.currentWeight = currentWeight;
    }

    public String getIp() {
        return ip;
    }

    public void setIp(String ip) {
        this.ip = ip;
    }

    public Integer getWeight() {
        return weight;
    }

    public Integer getEffectiveWeight() {
        return effectiveWeight;
    }

    public void setEffectiveWeight(Integer effectiveWeight) {
        this.effectiveWeight = effectiveWeight;
    }

    public Integer getCurrentWeight() {
        return currentWeight;
    }

    public void setCurrentWeight(Integer currentWeight) {
        this.currentWeight = currentWeight;
    }

    @Override
    public int compareTo(Node node) {
        return currentWeight > node.currentWeight ? 1 : (currentWeight.equals(node.currentWeight) ? 0 : -1);
    }

    @Override
    public String toString() {
        return "{ip='" + ip + "', weight=" + weight + ", effectiveWeight=" + effectiveWeight + ", currentWeight=" + currentWeight + "}";
    }
}

加權輪詢算法

/**
 * 加權輪詢算法
 */public class WeightedRoundRobin {

    private static List<Node> nodes = new ArrayList<>();
    // 權重之和
    private static Integer totalWeight = 0;
    // 准備模擬數據
    static {
        nodes.add(new Node("192.168.1.101",1));
        nodes.add(new Node("192.168.1.102",3));
        nodes.add(new Node("192.168.1.103",2));
        nodes.forEach(node -> totalWeight += node.getEffectiveWeight());
    }

    /**
     * 按照當前權重(currentWeight)最大值獲取IP
     * @return Node
     */
    public Node selectNode(){
        if (nodes ==null || nodes.size()<=0) return null;
        if (nodes.size() == 1)  return nodes.get(0);

        Node nodeOfMaxWeight = null; // 保存輪詢選中的節點信息
//      之前寫錯的代碼
//      synchronized (nodes){
        synchronized (WeightedRoundRobin.class){
            // 打印信息對象:避免並發時打印出來的信息太亂,不利於觀看結果
            StringBuffer sb = new StringBuffer();
            sb.append(Thread.currentThread().getName()+"==加權輪詢--[當前權重]值的變化:"+printCurrentWeight(nodes));

            // 選出當前權重最大的節點
            Node tempNodeOfMaxWeight = null;
            for (Node node : nodes) {
                if (tempNodeOfMaxWeight == null)
                    tempNodeOfMaxWeight = node;
                else
                    tempNodeOfMaxWeight = tempNodeOfMaxWeight.compareTo(node) > 0 ? tempNodeOfMaxWeight : node;
            }
            // 必須new個新的節點實例來保存信息,否則引用指向同一個堆實例,后面的set操作將會修改節點信息
            nodeOfMaxWeight = new Node(tempNodeOfMaxWeight.getIp(),tempNodeOfMaxWeight.getWeight(),tempNodeOfMaxWeight.getEffectiveWeight(),tempNodeOfMaxWeight.getCurrentWeight());

            // 調整當前權重比:按權重(effectiveWeight)的比例進行調整,確保請求分發合理。
            tempNodeOfMaxWeight.setCurrentWeight(tempNodeOfMaxWeight.getCurrentWeight() - totalWeight);
            sb.append(" -> "+printCurrentWeight(nodes));

            nodes.forEach(node -> node.setCurrentWeight(node.getCurrentWeight()+node.getEffectiveWeight()));

            sb.append(" -> "+printCurrentWeight(nodes));
            System.out.println(sb); //打印權重變化過程
        }
        return nodeOfMaxWeight;
    }

    // 格式化打印信息
    private String printCurrentWeight(List<Node> nodes){
        StringBuffer stringBuffer = new StringBuffer("[");
        nodes.forEach(node -> stringBuffer.append(node.getCurrentWeight()+",") );
        return stringBuffer.substring(0, stringBuffer.length() - 1) + "]";
    }

    // 並發測試:兩個線程循環獲取節點
    public static void main(String[] args){
        Thread thread = new Thread(() -> {
            WeightedRoundRobin weightedRoundRobin1 = new WeightedRoundRobin();
            for(int i=1;i<=totalWeight;i++){
                Node node = weightedRoundRobin1.selectNode();
                System.out.println(Thread.currentThread().getName()+"==第"+i+"次輪詢選中[當前權重最大]的節點:" + node + "\n");
            }
        });
        thread.start();
        //
        WeightedRoundRobin weightedRoundRobin2 = new WeightedRoundRobin();
        for(int i=1;i<=totalWeight;i++){
            Node node = weightedRoundRobin2.selectNode();
            System.out.println(Thread.currentThread().getName()+"==第"+i+"次輪詢選中[當前權重最大]的節點:" + node + "\n");
        }

    }
}

執行結果:

main==加權輪詢--[當前權重]值的變化:[1,3,2] -> [1,-3,2] -> [2,0,4] main==第1次輪詢選中[當前權重最大]的節點:{ip='192.168.1.102', weight=3, effectiveWeight=3, currentWeight=3}

Thread-0==加權輪詢--[當前權重]值的變化:[2,0,4] -> [2,0,-2] -> [3,3,0] Thread-0==第1次輪詢選中[當前權重最大]的節點:{ip='192.168.1.103', weight=2, effectiveWeight=2, currentWeight=4}

main==加權輪詢--[當前權重]值的變化:[3,3,0] -> [3,-3,0] -> [4,0,2] main==第2次輪詢選中[當前權重最大]的節點:{ip='192.168.1.102', weight=3, effectiveWeight=3, currentWeight=3}

main==加權輪詢--[當前權重]值的變化:[4,0,2] -> [-2,0,2] -> [-1,3,4] main==第3次輪詢選中[當前權重最大]的節點:{ip='192.168.1.101', weight=1, effectiveWeight=1, currentWeight=4}

Thread-0==加權輪詢--[當前權重]值的變化:[-1,3,4] -> [-1,3,-2] -> [0,6,0] Thread-0==第2次輪詢選中[當前權重最大]的節點:{ip='192.168.1.103', weight=2, effectiveWeight=2, currentWeight=4}

main==加權輪詢--[當前權重]值的變化:[0,6,0] -> [0,0,0] -> [1,3,2] main==第4次輪詢選中[當前權重最大]的節點:{ip='192.168.1.102', weight=3, effectiveWeight=3, currentWeight=6}

Thread-0==加權輪詢--[當前權重]值的變化:[1,3,2] -> [1,-3,2] -> [2,0,4] Thread-0==第3次輪詢選中[當前權重最大]的節點:{ip='192.168.1.102', weight=3, effectiveWeight=3, currentWeight=3}

main==加權輪詢--[當前權重]值的變化:[2,0,4] -> [2,0,-2] -> [3,3,0] main==第5次輪詢選中[當前權重最大]的節點:{ip='192.168.1.103', weight=2, effectiveWeight=2, currentWeight=4}

Thread-0==加權輪詢--[當前權重]值的變化:[3,3,0] -> [3,-3,0] -> [4,0,2] Thread-0==第4次輪詢選中[當前權重最大]的節點:{ip='192.168.1.102', weight=3, effectiveWeight=3, currentWeight=3}

main==加權輪詢--[當前權重]值的變化:[4,0,2] -> [-2,0,2] -> [-1,3,4] main==第6次輪詢選中[當前權重最大]的節點:{ip='192.168.1.101', weight=1, effectiveWeight=1, currentWeight=4}

Thread-0==加權輪詢--[當前權重]值的變化:[-1,3,4] -> [-1,3,-2] -> [0,6,0] Thread-0==第5次輪詢選中[當前權重最大]的節點:{ip='192.168.1.103', weight=2, effectiveWeight=2, currentWeight=4}

Thread-0==加權輪詢--[當前權重]值的變化:[0,6,0] -> [0,0,0] -> [1,3,2] Thread-0==第6次輪詢選中[當前權重最大]的節點:{ip='192.168.1.102', weight=3, effectiveWeight=3, currentWeight=6}

為了方便分析,簡化兩線程執行后的結果

[當前權重]值的變化:[1,3,2] -> [1,-3,2] -> [2,0,4]

[當前權重]值的變化:[2,0,4] -> [2,0,-2] -> [3,3,0]

[當前權重]值的變化:[3,3,0] -> [3,-3,0] -> [4,0,2]

[當前權重]值的變化:[4,0,2] -> [-2,0,2] -> [-1,3,4]

[當前權重]值的變化:[-1,3,4] -> [-1,3,-2] -> [0,6,0]

[當前權重]值的變化:[0,6,0] -> [0,0,0] -> [1,3,2]

[當前權重]值的變化:[1,3,2] -> [1,-3,2] -> [2,0,4]

[當前權重]值的變化:[2,0,4] -> [2,0,-2] -> [3,3,0]

[當前權重]值的變化:[3,3,0] -> [3,-3,0] -> [4,0,2]

[當前權重]值的變化:[4,0,2] -> [-2,0,2] -> [-1,3,4]

[當前權重]值的變化:[-1,3,4] -> [-1,3,-2] -> [0,6,0]

[當前權重]值的變化:[0,6,0] -> [0,0,0] -> [1,3,2]

因為整個過程只有當前權重發生變化,所以分析清楚它就明白了整個過程。

結論:

分配完成后當前權重發生變化,但權限之和還是等於最初值

每6輪(1+3+2權重)就出現權重全部為0,所以會出現重新循環,6正好等於權重之和,權重比等於1/6 : 3/6 : 2/6;

a=權重1,b=權重3,c=權重2,那么權重變化的6(a+b+c)次中,分配情況為:b c b a c b,很明顯,每個節點均勻按權重分配,節點分配不再是連續的。這也是最重要的結論,正是實現方式二在文初提到的要實現的關鍵點。

該算法在權重比相差很大時,比如:A=1,B=5,那這個算法的結果就跟方式一沒啥區別了,分配結果就變成了:{A,B,B,B,B,B},既然沒區別,那根據算法復雜情況,那肯定方式一更好了,所以方式一和方式二可以互補,可以根據權重比選擇不同的算法。

留下懸念

第一點:節點出現分配失敗時降低有效權重值,成功時提高有效權重值(但不能大於weight值)的功能。理解了方式二,后面再加這塊功能進去就很好理解了;

第二點:該算法實現的背后數學證明,用的是什么數學理論?

image

Java往期文章

Java 可重入鎖的那些事(一)

Java中的線程安全與線程同步

Java線程狀態(生命周期)--一篇入魂

自己編寫平滑加權輪詢算法,實現反向代理集群服務的平滑分配

Java實現平滑加權輪詢算法--降權和提權

Java實現負載均衡算法--輪詢和加權輪詢

Java全棧學習路線、學習資源和面試題一條龍

image


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM