Merge pull request #202 from modelix/indexer-health-check

feat(mps-model-server-plugin): 'indexer' health check
modelix · Aug 13, 2023 · 1f449e5 · 1f449e5
2 parents 84a8886 + cdcb83b
commit 1f449e5
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 0 deletions.
diff --git a/mps-model-server-plugin/build.gradle.kts b/mps-model-server-plugin/build.gradle.kts
@@ -9,6 +9,7 @@ dependencies {
     compileOnly("com.jetbrains:mps-openapi:2021.1.4")
     compileOnly("com.jetbrains:mps-core:2021.1.4")
     compileOnly("com.jetbrains:mps-environment:2021.1.4")
+    compileOnly("com.jetbrains:mps-platform:2021.1.4")
 }
 
 // Configure Gradle IntelliJ Plugin

diff --git a/mps-model-server-plugin/src/main/kotlin/org/modelix/model/server/mps/MPSModelServer.kt b/mps-model-server-plugin/src/main/kotlin/org/modelix/model/server/mps/MPSModelServer.kt
@@ -19,7 +19,9 @@ import com.intellij.ide.plugins.IdeaPluginDescriptor
 import com.intellij.openapi.Disposable
 import com.intellij.openapi.components.Service
 import com.intellij.openapi.components.service
+import com.intellij.openapi.project.DumbService
 import com.intellij.openapi.project.Project
+import jetbrains.mps.ide.project.ProjectHelper
 import jetbrains.mps.project.ProjectBase
 import jetbrains.mps.project.ProjectManager
 import jetbrains.mps.smodel.MPSModuleRepository
@@ -46,6 +48,28 @@ class MPSModelServer : Disposable {
             server = LightModelServer.builder()
                 .port(48305)
                 .rootNode(rootNodeProvider)
+                .healthCheck(object : LightModelServer.IHealthCheck {
+                    override val id: String
+                        get() = "indexer"
+                    override val enabledByDefault: Boolean
+                        get() = false
+
+                    override fun run(output: java.lang.StringBuilder): Boolean {
+                        var allSmart = true
+                        val projects = ProjectManager.getInstance().openedProjects
+                        for (project in projects) {
+                            project.repository.modelAccess.runReadAction {
+                                val indexerDone =
+                                    !DumbService.getInstance(ProjectHelper.toIdeaProject(project)).isDumb
+                                if (!indexerDone) {
+                                    output.append("  indexer running on project ").append(project.name)
+                                    allSmart = false
+                                }
+                            }
+                        }
+                        return allSmart
+                    }
+                })
                 .healthCheck(object : LightModelServer.IHealthCheck {
                     override val id: String
                         get() = "projects"