From 3d16f2ff5f75380c8a9fce846f35e92bb5bfb935 Mon Sep 17 00:00:00 2001
From: Dana Powers <dana.powers@gmail.com>
Date: Tue, 5 Apr 2016 13:04:24 -0700
Subject: KAFKA-2136: support Fetch and Produce v1 (throttle_time_ms)

---
 kafka/protocol/produce.py | 25 +++++++++++++++++++++++--
 1 file changed, 23 insertions(+), 2 deletions(-)

(limited to 'kafka/protocol/produce.py')

diff --git a/kafka/protocol/produce.py b/kafka/protocol/produce.py
index 5753f64..e0b8622 100644
--- a/kafka/protocol/produce.py
+++ b/kafka/protocol/produce.py
@@ -16,6 +16,20 @@ class ProduceResponse_v0(Struct):
     )
 
 
+class ProduceResponse_v1(Struct):
+    API_KEY = 0
+    API_VERSION = 1
+    SCHEMA = Schema(
+        ('topics', Array(
+            ('topic', String('utf-8')),
+            ('partitions', Array(
+                ('partition', Int32),
+                ('error_code', Int16),
+                ('offset', Int64))))),
+        ('throttle_time_ms', Int32)
+    )
+
+
 class ProduceRequest_v0(Struct):
     API_KEY = 0
     API_VERSION = 0
@@ -31,5 +45,12 @@ class ProduceRequest_v0(Struct):
     )
 
 
-ProduceRequest = [ProduceRequest_v0]
-ProduceResponse = [ProduceResponse_v0]
+class ProduceRequest_v1(Struct):
+    API_KEY = 0
+    API_VERSION = 1
+    RESPONSE_TYPE = ProduceResponse_v1
+    SCHEMA = ProduceRequest_v0.SCHEMA
+
+
+ProduceRequest = [ProduceRequest_v0, ProduceRequest_v1]
+ProduceResponse = [ProduceResponse_v0, ProduceResponse_v1]
-- 
cgit v1.2.1