Use version-indexed lists for request/response protocol structsprotocol_versions

1 files changed, 10 insertions, 4 deletions

diff --git a/kafka/protocol/produce.py b/kafka/protocol/produce.py
index ef2f96e..5753f64 100644
--- a/kafka/protocol/produce.py
+++ b/kafka/protocol/produce.py
@@ -1,9 +1,11 @@
 from .message import MessageSet
 from .struct import Struct
-from .types import Int8, Int16, Int32, Int64, Bytes, String, Array, Schema
+from .types import Int16, Int32, Int64, String, Array, Schema
 
 
-class ProduceResponse(Struct):
+class ProduceResponse_v0(Struct):
+    API_KEY = 0
+    API_VERSION = 0
     SCHEMA = Schema(
         ('topics', Array(
             ('topic', String('utf-8')),
@@ -14,10 +16,10 @@ class ProduceResponse(Struct):
     )
 
 
-class ProduceRequest(Struct):
+class ProduceRequest_v0(Struct):
     API_KEY = 0
     API_VERSION = 0
-    RESPONSE_TYPE = ProduceResponse
+    RESPONSE_TYPE = ProduceResponse_v0
     SCHEMA = Schema(
         ('required_acks', Int16),
         ('timeout', Int32),
@@ -27,3 +29,7 @@ class ProduceRequest(Struct):
                 ('partition', Int32),
                 ('messages', MessageSet)))))
     )
+
+
+ProduceRequest = [ProduceRequest_v0]
+ProduceResponse = [ProduceResponse_v0]