Ensure the Scala Ordering is used in the java PriorityQueue

scalameta · Aug 30, 2018 · 1f85bcb · 1f85bcb
1 parent 80593ad
commit 1f85bcb
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/scalafmt-core/jvm/src/main/scala/org/scalafmt/internal/PriorityQueue.scala b/scalafmt-core/jvm/src/main/scala/org/scalafmt/internal/PriorityQueue.scala
@@ -9,8 +9,9 @@ package org.scalafmt.internal
   *
   * @tparam T the values inside the queue
   */
-class PriorityQueue[T] {
-  private[this] val q = new java.util.PriorityQueue[T]
+class PriorityQueue[T <: Ordered[T]] {
+  private[this] val q =
+    new java.util.PriorityQueue[T](11, (t: T, t1: T) => t1.compare(t))
 
   def dequeueAll: Unit = q.clear()
 
@@ -20,9 +21,12 @@ class PriorityQueue[T] {
 
   def enqueue(x: T): Unit = q.add(x)
 
-  def +=(x: T): Unit = q.add(x)
+  def +=(x: T): PriorityQueue[T] = {
+    q.add(x)
+    this
+  }
 
-  def nonEmpty: Boolean = !q.isEmpty
+  def nonEmpty: Boolean = !isEmpty
 
   def isEmpty: Boolean = q.isEmpty