diff --git a/pykafka/protocol/offset.py b/pykafka/protocol/offset.py index 6064f218b..520174c01 100644 --- a/pykafka/protocol/offset.py +++ b/pykafka/protocol/offset.py @@ -42,7 +42,7 @@ class ListOffsetRequest(Request): @classmethod def get_versions(cls): # XXX use ListOffsetRequestV1 after 0.10 message format is supported - return {0: ListOffsetRequest, 1: ListOffsetRequest} + return {0: ListOffsetRequest, 1: ListOffsetRequestV1} def __init__(self, partition_requests): """Create a new offset request""" @@ -160,7 +160,7 @@ class ListOffsetResponse(Response): @classmethod def get_versions(cls): # XXX use ListOffsetResponseV1 after 0.10 message format is supported - return {0: ListOffsetResponse, 1: ListOffsetResponse} + return {0: ListOffsetResponse, 1: ListOffsetResponseV1} def __init__(self, buff): """Deserialize into a new Response @@ -195,7 +195,7 @@ def __init__(self, buff): :param buff: Serialized message :type buff: :class:`bytearray` """ - fmt = '[S [ihq [q] ] ]' + fmt = '[S [ihqq] ]' response = struct_helpers.unpack_from(fmt, buff, 0) self.topics = {}