mirror of
https://github.com/golang/net.git
synced 2026-04-01 02:47:08 +09:00
b7f5d985f9
Make the pipe code take an interface as the backing store. Now a pipe is something that's goroutine-safe and does the Cond waits but its underlying data is now an interface: anything that's a ReaderWriter with a Len method (such as a *bytes.Buffer), or a fixedBuffer (renamed in this CL from 'buffer'). This opens the ground to having a non-fixed buffer used with pipe. This also moves the CloseWithError code up into the pipe code, out of fixedBuffer. Change-Id: Ia3b853e8aa8920807b705ff4e41bed934a8c67b7 Reviewed-on: https://go-review.googlesource.com/16312 Reviewed-by: Blake Mizerany <blake.mizerany@gmail.com>
61 lines
1.3 KiB
Go
61 lines
1.3 KiB
Go
// Copyright 2014 The Go Authors. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package http2
|
|
|
|
import (
|
|
"errors"
|
|
)
|
|
|
|
// fixedBuffer is an io.ReadWriter backed by a fixed size buffer.
|
|
// It never allocates, but moves old data as new data is written.
|
|
type fixedBuffer struct {
|
|
buf []byte
|
|
r, w int
|
|
}
|
|
|
|
var (
|
|
errReadEmpty = errors.New("read from empty fixedBuffer")
|
|
errWriteFull = errors.New("write on full fixedBuffer")
|
|
)
|
|
|
|
// Read copies bytes from the buffer into p.
|
|
// It is an error to read when no data is available.
|
|
func (b *fixedBuffer) Read(p []byte) (n int, err error) {
|
|
if b.r == b.w {
|
|
return 0, errReadEmpty
|
|
}
|
|
n = copy(p, b.buf[b.r:b.w])
|
|
b.r += n
|
|
if b.r == b.w {
|
|
b.r = 0
|
|
b.w = 0
|
|
}
|
|
return n, nil
|
|
}
|
|
|
|
// Len returns the number of bytes of the unread portion of the buffer.
|
|
func (b *fixedBuffer) Len() int {
|
|
return b.w - b.r
|
|
}
|
|
|
|
// Write copies bytes from p into the buffer.
|
|
// It is an error to write more data than the buffer can hold.
|
|
func (b *fixedBuffer) Write(p []byte) (n int, err error) {
|
|
// Slide existing data to beginning.
|
|
if b.r > 0 && len(p) > len(b.buf)-b.w {
|
|
copy(b.buf, b.buf[b.r:b.w])
|
|
b.w -= b.r
|
|
b.r = 0
|
|
}
|
|
|
|
// Write new data.
|
|
n = copy(b.buf[b.w:], p)
|
|
b.w += n
|
|
if n < len(p) {
|
|
err = errWriteFull
|
|
}
|
|
return n, err
|
|
}
|