Welcome to OGeek Q&A Community for programmer and developer-Open, Learning and Share
Welcome To Ask or Share your Answers For Others

Categories

0 votes
265 views
in Technique[技术] by (71.8m points)

ruby - Create own Array#map like method

I want to create an instance method like Array#my_map and that method should behavior of the original Array#map method. I want to same output from new method as below:

arr = [1, 2, 3, 4]
arr.new_map_method do |x|
 x + 1
end # => [2, 3, 4, 5]

arr.new_map_method(&:to_s) # => ["1", "2", "3", "4"]
See Question&Answers more detail:os

与恶龙缠斗过久,自身亦成为恶龙;凝视深渊过久,深渊将回以凝视…
Welcome To Ask or Share your Answers For Others

1 Reply

0 votes
by (71.8m points)

The easiest way of creating a method with the same behavior of a different method is a simple alias:

class Array
  alias_method :new_map_method, :map
end

If, for whatever strange reason, you don't want to use map, you can use inject instead:

class Array
  def new_map_method
    return enum_for(__callee__) unless block_given?
    inject([]) {|acc, el| acc << yield(el) }
  end
end

与恶龙缠斗过久,自身亦成为恶龙;凝视深渊过久,深渊将回以凝视…
OGeek|极客中国-欢迎来到极客的世界,一个免费开放的程序员编程交流平台!开放,进步,分享!让技术改变生活,让极客改变未来! Welcome to OGeek Q&A Community for programmer and developer-Open, Learning and Share
Click Here to Ask a Question

...