Fix message timestamp_type (#828)

2016-09-15 16:29:19 -07:00
parent 1646edf7ff
commit 2a7aca1630
1 changed files with 4 additions and 1 deletions
--- 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