BestFirstSearch: minor optimization with HashMap

scalameta · May 4, 2024 · 7e32f2e · 7e32f2e
1 parent cee0e9c
commit 7e32f2e
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/scalafmt-core/shared/src/main/scala/org/scalafmt/internal/BestFirstSearch.scala b/scalafmt-core/shared/src/main/scala/org/scalafmt/internal/BestFirstSearch.scala
@@ -166,8 +166,7 @@ private class BestFirstSearch private (
           actualSplit.foreach { split =>
             val nextState = curr.next(split, allAltAreNL)
             val updateBest = !keepSlowStates && depth == 0 && split.isNL &&
-              !best.contains(curr.depth)
-            if (updateBest) best.update(curr.depth, nextState)
+              best.getOrElseUpdate(curr.depth, nextState).eq(nextState)
             style.runner.event(Enqueue(split))
             split.optimalAt match {
               case Some(OptimalToken(token, killOnFail))