你可以为 Ruby 中的 map(&：method) 语法提供参数吗

发布于05月17日

您可能熟悉以下Ruby速记(a是一个数组):

a.map(&:method)

例如，在irb中try 以下操作:

>> a=[:a, 'a', 1, 1.0]
=> [:a, "a", 1, 1.0]
>> a.map(&:class)
=> [Symbol, String, Fixnum, Float]

语法a.map(&:class)是a.map {|x| x.class}的简写.

在"What does map(&:name) mean in Ruby?"中阅读更多关于这种语法的内容.

通过语法&:class，可以对每个数组元素进行方法调用class.

我的问题是:你能为方法调用提供参数吗？如果是，怎么做？

例如，如何转换以下语法

a = [1,3,5,7,9]
a.map {|x| x + 2}

到&:语法？

我并不是说&:语法更好.

我假设你知道+是整数类上的一个方法.你可以在irb中try 以下方法:

>> a=1
=> 1
>> a+(1)
=> 2
>> a.send(:+, 1)
=> 2

推荐答案

您可以在Symbol上创建一个简单的补丁，如下所示:

class Symbol
  def with(*args, &block)
    ->(caller, *rest) { caller.send(self, *rest, *args, &block) }
  end
end

这不仅能让你做到:

a = [1,3,5,7,9]
a.map(&:+.with(2))
# => [3, 5, 7, 9, 11]

还有很多其他很酷的东西，比如传递多个参数:

arr = ["abc", "babc", "great", "fruit"]
arr.map(&:center.with(20, '*'))
# => ["********abc*********", "********babc********", "*******great********", "*******fruit********"]
arr.map(&:[].with(1, 3))
# => ["bc", "abc", "rea", "rui"]
arr.map(&:[].with(/a(.*)/))
# => ["abc", "abc", "at", nil] 
arr.map(&:[].with(/a(.*)/, 1))
# => ["bc", "bc", "t", nil]

甚至可以使用inject，它将两个参数传递给块:

%w(abecd ab cd).inject(&:gsub.with('cde'))
# => "cdeeecde"

或者一些超级酷的东西，比如通过[速记]模块to速记模块:

[['0', '1'], ['2', '3']].map(&:map.with(&:to_i))
# => [[0, 1], [2, 3]]
[%w(a b), %w(c d)].map(&:inject.with(&:+))
# => ["ab", "cd"] 
[(1..5), (6..10)].map(&:map.with(&:*.with(2)))
# => [[2, 4, 6, 8, 10], [12, 14, 16, 18, 20]]

Here is a conversation I had with @ArupRakshit explaining it further:
Can you supply arguments to the map(&:method) syntax in Ruby?

正如@amcaplan在comment below中所建议的，如果将with方法重命名为call，可以创建一个更短的语法.在本例中，ruby为这个特殊方法提供了一个内置的快捷方式.

所以你可以这样使用上面的内容:

class Symbol
  def call(*args, &block)
    ->(caller, *rest) { caller.send(self, *rest, *args, &block) }
  end
end

a = [1,3,5,7,9]
a.map(&:+.(2))
# => [3, 5, 7, 9, 11] 

[(1..5), (6..10)].map(&:map.(&:*.(2)))
# => [[2, 4, 6, 8, 10], [12, 14, 16, 18, 20]]