From 42d18ca1d3e6c1cf373df051445136e6896af7a3 Mon Sep 17 00:00:00 2001 From: "arraykeys@gmail.com" Date: Tue, 18 Sep 2018 16:08:59 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96=E8=B4=9F=E8=BD=BD=E5=9D=87?= =?UTF-8?q?=E8=A1=A1,=E5=BD=93=E5=8F=AA=E6=9C=89=E4=B8=80=E4=B8=AA?= =?UTF-8?q?=E8=8A=82=E7=82=B9=E7=9A=84=E6=97=B6=E5=80=99,=E4=B8=8D?= =?UTF-8?q?=E5=86=8D=E5=90=AF=E5=8A=A8=E5=81=A5=E5=BA=B7=E6=A3=80=E6=9F=A5?= =?UTF-8?q?.?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- utils/lb/lb.go | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/utils/lb/lb.go b/utils/lb/lb.go index 8d0b2b0..90b918c 100644 --- a/utils/lb/lb.go +++ b/utils/lb/lb.go @@ -36,6 +36,7 @@ type Group struct { lock *sync.Mutex last *Backend debug bool + bks []*Backend } 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, lock: &sync.Mutex{}, debug: debug, + bks: bks, } } func (g *Group) Select(srcAddr string, onlyHa bool) (addr string) { + if len(g.bks) == 1 { + return g.bks[0].Address + } if onlyHa { g.lock.Lock() defer g.lock.Unlock()