modify rpc调用记录

This commit is contained in:
sk 2025-01-18 16:10:20 +08:00
parent 17733f2477
commit 9b55e2640a
1 changed files with 14 additions and 7 deletions

View File

@ -115,7 +115,8 @@ func (this *RPClient) CallWithTimeout(serviceMethod string, args interface{}, re
}()
if this.client == nil {
return ErrRPClientNoConn
err = ErrRPClientNoConn
return err
}
if d <= time.Second {
@ -125,7 +126,8 @@ func (this *RPClient) CallWithTimeout(serviceMethod string, args interface{}, re
call := this.client.Go(serviceMethod, args, reply, make(chan *rpc.Call, 1))
select {
case <-time.After(d):
return ErrRPClientCallTimeout
err = ErrRPClientCallTimeout
return err
case call = <-call.Done:
err = call.Error
}
@ -142,25 +144,30 @@ func (this *RPClient) CallWithTimeout(serviceMethod string, args interface{}, re
dd := d - time.Now().Sub(start)
if dd <= 0 {
return ErrRPClientCallTimeout
err = ErrRPClientCallTimeout
return err
}
select {
case <-time.After(dd):
return ErrRPClientCallTimeout
err = ErrRPClientCallTimeout
return err
case <-dailed:
dd = d - time.Now().Sub(start)
if dd <= 0 {
return ErrRPClientCallTimeout
err = ErrRPClientCallTimeout
return err
}
call = this.client.Go(serviceMethod, args, reply, make(chan *rpc.Call, 1))
dd = d - time.Now().Sub(start)
if dd <= 0 {
return ErrRPClientCallTimeout
err = ErrRPClientCallTimeout
return err
}
select {
case <-time.After(dd):
return ErrRPClientCallTimeout
err = ErrRPClientCallTimeout
return err
case call = <-call.Done:
err = call.Error
}