Contents

require_relative './enum'
require_relative '../../exceptions'

module Dentaku
  module AST
    class Pluck < Enum
      def self.min_param_count
        2
      end

      def self.max_param_count
        2
      end

      def value(context = {})
        collection = Array(@args[0].value(context))
        unless collection.all? { |elem| elem.is_a?(Hash) }
          raise ArgumentError.for(:incompatible_type, value: collection),
                'PLUCK() requires first argument to be an array of hashes'
        end

        pluck_path = @args[1].identifier

        collection.map { |h| h.transform_keys(&:to_s)[pluck_path] }
      end
    end
  end
end

Dentaku::AST::Function.register_class(:pluck, Dentaku::AST::Pluck)

Version data entries

5 entries across 5 versions & 2 rubygems

Version	Path
dentaku-3.5.3	lib/dentaku/ast/functions/pluck.rb
dentaku-3.5.2	lib/dentaku/ast/functions/pluck.rb
dentaku_zevo-3.5.2	lib/dentaku/ast/functions/pluck.rb
dentaku-3.5.1	lib/dentaku/ast/functions/pluck.rb
dentaku-3.5.0	lib/dentaku/ast/functions/pluck.rb