Sha256: 9fcf8c7022be823a8315093daf445e8c2e6e96cea3b9b25a48d7db89102610cc

Contents?: true

Size: 1.3 KB

Versions: 19

Compression:

Stored size: 1.3 KB

Contents

# frozen_string_literal: true

require "yaml"

module ActiveRecord
  module Coders # :nodoc:
    class YAMLColumn # :nodoc:
      attr_accessor :object_class

      def initialize(attr_name, object_class = Object)
        @attr_name = attr_name
        @object_class = object_class
        check_arity_of_constructor
      end

      def dump(obj)
        return if obj.nil?

        assert_valid_value(obj, action: "dump")
        YAML.dump obj
      end

      def load(yaml)
        return object_class.new if object_class != Object && yaml.nil?
        return yaml unless yaml.is_a?(String) && /^---/.match?(yaml)
        obj = YAML.load(yaml)

        assert_valid_value(obj, action: "load")
        obj ||= object_class.new if object_class != Object

        obj
      end

      def assert_valid_value(obj, action:)
        unless obj.nil? || obj.is_a?(object_class)
          raise SerializationTypeMismatch,
            "can't #{action} `#{@attr_name}`: was supposed to be a #{object_class}, but was a #{obj.class}. -- #{obj.inspect}"
        end
      end

      private
        def check_arity_of_constructor
          load(nil)
        rescue ArgumentError
          raise ArgumentError, "Cannot serialize #{object_class}. Classes passed to `serialize` must have a 0 argument constructor."
        end
    end
  end
end

Version data entries

19 entries across 19 versions & 1 rubygems

Version Path
activerecord-6.0.5 lib/active_record/coders/yaml_column.rb
activerecord-6.0.4.8 lib/active_record/coders/yaml_column.rb
activerecord-6.0.4.7 lib/active_record/coders/yaml_column.rb
activerecord-6.0.4.6 lib/active_record/coders/yaml_column.rb
activerecord-6.0.4.5 lib/active_record/coders/yaml_column.rb
activerecord-6.0.4.4 lib/active_record/coders/yaml_column.rb
activerecord-6.0.4.3 lib/active_record/coders/yaml_column.rb
activerecord-6.0.4.2 lib/active_record/coders/yaml_column.rb
activerecord-6.0.4.1 lib/active_record/coders/yaml_column.rb
activerecord-6.0.4 lib/active_record/coders/yaml_column.rb
activerecord-6.0.3.7 lib/active_record/coders/yaml_column.rb
activerecord-6.0.3.6 lib/active_record/coders/yaml_column.rb
activerecord-6.0.3.5 lib/active_record/coders/yaml_column.rb
activerecord-6.0.3.4 lib/active_record/coders/yaml_column.rb
activerecord-6.0.3.3 lib/active_record/coders/yaml_column.rb
activerecord-6.0.3.2 lib/active_record/coders/yaml_column.rb
activerecord-6.0.3.1 lib/active_record/coders/yaml_column.rb
activerecord-6.0.3 lib/active_record/coders/yaml_column.rb
activerecord-6.0.3.rc1 lib/active_record/coders/yaml_column.rb