diff options
author | Kevin Deisz <kevin.deisz@gmail.com> | 2015-05-29 09:41:03 -0400 |
---|---|---|
committer | Kevin Deisz <kevin.deisz@gmail.com> | 2015-05-29 10:32:32 -0400 |
commit | 777fa257aaa962bf67f6e5522efaa420ea7dc88b (patch) | |
tree | 088161d48ab11cca58e09cb3d49c73a3f76db613 /activesupport/lib/active_support/core_ext | |
parent | 343dad9617e7448bf8effc7c70ae13bede38fd32 (diff) | |
download | rails-777fa257aaa962bf67f6e5522efaa420ea7dc88b.tar.gz rails-777fa257aaa962bf67f6e5522efaa420ea7dc88b.tar.bz2 rails-777fa257aaa962bf67f6e5522efaa420ea7dc88b.zip |
Allow Enumerable#pluck to take a splat.
This allows easier integration with ActiveRecord, such that
AR#pluck will now use Enumerable#pluck if the relation is loaded,
without needing to hit the database.
Diffstat (limited to 'activesupport/lib/active_support/core_ext')
-rw-r--r-- | activesupport/lib/active_support/core_ext/enumerable.rb | 11 |
1 files changed, 9 insertions, 2 deletions
diff --git a/activesupport/lib/active_support/core_ext/enumerable.rb b/activesupport/lib/active_support/core_ext/enumerable.rb index d28f26260e..fc7531d088 100644 --- a/activesupport/lib/active_support/core_ext/enumerable.rb +++ b/activesupport/lib/active_support/core_ext/enumerable.rb @@ -76,8 +76,15 @@ module Enumerable # # [{ name: "David" }, { name: "Rafael" }, { name: "Aaron" }].pluck(:name) # => ["David", "Rafael", "Aaron"] - def pluck(key) - map { |element| element[key] } + # + # [{ id: 1, name: "David" }, { id: 2, name: "Rafael" }].pluck(:id, :name) + # => [[1, "David"], [2, "Rafael"]] + def pluck(*keys) + if keys.many? + map { |element| keys.map { |key| element[key] } } + else + map { |element| element[keys.first] } + end end end |