添加收包超时处理

rpc/client.go

@@ -121,7 +121,7 @@ func NewClient(confFile string) (*Client, error) {
 	return c, nil
 }
 
-func (c *Client) Call(service, method string, args interface{}, reply interface{}) error {
+func (c *Client) Call(addr, protocol, service, method string, args interface{}, reply interface{}) error {
 	b := &GettyRPCRequest{}
 	b.header.Service = service
 	b.header.Method = method
@@ -134,7 +134,7 @@ func (c *Client) Call(service, method string, args interface{}, reply interface{
 	resp := NewPendingResponse()
 	resp.reply = reply
 
-	session := c.selectSession()
+	session := c.selectSession(protocol, addr)
 	if session == nil {
 		return errSessionNotExist
 	}
@@ -165,8 +165,12 @@ func (c *Client) Close() {
 	c.registry = nil
 }
 
-func (c *Client) selectSession() getty.Session {
-	return nil
+func (c *Client) selectSession(protocol, addr string) getty.Session {
+	rpcConn, err := c.pool.getConn(protocol, addr)
+	if err != nil {
+		return nil
+	}
+	return rpcConn.selectSession()
 }
 
 func (c *Client) heartbeat(session getty.Session) error {