Skip to content

Commit db2a898

Browse files
committed
Fix snappy comprassion on PyPy
1 parent fbea5f0 commit db2a898

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

kafka/record/legacy_records.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -412,15 +412,16 @@ def _encode_msg(self, start_pos, offset, timestamp, key, value,
412412

413413
def _maybe_compress(self):
414414
if self._compression_type:
415+
data = bytes(self._buffer)
415416
if self._compression_type == self.CODEC_GZIP:
416-
compressed = gzip_encode(bytes(self._buffer))
417+
compressed = gzip_encode(data)
417418
elif self._compression_type == self.CODEC_SNAPPY:
418-
compressed = snappy_encode(self._buffer)
419+
compressed = snappy_encode(data)
419420
elif self._compression_type == self.CODEC_LZ4:
420421
if self._magic == 0:
421-
compressed = lz4_encode_old_kafka(bytes(self._buffer))
422+
compressed = lz4_encode_old_kafka(data)
422423
else:
423-
compressed = lz4_encode(bytes(self._buffer))
424+
compressed = lz4_encode(data)
424425
size = self.size_in_bytes(
425426
0, timestamp=0, key=None, value=compressed)
426427
# We will try to reuse the same buffer if we have enough space

0 commit comments

Comments
 (0)