-
Notifications
You must be signed in to change notification settings - Fork 0
/
rtt_select.go
108 lines (102 loc) · 2.24 KB
/
rtt_select.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
package selector
import (
"github.com/bmclab-git/shuttle/log"
"github.com/bmclab-git/shuttle/proxy"
"sync/atomic"
"time"
)
const (
timerDulation = 10 * time.Minute
)
func init() {
proxy.RegisterSelector("rtt", func(group *proxy.ServerGroup) (proxy.ISelector, error) {
s := &rttSelector{
group: group,
timer: time.NewTimer(timerDulation),
selected: group.Servers[0].(proxy.IServer),
cancel: make(chan bool, 1),
}
go func() {
for {
select {
case <-s.timer.C:
case <-s.cancel:
return
}
s.autoTest()
s.timer.Reset(timerDulation)
}
}()
go s.Refresh()
return s, nil
})
}
type rttSelector struct {
group *proxy.ServerGroup
selected proxy.IServer
status uint32
timer *time.Timer
cancel chan bool
}
func (r *rttSelector) Get() (*proxy.Server, error) {
return r.selected.GetServer()
}
func (r *rttSelector) Select(name string) error {
return nil
}
func (r *rttSelector) Refresh() error {
r.autoTest()
return nil
}
func (r *rttSelector) Reset(group *proxy.ServerGroup) error {
r.group = group
r.selected = r.group.Servers[0].(proxy.IServer)
go r.autoTest()
return nil
}
func (r *rttSelector) Destroy() {
r.cancel <- true
}
func (r *rttSelector) autoTest() {
if r.status == 0 {
if ok := atomic.CompareAndSwapUint32(&r.status, 0, 1); !ok {
return
}
}
r.timer.Stop()
log.Logger.Debug("[Rtt-Selector] start testing ...")
var is proxy.IServer
var s *proxy.Server
var err error
c := make(chan *proxy.Server, 1)
for _, v := range r.group.Servers {
is = v.(proxy.IServer)
s, err = is.GetServer()
if err != nil {
continue
}
go urlTest(s, r.group.GetRttRrl(), c)
}
s = <-c
log.Logger.Infof("[Rtt-Select] rtt select server: [%s]", s.Name)
r.selected = s
r.timer.Reset(timerDulation)
atomic.CompareAndSwapUint32(&r.status, 1, 0)
}
func urlTest(s *proxy.Server, rttUrl string, c chan *proxy.Server) {
rtt, err := proxy.TestRTT(s, rttUrl)
if err != nil {
s.Rtt = -1
log.Logger.Debugf("[Rtt-Select] [%s] url test result: <failed> %v", s.Name, err)
return
}
s.Rtt = rtt
select {
case c <- s:
default:
}
log.Logger.Debugf("[Rtt-Select] [%s] Rtt:[%dms]", s.Name, s.Rtt.Nanoseconds()/1000000)
}
func (r *rttSelector) Current() proxy.IServer {
return r.selected
}