Sha256: 6ffe882905a15fe38b3b2f2829cc225c49187974013b792033f1428860511af1

Contents?: true

Size: 1.68 KB

Versions: 9

Compression:

Stored size: 1.68 KB

Contents

# frozen_string_literal: true

require 'rom/initializer'
require 'rom/data_proxy'

module ROM
  # A helper module that adds data-proxy behavior to an enumerable object
  #
  # This module is intended to be used by gateways
  #
  # Class that includes this module can define `row_proc` class method which
  # must return a proc-like object which will be used to process each element
  # in the enumerable
  #
  # @example
  #   class MyDataset
  #     include ROM::EnumerableDataset
  #
  #     def self.row_proc
  #       -> tuple { tuple.each_with_object({}) { |(k,v), h| h[k.to_sym] = v } }
  #     end
  #   end
  #
  #   ds = MyDataset.new([{ 'name' => 'Jane' }, [:name])
  #   ds.to_a # => { :name => 'Jane' }
  #
  # @api public
  module EnumerableDataset
    extend DataProxy::ClassMethods
    include Enumerable

    # Coerce a dataset to an array
    #
    # @return [Array]
    #
    # @api public
    alias_method :to_ary, :to_a

    # Included hook which extends a class with DataProxy behavior
    #
    # This module can also be included into other modules so we apply the
    # extension only for classes
    #
    # @api private
    def self.included(klass)
      return unless klass.is_a?(Class)

      klass.class_eval do
        extend Initializer
        include DataProxy

        param :data
      end
    end

    forward :take

    %i[
      chunk collect collect_concat drop_while find_all flat_map
      grep map reject select sort sort_by take_while
    ].each do |method|
      class_eval <<-RUBY, __FILE__, __LINE__ + 1
        def #{method}(*args, &block)
          return to_enum unless block
          self.class.new(super(*args, &block), **options)
        end
      RUBY
    end
  end
end

Version data entries

9 entries across 9 versions & 1 rubygems

Version Path
rom-core-5.3.2 lib/rom/enumerable_dataset.rb
rom-core-5.3.1 lib/rom/enumerable_dataset.rb
rom-core-5.3.0 lib/rom/enumerable_dataset.rb
rom-core-5.2.6 lib/rom/enumerable_dataset.rb
rom-core-5.2.5 lib/rom/enumerable_dataset.rb
rom-core-5.2.4 lib/rom/enumerable_dataset.rb
rom-core-5.2.3 lib/rom/enumerable_dataset.rb
rom-core-5.2.2 lib/rom/enumerable_dataset.rb
rom-core-5.2.1 lib/rom/enumerable_dataset.rb