mirror of https://github.com/grpc/grpc.git
Merge pull request #18991 from mhaidrygoog/change_cpp_plugin
Extracted the code of Cpp Generator into a headerreviewable/pr18937/r11
Moiz Haidry
6 years ago
committed by
GitHub
commit
6607cdfcf0
3 changed files with 156 additions and 131 deletions
@ -0,0 +1,154 @@ |
||||
/*
|
||||
* |
||||
* Copyright 2019 gRPC authors. |
||||
* |
||||
* 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. |
||||
* |
||||
*/ |
||||
|
||||
#ifndef GRPC_INTERNAL_COMPILER_CPP_PLUGIN_H |
||||
#define GRPC_INTERNAL_COMPILER_CPP_PLUGIN_H |
||||
|
||||
#include <memory> |
||||
#include <sstream> |
||||
|
||||
#include "src/compiler/config.h" |
||||
|
||||
#include "src/compiler/cpp_generator.h" |
||||
#include "src/compiler/generator_helpers.h" |
||||
#include "src/compiler/protobuf_plugin.h" |
||||
|
||||
// Cpp Generator for Protobug IDL
|
||||
class CppGrpcGenerator : public grpc::protobuf::compiler::CodeGenerator { |
||||
public: |
||||
CppGrpcGenerator() {} |
||||
virtual ~CppGrpcGenerator() {} |
||||
|
||||
virtual bool Generate(const grpc::protobuf::FileDescriptor* file, |
||||
const grpc::string& parameter, |
||||
grpc::protobuf::compiler::GeneratorContext* context, |
||||
grpc::string* error) const { |
||||
if (file->options().cc_generic_services()) { |
||||
*error = |
||||
"cpp grpc proto compiler plugin does not work with generic " |
||||
"services. To generate cpp grpc APIs, please set \"" |
||||
"cc_generic_service = false\"."; |
||||
return false; |
||||
} |
||||
|
||||
grpc_cpp_generator::Parameters generator_parameters; |
||||
generator_parameters.use_system_headers = true; |
||||
generator_parameters.generate_mock_code = false; |
||||
generator_parameters.include_import_headers = false; |
||||
|
||||
ProtoBufFile pbfile(file); |
||||
|
||||
if (!parameter.empty()) { |
||||
std::vector<grpc::string> parameters_list = |
||||
grpc_generator::tokenize(parameter, ","); |
||||
for (auto parameter_string = parameters_list.begin(); |
||||
parameter_string != parameters_list.end(); parameter_string++) { |
||||
std::vector<grpc::string> param = |
||||
grpc_generator::tokenize(*parameter_string, "="); |
||||
if (param[0] == "services_namespace") { |
||||
generator_parameters.services_namespace = param[1]; |
||||
} else if (param[0] == "use_system_headers") { |
||||
if (param[1] == "true") { |
||||
generator_parameters.use_system_headers = true; |
||||
} else if (param[1] == "false") { |
||||
generator_parameters.use_system_headers = false; |
||||
} else { |
||||
*error = grpc::string("Invalid parameter: ") + *parameter_string; |
||||
return false; |
||||
} |
||||
} else if (param[0] == "grpc_search_path") { |
||||
generator_parameters.grpc_search_path = param[1]; |
||||
} else if (param[0] == "generate_mock_code") { |
||||
if (param[1] == "true") { |
||||
generator_parameters.generate_mock_code = true; |
||||
} else if (param[1] != "false") { |
||||
*error = grpc::string("Invalid parameter: ") + *parameter_string; |
||||
return false; |
||||
} |
||||
} else if (param[0] == "gmock_search_path") { |
||||
generator_parameters.gmock_search_path = param[1]; |
||||
} else if (param[0] == "additional_header_includes") { |
||||
generator_parameters.additional_header_includes = |
||||
grpc_generator::tokenize(param[1], ":"); |
||||
} else if (param[0] == "message_header_extension") { |
||||
generator_parameters.message_header_extension = param[1]; |
||||
} else if (param[0] == "include_import_headers") { |
||||
if (param[1] == "true") { |
||||
generator_parameters.include_import_headers = true; |
||||
} else if (param[1] != "false") { |
||||
*error = grpc::string("Invalid parameter: ") + *parameter_string; |
||||
return false; |
||||
} |
||||
} else { |
||||
*error = grpc::string("Unknown parameter: ") + *parameter_string; |
||||
return false; |
||||
} |
||||
} |
||||
} |
||||
|
||||
grpc::string file_name = grpc_generator::StripProto(file->name()); |
||||
|
||||
grpc::string header_code = |
||||
grpc_cpp_generator::GetHeaderPrologue(&pbfile, generator_parameters) + |
||||
grpc_cpp_generator::GetHeaderIncludes(&pbfile, generator_parameters) + |
||||
grpc_cpp_generator::GetHeaderServices(&pbfile, generator_parameters) + |
||||
grpc_cpp_generator::GetHeaderEpilogue(&pbfile, generator_parameters); |
||||
std::unique_ptr<grpc::protobuf::io::ZeroCopyOutputStream> header_output( |
||||
context->Open(file_name + ".grpc.pb.h")); |
||||
grpc::protobuf::io::CodedOutputStream header_coded_out(header_output.get()); |
||||
header_coded_out.WriteRaw(header_code.data(), header_code.size()); |
||||
|
||||
grpc::string source_code = |
||||
grpc_cpp_generator::GetSourcePrologue(&pbfile, generator_parameters) + |
||||
grpc_cpp_generator::GetSourceIncludes(&pbfile, generator_parameters) + |
||||
grpc_cpp_generator::GetSourceServices(&pbfile, generator_parameters) + |
||||
grpc_cpp_generator::GetSourceEpilogue(&pbfile, generator_parameters); |
||||
std::unique_ptr<grpc::protobuf::io::ZeroCopyOutputStream> source_output( |
||||
context->Open(file_name + ".grpc.pb.cc")); |
||||
grpc::protobuf::io::CodedOutputStream source_coded_out(source_output.get()); |
||||
source_coded_out.WriteRaw(source_code.data(), source_code.size()); |
||||
|
||||
if (!generator_parameters.generate_mock_code) { |
||||
return true; |
||||
} |
||||
grpc::string mock_code = |
||||
grpc_cpp_generator::GetMockPrologue(&pbfile, generator_parameters) + |
||||
grpc_cpp_generator::GetMockIncludes(&pbfile, generator_parameters) + |
||||
grpc_cpp_generator::GetMockServices(&pbfile, generator_parameters) + |
||||
grpc_cpp_generator::GetMockEpilogue(&pbfile, generator_parameters); |
||||
std::unique_ptr<grpc::protobuf::io::ZeroCopyOutputStream> mock_output( |
||||
context->Open(file_name + "_mock.grpc.pb.h")); |
||||
grpc::protobuf::io::CodedOutputStream mock_coded_out(mock_output.get()); |
||||
mock_coded_out.WriteRaw(mock_code.data(), mock_code.size()); |
||||
|
||||
return true; |
||||
} |
||||
|
||||
private: |
||||
// Insert the given code into the given file at the given insertion point.
|
||||
void Insert(grpc::protobuf::compiler::GeneratorContext* context, |
||||
const grpc::string& filename, const grpc::string& insertion_point, |
||||
const grpc::string& code) const { |
||||
std::unique_ptr<grpc::protobuf::io::ZeroCopyOutputStream> output( |
||||
context->OpenForInsert(filename, insertion_point)); |
||||
grpc::protobuf::io::CodedOutputStream coded_out(output.get()); |
||||
coded_out.WriteRaw(code.data(), code.size()); |
||||
} |
||||
}; |
||||
|
||||
#endif // GRPC_INTERNAL_COMPILER_CPP_PLUGIN_H
|
||||
Loading…
Reference in new issue