Index: third_party/grpc/src/compiler/cpp_generator.cc |
diff --git a/third_party/grpc/src/compiler/cpp_generator.cc b/third_party/grpc/src/compiler/cpp_generator.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..97295bed4518b9325d8d420f2045c4b93baa14f0 |
--- /dev/null |
+++ b/third_party/grpc/src/compiler/cpp_generator.cc |
@@ -0,0 +1,1217 @@ |
+/* |
+ * |
+ * Copyright 2015-2016, Google Inc. |
+ * All rights reserved. |
+ * |
+ * 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. |
+ * |
+ */ |
+ |
+#include <map> |
+ |
+#include "src/compiler/cpp_generator.h" |
+#include "src/compiler/cpp_generator_helpers.h" |
+ |
+#include "src/compiler/config.h" |
+ |
+#include <sstream> |
+ |
+namespace grpc_cpp_generator { |
+namespace { |
+ |
+template <class T> |
+grpc::string as_string(T x) { |
+ std::ostringstream out; |
+ out << x; |
+ return out.str(); |
+} |
+ |
+bool NoStreaming(const grpc::protobuf::MethodDescriptor *method) { |
+ return !method->client_streaming() && !method->server_streaming(); |
+} |
+ |
+bool ClientOnlyStreaming(const grpc::protobuf::MethodDescriptor *method) { |
+ return method->client_streaming() && !method->server_streaming(); |
+} |
+ |
+bool ServerOnlyStreaming(const grpc::protobuf::MethodDescriptor *method) { |
+ return !method->client_streaming() && method->server_streaming(); |
+} |
+ |
+bool BidiStreaming(const grpc::protobuf::MethodDescriptor *method) { |
+ return method->client_streaming() && method->server_streaming(); |
+} |
+ |
+grpc::string FilenameIdentifier(const grpc::string &filename) { |
+ grpc::string result; |
+ for (unsigned i = 0; i < filename.size(); i++) { |
+ char c = filename[i]; |
+ if (isalnum(c)) { |
+ result.push_back(c); |
+ } else { |
+ static char hex[] = "0123456789abcdef"; |
+ result.push_back('_'); |
+ result.push_back(hex[(c >> 4) & 0xf]); |
+ result.push_back(hex[c & 0xf]); |
+ } |
+ } |
+ return result; |
+} |
+} // namespace |
+ |
+grpc::string GetHeaderPrologue(const grpc::protobuf::FileDescriptor *file, |
+ const Parameters ¶ms) { |
+ grpc::string output; |
+ { |
+ // Scope the output stream so it closes and finalizes output to the string. |
+ grpc::protobuf::io::StringOutputStream output_stream(&output); |
+ grpc::protobuf::io::Printer printer(&output_stream, '$'); |
+ std::map<grpc::string, grpc::string> vars; |
+ |
+ vars["filename"] = file->name(); |
+ vars["filename_identifier"] = FilenameIdentifier(file->name()); |
+ vars["filename_base"] = grpc_generator::StripProto(file->name()); |
+ |
+ printer.Print(vars, "// Generated by the gRPC protobuf plugin.\n"); |
+ printer.Print(vars, |
+ "// If you make any local change, they will be lost.\n"); |
+ printer.Print(vars, "// source: $filename$\n"); |
+ printer.Print(vars, "#ifndef GRPC_$filename_identifier$__INCLUDED\n"); |
+ printer.Print(vars, "#define GRPC_$filename_identifier$__INCLUDED\n"); |
+ printer.Print(vars, "\n"); |
+ printer.Print(vars, "#include \"$filename_base$.pb.h\"\n"); |
+ printer.Print(vars, "\n"); |
+ } |
+ return output; |
+} |
+ |
+grpc::string GetHeaderIncludes(const grpc::protobuf::FileDescriptor *file, |
+ const Parameters ¶ms) { |
+ grpc::string temp = |
+ "#include <grpc++/impl/codegen/async_stream.h>\n" |
+ "#include <grpc++/impl/codegen/async_unary_call.h>\n" |
+ "#include <grpc++/impl/codegen/proto_utils.h>\n" |
+ "#include <grpc++/impl/codegen/rpc_method.h>\n" |
+ "#include <grpc++/impl/codegen/service_type.h>\n" |
+ "#include <grpc++/impl/codegen/status.h>\n" |
+ "#include <grpc++/impl/codegen/stub_options.h>\n" |
+ "#include <grpc++/impl/codegen/sync_stream.h>\n" |
+ "\n" |
+ "namespace grpc {\n" |
+ "class CompletionQueue;\n" |
+ "class RpcService;\n" |
+ "class ServerCompletionQueue;\n" |
+ "class ServerContext;\n" |
+ "} // namespace grpc\n\n"; |
+ |
+ if (!file->package().empty()) { |
+ std::vector<grpc::string> parts = |
+ grpc_generator::tokenize(file->package(), "."); |
+ |
+ for (auto part = parts.begin(); part != parts.end(); part++) { |
+ temp.append("namespace "); |
+ temp.append(*part); |
+ temp.append(" {\n"); |
+ } |
+ temp.append("\n"); |
+ } |
+ |
+ return temp; |
+} |
+ |
+void PrintHeaderClientMethodInterfaces( |
+ grpc::protobuf::io::Printer *printer, |
+ const grpc::protobuf::MethodDescriptor *method, |
+ std::map<grpc::string, grpc::string> *vars, bool is_public) { |
+ (*vars)["Method"] = method->name(); |
+ (*vars)["Request"] = |
+ grpc_cpp_generator::ClassName(method->input_type(), true); |
+ (*vars)["Response"] = |
+ grpc_cpp_generator::ClassName(method->output_type(), true); |
+ |
+ if (is_public) { |
+ if (NoStreaming(method)) { |
+ printer->Print( |
+ *vars, |
+ "virtual ::grpc::Status $Method$(::grpc::ClientContext* context, " |
+ "const $Request$& request, $Response$* response) = 0;\n"); |
+ printer->Print(*vars, |
+ "std::unique_ptr< " |
+ "::grpc::ClientAsyncResponseReaderInterface< $Response$>> " |
+ "Async$Method$(::grpc::ClientContext* context, " |
+ "const $Request$& request, " |
+ "::grpc::CompletionQueue* cq) {\n"); |
+ printer->Indent(); |
+ printer->Print(*vars, |
+ "return std::unique_ptr< " |
+ "::grpc::ClientAsyncResponseReaderInterface< $Response$>>(" |
+ "Async$Method$Raw(context, request, cq));\n"); |
+ printer->Outdent(); |
+ printer->Print("}\n"); |
+ } else if (ClientOnlyStreaming(method)) { |
+ printer->Print( |
+ *vars, |
+ "std::unique_ptr< ::grpc::ClientWriterInterface< $Request$>>" |
+ " $Method$(" |
+ "::grpc::ClientContext* context, $Response$* response) {\n"); |
+ printer->Indent(); |
+ printer->Print( |
+ *vars, |
+ "return std::unique_ptr< ::grpc::ClientWriterInterface< $Request$>>" |
+ "($Method$Raw(context, response));\n"); |
+ printer->Outdent(); |
+ printer->Print("}\n"); |
+ printer->Print( |
+ *vars, |
+ "std::unique_ptr< ::grpc::ClientAsyncWriterInterface< $Request$>>" |
+ " Async$Method$(::grpc::ClientContext* context, $Response$* " |
+ "response, " |
+ "::grpc::CompletionQueue* cq, void* tag) {\n"); |
+ printer->Indent(); |
+ printer->Print(*vars, |
+ "return std::unique_ptr< " |
+ "::grpc::ClientAsyncWriterInterface< $Request$>>(" |
+ "Async$Method$Raw(context, response, cq, tag));\n"); |
+ printer->Outdent(); |
+ printer->Print("}\n"); |
+ } else if (ServerOnlyStreaming(method)) { |
+ printer->Print( |
+ *vars, |
+ "std::unique_ptr< ::grpc::ClientReaderInterface< $Response$>>" |
+ " $Method$(::grpc::ClientContext* context, const $Request$& request)" |
+ " {\n"); |
+ printer->Indent(); |
+ printer->Print( |
+ *vars, |
+ "return std::unique_ptr< ::grpc::ClientReaderInterface< $Response$>>" |
+ "($Method$Raw(context, request));\n"); |
+ printer->Outdent(); |
+ printer->Print("}\n"); |
+ printer->Print( |
+ *vars, |
+ "std::unique_ptr< ::grpc::ClientAsyncReaderInterface< $Response$>> " |
+ "Async$Method$(" |
+ "::grpc::ClientContext* context, const $Request$& request, " |
+ "::grpc::CompletionQueue* cq, void* tag) {\n"); |
+ printer->Indent(); |
+ printer->Print(*vars, |
+ "return std::unique_ptr< " |
+ "::grpc::ClientAsyncReaderInterface< $Response$>>(" |
+ "Async$Method$Raw(context, request, cq, tag));\n"); |
+ printer->Outdent(); |
+ printer->Print("}\n"); |
+ } else if (BidiStreaming(method)) { |
+ printer->Print(*vars, |
+ "std::unique_ptr< ::grpc::ClientReaderWriterInterface< " |
+ "$Request$, $Response$>> " |
+ "$Method$(::grpc::ClientContext* context) {\n"); |
+ printer->Indent(); |
+ printer->Print( |
+ *vars, |
+ "return std::unique_ptr< " |
+ "::grpc::ClientReaderWriterInterface< $Request$, $Response$>>(" |
+ "$Method$Raw(context));\n"); |
+ printer->Outdent(); |
+ printer->Print("}\n"); |
+ printer->Print( |
+ *vars, |
+ "std::unique_ptr< " |
+ "::grpc::ClientAsyncReaderWriterInterface< $Request$, $Response$>> " |
+ "Async$Method$(::grpc::ClientContext* context, " |
+ "::grpc::CompletionQueue* cq, void* tag) {\n"); |
+ printer->Indent(); |
+ printer->Print( |
+ *vars, |
+ "return std::unique_ptr< " |
+ "::grpc::ClientAsyncReaderWriterInterface< $Request$, $Response$>>(" |
+ "Async$Method$Raw(context, cq, tag));\n"); |
+ printer->Outdent(); |
+ printer->Print("}\n"); |
+ } |
+ } else { |
+ if (NoStreaming(method)) { |
+ printer->Print( |
+ *vars, |
+ "virtual ::grpc::ClientAsyncResponseReaderInterface< $Response$>* " |
+ "Async$Method$Raw(::grpc::ClientContext* context, " |
+ "const $Request$& request, " |
+ "::grpc::CompletionQueue* cq) = 0;\n"); |
+ } else if (ClientOnlyStreaming(method)) { |
+ printer->Print( |
+ *vars, |
+ "virtual ::grpc::ClientWriterInterface< $Request$>*" |
+ " $Method$Raw(" |
+ "::grpc::ClientContext* context, $Response$* response) = 0;\n"); |
+ printer->Print(*vars, |
+ "virtual ::grpc::ClientAsyncWriterInterface< $Request$>*" |
+ " Async$Method$Raw(::grpc::ClientContext* context, " |
+ "$Response$* response, " |
+ "::grpc::CompletionQueue* cq, void* tag) = 0;\n"); |
+ } else if (ServerOnlyStreaming(method)) { |
+ printer->Print( |
+ *vars, |
+ "virtual ::grpc::ClientReaderInterface< $Response$>* $Method$Raw(" |
+ "::grpc::ClientContext* context, const $Request$& request) = 0;\n"); |
+ printer->Print( |
+ *vars, |
+ "virtual ::grpc::ClientAsyncReaderInterface< $Response$>* " |
+ "Async$Method$Raw(" |
+ "::grpc::ClientContext* context, const $Request$& request, " |
+ "::grpc::CompletionQueue* cq, void* tag) = 0;\n"); |
+ } else if (BidiStreaming(method)) { |
+ printer->Print(*vars, |
+ "virtual ::grpc::ClientReaderWriterInterface< $Request$, " |
+ "$Response$>* " |
+ "$Method$Raw(::grpc::ClientContext* context) = 0;\n"); |
+ printer->Print(*vars, |
+ "virtual ::grpc::ClientAsyncReaderWriterInterface< " |
+ "$Request$, $Response$>* " |
+ "Async$Method$Raw(::grpc::ClientContext* context, " |
+ "::grpc::CompletionQueue* cq, void* tag) = 0;\n"); |
+ } |
+ } |
+} |
+ |
+void PrintHeaderClientMethod(grpc::protobuf::io::Printer *printer, |
+ const grpc::protobuf::MethodDescriptor *method, |
+ std::map<grpc::string, grpc::string> *vars, |
+ bool is_public) { |
+ (*vars)["Method"] = method->name(); |
+ (*vars)["Request"] = |
+ grpc_cpp_generator::ClassName(method->input_type(), true); |
+ (*vars)["Response"] = |
+ grpc_cpp_generator::ClassName(method->output_type(), true); |
+ if (is_public) { |
+ if (NoStreaming(method)) { |
+ printer->Print( |
+ *vars, |
+ "::grpc::Status $Method$(::grpc::ClientContext* context, " |
+ "const $Request$& request, $Response$* response) GRPC_OVERRIDE;\n"); |
+ printer->Print( |
+ *vars, |
+ "std::unique_ptr< ::grpc::ClientAsyncResponseReader< $Response$>> " |
+ "Async$Method$(::grpc::ClientContext* context, " |
+ "const $Request$& request, " |
+ "::grpc::CompletionQueue* cq) {\n"); |
+ printer->Indent(); |
+ printer->Print(*vars, |
+ "return std::unique_ptr< " |
+ "::grpc::ClientAsyncResponseReader< $Response$>>(" |
+ "Async$Method$Raw(context, request, cq));\n"); |
+ printer->Outdent(); |
+ printer->Print("}\n"); |
+ } else if (ClientOnlyStreaming(method)) { |
+ printer->Print( |
+ *vars, |
+ "std::unique_ptr< ::grpc::ClientWriter< $Request$>>" |
+ " $Method$(" |
+ "::grpc::ClientContext* context, $Response$* response) {\n"); |
+ printer->Indent(); |
+ printer->Print(*vars, |
+ "return std::unique_ptr< ::grpc::ClientWriter< $Request$>>" |
+ "($Method$Raw(context, response));\n"); |
+ printer->Outdent(); |
+ printer->Print("}\n"); |
+ printer->Print(*vars, |
+ "std::unique_ptr< ::grpc::ClientAsyncWriter< $Request$>>" |
+ " Async$Method$(::grpc::ClientContext* context, " |
+ "$Response$* response, " |
+ "::grpc::CompletionQueue* cq, void* tag) {\n"); |
+ printer->Indent(); |
+ printer->Print( |
+ *vars, |
+ "return std::unique_ptr< ::grpc::ClientAsyncWriter< $Request$>>(" |
+ "Async$Method$Raw(context, response, cq, tag));\n"); |
+ printer->Outdent(); |
+ printer->Print("}\n"); |
+ } else if (ServerOnlyStreaming(method)) { |
+ printer->Print( |
+ *vars, |
+ "std::unique_ptr< ::grpc::ClientReader< $Response$>>" |
+ " $Method$(::grpc::ClientContext* context, const $Request$& request)" |
+ " {\n"); |
+ printer->Indent(); |
+ printer->Print( |
+ *vars, |
+ "return std::unique_ptr< ::grpc::ClientReader< $Response$>>" |
+ "($Method$Raw(context, request));\n"); |
+ printer->Outdent(); |
+ printer->Print("}\n"); |
+ printer->Print( |
+ *vars, |
+ "std::unique_ptr< ::grpc::ClientAsyncReader< $Response$>> " |
+ "Async$Method$(" |
+ "::grpc::ClientContext* context, const $Request$& request, " |
+ "::grpc::CompletionQueue* cq, void* tag) {\n"); |
+ printer->Indent(); |
+ printer->Print( |
+ *vars, |
+ "return std::unique_ptr< ::grpc::ClientAsyncReader< $Response$>>(" |
+ "Async$Method$Raw(context, request, cq, tag));\n"); |
+ printer->Outdent(); |
+ printer->Print("}\n"); |
+ } else if (BidiStreaming(method)) { |
+ printer->Print( |
+ *vars, |
+ "std::unique_ptr< ::grpc::ClientReaderWriter< $Request$, $Response$>>" |
+ " $Method$(::grpc::ClientContext* context) {\n"); |
+ printer->Indent(); |
+ printer->Print(*vars, |
+ "return std::unique_ptr< " |
+ "::grpc::ClientReaderWriter< $Request$, $Response$>>(" |
+ "$Method$Raw(context));\n"); |
+ printer->Outdent(); |
+ printer->Print("}\n"); |
+ printer->Print(*vars, |
+ "std::unique_ptr< ::grpc::ClientAsyncReaderWriter< " |
+ "$Request$, $Response$>> " |
+ "Async$Method$(::grpc::ClientContext* context, " |
+ "::grpc::CompletionQueue* cq, void* tag) {\n"); |
+ printer->Indent(); |
+ printer->Print(*vars, |
+ "return std::unique_ptr< " |
+ "::grpc::ClientAsyncReaderWriter< $Request$, $Response$>>(" |
+ "Async$Method$Raw(context, cq, tag));\n"); |
+ printer->Outdent(); |
+ printer->Print("}\n"); |
+ } |
+ } else { |
+ if (NoStreaming(method)) { |
+ printer->Print(*vars, |
+ "::grpc::ClientAsyncResponseReader< $Response$>* " |
+ "Async$Method$Raw(::grpc::ClientContext* context, " |
+ "const $Request$& request, " |
+ "::grpc::CompletionQueue* cq) GRPC_OVERRIDE;\n"); |
+ } else if (ClientOnlyStreaming(method)) { |
+ printer->Print(*vars, |
+ "::grpc::ClientWriter< $Request$>* $Method$Raw(" |
+ "::grpc::ClientContext* context, $Response$* response) " |
+ "GRPC_OVERRIDE;\n"); |
+ printer->Print( |
+ *vars, |
+ "::grpc::ClientAsyncWriter< $Request$>* Async$Method$Raw(" |
+ "::grpc::ClientContext* context, $Response$* response, " |
+ "::grpc::CompletionQueue* cq, void* tag) GRPC_OVERRIDE;\n"); |
+ } else if (ServerOnlyStreaming(method)) { |
+ printer->Print(*vars, |
+ "::grpc::ClientReader< $Response$>* $Method$Raw(" |
+ "::grpc::ClientContext* context, const $Request$& request)" |
+ " GRPC_OVERRIDE;\n"); |
+ printer->Print( |
+ *vars, |
+ "::grpc::ClientAsyncReader< $Response$>* Async$Method$Raw(" |
+ "::grpc::ClientContext* context, const $Request$& request, " |
+ "::grpc::CompletionQueue* cq, void* tag) GRPC_OVERRIDE;\n"); |
+ } else if (BidiStreaming(method)) { |
+ printer->Print( |
+ *vars, |
+ "::grpc::ClientReaderWriter< $Request$, $Response$>* " |
+ "$Method$Raw(::grpc::ClientContext* context) GRPC_OVERRIDE;\n"); |
+ printer->Print( |
+ *vars, |
+ "::grpc::ClientAsyncReaderWriter< $Request$, $Response$>* " |
+ "Async$Method$Raw(::grpc::ClientContext* context, " |
+ "::grpc::CompletionQueue* cq, void* tag) GRPC_OVERRIDE;\n"); |
+ } |
+ } |
+} |
+ |
+void PrintHeaderClientMethodData(grpc::protobuf::io::Printer *printer, |
+ const grpc::protobuf::MethodDescriptor *method, |
+ std::map<grpc::string, grpc::string> *vars) { |
+ (*vars)["Method"] = method->name(); |
+ printer->Print(*vars, "const ::grpc::RpcMethod rpcmethod_$Method$_;\n"); |
+} |
+ |
+void PrintHeaderServerMethodSync(grpc::protobuf::io::Printer *printer, |
+ const grpc::protobuf::MethodDescriptor *method, |
+ std::map<grpc::string, grpc::string> *vars) { |
+ (*vars)["Method"] = method->name(); |
+ (*vars)["Request"] = |
+ grpc_cpp_generator::ClassName(method->input_type(), true); |
+ (*vars)["Response"] = |
+ grpc_cpp_generator::ClassName(method->output_type(), true); |
+ if (NoStreaming(method)) { |
+ printer->Print(*vars, |
+ "virtual ::grpc::Status $Method$(" |
+ "::grpc::ServerContext* context, const $Request$* request, " |
+ "$Response$* response);\n"); |
+ } else if (ClientOnlyStreaming(method)) { |
+ printer->Print(*vars, |
+ "virtual ::grpc::Status $Method$(" |
+ "::grpc::ServerContext* context, " |
+ "::grpc::ServerReader< $Request$>* reader, " |
+ "$Response$* response);\n"); |
+ } else if (ServerOnlyStreaming(method)) { |
+ printer->Print(*vars, |
+ "virtual ::grpc::Status $Method$(" |
+ "::grpc::ServerContext* context, const $Request$* request, " |
+ "::grpc::ServerWriter< $Response$>* writer);\n"); |
+ } else if (BidiStreaming(method)) { |
+ printer->Print( |
+ *vars, |
+ "virtual ::grpc::Status $Method$(" |
+ "::grpc::ServerContext* context, " |
+ "::grpc::ServerReaderWriter< $Response$, $Request$>* stream);" |
+ "\n"); |
+ } |
+} |
+ |
+void PrintHeaderServerMethodAsync( |
+ grpc::protobuf::io::Printer *printer, |
+ const grpc::protobuf::MethodDescriptor *method, |
+ std::map<grpc::string, grpc::string> *vars) { |
+ (*vars)["Method"] = method->name(); |
+ (*vars)["Request"] = |
+ grpc_cpp_generator::ClassName(method->input_type(), true); |
+ (*vars)["Response"] = |
+ grpc_cpp_generator::ClassName(method->output_type(), true); |
+ printer->Print(*vars, "template <class BaseClass>\n"); |
+ printer->Print(*vars, |
+ "class WithAsyncMethod_$Method$ : public BaseClass {\n"); |
+ printer->Print( |
+ " private:\n" |
+ " void BaseClassMustBeDerivedFromService(Service *service) {}\n"); |
+ printer->Print(" public:\n"); |
+ printer->Indent(); |
+ printer->Print(*vars, |
+ "WithAsyncMethod_$Method$() {\n" |
+ " ::grpc::Service::MarkMethodAsync($Idx$);\n" |
+ "}\n"); |
+ printer->Print(*vars, |
+ "~WithAsyncMethod_$Method$() GRPC_OVERRIDE {\n" |
+ " BaseClassMustBeDerivedFromService(this);\n" |
+ "}\n"); |
+ if (NoStreaming(method)) { |
+ printer->Print( |
+ *vars, |
+ "// disable synchronous version of this method\n" |
+ "::grpc::Status $Method$(" |
+ "::grpc::ServerContext* context, const $Request$* request, " |
+ "$Response$* response) GRPC_FINAL GRPC_OVERRIDE {\n" |
+ " abort();\n" |
+ " return ::grpc::Status(::grpc::StatusCode::UNIMPLEMENTED, \"\");\n" |
+ "}\n"); |
+ printer->Print( |
+ *vars, |
+ "void Request$Method$(" |
+ "::grpc::ServerContext* context, $Request$* request, " |
+ "::grpc::ServerAsyncResponseWriter< $Response$>* response, " |
+ "::grpc::CompletionQueue* new_call_cq, " |
+ "::grpc::ServerCompletionQueue* notification_cq, void *tag) {\n"); |
+ printer->Print(*vars, |
+ " ::grpc::Service::RequestAsyncUnary($Idx$, context, " |
+ "request, response, new_call_cq, notification_cq, tag);\n"); |
+ printer->Print("}\n"); |
+ } else if (ClientOnlyStreaming(method)) { |
+ printer->Print( |
+ *vars, |
+ "// disable synchronous version of this method\n" |
+ "::grpc::Status $Method$(" |
+ "::grpc::ServerContext* context, " |
+ "::grpc::ServerReader< $Request$>* reader, " |
+ "$Response$* response) GRPC_FINAL GRPC_OVERRIDE {\n" |
+ " abort();\n" |
+ " return ::grpc::Status(::grpc::StatusCode::UNIMPLEMENTED, \"\");\n" |
+ "}\n"); |
+ printer->Print( |
+ *vars, |
+ "void Request$Method$(" |
+ "::grpc::ServerContext* context, " |
+ "::grpc::ServerAsyncReader< $Response$, $Request$>* reader, " |
+ "::grpc::CompletionQueue* new_call_cq, " |
+ "::grpc::ServerCompletionQueue* notification_cq, void *tag) {\n"); |
+ printer->Print(*vars, |
+ " ::grpc::Service::RequestAsyncClientStreaming($Idx$, " |
+ "context, reader, new_call_cq, notification_cq, tag);\n"); |
+ printer->Print("}\n"); |
+ } else if (ServerOnlyStreaming(method)) { |
+ printer->Print( |
+ *vars, |
+ "// disable synchronous version of this method\n" |
+ "::grpc::Status $Method$(" |
+ "::grpc::ServerContext* context, const $Request$* request, " |
+ "::grpc::ServerWriter< $Response$>* writer) GRPC_FINAL GRPC_OVERRIDE " |
+ "{\n" |
+ " abort();\n" |
+ " return ::grpc::Status(::grpc::StatusCode::UNIMPLEMENTED, \"\");\n" |
+ "}\n"); |
+ printer->Print( |
+ *vars, |
+ "void Request$Method$(" |
+ "::grpc::ServerContext* context, $Request$* request, " |
+ "::grpc::ServerAsyncWriter< $Response$>* writer, " |
+ "::grpc::CompletionQueue* new_call_cq, " |
+ "::grpc::ServerCompletionQueue* notification_cq, void *tag) {\n"); |
+ printer->Print( |
+ *vars, |
+ " ::grpc::Service::RequestAsyncServerStreaming($Idx$, " |
+ "context, request, writer, new_call_cq, notification_cq, tag);\n"); |
+ printer->Print("}\n"); |
+ } else if (BidiStreaming(method)) { |
+ printer->Print( |
+ *vars, |
+ "// disable synchronous version of this method\n" |
+ "::grpc::Status $Method$(" |
+ "::grpc::ServerContext* context, " |
+ "::grpc::ServerReaderWriter< $Response$, $Request$>* stream) " |
+ "GRPC_FINAL GRPC_OVERRIDE {\n" |
+ " abort();\n" |
+ " return ::grpc::Status(::grpc::StatusCode::UNIMPLEMENTED, \"\");\n" |
+ "}\n"); |
+ printer->Print( |
+ *vars, |
+ "void Request$Method$(" |
+ "::grpc::ServerContext* context, " |
+ "::grpc::ServerAsyncReaderWriter< $Response$, $Request$>* stream, " |
+ "::grpc::CompletionQueue* new_call_cq, " |
+ "::grpc::ServerCompletionQueue* notification_cq, void *tag) {\n"); |
+ printer->Print(*vars, |
+ " ::grpc::Service::RequestAsyncBidiStreaming($Idx$, " |
+ "context, stream, new_call_cq, notification_cq, tag);\n"); |
+ printer->Print("}\n"); |
+ } |
+ printer->Outdent(); |
+ printer->Print(*vars, "};\n"); |
+} |
+ |
+void PrintHeaderServerMethodGeneric( |
+ grpc::protobuf::io::Printer *printer, |
+ const grpc::protobuf::MethodDescriptor *method, |
+ std::map<grpc::string, grpc::string> *vars) { |
+ (*vars)["Method"] = method->name(); |
+ (*vars)["Request"] = |
+ grpc_cpp_generator::ClassName(method->input_type(), true); |
+ (*vars)["Response"] = |
+ grpc_cpp_generator::ClassName(method->output_type(), true); |
+ printer->Print(*vars, "template <class BaseClass>\n"); |
+ printer->Print(*vars, |
+ "class WithGenericMethod_$Method$ : public BaseClass {\n"); |
+ printer->Print( |
+ " private:\n" |
+ " void BaseClassMustBeDerivedFromService(Service *service) {}\n"); |
+ printer->Print(" public:\n"); |
+ printer->Indent(); |
+ printer->Print(*vars, |
+ "WithGenericMethod_$Method$() {\n" |
+ " ::grpc::Service::MarkMethodGeneric($Idx$);\n" |
+ "}\n"); |
+ printer->Print(*vars, |
+ "~WithGenericMethod_$Method$() GRPC_OVERRIDE {\n" |
+ " BaseClassMustBeDerivedFromService(this);\n" |
+ "}\n"); |
+ if (NoStreaming(method)) { |
+ printer->Print( |
+ *vars, |
+ "// disable synchronous version of this method\n" |
+ "::grpc::Status $Method$(" |
+ "::grpc::ServerContext* context, const $Request$* request, " |
+ "$Response$* response) GRPC_FINAL GRPC_OVERRIDE {\n" |
+ " abort();\n" |
+ " return ::grpc::Status(::grpc::StatusCode::UNIMPLEMENTED, \"\");\n" |
+ "}\n"); |
+ } else if (ClientOnlyStreaming(method)) { |
+ printer->Print( |
+ *vars, |
+ "// disable synchronous version of this method\n" |
+ "::grpc::Status $Method$(" |
+ "::grpc::ServerContext* context, " |
+ "::grpc::ServerReader< $Request$>* reader, " |
+ "$Response$* response) GRPC_FINAL GRPC_OVERRIDE {\n" |
+ " abort();\n" |
+ " return ::grpc::Status(::grpc::StatusCode::UNIMPLEMENTED, \"\");\n" |
+ "}\n"); |
+ } else if (ServerOnlyStreaming(method)) { |
+ printer->Print( |
+ *vars, |
+ "// disable synchronous version of this method\n" |
+ "::grpc::Status $Method$(" |
+ "::grpc::ServerContext* context, const $Request$* request, " |
+ "::grpc::ServerWriter< $Response$>* writer) GRPC_FINAL GRPC_OVERRIDE " |
+ "{\n" |
+ " abort();\n" |
+ " return ::grpc::Status(::grpc::StatusCode::UNIMPLEMENTED, \"\");\n" |
+ "}\n"); |
+ } else if (BidiStreaming(method)) { |
+ printer->Print( |
+ *vars, |
+ "// disable synchronous version of this method\n" |
+ "::grpc::Status $Method$(" |
+ "::grpc::ServerContext* context, " |
+ "::grpc::ServerReaderWriter< $Response$, $Request$>* stream) " |
+ "GRPC_FINAL GRPC_OVERRIDE {\n" |
+ " abort();\n" |
+ " return ::grpc::Status(::grpc::StatusCode::UNIMPLEMENTED, \"\");\n" |
+ "}\n"); |
+ } |
+ printer->Outdent(); |
+ printer->Print(*vars, "};\n"); |
+} |
+ |
+void PrintHeaderService(grpc::protobuf::io::Printer *printer, |
+ const grpc::protobuf::ServiceDescriptor *service, |
+ std::map<grpc::string, grpc::string> *vars) { |
+ (*vars)["Service"] = service->name(); |
+ |
+ printer->Print(*vars, |
+ "class $Service$ GRPC_FINAL {\n" |
+ " public:\n"); |
+ printer->Indent(); |
+ |
+ // Client side |
+ printer->Print( |
+ "class StubInterface {\n" |
+ " public:\n"); |
+ printer->Indent(); |
+ printer->Print("virtual ~StubInterface() {}\n"); |
+ for (int i = 0; i < service->method_count(); ++i) { |
+ PrintHeaderClientMethodInterfaces(printer, service->method(i), vars, true); |
+ } |
+ printer->Outdent(); |
+ printer->Print("private:\n"); |
+ printer->Indent(); |
+ for (int i = 0; i < service->method_count(); ++i) { |
+ PrintHeaderClientMethodInterfaces(printer, service->method(i), vars, false); |
+ } |
+ printer->Outdent(); |
+ printer->Print("};\n"); |
+ printer->Print( |
+ "class Stub GRPC_FINAL : public StubInterface" |
+ " {\n public:\n"); |
+ printer->Indent(); |
+ printer->Print("Stub(const std::shared_ptr< ::grpc::ChannelInterface>& channel);\n"); |
+ for (int i = 0; i < service->method_count(); ++i) { |
+ PrintHeaderClientMethod(printer, service->method(i), vars, true); |
+ } |
+ printer->Outdent(); |
+ printer->Print("\n private:\n"); |
+ printer->Indent(); |
+ printer->Print("std::shared_ptr< ::grpc::ChannelInterface> channel_;\n"); |
+ for (int i = 0; i < service->method_count(); ++i) { |
+ PrintHeaderClientMethod(printer, service->method(i), vars, false); |
+ } |
+ for (int i = 0; i < service->method_count(); ++i) { |
+ PrintHeaderClientMethodData(printer, service->method(i), vars); |
+ } |
+ printer->Outdent(); |
+ printer->Print("};\n"); |
+ printer->Print( |
+ "static std::unique_ptr<Stub> NewStub(const std::shared_ptr< " |
+ "::grpc::ChannelInterface>& channel, " |
+ "const ::grpc::StubOptions& options = ::grpc::StubOptions());\n"); |
+ |
+ printer->Print("\n"); |
+ |
+ // Server side - base |
+ printer->Print( |
+ "class Service : public ::grpc::Service {\n" |
+ " public:\n"); |
+ printer->Indent(); |
+ printer->Print("Service();\n"); |
+ printer->Print("virtual ~Service();\n"); |
+ for (int i = 0; i < service->method_count(); ++i) { |
+ PrintHeaderServerMethodSync(printer, service->method(i), vars); |
+ } |
+ printer->Outdent(); |
+ printer->Print("};\n"); |
+ |
+ // Server side - Asynchronous |
+ for (int i = 0; i < service->method_count(); ++i) { |
+ (*vars)["Idx"] = as_string(i); |
+ PrintHeaderServerMethodAsync(printer, service->method(i), vars); |
+ } |
+ |
+ printer->Print("typedef "); |
+ |
+ for (int i = 0; i < service->method_count(); ++i) { |
+ (*vars)["method_name"] = service->method(i)->name(); |
+ printer->Print(*vars, "WithAsyncMethod_$method_name$<"); |
+ } |
+ printer->Print("Service"); |
+ for (int i = 0; i < service->method_count(); ++i) { |
+ printer->Print(" >"); |
+ } |
+ printer->Print(" AsyncService;\n"); |
+ |
+ // Server side - Generic |
+ for (int i = 0; i < service->method_count(); ++i) { |
+ (*vars)["Idx"] = as_string(i); |
+ PrintHeaderServerMethodGeneric(printer, service->method(i), vars); |
+ } |
+ |
+ printer->Outdent(); |
+ printer->Print("};\n"); |
+} |
+ |
+grpc::string GetHeaderServices(const grpc::protobuf::FileDescriptor *file, |
+ const Parameters ¶ms) { |
+ grpc::string output; |
+ { |
+ // Scope the output stream so it closes and finalizes output to the string. |
+ grpc::protobuf::io::StringOutputStream output_stream(&output); |
+ grpc::protobuf::io::Printer printer(&output_stream, '$'); |
+ std::map<grpc::string, grpc::string> vars; |
+ // Package string is empty or ends with a dot. It is used to fully qualify |
+ // method names. |
+ vars["Package"] = file->package(); |
+ if (!file->package().empty()) { |
+ vars["Package"].append("."); |
+ } |
+ |
+ if (!params.services_namespace.empty()) { |
+ vars["services_namespace"] = params.services_namespace; |
+ printer.Print(vars, "\nnamespace $services_namespace$ {\n\n"); |
+ } |
+ |
+ for (int i = 0; i < file->service_count(); ++i) { |
+ PrintHeaderService(&printer, file->service(i), &vars); |
+ printer.Print("\n"); |
+ } |
+ |
+ if (!params.services_namespace.empty()) { |
+ printer.Print(vars, "} // namespace $services_namespace$\n\n"); |
+ } |
+ } |
+ return output; |
+} |
+ |
+grpc::string GetHeaderEpilogue(const grpc::protobuf::FileDescriptor *file, |
+ const Parameters ¶ms) { |
+ grpc::string output; |
+ { |
+ // Scope the output stream so it closes and finalizes output to the string. |
+ grpc::protobuf::io::StringOutputStream output_stream(&output); |
+ grpc::protobuf::io::Printer printer(&output_stream, '$'); |
+ std::map<grpc::string, grpc::string> vars; |
+ |
+ vars["filename"] = file->name(); |
+ vars["filename_identifier"] = FilenameIdentifier(file->name()); |
+ |
+ if (!file->package().empty()) { |
+ std::vector<grpc::string> parts = |
+ grpc_generator::tokenize(file->package(), "."); |
+ |
+ for (auto part = parts.rbegin(); part != parts.rend(); part++) { |
+ vars["part"] = *part; |
+ printer.Print(vars, "} // namespace $part$\n"); |
+ } |
+ printer.Print(vars, "\n"); |
+ } |
+ |
+ printer.Print(vars, "\n"); |
+ printer.Print(vars, "#endif // GRPC_$filename_identifier$__INCLUDED\n"); |
+ } |
+ return output; |
+} |
+ |
+grpc::string GetSourcePrologue(const grpc::protobuf::FileDescriptor *file, |
+ const Parameters ¶ms) { |
+ grpc::string output; |
+ { |
+ // Scope the output stream so it closes and finalizes output to the string. |
+ grpc::protobuf::io::StringOutputStream output_stream(&output); |
+ grpc::protobuf::io::Printer printer(&output_stream, '$'); |
+ std::map<grpc::string, grpc::string> vars; |
+ |
+ vars["filename"] = file->name(); |
+ vars["filename_base"] = grpc_generator::StripProto(file->name()); |
+ |
+ printer.Print(vars, "// Generated by the gRPC protobuf plugin.\n"); |
+ printer.Print(vars, |
+ "// If you make any local change, they will be lost.\n"); |
+ printer.Print(vars, "// source: $filename$\n\n"); |
+ printer.Print(vars, "#include \"$filename_base$.pb.h\"\n"); |
+ printer.Print(vars, "#include \"$filename_base$.grpc.pb.h\"\n"); |
+ printer.Print(vars, "\n"); |
+ } |
+ return output; |
+} |
+ |
+grpc::string GetSourceIncludes(const grpc::protobuf::FileDescriptor *file, |
+ const Parameters ¶m) { |
+ grpc::string output; |
+ { |
+ // Scope the output stream so it closes and finalizes output to the string. |
+ grpc::protobuf::io::StringOutputStream output_stream(&output); |
+ grpc::protobuf::io::Printer printer(&output_stream, '$'); |
+ std::map<grpc::string, grpc::string> vars; |
+ |
+ printer.Print(vars, "#include <grpc++/impl/codegen/async_stream.h>\n"); |
+ printer.Print(vars, "#include <grpc++/impl/codegen/async_unary_call.h>\n"); |
+ printer.Print(vars, "#include <grpc++/impl/codegen/channel_interface.h>\n"); |
+ printer.Print(vars, "#include <grpc++/impl/codegen/client_unary_call.h>\n"); |
+ printer.Print(vars, |
+ "#include <grpc++/impl/codegen/method_handler_impl.h>\n"); |
+ printer.Print(vars, |
+ "#include <grpc++/impl/codegen/rpc_service_method.h>\n"); |
+ printer.Print(vars, "#include <grpc++/impl/codegen/service_type.h>\n"); |
+ printer.Print(vars, "#include <grpc++/impl/codegen/sync_stream.h>\n"); |
+ |
+ if (!file->package().empty()) { |
+ std::vector<grpc::string> parts = |
+ grpc_generator::tokenize(file->package(), "."); |
+ |
+ for (auto part = parts.begin(); part != parts.end(); part++) { |
+ vars["part"] = *part; |
+ printer.Print(vars, "namespace $part$ {\n"); |
+ } |
+ } |
+ |
+ printer.Print(vars, "\n"); |
+ } |
+ return output; |
+} |
+ |
+void PrintSourceClientMethod(grpc::protobuf::io::Printer *printer, |
+ const grpc::protobuf::MethodDescriptor *method, |
+ std::map<grpc::string, grpc::string> *vars) { |
+ (*vars)["Method"] = method->name(); |
+ (*vars)["Request"] = |
+ grpc_cpp_generator::ClassName(method->input_type(), true); |
+ (*vars)["Response"] = |
+ grpc_cpp_generator::ClassName(method->output_type(), true); |
+ if (NoStreaming(method)) { |
+ printer->Print(*vars, |
+ "::grpc::Status $ns$$Service$::Stub::$Method$(" |
+ "::grpc::ClientContext* context, " |
+ "const $Request$& request, $Response$* response) {\n"); |
+ printer->Print(*vars, |
+ " return ::grpc::BlockingUnaryCall(channel_.get(), " |
+ "rpcmethod_$Method$_, " |
+ "context, request, response);\n" |
+ "}\n\n"); |
+ printer->Print( |
+ *vars, |
+ "::grpc::ClientAsyncResponseReader< $Response$>* " |
+ "$ns$$Service$::Stub::Async$Method$Raw(::grpc::ClientContext* context, " |
+ "const $Request$& request, " |
+ "::grpc::CompletionQueue* cq) {\n"); |
+ printer->Print(*vars, |
+ " return new " |
+ "::grpc::ClientAsyncResponseReader< $Response$>(" |
+ "channel_.get(), cq, " |
+ "rpcmethod_$Method$_, " |
+ "context, request);\n" |
+ "}\n\n"); |
+ } else if (ClientOnlyStreaming(method)) { |
+ printer->Print(*vars, |
+ "::grpc::ClientWriter< $Request$>* " |
+ "$ns$$Service$::Stub::$Method$Raw(" |
+ "::grpc::ClientContext* context, $Response$* response) {\n"); |
+ printer->Print(*vars, |
+ " return new ::grpc::ClientWriter< $Request$>(" |
+ "channel_.get(), " |
+ "rpcmethod_$Method$_, " |
+ "context, response);\n" |
+ "}\n\n"); |
+ printer->Print(*vars, |
+ "::grpc::ClientAsyncWriter< $Request$>* " |
+ "$ns$$Service$::Stub::Async$Method$Raw(" |
+ "::grpc::ClientContext* context, $Response$* response, " |
+ "::grpc::CompletionQueue* cq, void* tag) {\n"); |
+ printer->Print(*vars, |
+ " return new ::grpc::ClientAsyncWriter< $Request$>(" |
+ "channel_.get(), cq, " |
+ "rpcmethod_$Method$_, " |
+ "context, response, tag);\n" |
+ "}\n\n"); |
+ } else if (ServerOnlyStreaming(method)) { |
+ printer->Print( |
+ *vars, |
+ "::grpc::ClientReader< $Response$>* " |
+ "$ns$$Service$::Stub::$Method$Raw(" |
+ "::grpc::ClientContext* context, const $Request$& request) {\n"); |
+ printer->Print(*vars, |
+ " return new ::grpc::ClientReader< $Response$>(" |
+ "channel_.get(), " |
+ "rpcmethod_$Method$_, " |
+ "context, request);\n" |
+ "}\n\n"); |
+ printer->Print(*vars, |
+ "::grpc::ClientAsyncReader< $Response$>* " |
+ "$ns$$Service$::Stub::Async$Method$Raw(" |
+ "::grpc::ClientContext* context, const $Request$& request, " |
+ "::grpc::CompletionQueue* cq, void* tag) {\n"); |
+ printer->Print(*vars, |
+ " return new ::grpc::ClientAsyncReader< $Response$>(" |
+ "channel_.get(), cq, " |
+ "rpcmethod_$Method$_, " |
+ "context, request, tag);\n" |
+ "}\n\n"); |
+ } else if (BidiStreaming(method)) { |
+ printer->Print( |
+ *vars, |
+ "::grpc::ClientReaderWriter< $Request$, $Response$>* " |
+ "$ns$$Service$::Stub::$Method$Raw(::grpc::ClientContext* context) {\n"); |
+ printer->Print(*vars, |
+ " return new ::grpc::ClientReaderWriter< " |
+ "$Request$, $Response$>(" |
+ "channel_.get(), " |
+ "rpcmethod_$Method$_, " |
+ "context);\n" |
+ "}\n\n"); |
+ printer->Print( |
+ *vars, |
+ "::grpc::ClientAsyncReaderWriter< $Request$, $Response$>* " |
+ "$ns$$Service$::Stub::Async$Method$Raw(::grpc::ClientContext* context, " |
+ "::grpc::CompletionQueue* cq, void* tag) {\n"); |
+ printer->Print(*vars, |
+ " return new " |
+ "::grpc::ClientAsyncReaderWriter< $Request$, $Response$>(" |
+ "channel_.get(), cq, " |
+ "rpcmethod_$Method$_, " |
+ "context, tag);\n" |
+ "}\n\n"); |
+ } |
+} |
+ |
+void PrintSourceServerMethod(grpc::protobuf::io::Printer *printer, |
+ const grpc::protobuf::MethodDescriptor *method, |
+ std::map<grpc::string, grpc::string> *vars) { |
+ (*vars)["Method"] = method->name(); |
+ (*vars)["Request"] = |
+ grpc_cpp_generator::ClassName(method->input_type(), true); |
+ (*vars)["Response"] = |
+ grpc_cpp_generator::ClassName(method->output_type(), true); |
+ if (NoStreaming(method)) { |
+ printer->Print(*vars, |
+ "::grpc::Status $ns$$Service$::Service::$Method$(" |
+ "::grpc::ServerContext* context, " |
+ "const $Request$* request, $Response$* response) {\n"); |
+ printer->Print(" (void) context;\n"); |
+ printer->Print(" (void) request;\n"); |
+ printer->Print(" (void) response;\n"); |
+ printer->Print( |
+ " return ::grpc::Status(" |
+ "::grpc::StatusCode::UNIMPLEMENTED, \"\");\n"); |
+ printer->Print("}\n\n"); |
+ } else if (ClientOnlyStreaming(method)) { |
+ printer->Print(*vars, |
+ "::grpc::Status $ns$$Service$::Service::$Method$(" |
+ "::grpc::ServerContext* context, " |
+ "::grpc::ServerReader< $Request$>* reader, " |
+ "$Response$* response) {\n"); |
+ printer->Print(" (void) context;\n"); |
+ printer->Print(" (void) reader;\n"); |
+ printer->Print(" (void) response;\n"); |
+ printer->Print( |
+ " return ::grpc::Status(" |
+ "::grpc::StatusCode::UNIMPLEMENTED, \"\");\n"); |
+ printer->Print("}\n\n"); |
+ } else if (ServerOnlyStreaming(method)) { |
+ printer->Print(*vars, |
+ "::grpc::Status $ns$$Service$::Service::$Method$(" |
+ "::grpc::ServerContext* context, " |
+ "const $Request$* request, " |
+ "::grpc::ServerWriter< $Response$>* writer) {\n"); |
+ printer->Print(" (void) context;\n"); |
+ printer->Print(" (void) request;\n"); |
+ printer->Print(" (void) writer;\n"); |
+ printer->Print( |
+ " return ::grpc::Status(" |
+ "::grpc::StatusCode::UNIMPLEMENTED, \"\");\n"); |
+ printer->Print("}\n\n"); |
+ } else if (BidiStreaming(method)) { |
+ printer->Print(*vars, |
+ "::grpc::Status $ns$$Service$::Service::$Method$(" |
+ "::grpc::ServerContext* context, " |
+ "::grpc::ServerReaderWriter< $Response$, $Request$>* " |
+ "stream) {\n"); |
+ printer->Print(" (void) context;\n"); |
+ printer->Print(" (void) stream;\n"); |
+ printer->Print( |
+ " return ::grpc::Status(" |
+ "::grpc::StatusCode::UNIMPLEMENTED, \"\");\n"); |
+ printer->Print("}\n\n"); |
+ } |
+} |
+ |
+void PrintSourceService(grpc::protobuf::io::Printer *printer, |
+ const grpc::protobuf::ServiceDescriptor *service, |
+ std::map<grpc::string, grpc::string> *vars) { |
+ (*vars)["Service"] = service->name(); |
+ |
+ printer->Print(*vars, |
+ "static const char* $prefix$$Service$_method_names[] = {\n"); |
+ for (int i = 0; i < service->method_count(); ++i) { |
+ (*vars)["Method"] = service->method(i)->name(); |
+ printer->Print(*vars, " \"/$Package$$Service$/$Method$\",\n"); |
+ } |
+ printer->Print(*vars, "};\n\n"); |
+ |
+ printer->Print(*vars, |
+ "std::unique_ptr< $ns$$Service$::Stub> $ns$$Service$::NewStub(" |
+ "const std::shared_ptr< ::grpc::ChannelInterface>& channel, " |
+ "const ::grpc::StubOptions& options) {\n" |
+ " std::unique_ptr< $ns$$Service$::Stub> stub(new " |
+ "$ns$$Service$::Stub(channel));\n" |
+ " return stub;\n" |
+ "}\n\n"); |
+ printer->Print(*vars, |
+ "$ns$$Service$::Stub::Stub(const std::shared_ptr< " |
+ "::grpc::ChannelInterface>& channel)\n"); |
+ printer->Indent(); |
+ printer->Print(": channel_(channel)"); |
+ for (int i = 0; i < service->method_count(); ++i) { |
+ const grpc::protobuf::MethodDescriptor *method = service->method(i); |
+ (*vars)["Method"] = method->name(); |
+ (*vars)["Idx"] = as_string(i); |
+ if (NoStreaming(method)) { |
+ (*vars)["StreamingType"] = "NORMAL_RPC"; |
+ } else if (ClientOnlyStreaming(method)) { |
+ (*vars)["StreamingType"] = "CLIENT_STREAMING"; |
+ } else if (ServerOnlyStreaming(method)) { |
+ (*vars)["StreamingType"] = "SERVER_STREAMING"; |
+ } else { |
+ (*vars)["StreamingType"] = "BIDI_STREAMING"; |
+ } |
+ printer->Print(*vars, |
+ ", rpcmethod_$Method$_(" |
+ "$prefix$$Service$_method_names[$Idx$], " |
+ "::grpc::RpcMethod::$StreamingType$, " |
+ "channel" |
+ ")\n"); |
+ } |
+ printer->Print("{}\n\n"); |
+ printer->Outdent(); |
+ |
+ for (int i = 0; i < service->method_count(); ++i) { |
+ (*vars)["Idx"] = as_string(i); |
+ PrintSourceClientMethod(printer, service->method(i), vars); |
+ } |
+ |
+ printer->Print(*vars, "$ns$$Service$::Service::Service() {\n"); |
+ printer->Indent(); |
+ printer->Print(*vars, "(void)$prefix$$Service$_method_names;\n"); |
+ for (int i = 0; i < service->method_count(); ++i) { |
+ const grpc::protobuf::MethodDescriptor *method = service->method(i); |
+ (*vars)["Idx"] = as_string(i); |
+ (*vars)["Method"] = method->name(); |
+ (*vars)["Request"] = |
+ grpc_cpp_generator::ClassName(method->input_type(), true); |
+ (*vars)["Response"] = |
+ grpc_cpp_generator::ClassName(method->output_type(), true); |
+ if (NoStreaming(method)) { |
+ printer->Print( |
+ *vars, |
+ "AddMethod(new ::grpc::RpcServiceMethod(\n" |
+ " $prefix$$Service$_method_names[$Idx$],\n" |
+ " ::grpc::RpcMethod::NORMAL_RPC,\n" |
+ " new ::grpc::RpcMethodHandler< $ns$$Service$::Service, " |
+ "$Request$, " |
+ "$Response$>(\n" |
+ " std::mem_fn(&$ns$$Service$::Service::$Method$), this)));\n"); |
+ } else if (ClientOnlyStreaming(method)) { |
+ printer->Print( |
+ *vars, |
+ "AddMethod(new ::grpc::RpcServiceMethod(\n" |
+ " $prefix$$Service$_method_names[$Idx$],\n" |
+ " ::grpc::RpcMethod::CLIENT_STREAMING,\n" |
+ " new ::grpc::ClientStreamingHandler< " |
+ "$ns$$Service$::Service, $Request$, $Response$>(\n" |
+ " std::mem_fn(&$ns$$Service$::Service::$Method$), this)));\n"); |
+ } else if (ServerOnlyStreaming(method)) { |
+ printer->Print( |
+ *vars, |
+ "AddMethod(new ::grpc::RpcServiceMethod(\n" |
+ " $prefix$$Service$_method_names[$Idx$],\n" |
+ " ::grpc::RpcMethod::SERVER_STREAMING,\n" |
+ " new ::grpc::ServerStreamingHandler< " |
+ "$ns$$Service$::Service, $Request$, $Response$>(\n" |
+ " std::mem_fn(&$ns$$Service$::Service::$Method$), this)));\n"); |
+ } else if (BidiStreaming(method)) { |
+ printer->Print( |
+ *vars, |
+ "AddMethod(new ::grpc::RpcServiceMethod(\n" |
+ " $prefix$$Service$_method_names[$Idx$],\n" |
+ " ::grpc::RpcMethod::BIDI_STREAMING,\n" |
+ " new ::grpc::BidiStreamingHandler< " |
+ "$ns$$Service$::Service, $Request$, $Response$>(\n" |
+ " std::mem_fn(&$ns$$Service$::Service::$Method$), this)));\n"); |
+ } |
+ } |
+ printer->Outdent(); |
+ printer->Print(*vars, "}\n\n"); |
+ printer->Print(*vars, |
+ "$ns$$Service$::Service::~Service() {\n" |
+ "}\n\n"); |
+ for (int i = 0; i < service->method_count(); ++i) { |
+ (*vars)["Idx"] = as_string(i); |
+ PrintSourceServerMethod(printer, service->method(i), vars); |
+ } |
+} |
+ |
+grpc::string GetSourceServices(const grpc::protobuf::FileDescriptor *file, |
+ const Parameters ¶ms) { |
+ grpc::string output; |
+ { |
+ // Scope the output stream so it closes and finalizes output to the string. |
+ grpc::protobuf::io::StringOutputStream output_stream(&output); |
+ grpc::protobuf::io::Printer printer(&output_stream, '$'); |
+ std::map<grpc::string, grpc::string> vars; |
+ // Package string is empty or ends with a dot. It is used to fully qualify |
+ // method names. |
+ vars["Package"] = file->package(); |
+ if (!file->package().empty()) { |
+ vars["Package"].append("."); |
+ } |
+ if (!params.services_namespace.empty()) { |
+ vars["ns"] = params.services_namespace + "::"; |
+ vars["prefix"] = params.services_namespace; |
+ } else { |
+ vars["ns"] = ""; |
+ vars["prefix"] = ""; |
+ } |
+ |
+ for (int i = 0; i < file->service_count(); ++i) { |
+ PrintSourceService(&printer, file->service(i), &vars); |
+ printer.Print("\n"); |
+ } |
+ } |
+ return output; |
+} |
+ |
+grpc::string GetSourceEpilogue(const grpc::protobuf::FileDescriptor *file, |
+ const Parameters ¶ms) { |
+ grpc::string temp; |
+ |
+ if (!file->package().empty()) { |
+ std::vector<grpc::string> parts = |
+ grpc_generator::tokenize(file->package(), "."); |
+ |
+ for (auto part = parts.begin(); part != parts.end(); part++) { |
+ temp.append("} // namespace "); |
+ temp.append(*part); |
+ temp.append("\n"); |
+ } |
+ temp.append("\n"); |
+ } |
+ |
+ return temp; |
+} |
+ |
+} // namespace grpc_cpp_generator |