Golang 實(shí)現(xiàn)隨機(jī)負(fù)載均衡

標(biāo)簽：

Go 面試

Golang 实现随机负载均衡

负载均衡简介

从设备角度实现负载均衡：

硬件负载均衡：

由专门的 负载均衡器 服务提供商，在你的服务器和外部网络之间架设服务商的 负载均衡器。

服务商提供设备和方案帮你处理流量的均衡，不过因为比较费钱包国内没什么人用
软件负载均衡：

包括家喻户晓的Nginx，LVS，Tengine（阿里版Nginx）。

优点就是成本比较低，但需要运维去配置、维护。会踩坑，但国内都在用基本上网找就有解决方案。

随机负载均衡

无论是Nginx，LVS 均衡负载的核心在于 网络 io 复用，负载均衡算法。其中均衡算法又分为 随机，轮询，加权轮询，哈希，而随机负载均衡算法就是本文的重点。

随机负载均衡：意味没有规律，随机在服务器队列中获得一台服务器处理请求。其代码实现如下：

package main

import (
	"fmt"
	"math/rand"
)

// 接口定义
type LoadBalance interface {
	//选择一个后端Server
	//参数remove是需要排除选择的后端Server
	Next(remove []string) *Server
	//更新可用Server列表
	UpdateServers(servers []*Server)
}

// 后端Server定义
type Server struct {
	//主机地址
	Host string
	//主机名
	Name string
	Id int
	//主机是否在线
	Online bool
}

type LoadBalanceRandom struct{
	servers []*Server
}

// 实例化 随机均衡负载
func NewLoadBalanceRandom(servers []*Server) *LoadBalanceRandom{
	newBalance := &LoadBalanceRandom{}
	newBalance.UpdateServers(servers)
	return newBalance
}

//选择一个后端Server
func (r *LoadBalanceRandom) Next() *Server {
	if len(r.servers) == 0 {
		return nil
	}

	curIndex := rand.Intn(len(r.servers))
	return r.servers[curIndex]
}

func (r *LoadBalanceRandom) Get(key string) (*Server, error) {
	return r.Next(), nil
}

//系统运行过程中，后端可用Server会更新
func (this *LoadBalanceRandom) UpdateServers(servers []*Server) {
	newServers:=make([]*Server,0)
	for _,e:=range servers {
		if e.Online==true {
			newServers=append(newServers,e)
		}
	}
	this.servers=newServers
}

测试随机负载均衡

func main() {
	count:=make([]int,4)
	servers:=make([]*Server,0)
	servers=append(servers,&Server{Host:"1",Id:0,Online:true})
	servers=append(servers,&Server{Host:"2",Id:1,Online:true})
	servers=append(servers,&Server{Host:"3",Id:2,Online:true})
	servers=append(servers,&Server{Host:"4",Id:3,Online:true})
	lb:=NewLoadBalanceRandom(servers)

	// 创建4个Server，随机选择100000次。查看4台机器 被选中次数
	for i:=0;i<100000;i++{
		c:=lb.Next()
		count[c.Id]++
	}
	fmt.Println(count)
}