Chiebot-Mirror
/
grpc
mirror of https://github.com/grpc/grpc.git
8
0
Fork 0
Code Issues Projects Releases Wiki Activity

Fix errors

Browse Source
pull/8063/head
Harsh Vardhan 8 years ago
parent 64741b279e
commit 6a9950dc0a
No known key found for this signature in database
GPG Key ID: 4A3BADA5BCFFEFD6
4 changed files with 98 additions and 84 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 19
      src/compiler/protobuf_plugin.h
  2. 122
      src/compiler/python_generator.cc
  3. 34
      src/compiler/python_private_generator.h
  4. 7
      src/compiler/schema_interface.h

19
src/compiler/protobuf_plugin.h
Unescape View File

@ -31,19 +31,26 @@ class ProtoBufMethod : public grpc_generator::Method {
return grpc_cpp_generator::ClassName(method_->output_type(), true); return grpc_cpp_generator::ClassName(method_->output_type(), true);
} }
bool get_module_and_message_path_input(grpc::string str, grpc::string get_input_type_name() const {
return method_->input_type()->file()->name();
}
grpc::string get_output_type_name() const {
return method_->output_type()->file()->name();
}
bool get_module_and_message_path_input(grpc::string *str,
grpc::string generator_file_name, grpc::string generator_file_name,
bool generate_in_pb2_grpc) const { bool generate_in_pb2_grpc) const {
return grpc_python_generator::GetModuleAndMessagePath(method_->input_type(), return grpc_python_generator::GetModuleAndMessagePath(method_->input_type(),
&str, generator_file_name, str, generator_file_name,
generate_in_pb2_grpc); generate_in_pb2_grpc);
} }
bool get_module_and_message_path_output(grpc::string str, bool get_module_and_message_path_output(grpc::string *str,
grpc::string generator_file_name, grpc::string generator_file_name,
bool generate_in_pb2_grpc) const { bool generate_in_pb2_grpc) const {
return grpc_python_generator::GetModuleAndMessagePath(method_->output_type(), return grpc_python_generator::GetModuleAndMessagePath(method_->output_type(),
&str, generator_file_name, str, generator_file_name,
generate_in_pb2_grpc); generate_in_pb2_grpc);
} }
@ -91,10 +98,6 @@ class ProtoBufService : public grpc_generator::Service {
return std::unique_ptr<const grpc_generator::Method>( return std::unique_ptr<const grpc_generator::Method>(
new ProtoBufMethod(service_->method(i))); new ProtoBufMethod(service_->method(i)));
}; };
std::unique_ptr<const grpc::protobuf::MethodDescriptor> get_method(int i) const {
return std::unique_ptr<const grpc::protobuf::MethodDescriptor>(
service_->method(i));
};
grpc::string GetLeadingComments(const grpc::string prefix) const { grpc::string GetLeadingComments(const grpc::string prefix) const {
return GetCommentsHelper(service_, true, prefix); return GetCommentsHelper(service_, true, prefix);

122
src/compiler/python_generator.cc
Unescape View File

@ -107,7 +107,8 @@ PrivateGenerator::PrivateGenerator(const GeneratorConfiguration& config,
const grpc_generator::File* file) const grpc_generator::File* file)
: config(config), file(file) {} : config(config), file(file) {}
void PrivateGenerator::PrintAllComments(StringVector comments) { void PrivateGenerator::PrintAllComments(StringVector comments,
grpc_generator::Printer* out) {
if (comments.empty()) { if (comments.empty()) {
return; return;
@ -124,7 +125,8 @@ void PrivateGenerator::PrintAllComments(StringVector comments) {
out->Print("\"\"\"\n"); out->Print("\"\"\"\n");
} }
bool PrivateGenerator::PrintBetaServicer(const grpc_generator::Service* service) { bool PrivateGenerator::PrintBetaServicer(const grpc_generator::Service* service,
grpc_generator::Printer* out) {
StringMap service_dict; StringMap service_dict;
service_dict["Service"] = service->name(); service_dict["Service"] = service->name();
out->Print("\n\n"); out->Print("\n\n");
@ -139,7 +141,7 @@ bool PrivateGenerator::PrintBetaServicer(const grpc_generator::Service* service)
"only to ease transition from grpcio<0.15.0 to " "only to ease transition from grpcio<0.15.0 to "
"grpcio>=0.15.0.\"\"\"\n"); "grpcio>=0.15.0.\"\"\"\n");
StringVector service_comments = service->GetAllComments(); StringVector service_comments = service->GetAllComments();
PrintAllComments(service_comments); PrintAllComments(service_comments, out);
for (int i = 0; i < service->method_count(); ++i) { for (int i = 0; i < service->method_count(); ++i) {
auto method = service->method(i); auto method = service->method(i);
grpc::string arg_name = grpc::string arg_name =
@ -151,7 +153,7 @@ bool PrivateGenerator::PrintBetaServicer(const grpc_generator::Service* service)
{ {
IndentScope raii_method_indent(out); IndentScope raii_method_indent(out);
StringVector method_comments = method.get()->GetAllComments(); StringVector method_comments = method.get()->GetAllComments();
PrintAllComments(method_comments); PrintAllComments(method_comments, out);
out->Print("context.code(beta_interfaces.StatusCode.UNIMPLEMENTED)\n"); out->Print("context.code(beta_interfaces.StatusCode.UNIMPLEMENTED)\n");
} }
} }
@ -159,7 +161,8 @@ bool PrivateGenerator::PrintBetaServicer(const grpc_generator::Service* service)
return true; return true;
} }
bool PrivateGenerator::PrintBetaStub(const grpc_generator::Service* service) { bool PrivateGenerator::PrintBetaStub(const grpc_generator::Service* service,
grpc_generator::Printer* out) {
StringMap service_dict; StringMap service_dict;
service_dict["Service"] = service->name(); service_dict["Service"] = service->name();
out->Print("\n\n"); out->Print("\n\n");
@ -174,7 +177,7 @@ bool PrivateGenerator::PrintBetaStub(const grpc_generator::Service* service) {
"only to ease transition from grpcio<0.15.0 to " "only to ease transition from grpcio<0.15.0 to "
"grpcio>=0.15.0.\"\"\"\n"); "grpcio>=0.15.0.\"\"\"\n");
StringVector service_comments = service->GetAllComments(); StringVector service_comments = service->GetAllComments();
PrintAllComments(service_comments); PrintAllComments(service_comments, out);
for (int i = 0; i < service->method_count(); ++i) { for (int i = 0; i < service->method_count(); ++i) {
auto method = service->method(i); auto method = service->method(i);
grpc::string arg_name = grpc::string arg_name =
@ -188,7 +191,7 @@ bool PrivateGenerator::PrintBetaStub(const grpc_generator::Service* service) {
{ {
IndentScope raii_method_indent(out); IndentScope raii_method_indent(out);
StringVector method_comments = method.get()->GetAllComments(); StringVector method_comments = method.get()->GetAllComments();
PrintAllComments(method_comments); PrintAllComments(method_comments, out);
out->Print("raise NotImplementedError()\n"); out->Print("raise NotImplementedError()\n");
} }
if (!method.get()->ServerStreaming()) { if (!method.get()->ServerStreaming()) {
@ -201,7 +204,7 @@ bool PrivateGenerator::PrintBetaStub(const grpc_generator::Service* service) {
bool PrivateGenerator::PrintBetaServerFactory( bool PrivateGenerator::PrintBetaServerFactory(
const grpc::string& package_qualified_service_name, const grpc::string& package_qualified_service_name,
const grpc_generator::Service* service) { const grpc_generator::Service* service, grpc_generator::Printer* out) {
StringMap service_dict; StringMap service_dict;
service_dict["Service"] = service->name(); service_dict["Service"] = service->name();
out->Print("\n\n"); out->Print("\n\n");
@ -226,12 +229,12 @@ bool PrivateGenerator::PrintBetaServerFactory(
grpc::string(method.get()->ServerStreaming() ? "stream_" : "unary_") + grpc::string(method.get()->ServerStreaming() ? "stream_" : "unary_") +
"inline"; "inline";
grpc::string input_message_module_and_class; grpc::string input_message_module_and_class;
if (!method.get()->get_module_and_message_path_input(input_message_module_and_class, if (!method.get()->get_module_and_message_path_input(&input_message_module_and_class,
generator_file_name, generate_in_pb2_grpc)) { generator_file_name, generate_in_pb2_grpc)) {
return false; return false;
} }
grpc::string output_message_module_and_class; grpc::string output_message_module_and_class;
if (!method.get()->get_module_and_message_path_output(output_message_module_and_class, if (!method.get()->get_module_and_message_path_output(&output_message_module_and_class,
generator_file_name, generate_in_pb2_grpc)) { generator_file_name, generate_in_pb2_grpc)) {
return false; return false;
} }
@ -304,7 +307,8 @@ bool PrivateGenerator::PrintBetaServerFactory(
bool PrivateGenerator::PrintBetaStubFactory( bool PrivateGenerator::PrintBetaStubFactory(
const grpc::string& package_qualified_service_name, const grpc::string& package_qualified_service_name,
const grpc_generator::Service* service) { const grpc_generator::Service* service,
grpc_generator::Printer* out) {
StringMap dict; StringMap dict;
dict["Service"] = service->name(); dict["Service"] = service->name();
out->Print("\n\n"); out->Print("\n\n");
@ -328,12 +332,12 @@ bool PrivateGenerator::PrintBetaStubFactory(
grpc::string(method.get()->ClientStreaming() ? "STREAM" : "UNARY") + "_" + grpc::string(method.get()->ClientStreaming() ? "STREAM" : "UNARY") + "_" +
grpc::string(method.get()->ServerStreaming() ? "STREAM" : "UNARY"); grpc::string(method.get()->ServerStreaming() ? "STREAM" : "UNARY");
grpc::string input_message_module_and_class; grpc::string input_message_module_and_class;
if (!method.get()->get_module_and_message_path_input(input_message_module_and_class, if (!method.get()->get_module_and_message_path_input(&input_message_module_and_class,
generator_file_name, generate_in_pb2_grpc)) { generator_file_name, generate_in_pb2_grpc)) {
return false; return false;
} }
grpc::string output_message_module_and_class; grpc::string output_message_module_and_class;
if (!method.get()->get_module_and_message_path_output(output_message_module_and_class, if (!method.get()->get_module_and_message_path_output(&output_message_module_and_class,
generator_file_name, generate_in_pb2_grpc)) { generator_file_name, generate_in_pb2_grpc)) {
return false; return false;
} }
@ -401,7 +405,8 @@ bool PrivateGenerator::PrintBetaStubFactory(
bool PrivateGenerator::PrintStub( bool PrivateGenerator::PrintStub(
const grpc::string& package_qualified_service_name, const grpc::string& package_qualified_service_name,
const grpc_generator::Service* service) { const grpc_generator::Service* service,
grpc_generator::Printer* out) {
StringMap dict; StringMap dict;
dict["Service"] = service->name(); dict["Service"] = service->name();
out->Print("\n\n"); out->Print("\n\n");
@ -409,7 +414,7 @@ bool PrivateGenerator::PrintStub(
{ {
IndentScope raii_class_indent(out); IndentScope raii_class_indent(out);
StringVector service_comments = service->GetAllComments(); StringVector service_comments = service->GetAllComments();
PrintAllComments(service_comments); PrintAllComments(service_comments, out);
out->Print("\n"); out->Print("\n");
out->Print("def __init__(self, channel):\n"); out->Print("def __init__(self, channel):\n");
{ {
@ -428,12 +433,12 @@ bool PrivateGenerator::PrintStub(
grpc::string(method.get()->ClientStreaming() ? "stream" : "unary") + grpc::string(method.get()->ClientStreaming() ? "stream" : "unary") +
"_" + grpc::string(method.get()->ServerStreaming() ? "stream" : "unary"); "_" + grpc::string(method.get()->ServerStreaming() ? "stream" : "unary");
grpc::string request_module_and_class; grpc::string request_module_and_class;
if (!method.get()->get_module_and_message_path_input(request_module_and_class, if (!method.get()->get_module_and_message_path_input(&request_module_and_class,
generator_file_name, generate_in_pb2_grpc)) { generator_file_name, generate_in_pb2_grpc)) {
return false; return false;
} }
grpc::string response_module_and_class; grpc::string response_module_and_class;
if (!method.get()->get_module_and_message_path_output(response_module_and_class, if (!method.get()->get_module_and_message_path_output(&response_module_and_class,
generator_file_name, generate_in_pb2_grpc)) { generator_file_name, generate_in_pb2_grpc)) {
return false; return false;
} }
@ -460,7 +465,8 @@ bool PrivateGenerator::PrintStub(
return true; return true;
} }
bool PrivateGenerator::PrintServicer(const grpc_generator::Service* service) { bool PrivateGenerator::PrintServicer(const grpc_generator::Service* service,
grpc_generator::Printer* out) {
StringMap service_dict; StringMap service_dict;
service_dict["Service"] = service->name(); service_dict["Service"] = service->name();
out->Print("\n\n"); out->Print("\n\n");
@ -468,7 +474,7 @@ bool PrivateGenerator::PrintServicer(const grpc_generator::Service* service) {
{ {
IndentScope raii_class_indent(out); IndentScope raii_class_indent(out);
StringVector service_comments = service->GetAllComments(); StringVector service_comments = service->GetAllComments();
PrintAllComments(service_comments); PrintAllComments(service_comments, out);
for (int i = 0; i < service->method_count(); ++i) { for (int i = 0; i < service->method_count(); ++i) {
auto method = service->method(i); auto method = service->method(i);
grpc::string arg_name = grpc::string arg_name =
@ -481,7 +487,7 @@ bool PrivateGenerator::PrintServicer(const grpc_generator::Service* service) {
{ {
IndentScope raii_method_indent(out); IndentScope raii_method_indent(out);
StringVector method_comments = method.get()->GetAllComments(); StringVector method_comments = method.get()->GetAllComments();
PrintAllComments(method_comments); PrintAllComments(method_comments, out);
out->Print("context.set_code(grpc.StatusCode.UNIMPLEMENTED)\n"); out->Print("context.set_code(grpc.StatusCode.UNIMPLEMENTED)\n");
out->Print("context.set_details('Method not implemented!')\n"); out->Print("context.set_details('Method not implemented!')\n");
out->Print("raise NotImplementedError('Method not implemented!')\n"); out->Print("raise NotImplementedError('Method not implemented!')\n");
@ -493,7 +499,7 @@ bool PrivateGenerator::PrintServicer(const grpc_generator::Service* service) {
bool PrivateGenerator::PrintAddServicerToServer( bool PrivateGenerator::PrintAddServicerToServer(
const grpc::string& package_qualified_service_name, const grpc::string& package_qualified_service_name,
const grpc_generator::Service* service) { const grpc_generator::Service* service, grpc_generator::Printer* out) {
StringMap service_dict; StringMap service_dict;
service_dict["Service"] = service->name(); service_dict["Service"] = service->name();
out->Print("\n\n"); out->Print("\n\n");
@ -512,12 +518,12 @@ bool PrivateGenerator::PrintAddServicerToServer(
grpc::string(method.get()->ServerStreaming() ? "stream" : "unary") + grpc::string(method.get()->ServerStreaming() ? "stream" : "unary") +
"_rpc_method_handler"; "_rpc_method_handler";
grpc::string request_module_and_class; grpc::string request_module_and_class;
if (!method.get()->get_module_and_message_path_input(request_module_and_class, if (!method.get()->get_module_and_message_path_input(&request_module_and_class,
generator_file_name, generate_in_pb2_grpc)) { generator_file_name, generate_in_pb2_grpc)) {
return false; return false;
} }
grpc::string response_module_and_class; grpc::string response_module_and_class;
if (!method.get()->get_module_and_message_path_output(response_module_and_class, if (!method.get()->get_module_and_message_path_output(&response_module_and_class,
generator_file_name, generate_in_pb2_grpc)) { generator_file_name, generate_in_pb2_grpc)) {
return false; return false;
} }
@ -554,7 +560,7 @@ bool PrivateGenerator::PrintAddServicerToServer(
return true; return true;
} }
bool PrivateGenerator::PrintBetaPreamble() { bool PrivateGenerator::PrintBetaPreamble(grpc_generator::Printer *out) {
StringMap var; StringMap var;
var["Package"] = config.beta_package_root; var["Package"] = config.beta_package_root;
out->Print(var, "from $Package$ import implementations as beta_implementations\n"); out->Print(var, "from $Package$ import implementations as beta_implementations\n");
@ -566,7 +572,7 @@ bool PrivateGenerator::PrintBetaPreamble() {
return true; return true;
} }
bool PrivateGenerator::PrintPreamble() { bool PrivateGenerator::PrintPreamble(grpc_generator::Printer* out) {
StringMap var; StringMap var;
var["Package"] = config.grpc_package_root; var["Package"] = config.grpc_package_root;
out->Print(var, "import $Package$\n"); out->Print(var, "import $Package$\n");
@ -574,23 +580,22 @@ bool PrivateGenerator::PrintPreamble() {
out->Print("\n"); out->Print("\n");
StringPairSet imports_set; StringPairSet imports_set;
for (int i = 0; i < file->service_count(); ++i) { for (int i = 0; i < file->service_count(); ++i) {
auto service = file->service(i).get(); auto service = file->service(i);
for (int j = 0; j < service->method_count(); ++j) { for (int j = 0; j < service->method_count(); ++j) {
const MethodDescriptor* method = service->get_method(j).get(); auto method = service.get()->method(j);
const Descriptor* types[2] = {method->input_type(),
method->output_type()}; grpc::string input_type_file_name = method->get_input_type_name();
for (int k = 0; k < 2; ++k) { grpc::string input_module_name = ModuleName(input_type_file_name);
const Descriptor* type = types[k]; grpc::string input_module_alias = ModuleAlias(input_type_file_name);
grpc::string type_file_name = type->file()->name(); imports_set.insert(std::make_tuple(input_module_name, input_module_alias));
grpc::string module_name =
ModuleName(type_file_name, config.import_prefix); grpc::string output_type_file_name = method->get_output_type_name();
grpc::string module_alias = grpc::string output_module_name = ModuleName(output_type_file_name);
ModuleAlias(type_file_name, config.import_prefix); grpc::string output_module_alias = ModuleAlias(output_type_file_name);
imports_set.insert(std::make_tuple(module_name, module_alias)); imports_set.insert(std::make_tuple(output_module_name, output_module_alias));
}
} }
} }
StringMap var;
for (StringPairSet::iterator it = imports_set.begin(); for (StringPairSet::iterator it = imports_set.begin();
it != imports_set.end(); ++it) { it != imports_set.end(); ++it) {
var["ModuleName"] = std::get<0>(*it); var["ModuleName"] = std::get<0>(*it);
@ -601,34 +606,34 @@ bool PrivateGenerator::PrintPreamble() {
return true; return true;
} }
bool PrivateGenerator::PrintGAServices() { bool PrivateGenerator::PrintGAServices(grpc_generator::Printer* out) {
grpc::string package = file->package(); grpc::string package = file->package();
if (!package.empty()) { if (!package.empty()) {
package = package.append("."); package = package.append(".");
} }
for (int i = 0; i < file->service_count(); ++i) { for (int i = 0; i < file->service_count(); ++i) {
auto service = file->service(i).get(); auto service = file->service(i);
grpc::string package_qualified_service_name = package + service->name(); grpc::string package_qualified_service_name = package + service->name();
if (!(PrintStub(package_qualified_service_name, service) && if (!(PrintStub(package_qualified_service_name, service.get(), out) &&
PrintServicer(service) && PrintServicer(service.get(), out) &&
PrintAddServicerToServer(package_qualified_service_name, service))) { PrintAddServicerToServer(package_qualified_service_name, service.get(), out))) {
return false; return false;
} }
} }
return true; return true;
} }
bool PrivateGenerator::PrintBetaServices() { bool PrivateGenerator::PrintBetaServices(grpc_generator::Printer* out) {
grpc::string package = file->package(); grpc::string package = file->package();
if (!package.empty()) { if (!package.empty()) {
package = package.append("."); package = package.append(".");
} }
for (int i = 0; i < file->service_count(); ++i) { for (int i = 0; i < file->service_count(); ++i) {
auto service = file->service(i).get(); auto service = file->service(i);
grpc::string package_qualified_service_name = package + service->name(); grpc::string package_qualified_service_name = package + service->name();
if (!(PrintBetaServicer(service) && PrintBetaStub(service) && if (!(PrintBetaServicer(service.get(), out) && PrintBetaStub(service.get(), out) &&
PrintBetaServerFactory(package_qualified_service_name, service) && PrintBetaServerFactory(package_qualified_service_name, service.get(), out) &&
PrintBetaStubFactory(package_qualified_service_name, service))) { PrintBetaStubFactory(package_qualified_service_name, service.get(), out))) {
return false; return false;
} }
} }
@ -639,41 +644,40 @@ pair<bool, grpc::string> PrivateGenerator::GetGrpcServices() {
grpc::string output; grpc::string output;
{ {
// Scope the output stream so it closes and finalizes output to the string. // Scope the output stream so it closes and finalizes output to the string.
auto out = file->CreatePrinter(&output).get(); auto out = file->CreatePrinter(&output);
if (generate_in_pb2_grpc) { if (generate_in_pb2_grpc) {
out->Print( out->Print(
"# Generated by the gRPC Python protocol compiler plugin. " "# Generated by the gRPC Python protocol compiler plugin. "
"DO NOT EDIT!\n"); "DO NOT EDIT!\n");
if (!PrintPreamble()) { if (!PrintPreamble(out.get())) {
return make_pair(false, ""); return make_pair(false, "");
} }
if (!PrintGAServices()) { if (!PrintGAServices(out.get())) {
return make_pair(false, ""); return make_pair(false, "");
} }
} else { } else {
out->Print("try:\n"); out->Print("try:\n");
{ {
IndentScope raii_dict_try_indent(out); IndentScope raii_dict_try_indent(out.get());
out->Print( out->Print(
"# THESE ELEMENTS WILL BE DEPRECATED.\n" "# THESE ELEMENTS WILL BE DEPRECATED.\n"
"# Please use the generated *_pb2_grpc.py files instead.\n"); "# Please use the generated *_pb2_grpc.py files instead.\n");
if (!PrintPreamble()) { if (!PrintPreamble(out.get())) {
return make_pair(false, ""); return make_pair(false, "");
} }
if (!PrintBetaPreamble()) { if (!PrintBetaPreamble(out.get())) {
return make_pair(false, ""); return make_pair(false, "");
} }
if (!PrintGAServices()) { if (!PrintGAServices(out.get())) {
return make_pair(false, ""); return make_pair(false, "");
} }
if (!PrintBetaServices()) { if (!PrintBetaServices(out.get())) {
return make_pair(false, ""); return make_pair(false, "");
} }
} }
out->Print("except ImportError:\n"); out->Print("except ImportError:\n");
{ {
IndentScope raii_dict_except_indent(out); IndentScope raii_dict_except_indent(out.get());
out->Print("pass"); out->Print("pass");
} }
} }

34
src/compiler/python_private_generator.h
Unescape View File

@ -54,38 +54,44 @@ struct PrivateGenerator {
bool generate_in_pb2_grpc; bool generate_in_pb2_grpc;
grpc_generator::Printer* out;
PrivateGenerator(const GeneratorConfiguration& config, PrivateGenerator(const GeneratorConfiguration& config,
const grpc_generator::File* file); const grpc_generator::File* file);
std::pair<bool, grpc::string> GetGrpcServices(); std::pair<bool, grpc::string> GetGrpcServices();
private: private:
bool PrintPreamble(); bool PrintPreamble(grpc_generator::Printer* out);
bool PrintBetaPreamble(); bool PrintBetaPreamble(grpc_generator::Printer *out);
bool PrintGAServices(); bool PrintGAServices(grpc_generator::Printer* out);
bool PrintBetaServices(); bool PrintBetaServices(grpc_generator::Printer* out);
bool PrintAddServicerToServer( bool PrintAddServicerToServer(
const grpc::string& package_qualified_service_name, const grpc::string& package_qualified_service_name,
const grpc_generator::Service* service); const grpc_generator::Service* service,
bool PrintServicer(const grpc_generator::Service* service); grpc_generator::Printer* out);
bool PrintServicer(const grpc_generator::Service* service,
grpc_generator::Printer* out);
bool PrintStub(const grpc::string& package_qualified_service_name, bool PrintStub(const grpc::string& package_qualified_service_name,
const grpc_generator::Service* service); const grpc_generator::Service* service,
grpc_generator::Printer* out);
bool PrintBetaServicer(const grpc_generator::Service* service); bool PrintBetaServicer(const grpc_generator::Service* service,
grpc_generator::Printer* out);
bool PrintBetaServerFactory( bool PrintBetaServerFactory(
const grpc::string& package_qualified_service_name, const grpc::string& package_qualified_service_name,
const grpc_generator::Service* service); const grpc_generator::Service* service,
bool PrintBetaStub(const grpc_generator::Service* service); grpc_generator::Printer* out);
bool PrintBetaStub(const grpc_generator::Service* service,
grpc_generator::Printer* out);
bool PrintBetaStubFactory(const grpc::string& package_qualified_service_name, bool PrintBetaStubFactory(const grpc::string& package_qualified_service_name,
const grpc_generator::Service* service); const grpc_generator::Service* service,
grpc_generator::Printer* out);
// Get all comments (leading, leading_detached, trailing) and print them as a // Get all comments (leading, leading_detached, trailing) and print them as a
// docstring. Any leading space of a line will be removed, but the line // docstring. Any leading space of a line will be removed, but the line
// wrapping will not be changed. // wrapping will not be changed.
void PrintAllComments(std::vector<grpc::string> comments); void PrintAllComments(std::vector<grpc::string> comments,
grpc_generator::Printer* out);
}; };

7
src/compiler/schema_interface.h
Unescape View File

@ -70,13 +70,15 @@ namespace grpc_generator {
virtual grpc::string input_type_name() const = 0; virtual grpc::string input_type_name() const = 0;
virtual grpc::string output_type_name() const = 0; virtual grpc::string output_type_name() const = 0;
virtual bool get_module_and_message_path_input(grpc::string str, virtual bool get_module_and_message_path_input(grpc::string *str,
grpc::string generator_file_name, grpc::string generator_file_name,
bool generate_in_pb2_grpc) const = 0; bool generate_in_pb2_grpc) const = 0;
virtual bool get_module_and_message_path_output(grpc::string str, virtual bool get_module_and_message_path_output(grpc::string *str,
grpc::string generator_file_name, grpc::string generator_file_name,
bool generate_in_pb2_grpc) const = 0; bool generate_in_pb2_grpc) const = 0;
virtual grpc::string get_input_type_name() const = 0;
virtual grpc::string get_output_type_name() const = 0;
virtual bool NoStreaming() const = 0; virtual bool NoStreaming() const = 0;
virtual bool ClientStreaming() const = 0; virtual bool ClientStreaming() const = 0;
virtual bool ServerStreaming() const = 0; virtual bool ServerStreaming() const = 0;
@ -91,7 +93,6 @@ namespace grpc_generator {
virtual int method_count() const = 0; virtual int method_count() const = 0;
virtual std::unique_ptr<const Method> method(int i) const = 0; virtual std::unique_ptr<const Method> method(int i) const = 0;
virtual std::unique_ptr<const grpc::protobuf::MethodDescriptor> get_method(int i) const = 0;
}; };
struct Printer { struct Printer {

Write Preview
Loading…
Cancel
Save