快捷導(dǎo)航

使用Golang實現(xiàn)加權(quán)負載均衡算法的實現(xiàn)代碼

更新時間：2021年09月09日 17:24:16 作者：Mr.YF

這篇文章主要介紹了使用Golang實現(xiàn)加權(quán)負載均衡算法的實現(xiàn)代碼,詳細說明權(quán)重轉(zhuǎn)發(fā)算法的實現(xiàn)，通過實例代碼給大家介紹的非常詳細，對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值，需要的朋友可以參考下

背景描述

如下圖所示，負載均衡做為反向代理，將請求方的請求轉(zhuǎn)發(fā)至后端的服務(wù)節(jié)點，實現(xiàn)服務(wù)的請求。

在nginx中可以通過upstream配置server時，設(shè)置weight表示對應(yīng)server的權(quán)重。

若存在多個服務(wù)節(jié)點時，負載均衡如何通過服務(wù)節(jié)點的權(quán)重進行轉(zhuǎn)發(fā)。

如下詳細說明權(quán)重轉(zhuǎn)發(fā)算法的實現(xiàn)。

用三個后端服務(wù)節(jié)點為例說明

設(shè)置三個后端服務(wù)ServerA，ServerB和ServerC，它們的權(quán)重分布是 5，3，1

按照加權(quán)負載均衡算法，在一輪（5+3+1=9次）中ServerA占5次，ServerB占3次，ServerC占1次，從而實現(xiàn)均衡。

如下圖所示：

為了實現(xiàn)這個功能，可以給每一個后端設(shè)置對應(yīng)的權(quán)重5,3,1

變量1：后端服務(wù)的權(quán)重 Weight

變量2：均衡器累計的總的有效權(quán)重EffectiveWeight

變量3：實時統(tǒng)計后端服務(wù)的當前權(quán)重 CurrentWeight

算法設(shè)計

第一步，向均衡器中增加后端服務(wù)標識

將三個后端服務(wù)標識和權(quán)重Weight增加到負載均衡器列表中。
每次增加后端服務(wù)時，累計總的有效權(quán)重EffectiveWeight。

第二步，每次獲取一個后端服務(wù)標識

對均衡器中的所有后端服務(wù)增加自己的權(quán)重Weight，即(5，3，1),計算ABC三個服務(wù)的當前權(quán)重。
選擇當前權(quán)重CurrentWeight最大的服務(wù)，做為本次期望的后端服務(wù)。
將期望的后端服務(wù)的當前權(quán)重CurrentWeight減小總的權(quán)重EffectiveWeight，供下一輪使用。

如下是一個一輪（5+3+1=9次）獲取的權(quán)重變化表：

從這個表中可以看到后端服務(wù)輪詢的順序是 A B A C A B A B A，其中A出現(xiàn)了5次，B出現(xiàn)了3次，C出現(xiàn)了1次，滿足三個服務(wù)的權(quán)重Weight設(shè)置。

完成9次獲取后，ABC三個服務(wù)的權(quán)重都歸0，因此下一輪的9次獲取也是均衡的，

算法實現(xiàn)

按照如上算法說明，使用Golang實現(xiàn)這個算法如下

package weightroundrobin

import (
    "fmt"
    "strings"
)

// 每一個后端服務(wù)定義
type BackendServer struct {
    // 實例權(quán)重
    Weight int
    // 當前的權(quán)重，初始為Weight
    currentWeight int
    // 后端服務(wù)名稱
    ServerName string
}

// 通過權(quán)重實現(xiàn)調(diào)用輪詢的定義
type WeightServerRoundRobin struct {
    // 所有有效的權(quán)重總和
    effectiveWeight int
    // 后端服務(wù)列表
    backendServerList []*BackendServer
}

// 創(chuàng)建一個負載輪詢器
func NewWeightServerRoundRobin() *WeightServerRoundRobin {
    return &WeightServerRoundRobin{
        effectiveWeight: 0,
    }
}

// 增加后端服務(wù)名稱和權(quán)重
func (r *WeightServerRoundRobin) AddBackendServer(backendServer *BackendServer) {
    r.effectiveWeight += backendServer.Weight
    r.backendServerList = append(r.backendServerList, backendServer)
}

// 更具權(quán)重獲取一個后端服務(wù)名稱
func (r *WeightServerRoundRobin) GetBackendServer() *BackendServer {
    var expectBackendServer *BackendServer
    for _, backendServer := range r.backendServerList {
        // 給每個后端服務(wù)增加自身權(quán)重
        backendServer.currentWeight += backendServer.Weight
        if expectBackendServer == nil {
            expectBackendServer = backendServer
        }
        if backendServer.currentWeight > expectBackendServer.currentWeight {
            expectBackendServer = backendServer
        }
    }
    r.VisitBackendServerCurrentWeight()
    // 把選擇的后端服務(wù)權(quán)重減掉總權(quán)重
    expectBackendServer.currentWeight -= r.effectiveWeight
    return expectBackendServer
}

// 打印后端服務(wù)的當前權(quán)重變化
func (r *WeightServerRoundRobin) VisitBackendServerCurrentWeight() {
    var serverListForLog []string
    for _, backendServer := range r.backendServerList {
        serverListForLog = append(serverListForLog,
            fmt.Sprintf("%v", backendServer.currentWeight))
    }
    fmt.Printf("(%v)\n", strings.Join(serverListForLog, ", "))
}

寫一個單測進行驗證

package weightroundrobin

import (
    "fmt"
    "testing"
)

func TestNewWeightServerRoundRobin(t *testing.T) {
    weightServerRoundRobin := NewWeightServerRoundRobin()
    weightServerRoundRobin.AddBackendServer(&BackendServer{
        ServerName: "ServerA",
        Weight: 5,
    })
    weightServerRoundRobin.AddBackendServer(&BackendServer{
        ServerName: "ServerB",
        Weight: 3,
    })
    weightServerRoundRobin.AddBackendServer(&BackendServer{
        ServerName: "ServerC",
        Weight: 1,
    })

    expectServerNameList := []string{
        "ServerA", "ServerB", "ServerA", "ServerC", "ServerA", "ServerB", "ServerA", "ServerB", "ServerA",
        //"ServerA", "ServerB", "ServerA", "ServerC", "ServerA", "ServerB", "ServerA", "ServerB", "ServerA",
    }
    fmt.Printf("(A, B, C)\n")
    for ii, expectServerName := range expectServerNameList {
        weightServerRoundRobin.VisitBackendServerCurrentWeight()
        backendServer := weightServerRoundRobin.GetBackendServer()
        if backendServer.ServerName != expectServerName {
            t.Errorf("%v.%v.expect:%v, actual:%v", t.Name(), ii, expectServerName, backendServer.ServerName)
            return
        }
    }
}

運行單元測試，觀察運行結(jié)果是否符合算法設(shè)計的預(yù)期

=== RUN TestNewWeightServerRoundRobin
(A, B, C)
(0, 0, 0)
(5, 3, 1)
(-4, 3, 1)
(1, 6, 2)
(1, -3, 2)
(6, 0, 3)
(-3, 0, 3)
(2, 3, 4)
(2, 3, -5)
(7, 6, -4)
(-2, 6, -4)
(3, 9, -3)
(3, 0, -3)
(8, 3, -2)
(-1, 3, -2)
(4, 6, -1)
(4, -3, -1)
(9, 0, 0)
--- PASS: TestNewWeightServerRoundRobin (0.00s)
PASS