在 ruby​​ 方法链的末尾获取对象

Question

已编辑

假设我有以下代码：

class Kid < ActiveRecord::Base
  has_one :calculator
  attr_accessible :favorite_num
end

class Teacher < ActiveRecord::Base
  has_one :calculator
  # does not have a favorite_num
end

class Calculator < ActiveRecord::Base
  belongs_to :kid
  belongs_to :teacher

  def compute(number)
    # takes the number passed in and multiplies it by
    # the favorite_num if it exists, otherwise returns 42
  end
end

我需要做的是在上面定义compute，以便像下面这样调用。

kid = Kid.first
=> #<User id: 1, favorite_num: 4, . . . >
kid.calculator.compute(5)
=> 20
teacher = Teacher.first
=> #<Teacher id: 1, . . . >
teacher.calculator.compute(5)
=> 42

一些额外的限制：

• 我不想在Kid 和Teacher 类中都添加compute 方法。这是因为：
  • 我想尽可能保持干燥，其他几个类的对象也有 Calculators 与之关联
  • 这样做需要对我正在使用的遗留代码进行大规模重构（代码周围散布着许多 calculator.compute 调用，通常以不太明显的方式调用）

原版

如果我有

object.foo

那么self 在foo 的上下文中的值是object。但如果我有：

object.foo.bar

如何在bar 的定义中引用object？ self 显然不起作用，因为它只是等于 object.foo 的返回值。

我无法更改foo 的实现。并且foo和bar不能在同一个类的对象上调用。

Answer 1

您所描述的方法链接类型要求方法foo 返回self。为了实现这一点，您必须更改方法定义或对其进行修补。更改方法定义很简单，只需将self 作为方法的最后一条语句即可。当然，这意味着您不能从该方法返回任何其他内容，因此这不是每个方法都可行的模式。我强烈建议不要使用猴子补丁，因为我认为由此产生的丑陋代码不值得。

然而，有一些方法可以解决这个问题，不需要修改原始方法。最简单的解决方案是将方法调用写在彼此的下面。它可能没有那么“性感”，但它可以完成工作。

object.foo
object.bar

如果您想省略重复的object，可以使用Object#tap。

object.tap{|o| o.foo }.bar

使用Symbol#to_proc 可以缩短一点。

object.tap(&:foo).bar