avoid unnecessary reads of file content during indexing
This commit is contained in:
+3
-3
@@ -25,7 +25,7 @@ import com.intellij.openapi.vfs.VirtualFile
|
||||
|
||||
class KotlinBinaryClassCache : Disposable {
|
||||
private class RequestCache {
|
||||
internal lateinit var virtualFile: VirtualFile
|
||||
internal var virtualFile: VirtualFile? = null
|
||||
internal var modificationStamp: Long = 0
|
||||
internal var virtualFileKotlinClass: VirtualFileKotlinClass? = null
|
||||
|
||||
@@ -53,7 +53,7 @@ class KotlinBinaryClassCache : Disposable {
|
||||
|
||||
companion object {
|
||||
|
||||
fun getKotlinBinaryClass(file: VirtualFile): KotlinJvmBinaryClass? {
|
||||
fun getKotlinBinaryClass(file: VirtualFile, fileContent: ByteArray? = null): KotlinJvmBinaryClass? {
|
||||
if (file.fileType !== JavaClassFileType.INSTANCE) return null
|
||||
|
||||
val service = ServiceManager.getService(KotlinBinaryClassCache::class.java)
|
||||
@@ -65,7 +65,7 @@ class KotlinBinaryClassCache : Disposable {
|
||||
else {
|
||||
val aClass = ApplicationManager.getApplication().runReadAction(Computable {
|
||||
//noinspection deprecation
|
||||
VirtualFileKotlinClass.create(file)
|
||||
VirtualFileKotlinClass.create(file, fileContent)
|
||||
})
|
||||
|
||||
return requestCache.cache(file, aClass)
|
||||
|
||||
+2
-2
@@ -54,12 +54,12 @@ public class VirtualFileKotlinClass private constructor(
|
||||
private val perfCounter = PerformanceCounter.create("Binary class from Kotlin file")
|
||||
|
||||
@Deprecated("Use KotlinBinaryClassCache")
|
||||
fun create(file: VirtualFile): VirtualFileKotlinClass? {
|
||||
fun create(file: VirtualFile, fileContent: ByteArray?): VirtualFileKotlinClass? {
|
||||
return perfCounter.time {
|
||||
assert(file.getFileType() == JavaClassFileType.INSTANCE) { "Trying to read binary data from a non-class file $file" }
|
||||
|
||||
try {
|
||||
val byteContent = file.contentsToByteArray(false)
|
||||
val byteContent = fileContent ?: file.contentsToByteArray(false)
|
||||
if (!byteContent.isEmpty()) {
|
||||
return@time FileBasedKotlinClass.create(byteContent) {
|
||||
name, header, innerClasses ->
|
||||
|
||||
@@ -18,7 +18,6 @@ package org.jetbrains.kotlin.idea.vfilefinder
|
||||
|
||||
import com.intellij.ide.highlighter.JavaClassFileType
|
||||
import com.intellij.openapi.diagnostic.Logger
|
||||
import com.intellij.openapi.vfs.VirtualFile
|
||||
import com.intellij.util.indexing.*
|
||||
import com.intellij.util.io.KeyDescriptor
|
||||
import org.jetbrains.kotlin.idea.decompiler.KotlinJavaScriptMetaFileType
|
||||
@@ -27,7 +26,7 @@ import org.jetbrains.kotlin.load.kotlin.KotlinBinaryClassCache
|
||||
import org.jetbrains.kotlin.name.FqName
|
||||
import java.io.DataInput
|
||||
import java.io.DataOutput
|
||||
import java.util.Collections
|
||||
import java.util.*
|
||||
|
||||
abstract class KotlinFileIndexBase<T>(private val classOfIndex: Class<T>) : ScalarIndexExtension<FqName>() {
|
||||
public val KEY: ID<FqName, Void> = ID.create(classOfIndex.canonicalName)
|
||||
@@ -50,10 +49,10 @@ abstract class KotlinFileIndexBase<T>(private val classOfIndex: Class<T>) : Scal
|
||||
|
||||
override fun getKeyDescriptor() = KEY_DESCRIPTOR
|
||||
|
||||
protected fun indexer(f: (VirtualFile) -> FqName?): DataIndexer<FqName, Void, FileContent> =
|
||||
protected fun indexer(f: (FileContent) -> FqName?): DataIndexer<FqName, Void, FileContent> =
|
||||
DataIndexer {
|
||||
try {
|
||||
val fqName = f(it.file)
|
||||
val fqName = f(it)
|
||||
if (fqName != null) {
|
||||
Collections.singletonMap<FqName, Void>(fqName, null)
|
||||
}
|
||||
@@ -78,8 +77,8 @@ public object KotlinClassFileIndex : KotlinFileIndexBase<KotlinClassFileIndex>(K
|
||||
|
||||
private val VERSION = 2
|
||||
|
||||
private val INDEXER = indexer() { file ->
|
||||
val kotlinClass = KotlinBinaryClassCache.getKotlinBinaryClass(file)
|
||||
private val INDEXER = indexer() { fileContent ->
|
||||
val kotlinClass = KotlinBinaryClassCache.getKotlinBinaryClass(fileContent.file, fileContent.content)
|
||||
if (kotlinClass != null && kotlinClass.classHeader.isCompatibleAbiVersion) kotlinClass.classId.asSingleFqName() else null
|
||||
}
|
||||
}
|
||||
@@ -94,7 +93,9 @@ public object KotlinJavaScriptMetaFileIndex : KotlinFileIndexBase<KotlinJavaScri
|
||||
|
||||
private val VERSION = 2
|
||||
|
||||
private val INDEXER = indexer() { file ->
|
||||
if (file.fileType == KotlinJavaScriptMetaFileType) JsMetaFileUtils.getClassFqName(file) else null
|
||||
private val INDEXER = indexer() { fileContent ->
|
||||
if (fileContent.fileType == KotlinJavaScriptMetaFileType) JsMetaFileUtils.getClassFqName(fileContent.file) else null
|
||||
}
|
||||
|
||||
override fun dependsOnFileContent(): Boolean = false
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user