bughz/glider
1
0
Fork 0
mirror of https://github.com/nadoo/glider.git synced 2025-11-04 07:42:38 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

strategy: optimize code

Browse Source
This commit is contained in:
nadoo 2018-01-13 20:08:49 +08:00
parent 77615f9f3d
commit 76a12aa8d1
1 changed files with 22 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
strategy.go
View File

@ -10,25 +10,27 @@ import (
)
// NewStrategyDialer returns a new Strategy Dialer
func NewStrategyDialer(strategy string, dialers []Dialer, website string, duration int) Dialer {
var dialer Dialer
func NewStrategyDialer(strategy string, dialers []Dialer, website string, interval int) Dialer {
if len(dialers) == 0 {
dialer = Direct
} else if len(dialers) == 1 {
dialer = dialers[0]
} else if len(dialers) > 1 {
return Direct
}
if len(dialers) == 1 {
return dialers[0]
}
var dialer Dialer
switch strategy {
case "rr":
dialer = newRRDialer(dialers, website, duration)
dialer = newRRDialer(dialers, website, interval)
logf("forward to remote servers in round robin mode.")
case "ha":
dialer = newHADialer(dialers, website, duration)
dialer = newHADialer(dialers, website, interval)
logf("forward to remote servers in high availability mode.")
default:
logf("not supported forward mode '%s', just use the first forward server.", conf.Strategy)
dialer = dialers[0]
}
}
return dialer
}
@ -42,15 +44,15 @@ type rrDialer struct {
// for checking
website string
duration int
interval int
}
// newRRDialer returns a new rrDialer
func newRRDialer(dialers []Dialer, website string, duration int) *rrDialer {
func newRRDialer(dialers []Dialer, website string, interval int) *rrDialer {
rr := &rrDialer{dialers: dialers}
rr.website = website
rr.duration = duration
rr.interval = interval
for k := range dialers {
rr.status.Store(k, true)
@ -104,7 +106,7 @@ func (rr *rrDialer) checkDialer(idx int) {
d := rr.dialers[idx]
for {
time.Sleep(time.Duration(rr.duration) * time.Second * time.Duration(retry>>1))
time.Sleep(time.Duration(rr.interval) * time.Second * time.Duration(retry>>1))
retry <<= 1
if retry > 16 {
@ -154,7 +156,6 @@ func (ha *haDialer) Dial(network, addr string) (net.Conn, error) {
d := ha.dialers[ha.idx]
result, ok := ha.status.Load(ha.idx)
if ok && !result.(bool) {
d = ha.NextDialer(addr)
}