Contents

# frozen_string_literal: true

require 'set'

module ROM
  module SQL
    class Schema < ROM::Schema
      # @api public
      class IndexDSL
        extend Initializer

        option :attr_class

        attr_reader :registry

        # @api private
        def initialize(*, **, &)
          super

          @registry = []

          instance_exec(&)
        end

        # @api public
        def index(*attributes, **options)
          registry << [attributes, options]
        end

        # @api private
        def call(schema_name, attrs)
          attributes = attrs.map do |attr|
            attr_class.new(attr[:type], **(attr[:options] || {})).meta(source: schema_name)
          end

          registry.to_set do |attr_names, options|
            build_index(attributes, attr_names, options)
          end
        end

        private

        # @api private
        def build_index(attributes, attr_names, options)
          index_attributes = attr_names.map do |name|
            attributes.find { |a| a.name == name }.unwrap
          end

          Index.new(index_attributes, **options)
        end
      end
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
rom-sql-3.7.0	lib/rom/sql/schema/index_dsl.rb