EM::PriorityQueue using RBTree

gistfile1.rb

module EventMachine
  class PriorityQueue < Queue
    def initialize(&comparator)
      super
      @items = Items.new(comparator)
    end

    class Items
      def initialize(comparator)
        require 'rbtree'
        @items = MultiRBTree.new
        @items.readjust(comparator)
      end
      def push(*args)
        args.each do |obj|
          @items[obj] = nil
        end
      end
      def shift
        value = @items.shift
        value ? value[0] : nil
      end
      def method_missing(name, *args, &block)
        @items.send(name, *args, &block)
      end
    end
  end
end