Custom class based range : Comparable « Collections « Ruby

1. Ruby
2. Collections
3. Comparable

Custom class based range

  class MyCapacity
    include Comparable

    attr :volume

    def initialize(volume)  # 0..9
      @volume = volume
    end

    def inspect
      '#' * @volume
    end

    def <=>(other)
      self.volume <=> other.volume
    end

    def succ
      raise(IndexError, "Volume too big") if @volume >= 9
      MyCapacity.new(@volume.succ)
    end
  end

medium_volume = MyCapacity.new(4)..MyCapacity.new(7)
medium_volume.to_a
medium_volume.include?(MyCapacity.new(3))

 

Related examples in the same category