Merge pull request confluentinc#251 from fpietka/fix/fastavro-complex-schema

ewencp · web-flow · commit e226538d90c1 · 2017-12-18T14:07:29.000-08:00
Use schemaless reader to handle complex schema
diff --git a/confluent_kafka/avro/serializer/message_serializer.py b/confluent_kafka/avro/serializer/message_serializer.py
@@ -39,7 +39,7 @@
 
 HAS_FAST = False
 try:
-    from fastavro.reader import read_data
+    from fastavro import schemaless_reader
 
     HAS_FAST = True
 except ImportError:
@@ -169,15 +169,15 @@ def _get_decoder_func(self, schema_id, payload):
             # try to use fast avro
             try:
                 schema_dict = schema.to_json()
-                read_data(payload, schema_dict)
+                schemaless_reader(payload, schema_dict)
 
                 # If we reach this point, this means we have fastavro and it can
                 # do this deserialization. Rewind since this method just determines
                 # the reader function and we need to deserialize again along the
                 # normal path.
                 payload.seek(curr_pos)
 
-                self.id_to_decoder_func[schema_id] = lambda p: read_data(p, schema_dict)
+                self.id_to_decoder_func[schema_id] = lambda p: schemaless_reader(p, schema_dict)
                 return self.id_to_decoder_func[schema_id]
             except Exception:
                 # Fast avro failed, fall thru to standard avro below.
