Ruby中如何覆盖太空船运算符<=>?

5

我正在尝试覆盖Ruby的<=>(太空舱)操作符,以便按重量排序苹果并将其放在第一位,其次是按甜度排序橙子。就像这样:

module Fruity
  attr_accessor :weight, :sweetness

  def <=>(other)
    # use Array#<=> to compare the attributes
    [self.weight, self.sweetness] <=> [other.weight, other.sweetness]
  end
  include Comparable
end

class Apple
include Fruity

def initialize(w)
  self.weight = w
end

end

class Orange
include Fruity

def initialize(s)
  self.sweetness = s
end

end

fruits = [Apple.new(2),Orange.new(4),Apple.new(6),Orange.new(9),Apple.new(1),Orange.new(22)]

p fruits

#should work?
p fruits.sort

但这并不起作用,有人能告诉我在这里做错了什么,或者有更好的方法吗?

更新:事实证明我走错了方向。提示:答案在这里——http://blog.hasmanythrough.com/2008/8/17/sorting-things-out - Eric Steen
2个回答

12

你的问题是你只初始化了一侧的属性,而另一侧仍将为nil。在Array#<=>方法中没有处理nil,这最终导致了排序失败。

有几种方法可以解决这个问题,第一种方法可能是这样的:

[self.weight.to_i, self.sweetness.to_i] <=> [other.weight.to_i, other.sweetness.to_i]

nil.to_i会得到0,这样就可以实现此功能。


谢谢,它起作用了。它们按重量和甜度排序,但是按降序排列。有没有一种升序排序的“最佳”方法? - Eric Steen
这给了我橙子先然后是苹果,但我想先得到按升序排序的苹果,然后是按升序排序的橙子。仍然没有成功。有人有任何想法吗? - Eric Steen
好的,那么为什么Array#sort方法不使用自己的<=>方法来执行比较,而是使用了我重写的方法?这难道不违反了Ruby的动态方法查找吗? - Eric Steen
你正在覆盖水果上的 <=>,而不是数组。 - Matt Briggs
更新:事实证明我走错了方向。提示:答案在这里——blog.hasmanythrough.com/2008/8/17/sorting-things-out - Eric Steen
显示剩余2条评论

-1

可能有点晚了,但是...

添加以下的猴子补丁

class Array
  def to_i(default=Float::INFINITY)
    self.map do |element|
      element.nil? ? default : element.to_i
    end
  end
end

并将 Fruity::<=> 的主体更改为

[self.weight, self.sweetness].to_i <=> [other.weight, other.sweetness].to_i

抱歉,但这是错误的建议。(猴子补丁总是如此,但这个问题可以更加优雅地解决)。像这样的猴子补丁可能会引起许多其他问题。 - Marian Theisen

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接