blob: 8c8725bedefa8a776a38dfa0c4c5c19ce364aaf7 (
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
39
40
41
42
43
44
|
package util
import (
"context"
"io"
)
const bufferSize = 512
type ChannelReader struct {
Out <-chan []byte
Err <-chan error
}
func NewChannelReader(ctx context.Context, reader io.Reader) ChannelReader {
outCh := make(chan []byte)
errCh := make(chan error)
go func(in chan<- []byte, inErr chan<- error) {
defer close(in)
defer close(inErr)
data := make([]byte, bufferSize)
outer:
for {
select {
case <-ctx.Done():
inErr <- ctx.Err()
break outer
default:
n, err := reader.Read(data)
if err != nil && err != io.EOF {
inErr <- err
break outer
}
in <- data[:n]
}
}
}(outCh, errCh)
return ChannelReader{
Out: outCh,
Err: errCh,
}
}
|
