使用 fastavro 从 Kafka 反序列化 Avro

Mic*_*hał 3 python avro apache-kafka fastavro

我正在构建一个从 Kafka 接收数据的应用程序。使用 Apache ( https://pypi.org/project/avro-python3/ )提供的标准 avro 库时,结果是正确的,但是,反序列化过程非常缓慢。

class KafkaReceiver:
    data = {}

    def __init__(self, bootstrap='192.168.1.111:9092'):
        self.client = KafkaConsumer(
            'topic',
            bootstrap_servers=bootstrap,
            client_id='app',
            api_version=(0, 10, 1)
        )
        self.schema = avro.schema.parse(open("Schema.avsc", "rb").read())
        self.reader = avro.io.DatumReader(self.schema)

    def do(self):
        for msg in self.client:
            bytes_reader = io.BytesIO(msg.value)
            decoder = BinaryDecoder(bytes_reader)

            self.data = self.reader.read(decoder)
Run Code Online (Sandbox Code Playgroud)

在阅读为什么这么慢时,我发现fastavro哪个应该快得多。我是这样使用的:

    def do(self):

        schema = fastavro.schema.load_schema('Schema.avsc')
        for msg in self.client:
            bytes_reader = io.BytesIO(msg.value)
            bytes_reader.seek(0)
            for record in reader(bytes_reader, schema):
                self.data = record
Run Code Online (Sandbox Code Playgroud)

而且,由于在使用 Apache 的库时一切正常,我希望一切都将与fastavro. 但是,在运行它时,我得到

  File "fastavro/_read.pyx", line 389, in fastavro._read.read_map
  File "fastavro/_read.pyx", line 290, in fastavro._read.read_utf8
  File "fastavro/_six.pyx", line 22, in fastavro._six.py3_btou
UnicodeDecodeError: 'utf-8' codec can't decode byte 0xfc in position 3: invalid start byte
Run Code Online (Sandbox Code Playgroud)

我通常不使用 Python 编程,所以我不完全知道如何处理这个问题。有任何想法吗?

小智 5

fastavro.reader期望,其包括标题中的阿夫罗文件格式。看起来您拥有的是没有标题的序列化记录。我认为您可以使用fastavro.schemaless_reader.

所以而不是:

for record in reader(bytes_reader, schema):
    self.data = record
Run Code Online (Sandbox Code Playgroud)

你会这样做:

self.data = schemaless_reader(bytes_reader, schema)
Run Code Online (Sandbox Code Playgroud)