fix: retry counting build tool digest

ckipp01 · Apr 19, 2024 · 73944e4 · 73944e4
1 parent dfa2bfe
commit 73944e4
Show file tree

Hide file tree

Showing 4 changed files with 22 additions and 4 deletions.
diff --git a/metals/src/main/scala/scala/meta/internal/builds/BloopInstall.scala b/metals/src/main/scala/scala/meta/internal/builds/BloopInstall.scala
@@ -161,7 +161,7 @@ final class BloopInstall(
       status: Status,
       buildTool: BloopInstallProvider,
   ): Unit = {
-    buildTool.digest(workspace).foreach { checksum =>
+    buildTool.digestWithRetry(workspace).foreach { checksum =>
       tables.digests.setStatus(checksum, status)
     }
   }

diff --git a/metals/src/main/scala/scala/meta/internal/builds/BuildTool.scala b/metals/src/main/scala/scala/meta/internal/builds/BuildTool.scala
@@ -1,5 +1,6 @@
 package scala.meta.internal.builds
 
+import java.io.IOException
 import java.nio.file.Files
 import java.nio.file.Path
 import java.nio.file.StandardCopyOption
@@ -9,7 +10,24 @@ import scala.meta.io.AbsolutePath
 
 trait BuildTool {
 
-  def digest(workspace: AbsolutePath): Option[String]
+  def digestWithRetry(
+      workspace: AbsolutePath,
+      retries: Int = 1,
+  ): Option[String] = {
+    try {
+      digest(workspace)
+    } catch {
+      case error: IOException =>
+        scribe.warn(
+          s"Failed to digest workspace $workspace for build tool $executableName with error: $error${if (retries > 0) ", will retry" else ""}"
+        )
+        if (retries > 0) {
+          digestWithRetry(workspace, retries = 0)
+        } else None
+    }
+  }
+
+  protected def digest(workspace: AbsolutePath): Option[String]
 
   protected lazy val tempDir: Path = {
     val dir = Files.createTempDirectory("metals")

diff --git a/metals/src/main/scala/scala/meta/internal/builds/WorkspaceReload.scala b/metals/src/main/scala/scala/meta/internal/builds/WorkspaceReload.scala
@@ -38,7 +38,7 @@ final class WorkspaceReload(
       status: Status,
       buildTool: BuildTool,
   ): Unit = {
-    buildTool.digest(workspace).foreach { checksum =>
+    buildTool.digestWithRetry(workspace).foreach { checksum =>
       tables.digests.setStatus(checksum, status)
     }
   }

diff --git a/metals/src/main/scala/scala/meta/internal/metals/MetalsLspService.scala b/metals/src/main/scala/scala/meta/internal/metals/MetalsLspService.scala
@@ -2091,7 +2091,7 @@ class MetalsLspService(
           ) =>
         BuildTool.IncompatibleVersion(buildTool)
       case _ =>
-        buildTool.digest(folder) match {
+        buildTool.digestWithRetry(folder) match {
           case Some(digest) =>
             BuildTool.Found(buildTool, digest)
           case None => BuildTool.NoChecksum(buildTool, folder)