Jelajahi Sumber

allow 0 timeout

Darien Raymond 8 tahun lalu
induk
melakukan
bb4daedd4c
1 mengubah file dengan 4 tambahan dan 0 penghapusan
  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