From d8bb5d2efa9f2df09421e77040edd0caa2643cec Mon Sep 17 00:00:00 2001
From: Dana Powers <dana.powers@gmail.com>
Date: Sun, 17 Jul 2016 11:48:19 -0700
Subject: Always encode size with MessageSet

---
 kafka/protocol/message.py | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

(limited to 'kafka')

diff --git a/kafka/protocol/message.py b/kafka/protocol/message.py
index 656c131..795495d 100644
--- a/kafka/protocol/message.py
+++ b/kafka/protocol/message.py
@@ -143,7 +143,7 @@ class MessageSet(AbstractType):
     HEADER_SIZE = 12 # offset + message_size
 
     @classmethod
-    def encode(cls, items, size=True, recalc_message_size=True):
+    def encode(cls, items):
         # RecordAccumulator encodes messagesets internally
         if isinstance(items, io.BytesIO):
             size = Int32.decode(items)
@@ -156,8 +156,6 @@ class MessageSet(AbstractType):
             encoded_values.append(Int64.encode(offset))
             encoded_values.append(Bytes.encode(message))
         encoded = b''.join(encoded_values)
-        if not size:
-            return encoded
         return Bytes.encode(encoded)
 
     @classmethod
-- 
cgit v1.2.1