activesupport/lib/active_support/core_ext/array/uniq_by.rb


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17

class Array
  # Return an unique array based on the criteria given as a proc.
  #
  #   [1, 2, 3, 4].uniq_by { |i| i.odd? }
  #   #=> [1, 2]
  #
  def uniq_by
    hash, array = {}, []
    each { |i| hash[yield(i)] ||= (array << i) }
    array
  end

  # Same as uniq_by, but modifies self.
  def uniq_by!
    replace(uniq_by{ |i| yield(i) })
  end
end