Переопределение оператора космического корабля Руби <=> - PullRequest
Новая
       23

Переопределение оператора космического корабля Руби <=>

4 голосов
/ 17 июня 2010

Я пытаюсь переопределить оператор Руби <=> (космический корабль), чтобы сортировать яблоки и апельсины так, чтобы яблоки были отсортированы по весу, а апельсины - по сладости. Вот так: 

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

Но это не работает, кто-то может сказать, что я здесь делаю неправильно, или лучший способ сделать это?

Ответы [ 2 ]

11 голосов
/ 17 июня 2010

Ваша проблема в том, что вы инициализируете только одно из свойств с обеих сторон, другое все равно будет nil. nil не обрабатывается в методе Array#<=>, что приводит к уничтожению сортировки.

Есть несколько способов решить проблему, во-первых, что-то вроде этого 

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

nil.to_i дает вам 0, что позволит этой работе.

0 голосов
/ 01 мая 2013

Вероятно поздно, тем не менее ...

добавить следующий monkeypatch 

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

И поменяй тело на Fruity::<=> to

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