Anulación de operador de la nave espacial de Rubí <=>
-
27-09-2019 - |
Pregunta
Estoy tratando de anular operador de Ruby <=> (nave espacial) para ordenar las manzanas y naranjas para que las manzanas vienen primero ordenados por peso, y las naranjas en segundo lugar, ordenadas según la dulzura. De esta manera:
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
Pero esto no funciona, puede que alguien diga lo que estoy haciendo mal aquí, o una mejor manera de hacer esto?
Solución
Su problema se va a inicializar sólo una de las propiedades de cada lado, el otro seguirá siendo nil
. nil
no se maneja en el método Array#<=>
, que acaba matando a la clase.
Hay algunas maneras de manejar el problema primero sería algo como esto
[self.weight.to_i, self.sweetness.to_i] <=> [other.weight.to_i, other.sweetness.to_i]
nil.to_i
le da 0
, que le permitirá a este trabajo.
Otros consejos
Probablemente tarde, sin embargo, ...
añadir lo siguiente monkeypatch
class Array
def to_i(default=Float::INFINITY)
self.map do |element|
element.nil? ? default : element.to_i
end
end
end
Y cambiar el cuerpo de Fruity::<=> to
[self.weight, self.sweetness].to_i <=> [other.weight, other.sweetness].to_i