You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
97 lines
2.4 KiB
97 lines
2.4 KiB
from __future__ import absolute_import
|
|
|
|
import abc
|
|
|
|
from kafka.protocol.struct import Struct
|
|
from kafka.protocol.types import Int16, Int32, String, Schema, Array
|
|
|
|
|
|
class RequestHeader(Struct):
|
|
SCHEMA = Schema(
|
|
('api_key', Int16),
|
|
('api_version', Int16),
|
|
('correlation_id', Int32),
|
|
('client_id', String('utf-8'))
|
|
)
|
|
|
|
def __init__(self, request, correlation_id=0, client_id='kafka-python'):
|
|
super(RequestHeader, self).__init__(
|
|
request.API_KEY, request.API_VERSION, correlation_id, client_id
|
|
)
|
|
|
|
|
|
class Request(Struct):
|
|
__metaclass__ = abc.ABCMeta
|
|
|
|
@abc.abstractproperty
|
|
def API_KEY(self):
|
|
"""Integer identifier for api request"""
|
|
pass
|
|
|
|
@abc.abstractproperty
|
|
def API_VERSION(self):
|
|
"""Integer of api request version"""
|
|
pass
|
|
|
|
@abc.abstractproperty
|
|
def SCHEMA(self):
|
|
"""An instance of Schema() representing the request structure"""
|
|
pass
|
|
|
|
@abc.abstractproperty
|
|
def RESPONSE_TYPE(self):
|
|
"""The Response class associated with the api request"""
|
|
pass
|
|
|
|
def expect_response(self):
|
|
"""Override this method if an api request does not always generate a response"""
|
|
return True
|
|
|
|
def to_object(self):
|
|
return _to_object(self.SCHEMA, self)
|
|
|
|
|
|
class Response(Struct):
|
|
__metaclass__ = abc.ABCMeta
|
|
|
|
@abc.abstractproperty
|
|
def API_KEY(self):
|
|
"""Integer identifier for api request/response"""
|
|
pass
|
|
|
|
@abc.abstractproperty
|
|
def API_VERSION(self):
|
|
"""Integer of api request/response version"""
|
|
pass
|
|
|
|
@abc.abstractproperty
|
|
def SCHEMA(self):
|
|
"""An instance of Schema() representing the response structure"""
|
|
pass
|
|
|
|
def to_object(self):
|
|
return _to_object(self.SCHEMA, self)
|
|
|
|
|
|
def _to_object(schema, data):
|
|
obj = {}
|
|
for idx, (name, _type) in enumerate(zip(schema.names, schema.fields)):
|
|
if isinstance(data, Struct):
|
|
val = data.get_item(name)
|
|
else:
|
|
val = data[idx]
|
|
|
|
if isinstance(_type, Schema):
|
|
obj[name] = _to_object(_type, val)
|
|
elif isinstance(_type, Array):
|
|
if isinstance(_type.array_of, (Array, Schema)):
|
|
obj[name] = [
|
|
_to_object(_type.array_of, x)
|
|
for x in val
|
|
]
|
|
else:
|
|
obj[name] = val
|
|
else:
|
|
obj[name] = val
|
|
|
|
return obj
|