12 lines
270 B
Ruby
12 lines
270 B
Ruby
|
module Enumerable
|
||
|
|
def flat_map
|
||
|
|
return to_enum(:flat_map) unless block_given?
|
||
|
|
r = []
|
||
|
|
each do |*args|
|
||
|
|
result = yield(*args)
|
||
|
|
result.respond_to?(:to_ary) ? r.concat(result) : r.push(result)
|
||
|
|
end
|
||
|
|
r
|
||
|
|
end unless method_defined?(:flat_map)
|
||
|
|
end
|