欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

ruby迭代map的简洁写法实现原理分析

程序员文章站 2022-05-25 22:32:57
简便方法的用法 现有一个字符串列表,需要对其中的每个字符串执行转换大写操作,我们可以用一个简便写法来完成。 复制代码 代码如下: name_list = ["char...

简便方法的用法

现有一个字符串列表,需要对其中的每个字符串执行转换大写操作,我们可以用一个简便写法来完成。

复制代码 代码如下:

name_list = ["chareice", "angel"]
name_list.map(&:upcase)
# => ["chareice", "angel"]

这个写法等同于
复制代码 代码如下:

name_list.map do {|name| name.upcase}

简便写法带来的是很明显的效率提升,可是这看似魔术一般的参数,背后的原理是怎样的呢?

&符号

如果把上面方法调用的&符号去掉,可以很明显得看到,是把:upcase这个符号传到方法中,作为方法的参数。

实际上,&符号代表的是块转变为proc(block-to-proc conversion)。我们看下面的一个例子。

复制代码 代码如下:

def capture_block(&block)
  block.call
end

capture_block { puts "我有一只小毛驴,我从来也不骑。" }
# => 我有一只小毛驴,我从来也不骑。

我们运行capture_block函数,给它传递一个代码块,代码块会经&符号的转换变为一个proc对象传递到函数中,在上面的例子中就是block变量。如果我们输出一下block的class,输出的结果会是proc。

你也可以将一个proc对象传递给capture_block来代替代码块.

复制代码 代码如下:

p = proc.new { puts "又给一只小毛驴" }
capture_block(&p)
# => 又给一只小毛驴

这里看来&符号是多余的,完全可以去掉&,运行的结果也是一样。

&符号做了什么?

以capture_block(&p)调用为例。

1.触发p的to_proc方法。
2.告诉ruby解释器,将to_proc方法返回的结果当做本次函数调用的block。

如果同时使用了&符号和传入了block给一个函数,ruby会报错。

复制代码 代码如下:

capture_block(&p) { puts "传给一个block" }
#=>syntaxerror: (irb):30: both block arg and actual block given

所以将一个proc对象传给&符号,它会调用proc对象的to_proc方法,返回它自己,然后把它当做方法调用的block传递给方法。

&:upcase是什么?

知道了&符号的作用后,我们可以看到,&:upcase是先调用了:upcase对象的to_proc方法。

:upcase的to_proc方法实现如下:

复制代码 代码如下:

class symbol
  def to_proc
    proc.new {|obj| obj.send(self) }
  end
end

这下结果就很清楚了,symbol#to_proc会返回一个带参数的proc对象,proc对象所做的是为使用这个proc对象的对象发送调用名字为该符号的方法。