Protocol updates for 0.11.0.0 (#1127)

1 files changed, 118 insertions, 10 deletions

diff --git a/kafka/protocol/admin.py b/kafka/protocol/admin.py
index c5142b3..09746bf 100644
--- a/kafka/protocol/admin.py
+++ b/kafka/protocol/admin.py
@@ -16,6 +16,19 @@ class ApiVersionResponse_v0(Response):
     )
 
 
+class ApiVersionResponse_v1(Response):
+    API_KEY = 18
+    API_VERSION = 1
+    SCHEMA = Schema(
+        ('error_code', Int16),
+        ('api_versions', Array(
+            ('api_key', Int16),
+            ('min_version', Int16),
+            ('max_version', Int16))),
+        ('throttle_time_ms', Int32)
+    )
+
+
 class ApiVersionRequest_v0(Request):
     API_KEY = 18
     API_VERSION = 0
@@ -23,8 +36,15 @@ class ApiVersionRequest_v0(Request):
     SCHEMA = Schema()
 
 
-ApiVersionRequest = [ApiVersionRequest_v0]
-ApiVersionResponse = [ApiVersionResponse_v0]
+class ApiVersionRequest_v1(Request):
+    API_KEY = 18
+    API_VERSION = 1
+    RESPONSE_TYPE = ApiVersionResponse_v1
+    SCHEMA = ApiVersionRequest_v0.SCHEMA
+
+
+ApiVersionRequest = [ApiVersionRequest_v0, ApiVersionRequest_v1]
+ApiVersionResponse = [ApiVersionResponse_v0, ApiVersionResponse_v1]
 
 
 class CreateTopicsResponse_v0(Response):
@@ -48,6 +68,18 @@ class CreateTopicsResponse_v1(Response):
     )
 
 
+class CreateTopicsResponse_v2(Response):
+    API_KEY = 19
+    API_VERSION = 2
+    SCHEMA = Schema(
+        ('throttle_time_ms', Int32),
+        ('topic_error_codes', Array(
+            ('topic', String('utf-8')),
+            ('error_code', Int16),
+            ('error_message', String('utf-8'))))
+    )
+
+
 class CreateTopicsRequest_v0(Request):
     API_KEY = 19
     API_VERSION = 0
@@ -87,8 +119,19 @@ class CreateTopicsRequest_v1(Request):
     )
 
 
-CreateTopicsRequest = [CreateTopicsRequest_v0, CreateTopicsRequest_v1]
-CreateTopicsResponse = [CreateTopicsResponse_v0, CreateTopicsRequest_v1]
+class CreateTopicsRequest_v2(Request):
+    API_KEY = 19
+    API_VERSION = 2
+    RESPONSE_TYPE = CreateTopicsResponse_v2
+    SCHEMA = CreateTopicsRequest_v1.SCHEMA
+
+
+CreateTopicsRequest = [
+    CreateTopicsRequest_v0, CreateTopicsRequest_v1, CreateTopicsRequest_v2
+]
+CreateTopicsResponse = [
+    CreateTopicsResponse_v0, CreateTopicsResponse_v1, CreateTopicsResponse_v2
+]
 
 
 class DeleteTopicsResponse_v0(Response):
@@ -101,6 +144,17 @@ class DeleteTopicsResponse_v0(Response):
     )
 
 
+class DeleteTopicsResponse_v1(Response):
+    API_KEY = 20
+    API_VERSION = 1
+    SCHEMA = Schema(
+        ('throttle_time_ms', Int32),
+        ('topic_error_codes', Array(
+            ('topic', String('utf-8')),
+            ('error_code', Int16)))
+    )
+
+
 class DeleteTopicsRequest_v0(Request):
     API_KEY = 20
     API_VERSION = 0
@@ -111,8 +165,15 @@ class DeleteTopicsRequest_v0(Request):
     )
 
 
-DeleteTopicsRequest = [DeleteTopicsRequest_v0]
-DeleteTopicsResponse = [DeleteTopicsResponse_v0]
+class DeleteTopicsRequest_v1(Request):
+    API_KEY = 20
+    API_VERSION = 1
+    RESPONSE_TYPE = DeleteTopicsResponse_v1
+    SCHEMA = DeleteTopicsRequest_v0.SCHEMA
+
+
+DeleteTopicsRequest = [DeleteTopicsRequest_v0, DeleteTopicsRequest_v1]
+DeleteTopicsResponse = [DeleteTopicsResponse_v0, DeleteTopicsResponse_v1]
 
 
 class ListGroupsResponse_v0(Response):
@@ -126,6 +187,18 @@ class ListGroupsResponse_v0(Response):
     )
 
 
+class ListGroupsResponse_v1(Response):
+    API_KEY = 16
+    API_VERSION = 1
+    SCHEMA = Schema(
+        ('throttle_time_ms', Int32),
+        ('error_code', Int16),
+        ('groups', Array(
+            ('group', String('utf-8')),
+            ('protocol_type', String('utf-8'))))
+    )
+
+
 class ListGroupsRequest_v0(Request):
     API_KEY = 16
     API_VERSION = 0
@@ -133,8 +206,15 @@ class ListGroupsRequest_v0(Request):
     SCHEMA = Schema()
 
 
-ListGroupsRequest = [ListGroupsRequest_v0]
-ListGroupsResponse = [ListGroupsResponse_v0]
+class ListGroupsRequest_v1(Request):
+    API_KEY = 16
+    API_VERSION = 1
+    RESPONSE_TYPE = ListGroupsResponse_v1
+    SCHEMA = ListGroupsRequest_v0.SCHEMA
+
+
+ListGroupsRequest = [ListGroupsRequest_v0, ListGroupsRequest_v1]
+ListGroupsResponse = [ListGroupsResponse_v0, ListGroupsResponse_v1]
 
 
 class DescribeGroupsResponse_v0(Response):
@@ -156,6 +236,27 @@ class DescribeGroupsResponse_v0(Response):
     )
 
 
+class DescribeGroupsResponse_v1(Response):
+    API_KEY = 15
+    API_VERSION = 1
+    SCHEMA = Schema(
+        ('throttle_time_ms', Int32),
+        ('groups', Array(
+            ('error_code', Int16),
+            ('group', String('utf-8')),
+            ('state', String('utf-8')),
+            ('protocol_type', String('utf-8')),
+            ('protocol', String('utf-8')),
+            ('members', Array(
+                ('member_id', String('utf-8')),
+                ('client_id', String('utf-8')),
+                ('client_host', String('utf-8')),
+                ('member_metadata', Bytes),
+                ('member_assignment', Bytes)))))
+    )
+
+
+
 class DescribeGroupsRequest_v0(Request):
     API_KEY = 15
     API_VERSION = 0
@@ -165,8 +266,15 @@ class DescribeGroupsRequest_v0(Request):
     )
 
 
-DescribeGroupsRequest = [DescribeGroupsRequest_v0]
-DescribeGroupsResponse = [DescribeGroupsResponse_v0]
+class DescribeGroupsRequest_v1(Request):
+    API_KEY = 15
+    API_VERSION = 1
+    RESPONSE_TYPE = DescribeGroupsResponse_v1
+    SCHEMA = DescribeGroupsRequest_v0.SCHEMA
+
+
+DescribeGroupsRequest = [DescribeGroupsRequest_v0, DescribeGroupsRequest_v1]
+DescribeGroupsResponse = [DescribeGroupsResponse_v0, DescribeGroupsResponse_v1]
 
 
 class SaslHandShakeResponse_v0(Response):