| Index: third_party/protobuf/src/google/protobuf/compiler/java/java_shared_code_generator.cc
|
| diff --git a/third_party/protobuf/src/google/protobuf/compiler/java/java_shared_code_generator.cc b/third_party/protobuf/src/google/protobuf/compiler/java/java_shared_code_generator.cc
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..70177367a250761dd074ff6ede9b590e9e0ead2a
|
| --- /dev/null
|
| +++ b/third_party/protobuf/src/google/protobuf/compiler/java/java_shared_code_generator.cc
|
| @@ -0,0 +1,205 @@
|
| +// Protocol Buffers - Google's data interchange format
|
| +// Copyright 2008 Google Inc. All rights reserved.
|
| +// 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
|
| +// met:
|
| +//
|
| +// * Redistributions of source code must retain the above copyright
|
| +// notice, this list of conditions and the following disclaimer.
|
| +// * Redistributions in binary form must reproduce the above
|
| +// copyright notice, this list of conditions and the following disclaimer
|
| +// in the documentation and/or other materials provided with the
|
| +// distribution.
|
| +// * Neither the name of Google Inc. nor the names of its
|
| +// contributors may be used to endorse or promote products derived from
|
| +// this software without specific prior written permission.
|
| +//
|
| +// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
| +// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
| +// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
| +// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
| +// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
| +// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
| +// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
| +// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
| +// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
| +// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
| +// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
| +
|
| +// Author: xiaofeng@google.com (Feng Xiao)
|
| +
|
| +#include <google/protobuf/compiler/java/java_shared_code_generator.h>
|
| +
|
| +#include <memory>
|
| +#ifndef _SHARED_PTR_H
|
| +#include <google/protobuf/stubs/shared_ptr.h>
|
| +#endif
|
| +
|
| +#include <google/protobuf/compiler/java/java_helpers.h>
|
| +#include <google/protobuf/compiler/java/java_name_resolver.h>
|
| +#include <google/protobuf/compiler/code_generator.h>
|
| +#include <google/protobuf/io/printer.h>
|
| +#include <google/protobuf/io/zero_copy_stream.h>
|
| +#include <google/protobuf/descriptor.pb.h>
|
| +#include <google/protobuf/descriptor.h>
|
| +#include <google/protobuf/stubs/strutil.h>
|
| +
|
| +namespace google {
|
| +namespace protobuf {
|
| +namespace compiler {
|
| +namespace java {
|
| +
|
| +SharedCodeGenerator::SharedCodeGenerator(const FileDescriptor* file)
|
| + : name_resolver_(new ClassNameResolver), file_(file) {
|
| +}
|
| +
|
| +SharedCodeGenerator::~SharedCodeGenerator() {
|
| +}
|
| +
|
| +void SharedCodeGenerator::Generate(GeneratorContext* context,
|
| + vector<string>* file_list) {
|
| + string java_package = FileJavaPackage(file_);
|
| + string package_dir = JavaPackageToDir(java_package);
|
| +
|
| + if (HasDescriptorMethods(file_)) {
|
| + // Generate descriptors.
|
| + string classname = name_resolver_->GetDescriptorClassName(file_);
|
| + string filename = package_dir + classname + ".java";
|
| + file_list->push_back(filename);
|
| + google::protobuf::scoped_ptr<io::ZeroCopyOutputStream> output(context->Open(filename));
|
| + google::protobuf::scoped_ptr<io::Printer> printer(new io::Printer(output.get(), '$'));
|
| +
|
| + printer->Print(
|
| + "// Generated by the protocol buffer compiler. DO NOT EDIT!\n"
|
| + "// source: $filename$\n"
|
| + "\n",
|
| + "filename", file_->name());
|
| + if (!java_package.empty()) {
|
| + printer->Print(
|
| + "package $package$;\n"
|
| + "\n",
|
| + "package", java_package);
|
| + }
|
| + printer->Print(
|
| + "public final class $classname$ {\n"
|
| + " public static com.google.protobuf.Descriptors.FileDescriptor\n"
|
| + " descriptor;\n"
|
| + " static {\n",
|
| + "classname", classname);
|
| + printer->Indent();
|
| + printer->Indent();
|
| + GenerateDescriptors(printer.get());
|
| + printer->Outdent();
|
| + printer->Outdent();
|
| + printer->Print(
|
| + " }\n"
|
| + "}\n");
|
| +
|
| + printer.reset();
|
| + output.reset();
|
| + }
|
| +}
|
| +
|
| +
|
| +void SharedCodeGenerator::GenerateDescriptors(io::Printer* printer) {
|
| + // Embed the descriptor. We simply serialize the entire FileDescriptorProto
|
| + // and embed it as a string literal, which is parsed and built into real
|
| + // descriptors at initialization time. We unfortunately have to put it in
|
| + // a string literal, not a byte array, because apparently using a literal
|
| + // byte array causes the Java compiler to generate *instructions* to
|
| + // initialize each and every byte of the array, e.g. as if you typed:
|
| + // b[0] = 123; b[1] = 456; b[2] = 789;
|
| + // This makes huge bytecode files and can easily hit the compiler's internal
|
| + // code size limits (error "code to large"). String literals are apparently
|
| + // embedded raw, which is what we want.
|
| + FileDescriptorProto file_proto;
|
| + file_->CopyTo(&file_proto);
|
| +
|
| +
|
| + string file_data;
|
| + file_proto.SerializeToString(&file_data);
|
| +
|
| + printer->Print(
|
| + "java.lang.String[] descriptorData = {\n");
|
| + printer->Indent();
|
| +
|
| + // Only write 40 bytes per line.
|
| + static const int kBytesPerLine = 40;
|
| + for (int i = 0; i < file_data.size(); i += kBytesPerLine) {
|
| + if (i > 0) {
|
| + // Every 400 lines, start a new string literal, in order to avoid the
|
| + // 64k length limit.
|
| + if (i % 400 == 0) {
|
| + printer->Print(",\n");
|
| + } else {
|
| + printer->Print(" +\n");
|
| + }
|
| + }
|
| + printer->Print("\"$data$\"",
|
| + "data", CEscape(file_data.substr(i, kBytesPerLine)));
|
| + }
|
| +
|
| + printer->Outdent();
|
| + printer->Print("\n};\n");
|
| +
|
| + // -----------------------------------------------------------------
|
| + // Create the InternalDescriptorAssigner.
|
| +
|
| + printer->Print(
|
| + "com.google.protobuf.Descriptors.FileDescriptor."
|
| + "InternalDescriptorAssigner assigner =\n"
|
| + " new com.google.protobuf.Descriptors.FileDescriptor."
|
| + " InternalDescriptorAssigner() {\n"
|
| + " public com.google.protobuf.ExtensionRegistry assignDescriptors(\n"
|
| + " com.google.protobuf.Descriptors.FileDescriptor root) {\n"
|
| + " descriptor = root;\n"
|
| + // Custom options will be handled when immutable messages' outer class is
|
| + // loaded. Here we just return null and let custom options be unknown
|
| + // fields.
|
| + " return null;\n"
|
| + " }\n"
|
| + " };\n");
|
| +
|
| + // -----------------------------------------------------------------
|
| + // Find out all dependencies.
|
| + vector<pair<string, string> > dependencies;
|
| + for (int i = 0; i < file_->dependency_count(); i++) {
|
| + if (ShouldIncludeDependency(file_->dependency(i))) {
|
| + string filename = file_->dependency(i)->name();
|
| + string classname = FileJavaPackage(file_->dependency(i)) + "." +
|
| + name_resolver_->GetDescriptorClassName(
|
| + file_->dependency(i));
|
| + dependencies.push_back(std::make_pair(filename, classname));
|
| + }
|
| + }
|
| +
|
| + // -----------------------------------------------------------------
|
| + // Invoke internalBuildGeneratedFileFrom() to build the file.
|
| + printer->Print(
|
| + "com.google.protobuf.Descriptors.FileDescriptor\n"
|
| + " .internalBuildGeneratedFileFrom(descriptorData,\n");
|
| + printer->Print(
|
| + " new com.google.protobuf.Descriptors.FileDescriptor[] {\n");
|
| +
|
| + for (int i = 0; i < dependencies.size(); i++) {
|
| + const string& dependency = dependencies[i].second;
|
| + printer->Print(
|
| + " $dependency$.getDescriptor(),\n",
|
| + "dependency", dependency);
|
| + }
|
| +
|
| + printer->Print(
|
| + " }, assigner);\n");
|
| +}
|
| +
|
| +bool SharedCodeGenerator::ShouldIncludeDependency(
|
| + const FileDescriptor* descriptor) {
|
| + return true;
|
| +}
|
| +
|
| +} // namespace java
|
| +} // namespace compiler
|
| +} // namespace protobuf
|
| +} // namespace google
|
|
|