| Index: third_party/protobuf/src/google/protobuf/compiler/java/java_extension.cc
|
| diff --git a/third_party/protobuf/src/google/protobuf/compiler/java/java_extension.cc b/third_party/protobuf/src/google/protobuf/compiler/java/java_extension.cc
|
| index 921fe6589c2b9d01fb60b27626200f5d8afd88bc..4db7085e01c6bf8835d978ea139aedad763cc837 100644
|
| --- a/third_party/protobuf/src/google/protobuf/compiler/java/java_extension.cc
|
| +++ b/third_party/protobuf/src/google/protobuf/compiler/java/java_extension.cc
|
| @@ -1,6 +1,6 @@
|
| // Protocol Buffers - Google's data interchange format
|
| // Copyright 2008 Google Inc. All rights reserved.
|
| -// http://code.google.com/p/protobuf/
|
| +// https://developers.google.com/protocol-buffers/
|
| //
|
| // Redistribution and use in source and binary forms, with or without
|
| // modification, are permitted provided that the following conditions are
|
| @@ -33,74 +33,50 @@
|
| // Sanjay Ghemawat, Jeff Dean, and others.
|
|
|
| #include <google/protobuf/compiler/java/java_extension.h>
|
| +
|
| +#include <google/protobuf/compiler/java/java_context.h>
|
| #include <google/protobuf/compiler/java/java_doc_comment.h>
|
| #include <google/protobuf/compiler/java/java_helpers.h>
|
| -#include <google/protobuf/stubs/strutil.h>
|
| +#include <google/protobuf/compiler/java/java_name_resolver.h>
|
| #include <google/protobuf/io/printer.h>
|
| +#include <google/protobuf/stubs/strutil.h>
|
|
|
| namespace google {
|
| namespace protobuf {
|
| namespace compiler {
|
| namespace java {
|
|
|
| -namespace {
|
| -
|
| -const char* TypeName(FieldDescriptor::Type field_type) {
|
| - switch (field_type) {
|
| - case FieldDescriptor::TYPE_INT32 : return "INT32";
|
| - case FieldDescriptor::TYPE_UINT32 : return "UINT32";
|
| - case FieldDescriptor::TYPE_SINT32 : return "SINT32";
|
| - case FieldDescriptor::TYPE_FIXED32 : return "FIXED32";
|
| - case FieldDescriptor::TYPE_SFIXED32: return "SFIXED32";
|
| - case FieldDescriptor::TYPE_INT64 : return "INT64";
|
| - case FieldDescriptor::TYPE_UINT64 : return "UINT64";
|
| - case FieldDescriptor::TYPE_SINT64 : return "SINT64";
|
| - case FieldDescriptor::TYPE_FIXED64 : return "FIXED64";
|
| - case FieldDescriptor::TYPE_SFIXED64: return "SFIXED64";
|
| - case FieldDescriptor::TYPE_FLOAT : return "FLOAT";
|
| - case FieldDescriptor::TYPE_DOUBLE : return "DOUBLE";
|
| - case FieldDescriptor::TYPE_BOOL : return "BOOL";
|
| - case FieldDescriptor::TYPE_STRING : return "STRING";
|
| - case FieldDescriptor::TYPE_BYTES : return "BYTES";
|
| - case FieldDescriptor::TYPE_ENUM : return "ENUM";
|
| - case FieldDescriptor::TYPE_GROUP : return "GROUP";
|
| - case FieldDescriptor::TYPE_MESSAGE : return "MESSAGE";
|
| -
|
| - // No default because we want the compiler to complain if any new
|
| - // types are added.
|
| - }
|
| -
|
| - GOOGLE_LOG(FATAL) << "Can't get here.";
|
| - return NULL;
|
| -}
|
| -
|
| -}
|
| -
|
| -ExtensionGenerator::ExtensionGenerator(const FieldDescriptor* descriptor)
|
| - : descriptor_(descriptor) {
|
| +ImmutableExtensionGenerator::ImmutableExtensionGenerator(
|
| + const FieldDescriptor* descriptor, Context* context)
|
| + : descriptor_(descriptor), context_(context),
|
| + name_resolver_(context->GetNameResolver()) {
|
| if (descriptor_->extension_scope() != NULL) {
|
| - scope_ = ClassName(descriptor_->extension_scope());
|
| + scope_ = name_resolver_->GetImmutableClassName(
|
| + descriptor_->extension_scope());
|
| } else {
|
| - scope_ = ClassName(descriptor_->file());
|
| + scope_ = name_resolver_->GetImmutableClassName(descriptor_->file());
|
| }
|
| }
|
|
|
| -ExtensionGenerator::~ExtensionGenerator() {}
|
| +ImmutableExtensionGenerator::~ImmutableExtensionGenerator() {}
|
|
|
| // Initializes the vars referenced in the generated code templates.
|
| -void InitTemplateVars(const FieldDescriptor* descriptor,
|
| - const string& scope,
|
| - map<string, string>* vars_pointer) {
|
| +void ExtensionGenerator::InitTemplateVars(const FieldDescriptor* descriptor,
|
| + const string& scope,
|
| + bool immutable,
|
| + ClassNameResolver* name_resolver,
|
| + map<string, string>* vars_pointer) {
|
| map<string, string> &vars = *vars_pointer;
|
| vars["scope"] = scope;
|
| vars["name"] = UnderscoresToCamelCase(descriptor);
|
| - vars["containing_type"] = ClassName(descriptor->containing_type());
|
| + vars["containing_type"] =
|
| + name_resolver->GetClassName(descriptor->containing_type(), immutable);
|
| vars["number"] = SimpleItoa(descriptor->number());
|
| vars["constant_name"] = FieldConstantName(descriptor);
|
| vars["index"] = SimpleItoa(descriptor->index());
|
| - vars["default"] =
|
| - descriptor->is_repeated() ? "" : DefaultValue(descriptor);
|
| - vars["type_constant"] = TypeName(GetType(descriptor));
|
| + vars["default"] = descriptor->is_repeated() ?
|
| + "" : DefaultValue(descriptor, immutable, name_resolver);
|
| + vars["type_constant"] = FieldTypeName(GetType(descriptor));
|
| vars["packed"] = descriptor->options().packed() ? "true" : "false";
|
| vars["enum_map"] = "null";
|
| vars["prototype"] = "null";
|
| @@ -109,13 +85,21 @@ void InitTemplateVars(const FieldDescriptor* descriptor,
|
| string singular_type;
|
| switch (java_type) {
|
| case JAVATYPE_MESSAGE:
|
| - singular_type = ClassName(descriptor->message_type());
|
| + singular_type = name_resolver->GetClassName(descriptor->message_type(),
|
| + immutable);
|
| vars["prototype"] = singular_type + ".getDefaultInstance()";
|
| break;
|
| case JAVATYPE_ENUM:
|
| - singular_type = ClassName(descriptor->enum_type());
|
| + singular_type = name_resolver->GetClassName(descriptor->enum_type(),
|
| + immutable);
|
| vars["enum_map"] = singular_type + ".internalGetValueMap()";
|
| break;
|
| + case JAVATYPE_STRING:
|
| + singular_type = "java.lang.String";
|
| + break;
|
| + case JAVATYPE_BYTES:
|
| + singular_type = immutable ? "com.google.protobuf.ByteString" : "byte[]";
|
| + break;
|
| default:
|
| singular_type = BoxedPrimitiveTypeName(java_type);
|
| break;
|
| @@ -125,9 +109,11 @@ void InitTemplateVars(const FieldDescriptor* descriptor,
|
| vars["singular_type"] = singular_type;
|
| }
|
|
|
| -void ExtensionGenerator::Generate(io::Printer* printer) {
|
| +void ImmutableExtensionGenerator::Generate(io::Printer* printer) {
|
| map<string, string> vars;
|
| - InitTemplateVars(descriptor_, scope_, &vars);
|
| + const bool kUseImmutableNames = true;
|
| + InitTemplateVars(descriptor_, scope_, kUseImmutableNames, name_resolver_,
|
| + &vars);
|
| printer->Print(vars,
|
| "public static final int $constant_name$ = $number$;\n");
|
|
|
| @@ -174,7 +160,8 @@ void ExtensionGenerator::Generate(io::Printer* printer) {
|
| " $enum_map$,\n"
|
| " $number$,\n"
|
| " com.google.protobuf.WireFormat.FieldType.$type_constant$,\n"
|
| - " $packed$);\n");
|
| + " $packed$,\n"
|
| + " $singular_type$.class);\n");
|
| } else {
|
| printer->Print(
|
| vars,
|
| @@ -188,13 +175,15 @@ void ExtensionGenerator::Generate(io::Printer* printer) {
|
| " $prototype$,\n"
|
| " $enum_map$,\n"
|
| " $number$,\n"
|
| - " com.google.protobuf.WireFormat.FieldType.$type_constant$);\n");
|
| + " com.google.protobuf.WireFormat.FieldType.$type_constant$,\n"
|
| + " $singular_type$.class);\n");
|
| }
|
| }
|
| }
|
|
|
| -void ExtensionGenerator::GenerateNonNestedInitializationCode(
|
| +int ImmutableExtensionGenerator::GenerateNonNestedInitializationCode(
|
| io::Printer* printer) {
|
| + int bytecode_estimate = 0;
|
| if (descriptor_->extension_scope() == NULL &&
|
| HasDescriptorMethods(descriptor_->file())) {
|
| // Only applies to non-nested, non-lite extensions.
|
| @@ -202,14 +191,18 @@ void ExtensionGenerator::GenerateNonNestedInitializationCode(
|
| "$name$.internalInit(descriptor.getExtensions().get($index$));\n",
|
| "name", UnderscoresToCamelCase(descriptor_),
|
| "index", SimpleItoa(descriptor_->index()));
|
| + bytecode_estimate += 21;
|
| }
|
| + return bytecode_estimate;
|
| }
|
|
|
| -void ExtensionGenerator::GenerateRegistrationCode(io::Printer* printer) {
|
| +int ImmutableExtensionGenerator::GenerateRegistrationCode(
|
| + io::Printer* printer) {
|
| printer->Print(
|
| "registry.add($scope$.$name$);\n",
|
| "scope", scope_,
|
| "name", UnderscoresToCamelCase(descriptor_));
|
| + return 7;
|
| }
|
|
|
| } // namespace java
|
|
|