lib/avromatic/io/datum_writer.rb in avromatic-2.2.5 vs lib/avromatic/io/datum_writer.rb in avromatic-2.2.6
- old
+ new
@@ -3,24 +3,18 @@
module Avromatic
module IO
# Subclass DatumWriter to use additional information about union member
# index.
class DatumWriter < Avro::IO::DatumWriter
- class << self
- attr_accessor :optimize
- end
-
def write_union(writers_schema, datum, encoder)
optional = writers_schema.schemas.first.type_sym == :null
if datum.is_a?(Hash) && datum.key?(Avromatic::IO::UNION_MEMBER_INDEX)
index_of_schema = datum[Avromatic::IO::UNION_MEMBER_INDEX]
# Avromatic does not treat the null of an optional field as part of the union
index_of_schema += 1 if optional
elsif optional && writers_schema.schemas.size == 2
# Optimize for the common case of a union that's just an optional field
index_of_schema = datum.nil? ? 0 : 1
- elsif self.class.optimize && optional && datum.nil?
- index_of_schema = 0
else
index_of_schema = writers_schema.schemas.find_index do |schema|
Avro::Schema.validate(schema, datum)
end
end