root/gost_software
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add retry mechanism for chain

Browse Source
This commit is contained in:
rui.zheng 2017-11-15 15:55:15 +08:00
parent cd26bcb81d
commit 6f935fc397
1 changed files with 26 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
chain.go
View File

@ -99,11 +99,21 @@ func (c *Chain) IsEmpty() bool {
// Dial connects to the target address addr through the chain. // Dial connects to the target address addr through the chain.
// If the chain is empty, it will use the net.Dial directly. // If the chain is empty, it will use the net.Dial directly.
func (c *Chain) Dial(addr string) (net.Conn, error) { func (c *Chain) Dial(addr string) (conn net.Conn, err error) {
if c.IsEmpty() { if c.IsEmpty() {
return net.DialTimeout("tcp", addr, DialTimeout) return net.DialTimeout("tcp", addr, DialTimeout)
} }
for i := 0; i < c.Retries+1; i++ {
conn, err = c.dial(addr)
if err == nil {
break
}
}
return
}
func (c *Chain) dial(addr string) (net.Conn, error) {
route, err := c.selectRoute() route, err := c.selectRoute()
if err != nil { if err != nil {
return nil, err return nil, err
@ -125,11 +135,19 @@ func (c *Chain) Dial(addr string) (net.Conn, error) {
// Conn obtains a handshaked connection to the last node of the chain. // Conn obtains a handshaked connection to the last node of the chain.
// If the chain is empty, it returns an ErrEmptyChain error. // If the chain is empty, it returns an ErrEmptyChain error.
func (c *Chain) Conn() (conn net.Conn, err error) { func (c *Chain) Conn() (conn net.Conn, err error) {
route, err := c.selectRoute() for i := 0; i < c.Retries+1; i++ {
if err != nil { var route *Chain
return nil, err route, err = c.selectRoute()
if err != nil {
continue
}
conn, err = route.getConn()
if err != nil {
continue
}
break
} }
conn, err = route.getConn()
return return
} }
@ -205,6 +223,8 @@ func (c *Chain) selectRoute() (route *Chain, err error) {
route.AddNode(node) route.AddNode(node)
} }
log.Log("select route:", buf.String()) if Debug {
log.Log("select route:", buf.String())
}
return return
} }