浏览代码

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