blob: 87a7da0c66851ac36194e6a5493658fb7962e06f (
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
58
59
60
61
62
63
64
65
66
|
package zip
import (
"bufio"
"compress/flate"
"errors"
"io"
"sync"
)
var ErrClosedReader = errors.New("deflatereader: reader is closed")
var deflateReaderPool sync.Pool
// deflateReader wrapper to support reading compressed files.
// Implements the io.ReadCloser interface.
type deflateReader struct {
reader *bufio.Reader
closer io.Closer
flateReader io.ReadCloser
}
// Read from flateReader
func (r *deflateReader) Read(p []byte) (n int, err error) {
if r.closer == nil {
return 0, ErrClosedReader
}
return r.flateReader.Read(p)
}
// Close all readers
func (r *deflateReader) Close() error {
if r.closer == nil {
return ErrClosedReader
}
defer func() {
r.closer.Close()
r.closer = nil
deflateReaderPool.Put(r)
}()
return r.flateReader.Close()
}
func (r *deflateReader) reset(rc io.ReadCloser) {
r.reader.Reset(rc)
r.closer = rc
r.flateReader.(flate.Resetter).Reset(r.reader, nil)
}
func newDeflateReader(r io.ReadCloser) *deflateReader {
if dr, ok := deflateReaderPool.Get().(*deflateReader); ok {
dr.reset(r)
return dr
}
br := bufio.NewReader(r)
return &deflateReader{
reader: br,
closer: r,
flateReader: flate.NewReader(br),
}
}
|
