forked from segmentio/kafka-go
-
Notifications
You must be signed in to change notification settings - Fork 0
/
lz4.go
68 lines (56 loc) · 1.28 KB
/
lz4.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
65
66
67
68
package lz4
import (
"io"
"sync"
"github.com/pierrec/lz4/v4"
)
var (
readerPool sync.Pool
writerPool sync.Pool
)
// Codec is the implementation of a compress.Codec which supports creating
// readers and writers for kafka messages compressed with lz4.
type Codec struct{}
// Code implements the compress.Codec interface.
func (c *Codec) Code() int8 { return 3 }
// Name implements the compress.Codec interface.
func (c *Codec) Name() string { return "lz4" }
// NewReader implements the compress.Codec interface.
func (c *Codec) NewReader(r io.Reader) io.ReadCloser {
z, _ := readerPool.Get().(*lz4.Reader)
if z != nil {
z.Reset(r)
} else {
z = lz4.NewReader(r)
}
return &reader{Reader: z}
}
// NewWriter implements the compress.Codec interface.
func (c *Codec) NewWriter(w io.Writer) io.WriteCloser {
z, _ := writerPool.Get().(*lz4.Writer)
if z != nil {
z.Reset(w)
} else {
z = lz4.NewWriter(w)
}
return &writer{Writer: z}
}
type reader struct{ *lz4.Reader }
func (r *reader) Close() (err error) {
if z := r.Reader; z != nil {
r.Reader = nil
z.Reset(nil)
readerPool.Put(z)
}
return
}
type writer struct{ *lz4.Writer }
func (w *writer) Close() (err error) {
if z := w.Writer; z != nil {
w.Writer = nil
err = z.Close()
z.Reset(nil)
writerPool.Put(z)
}
return
}