From 2a7aca1630b81669595d753083239ec9fbf66ff5 Mon Sep 17 00:00:00 2001
From: Josh Yudaken <j@yud.co.za>
Date: Thu, 15 Sep 2016 16:29:19 -0700
Subject: Fix message timestamp_type (#828)

---
 kafka/protocol/message.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

(limited to 'kafka/protocol')

diff --git a/kafka/protocol/message.py b/kafka/protocol/message.py
index f333800..36f03ca 100644
--- a/kafka/protocol/message.py
+++ b/kafka/protocol/message.py
@@ -64,7 +64,10 @@ class Message(Struct):
         """
         if self.magic == 0:
             return None
-        return self.attributes & self.TIMESTAMP_TYPE_MASK
+        elif self.attributes & self.TIMESTAMP_TYPE_MASK:
+            return 1
+        else:
+            return 0
 
     def _encode_self(self, recalc_crc=True):
         version = self.magic
-- 
cgit v1.2.1