diff options
Diffstat (limited to 'kafka/protocol/message.py')
| -rw-r--r-- | kafka/protocol/message.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/kafka/protocol/message.py b/kafka/protocol/message.py index 501ce47..3027ebd 100644 --- a/kafka/protocol/message.py +++ b/kafka/protocol/message.py @@ -1,3 +1,5 @@ +import io + from .struct import Struct from .types import ( Int8, Int32, Int64, Bytes, Schema, AbstractType @@ -34,7 +36,7 @@ class Message(Struct): @classmethod def decode(cls, data): if isinstance(data, bytes): - data = BytesIO(data) + data = io.BytesIO(data) fields = [field.decode(data) for field in cls.SCHEMA.fields] return cls(fields[4], key=fields[3], magic=fields[1], attributes=fields[2], crc=fields[0]) |