Where should I generate the stub tree?

yuriy.artamonov · April 21, 2025, 1:49pm

You also need to store soft reference here, take a look at example in Angular plugin

Angular/src/org/angular2/lang/metadata/psi/MetadataFileImpl.kt

master


      
          
          override fun getLanguage(): Language {
            return MetadataJsonLanguage
          }
          
          override fun getChildren(): Array<PsiElement> {
            val root = stubTree.root as MetadataFileStubImpl
            return root.childrenStubs.map { it.psi }.toTypedArray<PsiElement>()
          }
          
          override fun getStubTree(): StubTree {
            ApplicationManager.getApplication().assertReadAccessAllowed()
          
            myStub?.get()?.let { return it }
          
            // build newStub out of lock to avoid deadlock
            var newStubTree = StubTreeLoader.getInstance().readOrBuild(project, virtualFile, this) as StubTree?
            if (newStubTree == null) {
              if (LOG.isDebugEnabled) {
                LOG.debug("No stub for class file in index: " + virtualFile.presentableUrl)
              }