Working on client
This commit is contained in:
parent
ec9289224b
commit
212fa64be1
44
client.go
44
client.go
|
@ -2,12 +2,13 @@ package srpc
|
||||||
|
|
||||||
import (
|
import (
|
||||||
//"bytes"
|
//"bytes"
|
||||||
//"errors"
|
"errors"
|
||||||
//"fmt"
|
//"fmt"
|
||||||
"io"
|
"io"
|
||||||
"net"
|
"net"
|
||||||
"sync"
|
"sync"
|
||||||
//"time"
|
"sync/atomic"
|
||||||
|
"time"
|
||||||
)
|
)
|
||||||
|
|
||||||
//func (client *Client) Call(funcName string, args ...interface{}) (ret []byte, err error) {
|
//func (client *Client) Call(funcName string, args ...interface{}) (ret []byte, err error) {
|
||||||
|
@ -137,3 +138,42 @@ func (c *RPCClient) Stop() {
|
||||||
c.stopWg.Wait()
|
c.stopWg.Wait()
|
||||||
c.stopChan = nil
|
c.stopChan = nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func clientHandler(c *RPCClient) {
|
||||||
|
defer c.stopWg.Done()
|
||||||
|
|
||||||
|
var conn io.ReadWriteCloser
|
||||||
|
var err error
|
||||||
|
var stopping atomic.Value
|
||||||
|
|
||||||
|
for {
|
||||||
|
dialChan := make(chan struct{})
|
||||||
|
go func() {
|
||||||
|
if conn, err = c.DialHandler(c.Addr); err != nil {
|
||||||
|
if stopping.Load() == nil {
|
||||||
|
c.LogError("srpc - '%s' cannoc estable rpc connection: '%s'\n", c.Addr, err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
close(dialChan)
|
||||||
|
}()
|
||||||
|
|
||||||
|
select {
|
||||||
|
case <-c.stopChan:
|
||||||
|
stopping.Store(true)
|
||||||
|
<-dialChan
|
||||||
|
return
|
||||||
|
case <-dialChan:
|
||||||
|
}
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
select {
|
||||||
|
case <-c.stopChan:
|
||||||
|
return
|
||||||
|
case <-time.After(time.Second):
|
||||||
|
}
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
c.stopWg.Add(1)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
Loading…
Reference in New Issue