blob: 5c489ed574ff18b8d45170e60b4ead1ec42850ee (
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
45
46
47
48
49
50
51
52
53
54
55
56
57
|
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)
buffer := make([]byte, 2 * bufferSize)
isD1 := true
d1 := buffer[:bufferSize]
d2 := buffer[bufferSize:]
outer:
for {
select {
case <-ctx.Done():
inErr <- ctx.Err()
break outer
default:
var data []byte
if isD1 {
data = d1
} else {
data = d2
}
isD1 = !isD1
n, err := reader.Read(data)
if err != nil {
if err != io.EOF {
inErr <- err
}
break outer
}
in <- data[:n]
}
}
}(outCh, errCh)
return ChannelReader{
Out: outCh,
Err: errCh,
}
}
|
