Add simple BrokerConnection class; add request.RESPONSE_TYPE class vars

1 files changed, 11 insertions, 12 deletions

diff --git a/kafka/protocol/offset.py b/kafka/protocol/offset.py
index 942bdbf..776de39 100644
--- a/kafka/protocol/offset.py
+++ b/kafka/protocol/offset.py
@@ -2,31 +2,30 @@ from .struct import Struct
 from .types import Array, Int16, Int32, Int64, Schema, String
 
 
-class OffsetRequest(Struct):
-    API_KEY = 2
-    API_VERSION = 0
+class OffsetResponse(Struct):
     SCHEMA = Schema(
-        ('replica_id', Int32),
         ('topics', Array(
             ('topic', String('utf-8')),
             ('partitions', Array(
                 ('partition', Int32),
-                ('time', Int64),
-                ('max_offsets', Int32)))))
+                ('error_code', Int16),
+                ('offsets', Array(Int64))))))
     )
-    DEFAULTS = {
-        'replica_id': -1
-    }
 
 
-class OffsetResponse(Struct):
+class OffsetRequest(Struct):
     API_KEY = 2
     API_VERSION = 0
+    RESPONSE_TYPE = OffsetResponse
     SCHEMA = Schema(
+        ('replica_id', Int32),
         ('topics', Array(
             ('topic', String('utf-8')),
             ('partitions', Array(
                 ('partition', Int32),
-                ('error_code', Int16),
-                ('offsets', Array(Int64))))))
+                ('time', Int64),
+                ('max_offsets', Int32)))))
     )
+    DEFAULTS = {
+        'replica_id': -1
+    }