blob: 0e2485c97f26de7516b7c923d3b4465cd6205484 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
|
// Package bytesx implements bytes extensions.
package bytesx
import "io"
// ResetReader implements the io.Reader reading from a resettable byte slice.
type ResetReader struct {
buf []byte
head int
tail int
}
// NewResetReader returns a new ResetReader reading from b.
func NewResetReader(b []byte) *ResetReader {
r := &ResetReader{}
r.Reset(b)
return r
}
// Read reads bytes into p.
func (r *ResetReader) Read(p []byte) (int, error) {
if r.head == r.tail {
return 0, io.EOF
}
n := copy(p, r.buf[r.head:])
r.head += n
return n, nil
}
// Reset resets the byte slice being read from.
func (r *ResetReader) Reset(b []byte) {
r.buf = b
r.head = 0
r.tail = len(b)
}
|
