Ruby - 修改类

Question

我写了一个简单的方法来检查一个数字是否是 Armstrong number .但是现在我想修改默认的 Number 类放置方法。所以，我有代码:

def is_an(number)
    (number.to_s.split(//).map(&:to_i).inject{|x,y|x+y**(number.size-1)}) == number ? true : false
end
p is_an(153)

我想将它用作方法:153.is_a?那么，我该怎么做呢？

class Number
   def is_an
      ??? How to use object data over here? ???
   end
end

非常感谢阅读。

Answer 1

合并@mikej 的回答，加上用self 替换number:

class Fixnum
  def is_an
    digits = self.to_s.split(//).map(&:to_i)
    digits.inject(0) { |x,y| x+y**digits.size } == self
  end
end

但我建议更改名称，使其更像 Ruby。而不是 #is_an，它不是很具描述性，#armstrong? 怎么样，然后你可以调用:

153.armstrong?