Python: make functions suspend to obey the threading policy

GitOrigin-RevId: 90864b29bb9d986aa44717bdf178bbbfe4910f73

python/src/com/jetbrains/python/projectModel/BaseProjectModelService.kt

@@ -11,6 +11,8 @@ import com.intellij.platform.workspace.storage.EntityStorage
 import com.intellij.platform.workspace.storage.MutableEntityStorage
 import com.intellij.platform.workspace.storage.impl.url.toVirtualFileUrl
 import com.jetbrains.python.PyBundle
+import kotlinx.coroutines.Dispatchers
+import kotlinx.coroutines.withContext
 import org.jetbrains.annotations.NonNls
 import org.jetbrains.annotations.SystemIndependent
 import java.nio.file.Path
@@ -92,11 +94,11 @@ abstract class BaseProjectModelService<E : EntitySource, P : ExternalProject> {
     }
   }
 
-  private fun createProjectModel(
+  private suspend fun createProjectModel(
     project: Project,
     graph: List<ExternalProject>,
     source: EntitySource,
-  ): EntityStorage {
+  ): EntityStorage = withContext(Dispatchers.Default) {
     val fileUrlManager = project.workspaceModel.getVirtualFileUrlManager()
     val storage = MutableEntityStorage.create()
     for (extProject in graph) {
@@ -128,7 +130,7 @@ abstract class BaseProjectModelService<E : EntitySource, P : ExternalProject> {
         }
       }
     }
-    return storage
+    return@withContext storage
   }
 
   /**

python/src/com/jetbrains/python/projectModel/projectModel.kt

@@ -2,6 +2,9 @@
 package com.jetbrains.python.projectModel
 
 import com.intellij.openapi.util.io.FileUtil
+import com.intellij.util.concurrency.annotations.RequiresBackgroundThread
+import kotlinx.coroutines.Dispatchers
+import kotlinx.coroutines.withContext
 import java.nio.file.FileVisitResult
 import java.nio.file.Path
 import kotlin.io.path.ExperimentalPathApi
@@ -53,6 +56,7 @@ interface PythonProjectModelResolver<P : ExternalProject> {
    * this method should return `null` for `libs/` but module graphs containing *only* modules `project1` and `project2`
    * for the directories `project1/` and `project2` respectively, even if there is a dependency between them.
    */
+  @RequiresBackgroundThread
   fun discoverProjectRootSubgraph(root: Path): ExternalProjectGraph<P>?
 
   /**
@@ -73,7 +77,7 @@ interface PythonProjectModelResolver<P : ExternalProject> {
    * If these two projects are independents, there will be two graphs for `project1` and `project2` respectively.
    */
   @OptIn(ExperimentalPathApi::class)
-  fun discoverIndependentProjectGraphs(root: Path): List<ExternalProjectGraph<P>> {
+  suspend fun discoverIndependentProjectGraphs(root: Path): List<ExternalProjectGraph<P>> = withContext(Dispatchers.IO) {
     val graphs = mutableListOf<ExternalProjectGraph<P>>()
     root.visitFileTree {
       onPreVisitDirectory { dir, _ ->
@@ -87,7 +91,7 @@ interface PythonProjectModelResolver<P : ExternalProject> {
     }
 
     // TODO make sure that roots doesn't leave ijProjectRoot boundaries
-    return mergeRootsReferringToEachOther(graphs)
+    return@withContext mergeRootsReferringToEachOther(graphs)
   }
 }
 
@@ -103,6 +107,7 @@ interface ProjectModelSyncListener {
   fun onFinish(projectRoot: Path): Unit = Unit
 }
 
+@RequiresBackgroundThread
 private fun <P : ExternalProject> mergeRootsReferringToEachOther(roots: List<ExternalProjectGraph<P>>): List<ExternalProjectGraph<P>> {
   fun commonAncestorPath(paths: Iterable<Path>): Path {
     val normalized = paths.map { it.normalize() }