Skip to content

Commit 996ed2d

Browse files
author
Brigitte Lamarche
committed
PR 649: adding compression (second code review)
1 parent 477c9f8 commit 996ed2d

File tree

1 file changed

+5
-9
lines changed

1 file changed

+5
-9
lines changed

compress.go

+5-9
Original file line numberDiff line numberDiff line change
@@ -10,9 +10,6 @@ const (
1010
minCompressLength = 50
1111
)
1212

13-
var (
14-
blankHeader = []byte{0, 0, 0, 0, 0, 0, 0}
15-
)
1613

1714
type packetReader interface {
1815
readNext(need int) ([]byte, error)
@@ -146,17 +143,16 @@ func (cw *compressedWriter) Write(data []byte) (int, error) {
146143
}
147144

148145
totalBytes := len(data)
149-
150146
length := len(data) - 4
151-
152147
maxPayloadLength := maxPacketSize - 4
148+
blankHeader := make([]byte, 7)
153149

154150
for length >= maxPayloadLength {
155151
payload := data[:maxPayloadLength]
156152
payloadLen := len(payload)
157153

158154
bytesBuf := &bytes.Buffer{}
159-
bytesBuf.Write(blankHeader)
155+
bytesBuf.Write(blankHeader)
160156
cw.zw.Reset(bytesBuf)
161157
_, err := cw.zw.Write(payload)
162158
if err != nil {
@@ -167,7 +163,7 @@ func (cw *compressedWriter) Write(data []byte) (int, error) {
167163
// if compression expands the payload, do not compress
168164
compressedPayload := bytesBuf.Bytes()
169165
if len(compressedPayload) > maxPayloadLength {
170-
compressedPayload = append(blankHeader, payload...)
166+
compressedPayload = append(blankHeader, payload...)
171167
payloadLen = 0
172168
}
173169

@@ -184,7 +180,7 @@ func (cw *compressedWriter) Write(data []byte) (int, error) {
184180

185181
// do not attempt compression if packet is too small
186182
if payloadLen < minCompressLength {
187-
err := cw.writeToNetwork(append(blankHeader, data...), 0)
183+
err := cw.writeToNetwork(append(blankHeader, data...), 0)
188184
if err != nil {
189185
return 0, err
190186
}
@@ -203,7 +199,7 @@ func (cw *compressedWriter) Write(data []byte) (int, error) {
203199
compressedPayload := bytesBuf.Bytes()
204200

205201
if len(compressedPayload) > len(data) {
206-
compressedPayload = append(blankHeader, data...)
202+
compressedPayload = append(blankHeader, data...)
207203
payloadLen = 0
208204
}
209205

0 commit comments

Comments
 (0)