Add support for Record Batch compression codec flag

tulios · Jun 23, 2018 · 5b36598 · 5b36598
1 parent 2c60e74
commit 5b36598
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/src/protocol/message/compression/index.js b/src/protocol/message/compression/index.js
@@ -1,5 +1,8 @@
 const { KafkaJSNotImplemented } = require('../../../errors')
 
+const MESSAGE_CODEC_MASK = 0x3
+const RECORD_BATCH_CODEC_MASK = 0x07
+
 const Types = {
   None: 0,
   GZIP: 1,
@@ -19,7 +22,11 @@ const Codecs = {
 
 const lookupCodec = type => (Codecs[type] ? Codecs[type]() : null)
 const lookupCodecByAttributes = attributes => {
-  const codec = Codecs[attributes & 0x3]
+  const codec = Codecs[attributes & MESSAGE_CODEC_MASK]
+  return codec ? codec() : null
+}
+const lookupCodecByRecordBatchAttributes = attributes => {
+  const codec = Codecs[attributes & RECORD_BATCH_CODEC_MASK]
   return codec ? codec() : null
 }
 
@@ -28,4 +35,5 @@ module.exports = {
   Codecs,
   lookupCodec,
   lookupCodecByAttributes,
+  lookupCodecByRecordBatchAttributes,
 }