分享
  1. 首页
  2. 文章

Golang之实现一个负载均衡算法(随机,轮询)

pyyu · · 3179 次点击 · · 开始浏览
这是一个创建于 的文章,其中的信息可能已经有所发展或是发生改变。

代码记录

程序结构目录

--------程序包

package balance
type Balancer interface {
 DoBalance([]*Instance, ...string) (*Instance, error)
}
balance.go
package balance
import (
 "strconv"
)
type Instance struct {
 host string
 port int
}
func NewInstance(host string, port int) *Instance {
 return &Instance{
 host: host,
 port: port,
 }
}
//定义Instance结构体的方法GetHost()
func (p *Instance) GetHost() string {
 return p.host
}
//定义方法GetPort()
func (p *Instance) GetPort() int {
 return p.port
}
func (p *Instance) String() string {
 return p.host + ":" + strconv.Itoa(p.port)
}
instance.go
package balance
import "fmt"
type BalanceMgr struct {
 allBalancer map[string]Balancer
}
var mgr = BalanceMgr{
 allBalancer: make(map[string]Balancer),
}
func (p *BalanceMgr) registerBalancer(name string, b Balancer) {
 p.allBalancer[name] = b
}
func RegisterBalancer(name string, b Balancer) {
 mgr.registerBalancer(name, b)
}
func DoBalance(name string, insts []*Instance) (inst *Instance, err error) {
 balancer, ok := mgr.allBalancer[name]
 if !ok {
 err = fmt.Errorf("Not found %s balancer", name)
 return
 }
 fmt.Printf("use %s balancer\n", name)
 inst, err = balancer.DoBalance(insts)
 return
}
mgr.go
package balance
import (
 "errors"
 "math/rand"
)
func init() {
 RegisterBalancer("random", &RandomBalance{})
}
type RandomBalance struct {
}
func (p *RandomBalance) DoBalance(insts []*Instance, key ...string) (inst *Instance, err error) {
 if len(insts) == 0 {
 err = errors.New("No instance")
 return
 }
 lens := len(insts)
 index := rand.Intn(lens)
 inst = insts[index]
 return
}
random.go
package balance
import (
 "errors"
)
func init() {
 RegisterBalancer("roundrobin", &RoundRobinBalance{})
}
type RoundRobinBalance struct {
 curIndex int
}
func (p *RoundRobinBalance) DoBalance(insts []*Instance, key ...string) (inst *Instance, err error) {
 if len(insts) == 0 {
 err = errors.New("No instance")
 return
 }
 lens := len(insts)
 if p.curIndex >= lens {
 p.curIndex = 0
 }
 inst = insts[p.curIndex]
 p.curIndex = (p.curIndex + 1) % lens
 return
}
roundrobin.go

------入口

package main
import (
 "fmt"
 "go_dev/day7/example/example1/balance"
 "math/rand"
 "os"
 "time"
)
func main() {
 var insts []*balance.Instance
 for i := 0; i < 16; i++ {
 host := fmt.Sprintf("192.168.%d.%d", rand.Intn(255), rand.Intn(255))
 one := balance.NewInstance(host, 8080)
 insts = append(insts, one)
 }
 var balanceName = "random"
 if len(os.Args) > 1 {
 balanceName = os.Args[1]
 }
 for {
 inst, err := balance.DoBalance(balanceName, insts)
 if err != nil {
 fmt.Println("do balance err:", err)
 fmt.Fprintf(os.Stdout, "do balance err\n")
 continue
 }
 fmt.Println(inst)
 time.Sleep(time.Second)
 }
}
main.go
package main
import (
 "fmt"
 "go_dev/day7/example/example1/balance"
 "hash/crc32"
 "math/rand"
)
type HashBalance struct {
 Name string
 Age int
}
func init() {
 balance.RegisterBalancer("hash", &HashBalance{})
}
func (p *HashBalance) DoBalance(insts []*balance.Instance, key ...string) (inst *balance.Instance, err error) {
 var defKey string = fmt.Sprintf("%d", rand.Int())
 if len(key) > 0 {
 defKey = key[0]
 }
 lens := len(insts)
 if lens == 0 {
 err = fmt.Errorf("No backend instance")
 return
 }
 crcTable := crc32.MakeTable(crc32.IEEE)
 hashVal := crc32.Checksum([]byte(defKey), crcTable)
 index := int(hashVal) % lens
 inst = insts[index]
 return
}
hash.go

有疑问加站长微信联系(非本文作者)

本文来自:博客园

感谢作者:pyyu

查看原文:Golang之实现一个负载均衡算法(随机,轮询)

入群交流(和以上内容无关):加入Go大咖交流群,或添加微信:liuxiaoyan-s 备注:入群;或加QQ群:692541889

关注微信
3179 次点击
添加一条新回复 (您需要 后才能回复 没有账号 ?)
  • 请尽量让自己的回复能够对别人有帮助
  • 支持 Markdown 格式, **粗体**、~~删除线~~、`单行代码`
  • 支持 @ 本站用户;支持表情(输入 : 提示),见 Emoji cheat sheet
  • 图片支持拖拽、截图粘贴等方式上传

用户登录

没有账号?注册
或
(追記) (追記ここまで)

今日阅读排行

    加载中
(追記) (追記ここまで)

一周阅读排行

    加载中

关注我

  • 扫码关注领全套学习资料 关注微信公众号
  • 加入 QQ 群:
    • 192706294(已满)
    • 731990104(已满)
    • 798786647(已满)
    • 729884609(已满)
    • 977810755(已满)
    • 815126783(已满)
    • 812540095(已满)
    • 1006366459(已满)
    • 692541889

  • 关注微信公众号
  • 加入微信群:liuxiaoyan-s,备注入群
  • 也欢迎加入知识星球 Go粉丝们(免费)

给该专栏投稿 写篇新文章

每篇文章有总共有 5 次投稿机会

收入到我管理的专栏 新建专栏