go-lb provides a general load balancers library.
The stocked algorithm are:
- random
- round-robin
- weighted round-robin
- consistent hash
- weighted random
- weighted versioning
Use Register(...)/Unregister(...) to add the balancer with your algorithm and use it with our New(algorithm, opts...).
-
v0.5.1
- upgrade deps
-
v0.5.0
-
a tiny logger interface has been embedded. So
-
we removed the dep to
hedzr/logand free you from it -
you may still enable internal logging sentences in
hedzr/lbby setting up a custom Loggerimport "github.com/hedzr/lb/pkg/logger" logger.SetLogger(yoursLogger)
-
-
all codes reviewed
-
-
v0.3.3
- needs go modules 1.17 and higher
- upgraded deps
- remove unecessary deps
- remove the nest go.mod since it cannot work any more
- fix gin vuln report
-
v0.3.1, v0.3.0 : work for below go1.17
package main
import (
"fmt"
lb "github.com/hedzr/lb"
"github.com/hedzr/lb/lbapi"
)
func main() {
b := lb.New(lb.RoundRobin)
b.Add(exP("172.16.0.7:3500"), exP("172.16.0.8:3500"), exP("172.16.0.9:3500"))
sum := make(map[lbapi.Peer]int)
for i := 0; i < 300; i {
p, _ := b.Next(lbapi.DummyFactor)
sum[p]
}
for k, v := range sum {
fmt.Printf("%v: %v\n", k, v)
}
}
type exP string
func (s exP) String() string { return string(s) }With the Weighted versioning algorithm, a set of version constraints and weights can be put as the basic rule. Such as:
- "<= 1.1.x", weight: 2,
- "^1.2.x", weight: 4,
- "^2.x", weight: 11,
- ^3.x", weight: 3,
the peers will be picked out by its version. And, all versioning peers will be picked with weights specified in constraints totally.
The partial codes:
var testConstraints = []lbapi.Constrainable{
version.NewConstrainablePeer("<= 1.1.x", 2),
version.NewConstrainablePeer("^1.2.x", 4),
version.NewConstrainablePeer("^2.x", 11),
version.NewConstrainablePeer("^3.x", 3),
}
func TestVersionWRR2(t *testing.T) {
lb := version.New(version.WithConstrainedPeers(testConstraints...))
sum := make(map[lbapi.Peer]int)
hits := make(map[lbapi.Peer]map[lbapi.Constrainable]bool)
factor := initFactors()
for i := 0; i < 500; i {
peer, c := lb.Next(factor)
sum[peer]
if ps, ok := hits[peer]; ok {
if _, ok := ps[c]; !ok {
ps[c] = true
}
} else {
hits[peer] = make(map[lbapi.Constrainable]bool)
hits[peer][c] = true
}
}
// results
total := 0
for _, v := range sum {
total = v
}
for peer, v := range sum {
var keys []string
var w int
for c := range hits[peer] {
if kk, ok := c.(fmt.Stringer); ok {
keys = append(keys, kk.String())
}
if ww, ok := c.(lbapi.Weighted); ok {
w = ww.Weight()
}
}
// ex := findC(peer)
t.Logf("%v: %v/%0.2f%%/w:%v. [%v => weight: %v]",
peer, v, (float32(v)/float32(total))*100.0, w,
strings.Join(keys, ","), w)
}
}See the full codes at version/new_test.go,
For the full document of version constraints: Masterminds/semver .
Please check out the source codes:
And the command line to test its:
go run ./_examples/svc-small -port 8111 &
go run ./_examples/svc-small -port 8112 &
go run ./_examples/svc-small -port 8113 &
go run ./_examples/gw-lb-aware/main.go 8111 8112 8113 &
for ((i=0;i<5;i )); do curl http://localhost:8103/ ; doneMIT