Rename ClassDataProvider -> ClassDataWithSource

This commit is contained in:
Alexander Udalov
2015-09-22 17:45:00 +03:00
parent 0ef705f073
commit e52a9b9fb8
9 changed files with 43 additions and 57 deletions
@@ -26,8 +26,8 @@ import org.jetbrains.kotlin.name.Name;
import org.jetbrains.kotlin.resolve.scopes.ChainedScope; import org.jetbrains.kotlin.resolve.scopes.ChainedScope;
import org.jetbrains.kotlin.resolve.scopes.JetScope; import org.jetbrains.kotlin.resolve.scopes.JetScope;
import org.jetbrains.kotlin.serialization.ClassData; import org.jetbrains.kotlin.serialization.ClassData;
import org.jetbrains.kotlin.serialization.ClassDataWithSource;
import org.jetbrains.kotlin.serialization.PackageData; import org.jetbrains.kotlin.serialization.PackageData;
import org.jetbrains.kotlin.serialization.deserialization.ClassDataProvider;
import org.jetbrains.kotlin.serialization.deserialization.DeserializationComponents; import org.jetbrains.kotlin.serialization.deserialization.DeserializationComponents;
import org.jetbrains.kotlin.serialization.deserialization.ErrorReporter; import org.jetbrains.kotlin.serialization.deserialization.ErrorReporter;
import org.jetbrains.kotlin.serialization.deserialization.descriptors.DeserializedPackageMemberScope; import org.jetbrains.kotlin.serialization.deserialization.descriptors.DeserializedPackageMemberScope;
@@ -63,8 +63,10 @@ public final class DeserializedDescriptorResolver {
if (data != null) { if (data != null) {
ClassData classData = JvmProtoBufUtil.readClassDataFrom(data); ClassData classData = JvmProtoBufUtil.readClassDataFrom(data);
KotlinJvmBinarySourceElement sourceElement = new KotlinJvmBinarySourceElement(kotlinClass); KotlinJvmBinarySourceElement sourceElement = new KotlinJvmBinarySourceElement(kotlinClass);
ClassDataProvider classDataProvider = new ClassDataProvider(classData, sourceElement); return components.getClassDeserializer().deserializeClass(
return components.getClassDeserializer().deserializeClass(kotlinClass.getClassId(), classDataProvider); kotlinClass.getClassId(),
new ClassDataWithSource(classData, sourceElement)
);
} }
return null; return null;
} }
@@ -16,23 +16,22 @@
package org.jetbrains.kotlin.load.kotlin package org.jetbrains.kotlin.load.kotlin
import org.jetbrains.kotlin.load.kotlin.header.KotlinClassHeader
import org.jetbrains.kotlin.name.ClassId import org.jetbrains.kotlin.name.ClassId
import org.jetbrains.kotlin.serialization.ClassDataWithSource
import org.jetbrains.kotlin.serialization.deserialization.ClassDataFinder import org.jetbrains.kotlin.serialization.deserialization.ClassDataFinder
import org.jetbrains.kotlin.serialization.deserialization.ClassDataProvider
import org.jetbrains.kotlin.serialization.jvm.JvmProtoBufUtil import org.jetbrains.kotlin.serialization.jvm.JvmProtoBufUtil
public class JavaClassDataFinder( public class JavaClassDataFinder(
private val kotlinClassFinder: KotlinClassFinder, private val kotlinClassFinder: KotlinClassFinder,
private val deserializedDescriptorResolver: DeserializedDescriptorResolver private val deserializedDescriptorResolver: DeserializedDescriptorResolver
) : ClassDataFinder { ) : ClassDataFinder {
override fun findClassData(classId: ClassId): ClassDataProvider? { override fun findClassData(classId: ClassId): ClassDataWithSource? {
val kotlinJvmBinaryClass = kotlinClassFinder.findKotlinClass(classId) ?: return null val kotlinJvmBinaryClass = kotlinClassFinder.findKotlinClass(classId) ?: return null
assert(kotlinJvmBinaryClass.getClassId() == classId) { assert(kotlinJvmBinaryClass.getClassId() == classId) {
"Class with incorrect id found: expected $classId, actual ${kotlinJvmBinaryClass.getClassId()}" "Class with incorrect id found: expected $classId, actual ${kotlinJvmBinaryClass.getClassId()}"
} }
val data = deserializedDescriptorResolver.readData(kotlinJvmBinaryClass, DeserializedDescriptorResolver.KOTLIN_CLASS) ?: return null val data = deserializedDescriptorResolver.readData(kotlinJvmBinaryClass, DeserializedDescriptorResolver.KOTLIN_CLASS) ?: return null
val classData = JvmProtoBufUtil.readClassDataFrom(data) val classData = JvmProtoBufUtil.readClassDataFrom(data)
return ClassDataProvider(classData, KotlinJvmBinarySourceElement(kotlinJvmBinaryClass)) return ClassDataWithSource(classData, KotlinJvmBinarySourceElement(kotlinJvmBinaryClass))
} }
} }
@@ -16,6 +16,7 @@
package org.jetbrains.kotlin.serialization package org.jetbrains.kotlin.serialization
import org.jetbrains.kotlin.descriptors.SourceElement
import org.jetbrains.kotlin.serialization.deserialization.NameResolver import org.jetbrains.kotlin.serialization.deserialization.NameResolver
data class ClassData( data class ClassData(
@@ -23,6 +24,11 @@ data class ClassData(
val classProto: ProtoBuf.Class val classProto: ProtoBuf.Class
) )
data class ClassDataWithSource(
val classData: ClassData,
val sourceElement: SourceElement = SourceElement.NO_SOURCE
)
data class PackageData( data class PackageData(
val nameResolver: NameResolver, val nameResolver: NameResolver,
val packageProto: ProtoBuf.Package val packageProto: ProtoBuf.Package
@@ -19,8 +19,9 @@ package org.jetbrains.kotlin.serialization.deserialization;
import org.jetbrains.annotations.NotNull; import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable; import org.jetbrains.annotations.Nullable;
import org.jetbrains.kotlin.name.ClassId; import org.jetbrains.kotlin.name.ClassId;
import org.jetbrains.kotlin.serialization.ClassDataWithSource;
public interface ClassDataFinder { public interface ClassDataFinder {
@Nullable @Nullable
ClassDataProvider findClassData(@NotNull ClassId classId); ClassDataWithSource findClassData(@NotNull ClassId classId);
} }
@@ -1,25 +0,0 @@
/*
* Copyright 2010-2015 JetBrains s.r.o.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.jetbrains.kotlin.serialization.deserialization
import org.jetbrains.kotlin.descriptors.SourceElement
import org.jetbrains.kotlin.serialization.ClassData
public class ClassDataProvider(
public val classData: ClassData,
public val sourceElement: SourceElement = SourceElement.NO_SOURCE
)
@@ -18,50 +18,52 @@ package org.jetbrains.kotlin.serialization.deserialization
import org.jetbrains.kotlin.descriptors.ClassDescriptor import org.jetbrains.kotlin.descriptors.ClassDescriptor
import org.jetbrains.kotlin.name.ClassId import org.jetbrains.kotlin.name.ClassId
import org.jetbrains.kotlin.serialization.ClassDataWithSource
import org.jetbrains.kotlin.serialization.deserialization.descriptors.DeserializedClassDescriptor import org.jetbrains.kotlin.serialization.deserialization.descriptors.DeserializedClassDescriptor
public class ClassDeserializer(private val components: DeserializationComponents) { public class ClassDeserializer(private val components: DeserializationComponents) {
private val classes: (ClassKey) -> ClassDescriptor? = private val classes: (ClassKey) -> ClassDescriptor? =
components.storageManager.createMemoizedFunctionWithNullableValues { key -> createClass(key) } components.storageManager.createMemoizedFunctionWithNullableValues { key -> createClass(key) }
// Additional ClassData parameter is needed to avoid calling ClassDataFinder#findClassData() if it is already computed at call site // Additional ClassDataWithSource parameter is needed to avoid calling ClassDataFinder#findClassData()
public fun deserializeClass(classId: ClassId, classDataProvider: ClassDataProvider? = null): ClassDescriptor? = // if it is already computed at the call site
classes(ClassKey(classId, classDataProvider)) public fun deserializeClass(classId: ClassId, classDataWithSource: ClassDataWithSource? = null): ClassDescriptor? =
classes(ClassKey(classId, classDataWithSource))
private fun createClass(key: ClassKey): ClassDescriptor? { private fun createClass(key: ClassKey): ClassDescriptor? {
val classId = key.classId val classId = key.classId
components.fictitiousClassDescriptorFactory.createClass(classId)?.let { return it } components.fictitiousClassDescriptorFactory.createClass(classId)?.let { return it }
val classDataProvider = key.classDataProvider val (classData, sourceElement) = key.classDataWithSource
?: components.classDataFinder.findClassData(classId) ?: components.classDataFinder.findClassData(classId)
?: return null ?: return null
val (nameResolver, classProto) = classDataProvider.classData val (nameResolver, classProto) = classData
val outerContext = if (classId.isNestedClass()) { val outerContext = if (classId.isNestedClass) {
val outerClass = deserializeClass(classId.getOuterClassId()) as? DeserializedClassDescriptor ?: return null val outerClass = deserializeClass(classId.outerClassId) as? DeserializedClassDescriptor ?: return null
// Find the outer class first and check if he knows anything about the nested class we're looking for // Find the outer class first and check if he knows anything about the nested class we're looking for
if (!outerClass.hasNestedClass(classId.getShortClassName())) return null if (!outerClass.hasNestedClass(classId.shortClassName)) return null
outerClass.c outerClass.c
} }
else { else {
val fragments = components.packageFragmentProvider.getPackageFragments(classId.getPackageFqName()) val fragments = components.packageFragmentProvider.getPackageFragments(classId.packageFqName)
assert(fragments.size() == 1) { "There should be exactly one package: $fragments, class id is $classId" } assert(fragments.size() == 1) { "There should be exactly one package: $fragments, class id is $classId" }
val fragment = fragments.single() val fragment = fragments.single()
if (fragment is DeserializedPackageFragment) { if (fragment is DeserializedPackageFragment) {
// Similarly, verify that the containing package has information about this class // Similarly, verify that the containing package has information about this class
if (!fragment.hasTopLevelClass(classId.getShortClassName())) return null if (!fragment.hasTopLevelClass(classId.shortClassName)) return null
} }
components.createContext(fragment, nameResolver) components.createContext(fragment, nameResolver)
} }
return DeserializedClassDescriptor(outerContext, classProto, nameResolver, classDataProvider.sourceElement) return DeserializedClassDescriptor(outerContext, classProto, nameResolver, sourceElement)
} }
private data class ClassKey(val classId: ClassId, classDataProvider: ClassDataProvider?) { private data class ClassKey(val classId: ClassId, classDataWithSource: ClassDataWithSource?) {
// This property is not declared in the constructor because it shouldn't participate in equals/hashCode // This property is not declared in the constructor because it shouldn't participate in equals/hashCode
val classDataProvider: ClassDataProvider? = classDataProvider val classDataWithSource: ClassDataWithSource? = classDataWithSource
} }
} }
@@ -19,6 +19,7 @@ package org.jetbrains.kotlin.serialization.deserialization
import org.jetbrains.kotlin.descriptors.PackageFragmentProvider import org.jetbrains.kotlin.descriptors.PackageFragmentProvider
import org.jetbrains.kotlin.name.ClassId import org.jetbrains.kotlin.name.ClassId
import org.jetbrains.kotlin.serialization.ClassData import org.jetbrains.kotlin.serialization.ClassData
import org.jetbrains.kotlin.serialization.ClassDataWithSource
import org.jetbrains.kotlin.serialization.ProtoBuf import org.jetbrains.kotlin.serialization.ProtoBuf
import org.jetbrains.kotlin.serialization.SerializedResourcePaths import org.jetbrains.kotlin.serialization.SerializedResourcePaths
import java.io.InputStream import java.io.InputStream
@@ -28,7 +29,7 @@ public open class ResourceLoadingClassDataFinder(
private val serializedResourcePaths: SerializedResourcePaths, private val serializedResourcePaths: SerializedResourcePaths,
private val loadResource: (path: String) -> InputStream? private val loadResource: (path: String) -> InputStream?
) : ClassDataFinder { ) : ClassDataFinder {
override fun findClassData(classId: ClassId): ClassDataProvider? { override fun findClassData(classId: ClassId): ClassDataWithSource? {
val packageFragment = packageFragmentProvider.getPackageFragments(classId.getPackageFqName()).singleOrNull() val packageFragment = packageFragmentProvider.getPackageFragments(classId.getPackageFqName()).singleOrNull()
as? DeserializedPackageFragment ?: return null as? DeserializedPackageFragment ?: return null
@@ -38,6 +39,6 @@ public open class ResourceLoadingClassDataFinder(
packageFragment.nameResolver, packageFragment.nameResolver,
ProtoBuf.Class.parseFrom(stream, serializedResourcePaths.extensionRegistry) ProtoBuf.Class.parseFrom(stream, serializedResourcePaths.extensionRegistry)
) )
return ClassDataProvider(classData) return ClassDataWithSource(classData)
} }
} }
@@ -220,8 +220,8 @@ private class ClassClsStubBuilder(
} }
private fun createNestedClassStub(classBody: StubElement<out PsiElement>, nestedClassId: ClassId) { private fun createNestedClassStub(classBody: StubElement<out PsiElement>, nestedClassId: ClassId) {
val classDataProvider = c.components.classDataFinder.findClassData(nestedClassId)!! val classDataWithSource = c.components.classDataFinder.findClassData(nestedClassId)!!
val (nameResolver, classProto) = classDataProvider.classData val (nameResolver, classProto) = classDataWithSource.classData
createClassStub(classBody, classProto, nestedClassId, c.child(nameResolver)) createClassStub(classBody, classProto, nestedClassId, c.child(nameResolver))
} }
} }
@@ -27,8 +27,8 @@ import org.jetbrains.kotlin.load.kotlin.KotlinClassFinder
import org.jetbrains.kotlin.load.kotlin.KotlinJvmBinaryClass import org.jetbrains.kotlin.load.kotlin.KotlinJvmBinaryClass
import org.jetbrains.kotlin.name.ClassId import org.jetbrains.kotlin.name.ClassId
import org.jetbrains.kotlin.name.FqName import org.jetbrains.kotlin.name.FqName
import org.jetbrains.kotlin.serialization.ClassDataWithSource
import org.jetbrains.kotlin.serialization.deserialization.ClassDataFinder import org.jetbrains.kotlin.serialization.deserialization.ClassDataFinder
import org.jetbrains.kotlin.serialization.deserialization.ClassDataProvider
import org.jetbrains.kotlin.serialization.deserialization.LocalClassResolver import org.jetbrains.kotlin.serialization.deserialization.LocalClassResolver
import org.jetbrains.kotlin.serialization.deserialization.NameResolver import org.jetbrains.kotlin.serialization.deserialization.NameResolver
import org.jetbrains.kotlin.serialization.js.KotlinJavascriptSerializationUtil import org.jetbrains.kotlin.serialization.js.KotlinJavascriptSerializationUtil
@@ -71,7 +71,7 @@ class DirectoryBasedDataFinder(
val classFinder: DirectoryBasedClassFinder, val classFinder: DirectoryBasedClassFinder,
val log: Logger val log: Logger
) : ClassDataFinder { ) : ClassDataFinder {
override fun findClassData(classId: ClassId): ClassDataProvider? { override fun findClassData(classId: ClassId): ClassDataWithSource? {
val binaryClass = classFinder.findKotlinClass(classId) ?: return null val binaryClass = classFinder.findKotlinClass(classId) ?: return null
val data = binaryClass.getClassHeader().annotationData val data = binaryClass.getClassHeader().annotationData
if (data == null) { if (data == null) {
@@ -80,7 +80,7 @@ class DirectoryBasedDataFinder(
} }
val classData = JvmProtoBufUtil.readClassDataFrom(data) val classData = JvmProtoBufUtil.readClassDataFrom(data)
return ClassDataProvider(classData) return ClassDataWithSource(classData)
} }
} }
@@ -88,12 +88,12 @@ class DirectoryBasedKotlinJavaScriptDataFinder(
val classFinder: DirectoryBasedKotlinJavaScriptMetaFileFinder, val classFinder: DirectoryBasedKotlinJavaScriptMetaFileFinder,
val log: Logger val log: Logger
) : ClassDataFinder { ) : ClassDataFinder {
override fun findClassData(classId: ClassId): ClassDataProvider? { override fun findClassData(classId: ClassId): ClassDataWithSource? {
val file = classFinder.findKotlinJavascriptMetaFile(classId) ?: return null val file = classFinder.findKotlinJavascriptMetaFile(classId) ?: return null
val content = file.contentsToByteArray(false) val content = file.contentsToByteArray(false)
val classData = content.toClassData(classFinder.nameResolver) val classData = content.toClassData(classFinder.nameResolver)
return ClassDataProvider(classData) return ClassDataWithSource(classData)
} }
} }