cy/goproxy
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

优化负载均衡,当只有一个节点的时候,不再启动健康检查.

Browse Source
This commit is contained in:
arraykeys@gmail.com
2018-09-18 16:08:59 +08:00
parent 08b3715bda
commit 42d18ca1d3
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
utils/lb/lb.go
View File

@ -36,6 +36,7 @@ type Group struct {
lock *sync.Mutex lock *sync.Mutex
last *Backend last *Backend
debug bool debug bool
bks []*Backend
} }
func NewGroup(selectType int, configs BackendsConfig, dr *dnsx.DomainResolver, log *log.Logger, debug bool) Group { func NewGroup(selectType int, configs BackendsConfig, dr *dnsx.DomainResolver, log *log.Logger, debug bool) Group {
@ -68,9 +69,13 @@ func NewGroup(selectType int, configs BackendsConfig, dr *dnsx.DomainResolver, l
dr: dr, dr: dr,
lock: &sync.Mutex{}, lock: &sync.Mutex{},
debug: debug, debug: debug,
bks: bks,
} }
} }
func (g *Group) Select(srcAddr string, onlyHa bool) (addr string) { func (g *Group) Select(srcAddr string, onlyHa bool) (addr string) {
if len(g.bks) == 1 {
return g.bks[0].Address
}
if onlyHa { if onlyHa {
g.lock.Lock() g.lock.Lock()
defer g.lock.Unlock() defer g.lock.Unlock()