Override operatore nave spaziale di Ruby <=>
-
27-09-2019 - |
Domanda
sto cercando di ignorare operatore di Ruby <=> (astronave) per le mele e le arance di ordinamento in modo che le mele vengono prima ordinati in peso, e le arance secondo, scelte per la dolcezza. In questo modo:
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
Ma questo non funziona, qualcuno può dire quello che sto facendo male qui, o un modo migliore per fare questo?
Soluzione
Il tuo problema è che si sta solo in fase di inizializzazione una delle proprietà su entrambi i lati, l'altra sarà ancora nil
. nil
non viene gestita nel metodo Array#<=>
, che finisce per uccidere il genere.
Ci sono alcuni modi per gestire il problema in primo luogo sarebbe qualcosa di simile
[self.weight.to_i, self.sweetness.to_i] <=> [other.weight.to_i, other.sweetness.to_i]
nil.to_i
dà 0
, che vi permetterà di questo lavoro.
Altri suggerimenti
Probabilmente in ritardo, comunque ...
aggiungere il seguente monkeypatch
class Array
def to_i(default=Float::INFINITY)
self.map do |element|
element.nil? ? default : element.to_i
end
end
end
E cambiare il corpo di Fruity::<=> to
[self.weight, self.sweetness].to_i <=> [other.weight, other.sweetness].to_i