lib/avromatic/model/message_decoder.rb in avromatic-0.29.1 vs lib/avromatic/model/message_decoder.rb in avromatic-0.30.0

@@ -55,9 +55,14 @@
       def decode_hash(*args)
         model, message_key, message_value = extract_decode_args(*args)
         model.avro_message_attributes(message_key, message_value)
       end
 
+      # @return [Avromatic model class]
+      def model(*args)
+        extract_decode_args(*args).first
+      end
+
       private
 
       attr_reader :schema_names_by_id, :model_map, :schema_registry
 
       def extract_key_and_value(*args)