這是一個建立於 的文章,其中的資訊可能已經有所發展或是發生改變。
最近在看golang串連mysql資料庫的例子(Go-MySQL-Driver),但是想到串連資料庫肯定會有串連多從庫的需求,而且每個從庫的權重也會不一樣,參考<<權重輪詢調度演算法(Weighted Round-Robin Scheduling)>>這篇文章寫了一個golang版本的權重輪詢調度演算法,請大家指教,不多說了直接上代碼:
package main import ( "fmt" "time" ) var slaveDns = map[int]map[string]interface{}{ 0: {"connectstring": "root@tcp(172.16.0.164:3306)/shiqu_tools?charset=utf8", "weight": 2}, 1: {"connectstring": "root@tcp(172.16.0.165:3306)/shiqu_tools?charset=utf8", "weight": 4}, 2: {"connectstring": "root@tcp(172.16.0.166:3306)/shiqu_tools?charset=utf8", "weight": 8}, } var i int = -1 //表示上一次選擇的伺服器 var cw int = 0 //表示當前調度的權值 var gcd int = 2 //當前所有權重的最大公約數 比如 2,4,8 的最大公約數為:2 func getDns() string { for { i = (i + 1) % len(slaveDns) if i == 0 { cw = cw - gcd if cw <= 0 { cw = getMaxWeight() if cw == 0 { return "" } } } if weight, _ := slaveDns[i]["weight"].(int); weight >= cw { return slaveDns[i]["connectstring"].(string) } } } func getMaxWeight() int { max := 0 for _, v := range slaveDns { if weight, _ := v["weight"].(int); weight >= max { max = weight } } return max } func main() { note := map[string]int{} s_time := time.Now().Unix() for i := 0; i < 100; i++ { s := getDns() fmt.Println(s) if note[s] != 0 { note[s]++ } else { note[s] = 1 } } e_time := time.Now().Unix() fmt.Println("total time: ", e_time-s_time) fmt.Println("--------------------------------------------------") for k, v := range note { fmt.Println(k, " ", v) } }
100次結果是: