/
gwioutil.go
64 lines (53 loc) · 1.09 KB
/
gwioutil.go
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
package gwioutil
import (
"io"
"github.com/pkg/errors"
)
type timeoutError interface {
Timeout() bool // Is it a timeout error
}
// IsTimeoutError checks if the error is a timeout error
func IsTimeoutError(err error) bool {
if err == nil {
return false
}
err = errors.Cause(err)
ne, ok := err.(timeoutError)
return ok && ne.Timeout()
}
// WriteAll write all bytes of data to the writer
func WriteAll(conn io.Writer, data []byte) error {
left := len(data)
for left > 0 {
n, err := conn.Write(data)
if n == left && err == nil { // handle most common case first
return nil
}
if n > 0 {
data = data[n:]
left -= n
}
if err != nil && !IsTimeoutError(err) {
return err
}
}
return nil
}
// ReadAll reads from the reader until all bytes in data is filled
func ReadAll(conn io.Reader, data []byte) error {
left := len(data)
for left > 0 {
n, err := conn.Read(data)
if n == left && err == nil { // handle most common case first
return nil
}
if n > 0 {
data = data[n:]
left -= n
}
if err != nil && !IsTimeoutError(err) {
return err
}
}
return nil
}