소스 검색

allow 0 timeout

Darien Raymond 8 년 전
부모
커밋
bb4daedd4c
1개의 변경된 파일4개의 추가작업 그리고 0개의 파일을 삭제
  1. 4 0
      transport/ray/direct.go

+ 4 - 0
transport/ray/direct.go

@@ -91,6 +91,10 @@ func (v *Stream) ReadTimeout(timeout time.Duration) (*buf.Buffer, error) {
 	case b := <-v.buffer:
 		return b, nil
 	default:
+		if timeout == 0 {
+			return nil, ErrReadTimeout
+		}
+
 		select {
 		case <-v.ctx.Done():
 			return nil, io.ErrClosedPipe