Sha256: 71e2fc6b59b97d5899f4a5b34ded6c1f03a1c8aea403125c13cbd2b0e23574fb

Contents?: true

Size: 1.4 KB

Versions: 48

Compression:

Stored size: 1.4 KB

Contents

require 'thread_safe'

module ActiveRecord
  module Type
    class TypeMap # :nodoc:
      def initialize
        @mapping = {}
        @cache = ThreadSafe::Cache.new do |h, key|
          h.fetch_or_store(key, ThreadSafe::Cache.new)
        end
      end

      def lookup(lookup_key, *args)
        fetch(lookup_key, *args) { default_value }
      end

      def fetch(lookup_key, *args, &block)
        @cache[lookup_key].fetch_or_store(args) do
          perform_fetch(lookup_key, *args, &block)
        end
      end

      def register_type(key, value = nil, &block)
        raise ::ArgumentError unless value || block
        @cache.clear

        if block
          @mapping[key] = block
        else
          @mapping[key] = proc { value }
        end
      end

      def alias_type(key, target_key)
        register_type(key) do |sql_type, *args|
          metadata = sql_type[/\(.*\)/, 0]
          lookup("#{target_key}#{metadata}", *args)
        end
      end

      def clear
        @mapping.clear
      end

      private

      def perform_fetch(lookup_key, *args)
        matching_pair = @mapping.reverse_each.detect do |key, _|
          key === lookup_key
        end

        if matching_pair
          matching_pair.last.call(lookup_key, *args)
        else
          yield lookup_key, *args
        end
      end

      def default_value
        @default_value ||= Value.new
      end
    end
  end
end

Version data entries

48 entries across 47 versions & 8 rubygems

Version Path
activerecord-4.2.11.3 lib/active_record/type/type_map.rb
activerecord-4.2.11.2 lib/active_record/type/type_map.rb
activerecord-4.2.11.1 lib/active_record/type/type_map.rb
activerecord-4.2.11 lib/active_record/type/type_map.rb
activerecord-4.2.10 lib/active_record/type/type_map.rb
activerecord-4.2.10.rc1 lib/active_record/type/type_map.rb
activerecord-4.2.9 lib/active_record/type/type_map.rb
activerecord-4.2.9.rc2 lib/active_record/type/type_map.rb
activerecord-4.2.9.rc1 lib/active_record/type/type_map.rb
enju_leaf-1.2.1 vendor/bundle/ruby/2.3/gems/activerecord-4.2.8/lib/active_record/type/type_map.rb
activerecord-4.2.8 lib/active_record/type/type_map.rb
activerecord-4.2.8.rc1 lib/active_record/type/type_map.rb
activerecord-4.2.7.1 lib/active_record/type/type_map.rb
activerecord-4.2.7 lib/active_record/type/type_map.rb
activerecord-4.2.7.rc1 lib/active_record/type/type_map.rb
ish_lib_manager-0.0.1 test/dummy/vendor/bundle/ruby/2.3.0/gems/activerecord-4.2.6/lib/active_record/type/type_map.rb
activerecord-4.2.6 lib/active_record/type/type_map.rb
activerecord-4.2.6.rc1 lib/active_record/type/type_map.rb
activerecord-4.2.5.2 lib/active_record/type/type_map.rb
activejob-lock-0.0.2 rails/activerecord/lib/active_record/type/type_map.rb