gian/cubefs
1
0
Fork 0
mirror of https://github.com/cubefs/cubefs.git synced 2026-08-02 02:00:56 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix(rpc): timeout with response in rpc client for timer reset

Browse Source 
@formatter:off

Signed-off-by: JasonHu520 <huzongchao@oppo.com>
This commit is contained in:
JasonHu520 2025-09-10 17:47:03 +08:00 committed by 梁曟風
parent d076024883
commit b4309236a2
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
blobstore/common/rpc/client.go
View File

@ -357,6 +357,7 @@ type timeoutReadCloser struct {
timer *time.Timer
timeout time.Duration
closeOnce sync.Once
hasRead bool
}
func (tr *timeoutReadCloser) Close() (err error) {
@ -371,6 +372,14 @@ func (tr *timeoutReadCloser) Read(p []byte) (int, error) {
if tr.timeout <= 0 {
return 0, ErrBodyReadTimeout
}
if tr.hasRead {
select {
case <-tr.timer.C:
default:
}
}
tr.hasRead = true
tr.timer.Reset(tr.timeout)
start := time.Now()