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
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
|
// Package util implements various utility types and interfaces.
package util
import "io"
// SizeReadSeekCloser is an io.Reader, io.Seeker, and io.Closer with a Size
// method.
type SizeReadSeekCloser interface {
io.Reader
io.Seeker
io.Closer
Size() int64
}
// Reader is both an io.Reader and io.ByteReader.
type Reader interface {
io.Reader
io.ByteReader
}
// ReadCloser is a Reader that is also an io.Closer.
type ReadCloser interface {
Reader
io.Closer
}
type nopCloser struct {
Reader
}
func (nopCloser) Close() error {
return nil
}
// NopCloser returns a ReadCloser with a no-op Close method wrapping the
// provided Reader r.
func NopCloser(r Reader) ReadCloser {
return &nopCloser{r}
}
type byteReadCloser struct {
io.ReadCloser
}
func (rc *byteReadCloser) ReadByte() (byte, error) {
var b [1]byte
n, err := rc.Read(b[:])
if err != nil {
return 0, err //nolint:wrapcheck
}
if n == 0 {
return 0, io.ErrNoProgress
}
return b[0], nil
}
// ByteReadCloser returns a ReadCloser either by returning the io.ReadCloser
// r if it implements the interface, or wrapping it with a ReadByte method.
func ByteReadCloser(r io.ReadCloser) ReadCloser {
if rc, ok := r.(ReadCloser); ok {
return rc
}
return &byteReadCloser{r}
}
|
