Rename ClassDataProvider -> ClassDataWithSource
This commit is contained in:
@@ -16,6 +16,7 @@
|
||||
|
||||
package org.jetbrains.kotlin.serialization
|
||||
|
||||
import org.jetbrains.kotlin.descriptors.SourceElement
|
||||
import org.jetbrains.kotlin.serialization.deserialization.NameResolver
|
||||
|
||||
data class ClassData(
|
||||
@@ -23,6 +24,11 @@ data class ClassData(
|
||||
val classProto: ProtoBuf.Class
|
||||
)
|
||||
|
||||
data class ClassDataWithSource(
|
||||
val classData: ClassData,
|
||||
val sourceElement: SourceElement = SourceElement.NO_SOURCE
|
||||
)
|
||||
|
||||
data class PackageData(
|
||||
val nameResolver: NameResolver,
|
||||
val packageProto: ProtoBuf.Package
|
||||
|
||||
+2
-1
@@ -19,8 +19,9 @@ package org.jetbrains.kotlin.serialization.deserialization;
|
||||
import org.jetbrains.annotations.NotNull;
|
||||
import org.jetbrains.annotations.Nullable;
|
||||
import org.jetbrains.kotlin.name.ClassId;
|
||||
import org.jetbrains.kotlin.serialization.ClassDataWithSource;
|
||||
|
||||
public interface ClassDataFinder {
|
||||
@Nullable
|
||||
ClassDataProvider findClassData(@NotNull ClassId classId);
|
||||
ClassDataWithSource findClassData(@NotNull ClassId classId);
|
||||
}
|
||||
|
||||
-25
@@ -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
|
||||
)
|
||||
+17
-15
@@ -18,50 +18,52 @@ package org.jetbrains.kotlin.serialization.deserialization
|
||||
|
||||
import org.jetbrains.kotlin.descriptors.ClassDescriptor
|
||||
import org.jetbrains.kotlin.name.ClassId
|
||||
import org.jetbrains.kotlin.serialization.ClassDataWithSource
|
||||
import org.jetbrains.kotlin.serialization.deserialization.descriptors.DeserializedClassDescriptor
|
||||
|
||||
public class ClassDeserializer(private val components: DeserializationComponents) {
|
||||
private val classes: (ClassKey) -> ClassDescriptor? =
|
||||
components.storageManager.createMemoizedFunctionWithNullableValues { key -> createClass(key) }
|
||||
|
||||
// Additional ClassData parameter is needed to avoid calling ClassDataFinder#findClassData() if it is already computed at call site
|
||||
public fun deserializeClass(classId: ClassId, classDataProvider: ClassDataProvider? = null): ClassDescriptor? =
|
||||
classes(ClassKey(classId, classDataProvider))
|
||||
// Additional ClassDataWithSource parameter is needed to avoid calling ClassDataFinder#findClassData()
|
||||
// if it is already computed at the call site
|
||||
public fun deserializeClass(classId: ClassId, classDataWithSource: ClassDataWithSource? = null): ClassDescriptor? =
|
||||
classes(ClassKey(classId, classDataWithSource))
|
||||
|
||||
private fun createClass(key: ClassKey): ClassDescriptor? {
|
||||
val classId = key.classId
|
||||
components.fictitiousClassDescriptorFactory.createClass(classId)?.let { return it }
|
||||
val classDataProvider = key.classDataProvider
|
||||
?: components.classDataFinder.findClassData(classId)
|
||||
?: return null
|
||||
val (nameResolver, classProto) = classDataProvider.classData
|
||||
val (classData, sourceElement) = key.classDataWithSource
|
||||
?: components.classDataFinder.findClassData(classId)
|
||||
?: return null
|
||||
val (nameResolver, classProto) = classData
|
||||
|
||||
val outerContext = if (classId.isNestedClass()) {
|
||||
val outerClass = deserializeClass(classId.getOuterClassId()) as? DeserializedClassDescriptor ?: return null
|
||||
val outerContext = if (classId.isNestedClass) {
|
||||
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
|
||||
if (!outerClass.hasNestedClass(classId.getShortClassName())) return null
|
||||
if (!outerClass.hasNestedClass(classId.shortClassName)) return null
|
||||
|
||||
outerClass.c
|
||||
}
|
||||
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" }
|
||||
|
||||
val fragment = fragments.single()
|
||||
if (fragment is DeserializedPackageFragment) {
|
||||
// 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)
|
||||
}
|
||||
|
||||
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
|
||||
val classDataProvider: ClassDataProvider? = classDataProvider
|
||||
val classDataWithSource: ClassDataWithSource? = classDataWithSource
|
||||
}
|
||||
}
|
||||
|
||||
+3
-2
@@ -19,6 +19,7 @@ package org.jetbrains.kotlin.serialization.deserialization
|
||||
import org.jetbrains.kotlin.descriptors.PackageFragmentProvider
|
||||
import org.jetbrains.kotlin.name.ClassId
|
||||
import org.jetbrains.kotlin.serialization.ClassData
|
||||
import org.jetbrains.kotlin.serialization.ClassDataWithSource
|
||||
import org.jetbrains.kotlin.serialization.ProtoBuf
|
||||
import org.jetbrains.kotlin.serialization.SerializedResourcePaths
|
||||
import java.io.InputStream
|
||||
@@ -28,7 +29,7 @@ public open class ResourceLoadingClassDataFinder(
|
||||
private val serializedResourcePaths: SerializedResourcePaths,
|
||||
private val loadResource: (path: String) -> InputStream?
|
||||
) : ClassDataFinder {
|
||||
override fun findClassData(classId: ClassId): ClassDataProvider? {
|
||||
override fun findClassData(classId: ClassId): ClassDataWithSource? {
|
||||
val packageFragment = packageFragmentProvider.getPackageFragments(classId.getPackageFqName()).singleOrNull()
|
||||
as? DeserializedPackageFragment ?: return null
|
||||
|
||||
@@ -38,6 +39,6 @@ public open class ResourceLoadingClassDataFinder(
|
||||
packageFragment.nameResolver,
|
||||
ProtoBuf.Class.parseFrom(stream, serializedResourcePaths.extensionRegistry)
|
||||
)
|
||||
return ClassDataProvider(classData)
|
||||
return ClassDataWithSource(classData)
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user