Serialization of flexible types supported
This commit is contained in:
@@ -20,14 +20,15 @@ option java_outer_classname = "ProtoBuf";
|
||||
option optimize_for = LITE_RUNTIME; // Smaller runtime
|
||||
option java_generic_services = false; // Less code
|
||||
|
||||
message SimpleNameTable {
|
||||
repeated string name = 1;
|
||||
message StringTable {
|
||||
repeated string string = 1;
|
||||
}
|
||||
|
||||
message QualifiedNameTable {
|
||||
|
||||
message QualifiedName {
|
||||
optional int32 parent_qualified_name = 1 [default = -1];
|
||||
// Id in the StringTable
|
||||
required int32 short_name = 2;
|
||||
optional Kind kind = 3 [default = PACKAGE];
|
||||
|
||||
@@ -68,10 +69,21 @@ message Type {
|
||||
repeated Argument argument = 2;
|
||||
|
||||
optional bool nullable = 3 [default = false];
|
||||
|
||||
// Id in the StringTable
|
||||
// If this field is set, the type is flexible.
|
||||
// All the fields above represent its lower bound, and flexible_upper_bound must be set and represents its upper bound.
|
||||
optional int32 flexible_type_capabilities_id = 4;
|
||||
|
||||
// While such an "indirect" encoding helps backwards compatibility with pre-flexible-types versions of this format,
|
||||
// we use it mainly to save space: having a special mandatory tag on each an every type just to have an option
|
||||
// to represent flexible types is too many wasted bytes.
|
||||
optional Type flexible_upper_bound = 5;
|
||||
}
|
||||
|
||||
message TypeParameter {
|
||||
required int32 id = 1;
|
||||
// Id in the StringTable
|
||||
required int32 name = 2;
|
||||
|
||||
optional bool reified = 3 [default = false];
|
||||
@@ -188,6 +200,7 @@ message Callable {
|
||||
|
||||
optional Type receiver_type = 5;
|
||||
|
||||
// Id in the StringTable
|
||||
required int32 name = 6;
|
||||
|
||||
message ValueParameter {
|
||||
@@ -196,6 +209,8 @@ message Callable {
|
||||
has_annotations
|
||||
*/
|
||||
optional int32 flags = 1;
|
||||
|
||||
// Id in the StringTable
|
||||
required int32 name = 2;
|
||||
required Type type = 3;
|
||||
optional Type vararg_element_type = 4;
|
||||
|
||||
+15
-4
@@ -20,10 +20,7 @@ import org.jetbrains.annotations.NotNull;
|
||||
import org.jetbrains.jet.lang.descriptors.*;
|
||||
import org.jetbrains.jet.lang.descriptors.annotations.Annotated;
|
||||
import org.jetbrains.jet.lang.resolve.DescriptorFactory;
|
||||
import org.jetbrains.jet.lang.types.JetType;
|
||||
import org.jetbrains.jet.lang.types.TypeConstructor;
|
||||
import org.jetbrains.jet.lang.types.TypeProjection;
|
||||
import org.jetbrains.jet.lang.types.Variance;
|
||||
import org.jetbrains.jet.lang.types.*;
|
||||
import org.jetbrains.jet.lang.types.lang.KotlinBuiltIns;
|
||||
import org.jetbrains.jet.renderer.DescriptorRenderer;
|
||||
|
||||
@@ -343,6 +340,8 @@ public class DescriptorSerializer {
|
||||
public ProtoBuf.Type.Builder type(@NotNull JetType type) {
|
||||
assert !type.isError() : "Can't serialize error types: " + type; // TODO
|
||||
|
||||
if (TypesPackage.isFlexible(type)) return flexibleType(type);
|
||||
|
||||
ProtoBuf.Type.Builder builder = ProtoBuf.Type.newBuilder();
|
||||
|
||||
builder.setConstructor(typeConstructor(type.getConstructor()));
|
||||
@@ -359,6 +358,18 @@ public class DescriptorSerializer {
|
||||
return builder;
|
||||
}
|
||||
|
||||
private ProtoBuf.Type.Builder flexibleType(@NotNull JetType type) {
|
||||
Flexibility flexibility = TypesPackage.flexibility(type);
|
||||
|
||||
ProtoBuf.Type.Builder builder = type(flexibility.getLowerBound());
|
||||
|
||||
builder.setFlexibleTypeCapabilitiesId(nameTable.getStringIndex(flexibility.getExtraCapabilities().getId()));
|
||||
|
||||
builder.setFlexibleUpperBound(type(flexibility.getUpperBound()));
|
||||
|
||||
return builder;
|
||||
}
|
||||
|
||||
@NotNull
|
||||
private ProtoBuf.Type.Argument.Builder typeArgument(@NotNull TypeProjection typeProjection) {
|
||||
ProtoBuf.Type.Argument.Builder builder = ProtoBuf.Type.Argument.newBuilder();
|
||||
|
||||
+30
@@ -0,0 +1,30 @@
|
||||
/*
|
||||
* Copyright 2010-2014 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.jet.descriptors.serialization
|
||||
|
||||
import org.jetbrains.jet.lang.resolve.name.FqName
|
||||
import org.jetbrains.jet.lang.types.FlexibleTypeCapabilities
|
||||
|
||||
trait FlexibleTypeCapabilitiesDeserializer {
|
||||
object ThrowException : FlexibleTypeCapabilitiesDeserializer {
|
||||
override fun capabilitiesById(id: String): FlexibleTypeCapabilities? {
|
||||
throw IllegalArgumentException("Capabilities not found by ThrowException manager: $id")
|
||||
}
|
||||
}
|
||||
|
||||
fun capabilitiesById(id: String): FlexibleTypeCapabilities?
|
||||
}
|
||||
+12
-7
@@ -26,20 +26,20 @@ import org.jetbrains.jet.lang.resolve.name.Name;
|
||||
import static org.jetbrains.jet.descriptors.serialization.ProtoBuf.QualifiedNameTable.QualifiedName;
|
||||
|
||||
public class NameResolver {
|
||||
private final ProtoBuf.SimpleNameTable simpleNames;
|
||||
private final ProtoBuf.StringTable strings;
|
||||
private final ProtoBuf.QualifiedNameTable qualifiedNames;
|
||||
|
||||
public NameResolver(
|
||||
@NotNull ProtoBuf.SimpleNameTable simpleNames,
|
||||
@NotNull ProtoBuf.StringTable strings,
|
||||
@NotNull ProtoBuf.QualifiedNameTable qualifiedNames
|
||||
) {
|
||||
this.simpleNames = simpleNames;
|
||||
this.strings = strings;
|
||||
this.qualifiedNames = qualifiedNames;
|
||||
}
|
||||
|
||||
@NotNull
|
||||
public ProtoBuf.SimpleNameTable getSimpleNameTable() {
|
||||
return simpleNames;
|
||||
public ProtoBuf.StringTable getStringTable() {
|
||||
return strings;
|
||||
}
|
||||
|
||||
@NotNull
|
||||
@@ -47,9 +47,14 @@ public class NameResolver {
|
||||
return qualifiedNames;
|
||||
}
|
||||
|
||||
@NotNull
|
||||
public String getString(int index) {
|
||||
return strings.getString(index);
|
||||
}
|
||||
|
||||
@NotNull
|
||||
public Name getName(int index) {
|
||||
String name = simpleNames.getName(index);
|
||||
String name = strings.getString(index);
|
||||
return Name.guess(name);
|
||||
}
|
||||
|
||||
@@ -89,7 +94,7 @@ public class NameResolver {
|
||||
result = parentProto;
|
||||
}
|
||||
}
|
||||
sb.append(simpleNames.getName(fqNameProto.getShortName()));
|
||||
sb.append(strings.getString(fqNameProto.getShortName()));
|
||||
return result;
|
||||
}
|
||||
|
||||
|
||||
+10
-10
@@ -30,7 +30,7 @@ public class NameSerializationUtil {
|
||||
@NotNull
|
||||
public static NameResolver deserializeNameResolver(@NotNull InputStream in) {
|
||||
try {
|
||||
ProtoBuf.SimpleNameTable simpleNames = ProtoBuf.SimpleNameTable.parseDelimitedFrom(in);
|
||||
ProtoBuf.StringTable simpleNames = ProtoBuf.StringTable.parseDelimitedFrom(in);
|
||||
ProtoBuf.QualifiedNameTable qualifiedNames = ProtoBuf.QualifiedNameTable.parseDelimitedFrom(in);
|
||||
return new NameResolver(simpleNames, qualifiedNames);
|
||||
}
|
||||
@@ -40,20 +40,20 @@ public class NameSerializationUtil {
|
||||
}
|
||||
|
||||
public static void serializeNameResolver(@NotNull OutputStream out, @NotNull NameResolver nameResolver) {
|
||||
serializeNameTable(out, nameResolver.getSimpleNameTable(), nameResolver.getQualifiedNameTable());
|
||||
serializeNameTable(out, nameResolver.getStringTable(), nameResolver.getQualifiedNameTable());
|
||||
}
|
||||
|
||||
public static void serializeNameTable(@NotNull OutputStream out, @NotNull NameTable nameTable) {
|
||||
serializeNameTable(out, toSimpleNameTable(nameTable), toQualifiedNameTable(nameTable));
|
||||
serializeNameTable(out, toStringTable(nameTable), toQualifiedNameTable(nameTable));
|
||||
}
|
||||
|
||||
private static void serializeNameTable(
|
||||
@NotNull OutputStream out,
|
||||
@NotNull ProtoBuf.SimpleNameTable simpleNameTable,
|
||||
@NotNull ProtoBuf.StringTable stringTable,
|
||||
@NotNull ProtoBuf.QualifiedNameTable qualifiedNameTable
|
||||
) {
|
||||
try {
|
||||
simpleNameTable.writeDelimitedTo(out);
|
||||
stringTable.writeDelimitedTo(out);
|
||||
qualifiedNameTable.writeDelimitedTo(out);
|
||||
}
|
||||
catch (IOException e) {
|
||||
@@ -62,10 +62,10 @@ public class NameSerializationUtil {
|
||||
}
|
||||
|
||||
@NotNull
|
||||
public static ProtoBuf.SimpleNameTable toSimpleNameTable(@NotNull NameTable nameTable) {
|
||||
ProtoBuf.SimpleNameTable.Builder simpleNames = ProtoBuf.SimpleNameTable.newBuilder();
|
||||
for (String simpleName : nameTable.getSimpleNames()) {
|
||||
simpleNames.addName(simpleName);
|
||||
public static ProtoBuf.StringTable toStringTable(@NotNull NameTable nameTable) {
|
||||
ProtoBuf.StringTable.Builder simpleNames = ProtoBuf.StringTable.newBuilder();
|
||||
for (String simpleName : nameTable.getStrings()) {
|
||||
simpleNames.addString(simpleName);
|
||||
}
|
||||
return simpleNames.build();
|
||||
}
|
||||
@@ -81,6 +81,6 @@ public class NameSerializationUtil {
|
||||
|
||||
@NotNull
|
||||
public static NameResolver createNameResolver(@NotNull NameTable table) {
|
||||
return new NameResolver(toSimpleNameTable(table), toQualifiedNameTable(table));
|
||||
return new NameResolver(toStringTable(table), toQualifiedNameTable(table));
|
||||
}
|
||||
}
|
||||
|
||||
@@ -46,15 +46,15 @@ public class NameTable {
|
||||
}
|
||||
};
|
||||
|
||||
private final Interner<String> simpleNames = new Interner<String>();
|
||||
private final Interner<String> strings = new Interner<String>();
|
||||
private final Interner<QualifiedName.Builder> qualifiedNames = new Interner<QualifiedName.Builder>(QUALIFIED_NAME_BUILDER_HASHING);
|
||||
|
||||
public NameTable() {
|
||||
}
|
||||
|
||||
@NotNull
|
||||
public List<String> getSimpleNames() {
|
||||
return simpleNames.getAllInternedObjects();
|
||||
public List<String> getStrings() {
|
||||
return strings.getAllInternedObjects();
|
||||
}
|
||||
|
||||
@NotNull
|
||||
@@ -63,7 +63,11 @@ public class NameTable {
|
||||
}
|
||||
|
||||
public int getSimpleNameIndex(@NotNull Name name) {
|
||||
return simpleNames.intern(name.asString());
|
||||
return getStringIndex(name.asString());
|
||||
}
|
||||
|
||||
public int getStringIndex(@NotNull String string) {
|
||||
return strings.intern(string);
|
||||
}
|
||||
|
||||
public int getFqNameIndex(@NotNull ClassOrPackageFragmentDescriptor descriptor) {
|
||||
|
||||
+567
-121
File diff suppressed because it is too large
Load Diff
+12
@@ -100,6 +100,18 @@ public class TypeDeserializer {
|
||||
|
||||
@NotNull
|
||||
public JetType type(@NotNull ProtoBuf.Type proto) {
|
||||
if (proto.hasFlexibleTypeCapabilitiesId()) {
|
||||
String id = context.getNameResolver().getString(proto.getFlexibleTypeCapabilitiesId());
|
||||
FlexibleTypeCapabilities capabilities = context.getFlexibleTypeCapabilitiesDeserializer().capabilitiesById(id);
|
||||
|
||||
if (capabilities == null) return ErrorUtils.createErrorType(new DeserializedType(proto) + ": Capabilities not found for id " + id);
|
||||
|
||||
return DelegatingFlexibleType.OBJECT$.create(
|
||||
new DeserializedType(proto),
|
||||
new DeserializedType(proto.getFlexibleUpperBound()),
|
||||
capabilities
|
||||
);
|
||||
}
|
||||
return new DeserializedType(proto);
|
||||
}
|
||||
|
||||
|
||||
+8
-4
@@ -32,6 +32,7 @@ import org.jetbrains.jet.lang.descriptors.ModuleDescriptor
|
||||
import org.jetbrains.jet.lang.descriptors.ClassDescriptor
|
||||
import org.jetbrains.jet.lang.resolve.name.ClassId
|
||||
import org.jetbrains.jet.descriptors.serialization.ClassDeserializer
|
||||
import org.jetbrains.jet.descriptors.serialization.FlexibleTypeCapabilitiesDeserializer
|
||||
|
||||
public open class DeserializationGlobalContext(
|
||||
public val storageManager: StorageManager,
|
||||
@@ -40,6 +41,7 @@ public open class DeserializationGlobalContext(
|
||||
public val annotationLoader: AnnotationLoader,
|
||||
public val constantLoader: ConstantLoader,
|
||||
public val packageFragmentProvider: PackageFragmentProvider,
|
||||
public val flexibleTypeCapabilitiesDeserializer: FlexibleTypeCapabilitiesDeserializer,
|
||||
public val classDeserializer: ClassDeserializer = ClassDeserializer(storageManager, classDataFinder)
|
||||
) {
|
||||
{
|
||||
@@ -48,7 +50,7 @@ public open class DeserializationGlobalContext(
|
||||
|
||||
public fun withNameResolver(nameResolver: NameResolver): DeserializationContext {
|
||||
return DeserializationContext(storageManager, moduleDescriptor, classDataFinder, annotationLoader,
|
||||
constantLoader, packageFragmentProvider, classDeserializer, nameResolver)
|
||||
constantLoader, packageFragmentProvider, flexibleTypeCapabilitiesDeserializer, classDeserializer, nameResolver)
|
||||
}
|
||||
}
|
||||
|
||||
@@ -60,10 +62,11 @@ public open class DeserializationContext(
|
||||
annotationLoader: AnnotationLoader,
|
||||
constantLoader: ConstantLoader,
|
||||
packageFragmentProvider: PackageFragmentProvider,
|
||||
flexibleTypeCapabilitiesDeserializer: FlexibleTypeCapabilitiesDeserializer,
|
||||
classDeserializer: ClassDeserializer,
|
||||
public val nameResolver: NameResolver
|
||||
) : DeserializationGlobalContext(storageManager, moduleDescriptor, classDataFinder, annotationLoader,
|
||||
constantLoader, packageFragmentProvider, classDeserializer) {
|
||||
constantLoader, packageFragmentProvider, flexibleTypeCapabilitiesDeserializer, classDeserializer) {
|
||||
fun withTypes(containingDeclaration: DeclarationDescriptor): DeserializationContextWithTypes {
|
||||
val typeDeserializer = TypeDeserializer(this, null, "Deserializer for ${containingDeclaration.getName()}",
|
||||
TypeDeserializer.TypeParameterResolver.NONE)
|
||||
@@ -73,7 +76,7 @@ public open class DeserializationContext(
|
||||
fun withTypes(containingDeclaration: DeclarationDescriptor, typeDeserializer: TypeDeserializer): DeserializationContextWithTypes {
|
||||
return DeserializationContextWithTypes(storageManager, moduleDescriptor, classDataFinder, annotationLoader,
|
||||
constantLoader, packageFragmentProvider, classDeserializer,
|
||||
nameResolver, containingDeclaration, typeDeserializer)
|
||||
nameResolver, flexibleTypeCapabilitiesDeserializer, containingDeclaration, typeDeserializer)
|
||||
}
|
||||
|
||||
}
|
||||
@@ -88,10 +91,11 @@ class DeserializationContextWithTypes(
|
||||
packageFragmentProvider: PackageFragmentProvider,
|
||||
classDeserializer: ClassDeserializer,
|
||||
nameResolver: NameResolver,
|
||||
flexibleTypeCapabilitiesDeserializer: FlexibleTypeCapabilitiesDeserializer,
|
||||
val containingDeclaration: DeclarationDescriptor,
|
||||
val typeDeserializer: TypeDeserializer
|
||||
) : DeserializationContext(storageManager, moduleDescriptor, classDataFinder, annotationLoader,
|
||||
constantLoader, packageFragmentProvider, classDeserializer, nameResolver) {
|
||||
constantLoader, packageFragmentProvider, flexibleTypeCapabilitiesDeserializer, classDeserializer, nameResolver) {
|
||||
val deserializer: MemberDeserializer = MemberDeserializer(this)
|
||||
|
||||
public fun childContext(
|
||||
|
||||
Reference in New Issue
Block a user