blob: fa72f9b0d97aeb2b8abb51a65fe13e82f71f2228 [file] [log] [blame]
nnobleebebb7e2014-12-10 16:31:01 -08001/*
2 *
Craig Tiller6169d5f2016-03-31 07:46:18 -07003 * Copyright 2015, Google Inc.
nnobleebebb7e2014-12-10 16:31:01 -08004 * All rights reserved.
5 *
6 * Redistribution and use in source and binary forms, with or without
7 * modification, are permitted provided that the following conditions are
8 * met:
9 *
10 * * Redistributions of source code must retain the above copyright
11 * notice, this list of conditions and the following disclaimer.
12 * * Redistributions in binary form must reproduce the above
13 * copyright notice, this list of conditions and the following disclaimer
14 * in the documentation and/or other materials provided with the
15 * distribution.
16 * * Neither the name of Google Inc. nor the names of its
17 * contributors may be used to endorse or promote products derived from
18 * this software without specific prior written permission.
19 *
20 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
21 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
22 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
23 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
24 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
25 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
26 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
27 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
28 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
29 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
30 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
31 *
32 */
33
Nicolas "Pixel" Noble36f53232015-01-16 06:39:58 +010034#include <map>
35
nnobleebebb7e2014-12-10 16:31:01 -080036#include "src/compiler/cpp_generator.h"
Nicolas Nobled446eb82015-03-12 17:22:33 -070037
Craig Tiller8c8d0aa2015-02-12 11:38:36 -080038#include <sstream>
nnobleebebb7e2014-12-10 16:31:01 -080039
40namespace grpc_cpp_generator {
41namespace {
42
Craig Tiller8c8d0aa2015-02-12 11:38:36 -080043template <class T>
Nicolas Nobled446eb82015-03-12 17:22:33 -070044grpc::string as_string(T x) {
Craig Tiller8c8d0aa2015-02-12 11:38:36 -080045 std::ostringstream out;
46 out << x;
47 return out.str();
48}
49
Craig Tiller277d3cf2015-04-14 14:04:51 -070050grpc::string FilenameIdentifier(const grpc::string &filename) {
Nicolas "Pixel" Noble0caebbf2015-04-09 23:08:51 +020051 grpc::string result;
52 for (unsigned i = 0; i < filename.size(); i++) {
53 char c = filename[i];
54 if (isalnum(c)) {
55 result.push_back(c);
56 } else {
57 static char hex[] = "0123456789abcdef";
58 result.push_back('_');
59 result.push_back(hex[(c >> 4) & 0xf]);
60 result.push_back(hex[c & 0xf]);
61 }
62 }
63 return result;
64}
nnobleebebb7e2014-12-10 16:31:01 -080065} // namespace
66
Masood Malekghassemiac592452016-07-01 11:58:04 -070067template <class T, size_t N>
68T *array_end(T (&array)[N]) {
69 return array + N;
70}
Nicolas "Pixel" Noble1ba1f632016-03-17 19:57:54 +010071
Masood Malekghassemiac592452016-07-01 11:58:04 -070072void PrintIncludes(Printer *printer, const std::vector<grpc::string> &headers,
73 const Parameters &params) {
Nicolas "Pixel" Noble1ba1f632016-03-17 19:57:54 +010074 std::map<grpc::string, grpc::string> vars;
75
76 vars["l"] = params.use_system_headers ? '<' : '"';
77 vars["r"] = params.use_system_headers ? '>' : '"';
78
Masood Malekghassemiac592452016-07-01 11:58:04 -070079 auto &s = params.grpc_search_path;
Vijay Paice350a52016-06-09 17:22:35 -070080 if (!s.empty()) {
81 vars["l"] += s;
Masood Malekghassemiac592452016-07-01 11:58:04 -070082 if (s[s.size() - 1] != '/') {
Nicolas "Pixel" Noble1ba1f632016-03-17 19:57:54 +010083 vars["l"] += '/';
84 }
85 }
86
87 for (auto i = headers.begin(); i != headers.end(); i++) {
88 vars["h"] = *i;
89 printer->Print(vars, "#include $l$$h$$r$\n");
90 }
91}
92
Wouter van Oortmerssenb695b9b2016-04-13 18:18:08 -070093grpc::string GetHeaderPrologue(File *file, const Parameters & /*params*/) {
Nicolas "Pixel" Noble0caebbf2015-04-09 23:08:51 +020094 grpc::string output;
Jan Tattermusch5dcebd92015-05-27 15:30:59 -070095 {
96 // Scope the output stream so it closes and finalizes output to the string.
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -080097 auto printer = file->CreatePrinter(&output);
Jan Tattermusch5dcebd92015-05-27 15:30:59 -070098 std::map<grpc::string, grpc::string> vars;
Nicolas "Pixel" Noble0caebbf2015-04-09 23:08:51 +020099
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -0800100 vars["filename"] = file->filename();
101 vars["filename_identifier"] = FilenameIdentifier(file->filename());
102 vars["filename_base"] = file->filename_without_ext();
Wouter van Oortmerssenb695b9b2016-04-13 18:18:08 -0700103 vars["message_header_ext"] = file->message_header_ext();
Nicolas "Pixel" Noble0caebbf2015-04-09 23:08:51 +0200104
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -0800105 printer->Print(vars, "// Generated by the gRPC protobuf plugin.\n");
106 printer->Print(vars,
Masood Malekghassemiac592452016-07-01 11:58:04 -0700107 "// If you make any local change, they will be lost.\n");
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -0800108 printer->Print(vars, "// source: $filename$\n");
yang-g2e089412016-04-15 10:46:41 -0700109 grpc::string leading_comments = file->GetLeadingComments();
110 if (!leading_comments.empty()) {
111 printer->Print(vars, "// Original file comments:\n");
112 printer->Print(leading_comments.c_str());
113 }
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -0800114 printer->Print(vars, "#ifndef GRPC_$filename_identifier$__INCLUDED\n");
115 printer->Print(vars, "#define GRPC_$filename_identifier$__INCLUDED\n");
116 printer->Print(vars, "\n");
Wouter van Oortmerssenb695b9b2016-04-13 18:18:08 -0700117 printer->Print(vars, "#include \"$filename_base$$message_header_ext$\"\n");
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -0800118 printer->Print(vars, "\n");
Jan Tattermusch5dcebd92015-05-27 15:30:59 -0700119 }
Nicolas "Pixel" Noble0caebbf2015-04-09 23:08:51 +0200120 return output;
121}
122
Masood Malekghassemiac592452016-07-01 11:58:04 -0700123grpc::string GetHeaderIncludes(File *file, const Parameters &params) {
Nicolas "Pixel" Noble931bdce2016-01-12 03:08:11 +0100124 grpc::string output;
125 {
126 // Scope the output stream so it closes and finalizes output to the string.
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -0800127 auto printer = file->CreatePrinter(&output);
Nicolas "Pixel" Noble931bdce2016-01-12 03:08:11 +0100128 std::map<grpc::string, grpc::string> vars;
Nicolas "Pixel" Noble0caebbf2015-04-09 23:08:51 +0200129
Nicolas "Pixel" Noble1ba1f632016-03-17 19:57:54 +0100130 static const char *headers_strs[] = {
Masood Malekghassemiac592452016-07-01 11:58:04 -0700131 "grpc++/impl/codegen/async_stream.h",
132 "grpc++/impl/codegen/async_unary_call.h",
Vijay Pai581097f2016-08-04 10:50:51 -0700133 "grpc++/impl/codegen/method_handler_impl.h",
Masood Malekghassemiac592452016-07-01 11:58:04 -0700134 "grpc++/impl/codegen/proto_utils.h",
135 "grpc++/impl/codegen/rpc_method.h",
136 "grpc++/impl/codegen/service_type.h",
137 "grpc++/impl/codegen/status.h",
138 "grpc++/impl/codegen/stub_options.h",
139 "grpc++/impl/codegen/sync_stream.h"};
Nicolas "Pixel" Noble1ba1f632016-03-17 19:57:54 +0100140 std::vector<grpc::string> headers(headers_strs, array_end(headers_strs));
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -0800141 PrintIncludes(printer.get(), headers, params);
142 printer->Print(vars, "\n");
143 printer->Print(vars, "namespace grpc {\n");
144 printer->Print(vars, "class CompletionQueue;\n");
145 printer->Print(vars, "class Channel;\n");
146 printer->Print(vars, "class RpcService;\n");
147 printer->Print(vars, "class ServerCompletionQueue;\n");
148 printer->Print(vars, "class ServerContext;\n");
149 printer->Print(vars, "} // namespace grpc\n\n");
Nicolas "Pixel" Noble931bdce2016-01-12 03:08:11 +0100150
151 if (!file->package().empty()) {
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -0800152 std::vector<grpc::string> parts = file->package_parts();
Nicolas "Pixel" Noble931bdce2016-01-12 03:08:11 +0100153
154 for (auto part = parts.begin(); part != parts.end(); part++) {
155 vars["part"] = *part;
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -0800156 printer->Print(vars, "namespace $part$ {\n");
Nicolas "Pixel" Noble931bdce2016-01-12 03:08:11 +0100157 }
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -0800158 printer->Print(vars, "\n");
Nicolas "Pixel" Noble931bdce2016-01-12 03:08:11 +0100159 }
160 }
161 return output;
nnobleebebb7e2014-12-10 16:31:01 -0800162}
163
Craig Tillerce40de52015-06-05 07:14:58 -0700164void PrintHeaderClientMethodInterfaces(
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -0800165 Printer *printer, const Method *method,
Craig Tillerce40de52015-06-05 07:14:58 -0700166 std::map<grpc::string, grpc::string> *vars, bool is_public) {
nnobleebebb7e2014-12-10 16:31:01 -0800167 (*vars)["Method"] = method->name();
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -0800168 (*vars)["Request"] = method->input_type_name();
169 (*vars)["Response"] = method->output_type_name();
Yang Gaoc6924c82015-05-05 10:42:51 -0700170
171 if (is_public) {
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -0800172 if (method->NoStreaming()) {
Yang Gaoc6924c82015-05-05 10:42:51 -0700173 printer->Print(
174 *vars,
175 "virtual ::grpc::Status $Method$(::grpc::ClientContext* context, "
176 "const $Request$& request, $Response$* response) = 0;\n");
Craig Tillerce40de52015-06-05 07:14:58 -0700177 printer->Print(*vars,
178 "std::unique_ptr< "
179 "::grpc::ClientAsyncResponseReaderInterface< $Response$>> "
180 "Async$Method$(::grpc::ClientContext* context, "
181 "const $Request$& request, "
182 "::grpc::CompletionQueue* cq) {\n");
Yang Gaoc6924c82015-05-05 10:42:51 -0700183 printer->Indent();
Craig Tillerce40de52015-06-05 07:14:58 -0700184 printer->Print(*vars,
185 "return std::unique_ptr< "
186 "::grpc::ClientAsyncResponseReaderInterface< $Response$>>("
187 "Async$Method$Raw(context, request, cq));\n");
Yang Gaoc6924c82015-05-05 10:42:51 -0700188 printer->Outdent();
189 printer->Print("}\n");
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -0800190 } else if (method->ClientOnlyStreaming()) {
Yang Gaoc6924c82015-05-05 10:42:51 -0700191 printer->Print(
192 *vars,
193 "std::unique_ptr< ::grpc::ClientWriterInterface< $Request$>>"
194 " $Method$("
195 "::grpc::ClientContext* context, $Response$* response) {\n");
196 printer->Indent();
197 printer->Print(
198 *vars,
199 "return std::unique_ptr< ::grpc::ClientWriterInterface< $Request$>>"
200 "($Method$Raw(context, response));\n");
201 printer->Outdent();
202 printer->Print("}\n");
203 printer->Print(
204 *vars,
205 "std::unique_ptr< ::grpc::ClientAsyncWriterInterface< $Request$>>"
Craig Tillerce40de52015-06-05 07:14:58 -0700206 " Async$Method$(::grpc::ClientContext* context, $Response$* "
207 "response, "
Yang Gaoc6924c82015-05-05 10:42:51 -0700208 "::grpc::CompletionQueue* cq, void* tag) {\n");
209 printer->Indent();
Craig Tillerce40de52015-06-05 07:14:58 -0700210 printer->Print(*vars,
211 "return std::unique_ptr< "
212 "::grpc::ClientAsyncWriterInterface< $Request$>>("
213 "Async$Method$Raw(context, response, cq, tag));\n");
Yang Gaoc6924c82015-05-05 10:42:51 -0700214 printer->Outdent();
215 printer->Print("}\n");
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -0800216 } else if (method->ServerOnlyStreaming()) {
Yang Gaoc6924c82015-05-05 10:42:51 -0700217 printer->Print(
218 *vars,
219 "std::unique_ptr< ::grpc::ClientReaderInterface< $Response$>>"
220 " $Method$(::grpc::ClientContext* context, const $Request$& request)"
221 " {\n");
222 printer->Indent();
223 printer->Print(
224 *vars,
225 "return std::unique_ptr< ::grpc::ClientReaderInterface< $Response$>>"
226 "($Method$Raw(context, request));\n");
227 printer->Outdent();
228 printer->Print("}\n");
229 printer->Print(
230 *vars,
231 "std::unique_ptr< ::grpc::ClientAsyncReaderInterface< $Response$>> "
232 "Async$Method$("
233 "::grpc::ClientContext* context, const $Request$& request, "
234 "::grpc::CompletionQueue* cq, void* tag) {\n");
235 printer->Indent();
Craig Tillerce40de52015-06-05 07:14:58 -0700236 printer->Print(*vars,
237 "return std::unique_ptr< "
238 "::grpc::ClientAsyncReaderInterface< $Response$>>("
239 "Async$Method$Raw(context, request, cq, tag));\n");
Yang Gaoc6924c82015-05-05 10:42:51 -0700240 printer->Outdent();
241 printer->Print("}\n");
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -0800242 } else if (method->BidiStreaming()) {
Craig Tillerce40de52015-06-05 07:14:58 -0700243 printer->Print(*vars,
244 "std::unique_ptr< ::grpc::ClientReaderWriterInterface< "
245 "$Request$, $Response$>> "
246 "$Method$(::grpc::ClientContext* context) {\n");
Yang Gaoc6924c82015-05-05 10:42:51 -0700247 printer->Indent();
248 printer->Print(
249 *vars,
250 "return std::unique_ptr< "
251 "::grpc::ClientReaderWriterInterface< $Request$, $Response$>>("
252 "$Method$Raw(context));\n");
253 printer->Outdent();
254 printer->Print("}\n");
255 printer->Print(
256 *vars,
257 "std::unique_ptr< "
258 "::grpc::ClientAsyncReaderWriterInterface< $Request$, $Response$>> "
259 "Async$Method$(::grpc::ClientContext* context, "
260 "::grpc::CompletionQueue* cq, void* tag) {\n");
261 printer->Indent();
262 printer->Print(
263 *vars,
264 "return std::unique_ptr< "
265 "::grpc::ClientAsyncReaderWriterInterface< $Request$, $Response$>>("
266 "Async$Method$Raw(context, cq, tag));\n");
267 printer->Outdent();
268 printer->Print("}\n");
269 }
270 } else {
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -0800271 if (method->NoStreaming()) {
Yang Gaoc6924c82015-05-05 10:42:51 -0700272 printer->Print(
273 *vars,
274 "virtual ::grpc::ClientAsyncResponseReaderInterface< $Response$>* "
275 "Async$Method$Raw(::grpc::ClientContext* context, "
276 "const $Request$& request, "
Craig Tiller5f871ac2015-05-08 13:05:51 -0700277 "::grpc::CompletionQueue* cq) = 0;\n");
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -0800278 } else if (method->ClientOnlyStreaming()) {
Yang Gaoc6924c82015-05-05 10:42:51 -0700279 printer->Print(
280 *vars,
281 "virtual ::grpc::ClientWriterInterface< $Request$>*"
282 " $Method$Raw("
283 "::grpc::ClientContext* context, $Response$* response) = 0;\n");
Craig Tillerce40de52015-06-05 07:14:58 -0700284 printer->Print(*vars,
285 "virtual ::grpc::ClientAsyncWriterInterface< $Request$>*"
286 " Async$Method$Raw(::grpc::ClientContext* context, "
287 "$Response$* response, "
288 "::grpc::CompletionQueue* cq, void* tag) = 0;\n");
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -0800289 } else if (method->ServerOnlyStreaming()) {
Yang Gaoc6924c82015-05-05 10:42:51 -0700290 printer->Print(
291 *vars,
292 "virtual ::grpc::ClientReaderInterface< $Response$>* $Method$Raw("
293 "::grpc::ClientContext* context, const $Request$& request) = 0;\n");
294 printer->Print(
295 *vars,
296 "virtual ::grpc::ClientAsyncReaderInterface< $Response$>* "
297 "Async$Method$Raw("
298 "::grpc::ClientContext* context, const $Request$& request, "
299 "::grpc::CompletionQueue* cq, void* tag) = 0;\n");
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -0800300 } else if (method->BidiStreaming()) {
Craig Tillerce40de52015-06-05 07:14:58 -0700301 printer->Print(*vars,
302 "virtual ::grpc::ClientReaderWriterInterface< $Request$, "
303 "$Response$>* "
304 "$Method$Raw(::grpc::ClientContext* context) = 0;\n");
305 printer->Print(*vars,
306 "virtual ::grpc::ClientAsyncReaderWriterInterface< "
307 "$Request$, $Response$>* "
308 "Async$Method$Raw(::grpc::ClientContext* context, "
309 "::grpc::CompletionQueue* cq, void* tag) = 0;\n");
Yang Gaoc6924c82015-05-05 10:42:51 -0700310 }
311 }
312}
313
Masood Malekghassemiac592452016-07-01 11:58:04 -0700314void PrintHeaderClientMethod(Printer *printer, const Method *method,
Yang Gaoc6924c82015-05-05 10:42:51 -0700315 std::map<grpc::string, grpc::string> *vars,
316 bool is_public) {
317 (*vars)["Method"] = method->name();
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -0800318 (*vars)["Request"] = method->input_type_name();
319 (*vars)["Response"] = method->output_type_name();
Yang Gaoc6924c82015-05-05 10:42:51 -0700320 if (is_public) {
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -0800321 if (method->NoStreaming()) {
Yang Gaoc6924c82015-05-05 10:42:51 -0700322 printer->Print(
323 *vars,
324 "::grpc::Status $Method$(::grpc::ClientContext* context, "
325 "const $Request$& request, $Response$* response) GRPC_OVERRIDE;\n");
326 printer->Print(
327 *vars,
328 "std::unique_ptr< ::grpc::ClientAsyncResponseReader< $Response$>> "
329 "Async$Method$(::grpc::ClientContext* context, "
330 "const $Request$& request, "
Craig Tiller5f871ac2015-05-08 13:05:51 -0700331 "::grpc::CompletionQueue* cq) {\n");
Yang Gaoc6924c82015-05-05 10:42:51 -0700332 printer->Indent();
Craig Tillerce40de52015-06-05 07:14:58 -0700333 printer->Print(*vars,
334 "return std::unique_ptr< "
335 "::grpc::ClientAsyncResponseReader< $Response$>>("
336 "Async$Method$Raw(context, request, cq));\n");
Yang Gaoc6924c82015-05-05 10:42:51 -0700337 printer->Outdent();
338 printer->Print("}\n");
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -0800339 } else if (method->ClientOnlyStreaming()) {
Yang Gaoc6924c82015-05-05 10:42:51 -0700340 printer->Print(
341 *vars,
342 "std::unique_ptr< ::grpc::ClientWriter< $Request$>>"
343 " $Method$("
344 "::grpc::ClientContext* context, $Response$* response) {\n");
345 printer->Indent();
Craig Tillerce40de52015-06-05 07:14:58 -0700346 printer->Print(*vars,
347 "return std::unique_ptr< ::grpc::ClientWriter< $Request$>>"
348 "($Method$Raw(context, response));\n");
Yang Gaoc6924c82015-05-05 10:42:51 -0700349 printer->Outdent();
350 printer->Print("}\n");
Craig Tillerce40de52015-06-05 07:14:58 -0700351 printer->Print(*vars,
352 "std::unique_ptr< ::grpc::ClientAsyncWriter< $Request$>>"
353 " Async$Method$(::grpc::ClientContext* context, "
354 "$Response$* response, "
355 "::grpc::CompletionQueue* cq, void* tag) {\n");
Yang Gaoc6924c82015-05-05 10:42:51 -0700356 printer->Indent();
357 printer->Print(
358 *vars,
359 "return std::unique_ptr< ::grpc::ClientAsyncWriter< $Request$>>("
360 "Async$Method$Raw(context, response, cq, tag));\n");
361 printer->Outdent();
362 printer->Print("}\n");
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -0800363 } else if (method->ServerOnlyStreaming()) {
Yang Gaoc6924c82015-05-05 10:42:51 -0700364 printer->Print(
365 *vars,
366 "std::unique_ptr< ::grpc::ClientReader< $Response$>>"
367 " $Method$(::grpc::ClientContext* context, const $Request$& request)"
368 " {\n");
369 printer->Indent();
370 printer->Print(
371 *vars,
372 "return std::unique_ptr< ::grpc::ClientReader< $Response$>>"
373 "($Method$Raw(context, request));\n");
374 printer->Outdent();
375 printer->Print("}\n");
376 printer->Print(
377 *vars,
378 "std::unique_ptr< ::grpc::ClientAsyncReader< $Response$>> "
379 "Async$Method$("
380 "::grpc::ClientContext* context, const $Request$& request, "
381 "::grpc::CompletionQueue* cq, void* tag) {\n");
382 printer->Indent();
383 printer->Print(
384 *vars,
385 "return std::unique_ptr< ::grpc::ClientAsyncReader< $Response$>>("
386 "Async$Method$Raw(context, request, cq, tag));\n");
387 printer->Outdent();
388 printer->Print("}\n");
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -0800389 } else if (method->BidiStreaming()) {
Yang Gaoc6924c82015-05-05 10:42:51 -0700390 printer->Print(
391 *vars,
392 "std::unique_ptr< ::grpc::ClientReaderWriter< $Request$, $Response$>>"
393 " $Method$(::grpc::ClientContext* context) {\n");
394 printer->Indent();
Craig Tillerce40de52015-06-05 07:14:58 -0700395 printer->Print(*vars,
396 "return std::unique_ptr< "
397 "::grpc::ClientReaderWriter< $Request$, $Response$>>("
398 "$Method$Raw(context));\n");
Yang Gaoc6924c82015-05-05 10:42:51 -0700399 printer->Outdent();
400 printer->Print("}\n");
Craig Tillerce40de52015-06-05 07:14:58 -0700401 printer->Print(*vars,
402 "std::unique_ptr< ::grpc::ClientAsyncReaderWriter< "
403 "$Request$, $Response$>> "
404 "Async$Method$(::grpc::ClientContext* context, "
405 "::grpc::CompletionQueue* cq, void* tag) {\n");
Yang Gaoc6924c82015-05-05 10:42:51 -0700406 printer->Indent();
Craig Tillerce40de52015-06-05 07:14:58 -0700407 printer->Print(*vars,
408 "return std::unique_ptr< "
409 "::grpc::ClientAsyncReaderWriter< $Request$, $Response$>>("
410 "Async$Method$Raw(context, cq, tag));\n");
Yang Gaoc6924c82015-05-05 10:42:51 -0700411 printer->Outdent();
412 printer->Print("}\n");
413 }
414 } else {
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -0800415 if (method->NoStreaming()) {
Craig Tillerce40de52015-06-05 07:14:58 -0700416 printer->Print(*vars,
417 "::grpc::ClientAsyncResponseReader< $Response$>* "
418 "Async$Method$Raw(::grpc::ClientContext* context, "
419 "const $Request$& request, "
420 "::grpc::CompletionQueue* cq) GRPC_OVERRIDE;\n");
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -0800421 } else if (method->ClientOnlyStreaming()) {
Craig Tillerce40de52015-06-05 07:14:58 -0700422 printer->Print(*vars,
423 "::grpc::ClientWriter< $Request$>* $Method$Raw("
424 "::grpc::ClientContext* context, $Response$* response) "
425 "GRPC_OVERRIDE;\n");
Yang Gaoc6924c82015-05-05 10:42:51 -0700426 printer->Print(
427 *vars,
428 "::grpc::ClientAsyncWriter< $Request$>* Async$Method$Raw("
429 "::grpc::ClientContext* context, $Response$* response, "
430 "::grpc::CompletionQueue* cq, void* tag) GRPC_OVERRIDE;\n");
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -0800431 } else if (method->ServerOnlyStreaming()) {
Craig Tillerce40de52015-06-05 07:14:58 -0700432 printer->Print(*vars,
433 "::grpc::ClientReader< $Response$>* $Method$Raw("
434 "::grpc::ClientContext* context, const $Request$& request)"
435 " GRPC_OVERRIDE;\n");
Yang Gaoc6924c82015-05-05 10:42:51 -0700436 printer->Print(
437 *vars,
438 "::grpc::ClientAsyncReader< $Response$>* Async$Method$Raw("
439 "::grpc::ClientContext* context, const $Request$& request, "
440 "::grpc::CompletionQueue* cq, void* tag) GRPC_OVERRIDE;\n");
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -0800441 } else if (method->BidiStreaming()) {
Yang Gaoc6924c82015-05-05 10:42:51 -0700442 printer->Print(
443 *vars,
444 "::grpc::ClientReaderWriter< $Request$, $Response$>* "
445 "$Method$Raw(::grpc::ClientContext* context) GRPC_OVERRIDE;\n");
446 printer->Print(
447 *vars,
448 "::grpc::ClientAsyncReaderWriter< $Request$, $Response$>* "
449 "Async$Method$Raw(::grpc::ClientContext* context, "
450 "::grpc::CompletionQueue* cq, void* tag) GRPC_OVERRIDE;\n");
451 }
nnobleebebb7e2014-12-10 16:31:01 -0800452 }
453}
454
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -0800455void PrintHeaderClientMethodData(Printer *printer, const Method *method,
Craig Tillerbd6c6182015-04-10 17:08:15 -0700456 std::map<grpc::string, grpc::string> *vars) {
457 (*vars)["Method"] = method->name();
458 printer->Print(*vars, "const ::grpc::RpcMethod rpcmethod_$Method$_;\n");
459}
460
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -0800461void PrintHeaderServerMethodSync(Printer *printer, const Method *method,
Craig Tiller277d3cf2015-04-14 14:04:51 -0700462 std::map<grpc::string, grpc::string> *vars) {
nnobleebebb7e2014-12-10 16:31:01 -0800463 (*vars)["Method"] = method->name();
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -0800464 (*vars)["Request"] = method->input_type_name();
465 (*vars)["Response"] = method->output_type_name();
yang-g9efec8e2016-04-14 14:34:55 -0700466 printer->Print(method->GetLeadingComments().c_str());
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -0800467 if (method->NoStreaming()) {
nnobleebebb7e2014-12-10 16:31:01 -0800468 printer->Print(*vars,
yangga4b6f5d2014-12-17 15:53:12 -0800469 "virtual ::grpc::Status $Method$("
470 "::grpc::ServerContext* context, const $Request$* request, "
nnobleebebb7e2014-12-10 16:31:01 -0800471 "$Response$* response);\n");
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -0800472 } else if (method->ClientOnlyStreaming()) {
nnobleebebb7e2014-12-10 16:31:01 -0800473 printer->Print(*vars,
474 "virtual ::grpc::Status $Method$("
yangga4b6f5d2014-12-17 15:53:12 -0800475 "::grpc::ServerContext* context, "
Yang Gao1ff11f62015-01-14 11:45:32 -0800476 "::grpc::ServerReader< $Request$>* reader, "
nnobleebebb7e2014-12-10 16:31:01 -0800477 "$Response$* response);\n");
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -0800478 } else if (method->ServerOnlyStreaming()) {
nnobleebebb7e2014-12-10 16:31:01 -0800479 printer->Print(*vars,
yangga4b6f5d2014-12-17 15:53:12 -0800480 "virtual ::grpc::Status $Method$("
481 "::grpc::ServerContext* context, const $Request$* request, "
Yang Gao1ff11f62015-01-14 11:45:32 -0800482 "::grpc::ServerWriter< $Response$>* writer);\n");
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -0800483 } else if (method->BidiStreaming()) {
Yang Gao5680ff42015-01-14 12:14:21 -0800484 printer->Print(
485 *vars,
486 "virtual ::grpc::Status $Method$("
487 "::grpc::ServerContext* context, "
488 "::grpc::ServerReaderWriter< $Response$, $Request$>* stream);"
489 "\n");
nnobleebebb7e2014-12-10 16:31:01 -0800490 }
yang-g9efec8e2016-04-14 14:34:55 -0700491 printer->Print(method->GetTrailingComments().c_str());
nnobleebebb7e2014-12-10 16:31:01 -0800492}
493
Masood Malekghassemiac592452016-07-01 11:58:04 -0700494void PrintHeaderServerMethodAsync(Printer *printer, const Method *method,
495 std::map<grpc::string, grpc::string> *vars) {
Craig Tiller2dff17d2015-02-09 12:42:23 -0800496 (*vars)["Method"] = method->name();
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -0800497 (*vars)["Request"] = method->input_type_name();
498 (*vars)["Response"] = method->output_type_name();
Craig Tiller15f383c2016-01-07 12:45:32 -0800499 printer->Print(*vars, "template <class BaseClass>\n");
500 printer->Print(*vars,
501 "class WithAsyncMethod_$Method$ : public BaseClass {\n");
502 printer->Print(
503 " private:\n"
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -0800504 " void BaseClassMustBeDerivedFromService(const Service *service) {}\n");
Craig Tiller15f383c2016-01-07 12:45:32 -0800505 printer->Print(" public:\n");
506 printer->Indent();
507 printer->Print(*vars,
yang-gbef0d872016-01-13 15:27:33 -0800508 "WithAsyncMethod_$Method$() {\n"
yang-g0bbc87f2016-01-15 09:22:50 -0800509 " ::grpc::Service::MarkMethodAsync($Idx$);\n"
yang-gbef0d872016-01-13 15:27:33 -0800510 "}\n");
511 printer->Print(*vars,
512 "~WithAsyncMethod_$Method$() GRPC_OVERRIDE {\n"
Craig Tiller15f383c2016-01-07 12:45:32 -0800513 " BaseClassMustBeDerivedFromService(this);\n"
514 "}\n");
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -0800515 if (method->NoStreaming()) {
Craig Tillerf9e6adf2015-05-06 11:45:59 -0700516 printer->Print(
517 *vars,
Craig Tiller15f383c2016-01-07 12:45:32 -0800518 "// disable synchronous version of this method\n"
519 "::grpc::Status $Method$("
520 "::grpc::ServerContext* context, const $Request$* request, "
521 "$Response$* response) GRPC_FINAL GRPC_OVERRIDE {\n"
522 " abort();\n"
523 " return ::grpc::Status(::grpc::StatusCode::UNIMPLEMENTED, \"\");\n"
524 "}\n");
525 printer->Print(
526 *vars,
Craig Tillerf9e6adf2015-05-06 11:45:59 -0700527 "void Request$Method$("
528 "::grpc::ServerContext* context, $Request$* request, "
529 "::grpc::ServerAsyncResponseWriter< $Response$>* response, "
530 "::grpc::CompletionQueue* new_call_cq, "
Craig Tiller15f383c2016-01-07 12:45:32 -0800531 "::grpc::ServerCompletionQueue* notification_cq, void *tag) {\n");
532 printer->Print(*vars,
533 " ::grpc::Service::RequestAsyncUnary($Idx$, context, "
534 "request, response, new_call_cq, notification_cq, tag);\n");
535 printer->Print("}\n");
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -0800536 } else if (method->ClientOnlyStreaming()) {
Craig Tillerf9e6adf2015-05-06 11:45:59 -0700537 printer->Print(
538 *vars,
Craig Tiller15f383c2016-01-07 12:45:32 -0800539 "// disable synchronous version of this method\n"
540 "::grpc::Status $Method$("
541 "::grpc::ServerContext* context, "
542 "::grpc::ServerReader< $Request$>* reader, "
543 "$Response$* response) GRPC_FINAL GRPC_OVERRIDE {\n"
544 " abort();\n"
545 " return ::grpc::Status(::grpc::StatusCode::UNIMPLEMENTED, \"\");\n"
546 "}\n");
547 printer->Print(
548 *vars,
Craig Tillerf9e6adf2015-05-06 11:45:59 -0700549 "void Request$Method$("
550 "::grpc::ServerContext* context, "
551 "::grpc::ServerAsyncReader< $Response$, $Request$>* reader, "
552 "::grpc::CompletionQueue* new_call_cq, "
Craig Tiller15f383c2016-01-07 12:45:32 -0800553 "::grpc::ServerCompletionQueue* notification_cq, void *tag) {\n");
554 printer->Print(*vars,
555 " ::grpc::Service::RequestAsyncClientStreaming($Idx$, "
556 "context, reader, new_call_cq, notification_cq, tag);\n");
557 printer->Print("}\n");
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -0800558 } else if (method->ServerOnlyStreaming()) {
Craig Tillerf9e6adf2015-05-06 11:45:59 -0700559 printer->Print(
560 *vars,
Craig Tiller15f383c2016-01-07 12:45:32 -0800561 "// disable synchronous version of this method\n"
562 "::grpc::Status $Method$("
563 "::grpc::ServerContext* context, const $Request$* request, "
564 "::grpc::ServerWriter< $Response$>* writer) GRPC_FINAL GRPC_OVERRIDE "
565 "{\n"
566 " abort();\n"
567 " return ::grpc::Status(::grpc::StatusCode::UNIMPLEMENTED, \"\");\n"
568 "}\n");
569 printer->Print(
570 *vars,
Craig Tillerf9e6adf2015-05-06 11:45:59 -0700571 "void Request$Method$("
572 "::grpc::ServerContext* context, $Request$* request, "
573 "::grpc::ServerAsyncWriter< $Response$>* writer, "
574 "::grpc::CompletionQueue* new_call_cq, "
Craig Tiller15f383c2016-01-07 12:45:32 -0800575 "::grpc::ServerCompletionQueue* notification_cq, void *tag) {\n");
576 printer->Print(
577 *vars,
578 " ::grpc::Service::RequestAsyncServerStreaming($Idx$, "
579 "context, request, writer, new_call_cq, notification_cq, tag);\n");
580 printer->Print("}\n");
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -0800581 } else if (method->BidiStreaming()) {
Craig Tiller2dff17d2015-02-09 12:42:23 -0800582 printer->Print(
583 *vars,
Craig Tiller15f383c2016-01-07 12:45:32 -0800584 "// disable synchronous version of this method\n"
585 "::grpc::Status $Method$("
586 "::grpc::ServerContext* context, "
587 "::grpc::ServerReaderWriter< $Response$, $Request$>* stream) "
588 "GRPC_FINAL GRPC_OVERRIDE {\n"
589 " abort();\n"
590 " return ::grpc::Status(::grpc::StatusCode::UNIMPLEMENTED, \"\");\n"
591 "}\n");
592 printer->Print(
593 *vars,
Yang Gaoca3cb3e2015-02-12 00:05:11 -0800594 "void Request$Method$("
Craig Tiller2dff17d2015-02-09 12:42:23 -0800595 "::grpc::ServerContext* context, "
Craig Tiller225f7be2015-02-09 22:32:37 -0800596 "::grpc::ServerAsyncReaderWriter< $Response$, $Request$>* stream, "
Craig Tillerf9e6adf2015-05-06 11:45:59 -0700597 "::grpc::CompletionQueue* new_call_cq, "
Craig Tiller15f383c2016-01-07 12:45:32 -0800598 "::grpc::ServerCompletionQueue* notification_cq, void *tag) {\n");
599 printer->Print(*vars,
600 " ::grpc::Service::RequestAsyncBidiStreaming($Idx$, "
601 "context, stream, new_call_cq, notification_cq, tag);\n");
602 printer->Print("}\n");
Craig Tiller2dff17d2015-02-09 12:42:23 -0800603 }
Craig Tiller15f383c2016-01-07 12:45:32 -0800604 printer->Outdent();
605 printer->Print(*vars, "};\n");
Craig Tiller2dff17d2015-02-09 12:42:23 -0800606}
607
Vijay Pai96e03112016-10-13 14:09:16 -0700608void PrintHeaderServerMethodStreamedUnary(
Vijay Pai5d941182016-08-04 11:48:47 -0700609 Printer *printer, const Method *method,
vjpai726561e2016-06-22 11:14:24 -0700610 std::map<grpc::string, grpc::string> *vars) {
611 (*vars)["Method"] = method->name();
612 (*vars)["Request"] = method->input_type_name();
613 (*vars)["Response"] = method->output_type_name();
614 if (method->NoStreaming()) {
615 printer->Print(*vars, "template <class BaseClass>\n");
616 printer->Print(*vars,
Vijay Paia9c0d7f2016-08-29 16:42:04 -0700617 "class WithStreamedUnaryMethod_$Method$ : "
Vijay Paid4d5f4c2016-08-29 16:59:21 -0700618 "public BaseClass {\n");
vjpai726561e2016-06-22 11:14:24 -0700619 printer->Print(
Vijay Pai5d941182016-08-04 11:48:47 -0700620 " private:\n"
621 " void BaseClassMustBeDerivedFromService(const Service *service) "
622 "{}\n");
vjpai726561e2016-06-22 11:14:24 -0700623 printer->Print(" public:\n");
624 printer->Indent();
625 printer->Print(*vars,
Vijay Paia9c0d7f2016-08-29 16:42:04 -0700626 "WithStreamedUnaryMethod_$Method$() {\n"
Vijay Paidb5b1cb2016-10-03 09:18:46 -0700627 " ::grpc::Service::MarkMethodStreamed($Idx$,\n"
Vijay Pai5e6edbb2016-08-30 11:43:32 -0700628 " new ::grpc::StreamedUnaryHandler< $Request$, "
Vijay Paia9c0d7f2016-08-29 16:42:04 -0700629 "$Response$>(std::bind"
Vijay Paid4d5f4c2016-08-29 16:59:21 -0700630 "(&WithStreamedUnaryMethod_$Method$<BaseClass>::"
Vijay Paia9c0d7f2016-08-29 16:42:04 -0700631 "Streamed$Method$, this, std::placeholders::_1, "
Vijay Paid4d5f4c2016-08-29 16:59:21 -0700632 "std::placeholders::_2)));\n"
Vijay Pai5d941182016-08-04 11:48:47 -0700633 "}\n");
vjpai726561e2016-06-22 11:14:24 -0700634 printer->Print(*vars,
Vijay Paia9c0d7f2016-08-29 16:42:04 -0700635 "~WithStreamedUnaryMethod_$Method$() GRPC_OVERRIDE {\n"
Vijay Pai5d941182016-08-04 11:48:47 -0700636 " BaseClassMustBeDerivedFromService(this);\n"
637 "}\n");
vjpai726561e2016-06-22 11:14:24 -0700638 printer->Print(
639 *vars,
640 "// disable regular version of this method\n"
641 "::grpc::Status $Method$("
642 "::grpc::ServerContext* context, const $Request$* request, "
643 "$Response$* response) GRPC_FINAL GRPC_OVERRIDE {\n"
644 " abort();\n"
645 " return ::grpc::Status(::grpc::StatusCode::UNIMPLEMENTED, \"\");\n"
646 "}\n");
Vijay Pai5d941182016-08-04 11:48:47 -0700647 printer->Print(*vars,
Vijay Paia9c0d7f2016-08-29 16:42:04 -0700648 "// replace default version of method with streamed unary\n"
649 "virtual ::grpc::Status Streamed$Method$("
650 "::grpc::ServerContext* context, "
Vijay Paid4d5f4c2016-08-29 16:59:21 -0700651 "::grpc::ServerUnaryStreamer< "
Vijay Paia9c0d7f2016-08-29 16:42:04 -0700652 "$Request$,$Response$>* server_unary_streamer)"
Vijay Pai5d941182016-08-04 11:48:47 -0700653 " = 0;\n");
vjpai726561e2016-06-22 11:14:24 -0700654 printer->Outdent();
655 printer->Print(*vars, "};\n");
656 }
657}
658
Vijay Paidb5b1cb2016-10-03 09:18:46 -0700659void PrintHeaderServerMethodSplitStreaming(
660 Printer *printer, const Method *method,
661 std::map<grpc::string, grpc::string> *vars) {
662 (*vars)["Method"] = method->name();
663 (*vars)["Request"] = method->input_type_name();
664 (*vars)["Response"] = method->output_type_name();
665 if (method->ServerOnlyStreaming()) {
666 printer->Print(*vars, "template <class BaseClass>\n");
667 printer->Print(*vars,
668 "class WithSplitStreamingMethod_$Method$ : "
669 "public BaseClass {\n");
670 printer->Print(
671 " private:\n"
672 " void BaseClassMustBeDerivedFromService(const Service *service) "
673 "{}\n");
674 printer->Print(" public:\n");
675 printer->Indent();
676 printer->Print(*vars,
677 "WithSplitStreamingMethod_$Method$() {\n"
678 " ::grpc::Service::MarkMethodStreamed($Idx$,\n"
679 " new ::grpc::SplitServerStreamingHandler< $Request$, "
680 "$Response$>(std::bind"
681 "(&WithSplitStreamingMethod_$Method$<BaseClass>::"
682 "Streamed$Method$, this, std::placeholders::_1, "
683 "std::placeholders::_2)));\n"
684 "}\n");
685 printer->Print(*vars,
686 "~WithSplitStreamingMethod_$Method$() GRPC_OVERRIDE {\n"
687 " BaseClassMustBeDerivedFromService(this);\n"
688 "}\n");
689 printer->Print(
690 *vars,
691 "// disable regular version of this method\n"
692 "::grpc::Status $Method$("
693 "::grpc::ServerContext* context, const $Request$* request, "
Vijay Pai96e03112016-10-13 14:09:16 -0700694 "::grpc::ServerWriter< $Response$>* writer) GRPC_FINAL GRPC_OVERRIDE "
695 "{\n"
Vijay Paidb5b1cb2016-10-03 09:18:46 -0700696 " abort();\n"
697 " return ::grpc::Status(::grpc::StatusCode::UNIMPLEMENTED, \"\");\n"
698 "}\n");
699 printer->Print(*vars,
700 "// replace default version of method with split streamed\n"
701 "virtual ::grpc::Status Streamed$Method$("
702 "::grpc::ServerContext* context, "
703 "::grpc::ServerSplitStreamer< "
704 "$Request$,$Response$>* server_split_streamer)"
705 " = 0;\n");
706 printer->Outdent();
707 printer->Print(*vars, "};\n");
708 }
709}
710
yang-g0bbc87f2016-01-15 09:22:50 -0800711void PrintHeaderServerMethodGeneric(
Masood Malekghassemiac592452016-07-01 11:58:04 -0700712 Printer *printer, const Method *method,
yang-g0bbc87f2016-01-15 09:22:50 -0800713 std::map<grpc::string, grpc::string> *vars) {
714 (*vars)["Method"] = method->name();
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -0800715 (*vars)["Request"] = method->input_type_name();
716 (*vars)["Response"] = method->output_type_name();
yang-g0bbc87f2016-01-15 09:22:50 -0800717 printer->Print(*vars, "template <class BaseClass>\n");
718 printer->Print(*vars,
719 "class WithGenericMethod_$Method$ : public BaseClass {\n");
720 printer->Print(
721 " private:\n"
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -0800722 " void BaseClassMustBeDerivedFromService(const Service *service) {}\n");
yang-g0bbc87f2016-01-15 09:22:50 -0800723 printer->Print(" public:\n");
724 printer->Indent();
725 printer->Print(*vars,
726 "WithGenericMethod_$Method$() {\n"
727 " ::grpc::Service::MarkMethodGeneric($Idx$);\n"
728 "}\n");
729 printer->Print(*vars,
730 "~WithGenericMethod_$Method$() GRPC_OVERRIDE {\n"
731 " BaseClassMustBeDerivedFromService(this);\n"
732 "}\n");
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -0800733 if (method->NoStreaming()) {
yang-g0bbc87f2016-01-15 09:22:50 -0800734 printer->Print(
735 *vars,
736 "// disable synchronous version of this method\n"
737 "::grpc::Status $Method$("
738 "::grpc::ServerContext* context, const $Request$* request, "
739 "$Response$* response) GRPC_FINAL GRPC_OVERRIDE {\n"
740 " abort();\n"
741 " return ::grpc::Status(::grpc::StatusCode::UNIMPLEMENTED, \"\");\n"
742 "}\n");
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -0800743 } else if (method->ClientOnlyStreaming()) {
yang-g0bbc87f2016-01-15 09:22:50 -0800744 printer->Print(
745 *vars,
746 "// disable synchronous version of this method\n"
747 "::grpc::Status $Method$("
748 "::grpc::ServerContext* context, "
749 "::grpc::ServerReader< $Request$>* reader, "
750 "$Response$* response) GRPC_FINAL GRPC_OVERRIDE {\n"
751 " abort();\n"
752 " return ::grpc::Status(::grpc::StatusCode::UNIMPLEMENTED, \"\");\n"
753 "}\n");
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -0800754 } else if (method->ServerOnlyStreaming()) {
yang-g0bbc87f2016-01-15 09:22:50 -0800755 printer->Print(
756 *vars,
757 "// disable synchronous version of this method\n"
758 "::grpc::Status $Method$("
759 "::grpc::ServerContext* context, const $Request$* request, "
760 "::grpc::ServerWriter< $Response$>* writer) GRPC_FINAL GRPC_OVERRIDE "
761 "{\n"
762 " abort();\n"
763 " return ::grpc::Status(::grpc::StatusCode::UNIMPLEMENTED, \"\");\n"
764 "}\n");
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -0800765 } else if (method->BidiStreaming()) {
yang-g0bbc87f2016-01-15 09:22:50 -0800766 printer->Print(
767 *vars,
768 "// disable synchronous version of this method\n"
769 "::grpc::Status $Method$("
770 "::grpc::ServerContext* context, "
771 "::grpc::ServerReaderWriter< $Response$, $Request$>* stream) "
772 "GRPC_FINAL GRPC_OVERRIDE {\n"
773 " abort();\n"
774 " return ::grpc::Status(::grpc::StatusCode::UNIMPLEMENTED, \"\");\n"
775 "}\n");
776 }
777 printer->Outdent();
778 printer->Print(*vars, "};\n");
nnobleebebb7e2014-12-10 16:31:01 -0800779}
780
Masood Malekghassemiac592452016-07-01 11:58:04 -0700781void PrintHeaderService(Printer *printer, const Service *service,
Nicolas Nobled446eb82015-03-12 17:22:33 -0700782 std::map<grpc::string, grpc::string> *vars) {
nnobleebebb7e2014-12-10 16:31:01 -0800783 (*vars)["Service"] = service->name();
784
yang-g9efec8e2016-04-14 14:34:55 -0700785 printer->Print(service->GetLeadingComments().c_str());
nnobleebebb7e2014-12-10 16:31:01 -0800786 printer->Print(*vars,
Craig Tillercf133f42015-02-26 14:05:56 -0800787 "class $Service$ GRPC_FINAL {\n"
nnobleebebb7e2014-12-10 16:31:01 -0800788 " public:\n");
789 printer->Indent();
790
791 // Client side
Craig Tillerb5dcec52015-01-13 11:13:42 -0800792 printer->Print(
Yang Gao72e0fb82015-05-01 16:24:07 -0700793 "class StubInterface {\n"
Craig Tillerb5dcec52015-01-13 11:13:42 -0800794 " public:\n");
nnobleebebb7e2014-12-10 16:31:01 -0800795 printer->Indent();
Yang Gao72e0fb82015-05-01 16:24:07 -0700796 printer->Print("virtual ~StubInterface() {}\n");
797 for (int i = 0; i < service->method_count(); ++i) {
yang-g9efec8e2016-04-14 14:34:55 -0700798 printer->Print(service->method(i)->GetLeadingComments().c_str());
Masood Malekghassemiac592452016-07-01 11:58:04 -0700799 PrintHeaderClientMethodInterfaces(printer, service->method(i).get(), vars,
800 true);
yang-g9efec8e2016-04-14 14:34:55 -0700801 printer->Print(service->method(i)->GetTrailingComments().c_str());
Yang Gaoc6924c82015-05-05 10:42:51 -0700802 }
803 printer->Outdent();
804 printer->Print("private:\n");
805 printer->Indent();
806 for (int i = 0; i < service->method_count(); ++i) {
Masood Malekghassemiac592452016-07-01 11:58:04 -0700807 PrintHeaderClientMethodInterfaces(printer, service->method(i).get(), vars,
808 false);
Yang Gao72e0fb82015-05-01 16:24:07 -0700809 }
810 printer->Outdent();
811 printer->Print("};\n");
812 printer->Print(
yang-gef003082015-08-20 11:40:51 -0700813 "class Stub GRPC_FINAL : public StubInterface"
814 " {\n public:\n");
Yang Gao72e0fb82015-05-01 16:24:07 -0700815 printer->Indent();
Masood Malekghassemiac592452016-07-01 11:58:04 -0700816 printer->Print(
817 "Stub(const std::shared_ptr< ::grpc::ChannelInterface>& channel);\n");
nnobleebebb7e2014-12-10 16:31:01 -0800818 for (int i = 0; i < service->method_count(); ++i) {
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -0800819 PrintHeaderClientMethod(printer, service->method(i).get(), vars, true);
nnobleebebb7e2014-12-10 16:31:01 -0800820 }
821 printer->Outdent();
Yang Gaoc6924c82015-05-05 10:42:51 -0700822 printer->Print("\n private:\n");
Craig Tillerbd6c6182015-04-10 17:08:15 -0700823 printer->Indent();
David Garcia Quintase50c6c22016-01-13 16:02:00 -0800824 printer->Print("std::shared_ptr< ::grpc::ChannelInterface> channel_;\n");
Craig Tillerbd6c6182015-04-10 17:08:15 -0700825 for (int i = 0; i < service->method_count(); ++i) {
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -0800826 PrintHeaderClientMethod(printer, service->method(i).get(), vars, false);
Yang Gaoc6924c82015-05-05 10:42:51 -0700827 }
828 for (int i = 0; i < service->method_count(); ++i) {
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -0800829 PrintHeaderClientMethodData(printer, service->method(i).get(), vars);
Craig Tillerbd6c6182015-04-10 17:08:15 -0700830 }
831 printer->Outdent();
nnobleebebb7e2014-12-10 16:31:01 -0800832 printer->Print("};\n");
833 printer->Print(
Craig Tillerfd1b49b2015-02-23 12:53:39 -0800834 "static std::unique_ptr<Stub> NewStub(const std::shared_ptr< "
David Garcia Quintase50c6c22016-01-13 16:02:00 -0800835 "::grpc::ChannelInterface>& channel, "
yang-g297a25b2015-08-03 16:43:46 -0700836 "const ::grpc::StubOptions& options = ::grpc::StubOptions());\n");
nnobleebebb7e2014-12-10 16:31:01 -0800837
838 printer->Print("\n");
839
Craig Tiller15f383c2016-01-07 12:45:32 -0800840 // Server side - base
Craig Tillerb5dcec52015-01-13 11:13:42 -0800841 printer->Print(
Craig Tiller15f383c2016-01-07 12:45:32 -0800842 "class Service : public ::grpc::Service {\n"
Craig Tillerb5dcec52015-01-13 11:13:42 -0800843 " public:\n");
nnobleebebb7e2014-12-10 16:31:01 -0800844 printer->Indent();
Vijay Pai7f715702015-10-12 22:47:58 +0000845 printer->Print("Service();\n");
nnobleebebb7e2014-12-10 16:31:01 -0800846 printer->Print("virtual ~Service();\n");
847 for (int i = 0; i < service->method_count(); ++i) {
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -0800848 PrintHeaderServerMethodSync(printer, service->method(i).get(), vars);
Craig Tiller2dff17d2015-02-09 12:42:23 -0800849 }
Craig Tiller2dff17d2015-02-09 12:42:23 -0800850 printer->Outdent();
Craig Tiller2dff17d2015-02-09 12:42:23 -0800851 printer->Print("};\n");
852
853 // Server side - Asynchronous
Craig Tiller2dff17d2015-02-09 12:42:23 -0800854 for (int i = 0; i < service->method_count(); ++i) {
Craig Tiller15f383c2016-01-07 12:45:32 -0800855 (*vars)["Idx"] = as_string(i);
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -0800856 PrintHeaderServerMethodAsync(printer, service->method(i).get(), vars);
nnobleebebb7e2014-12-10 16:31:01 -0800857 }
Craig Tiller15f383c2016-01-07 12:45:32 -0800858
859 printer->Print("typedef ");
860
861 for (int i = 0; i < service->method_count(); ++i) {
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -0800862 (*vars)["method_name"] = service->method(i).get()->name();
Craig Tiller15f383c2016-01-07 12:45:32 -0800863 printer->Print(*vars, "WithAsyncMethod_$method_name$<");
864 }
865 printer->Print("Service");
866 for (int i = 0; i < service->method_count(); ++i) {
867 printer->Print(" >");
868 }
869 printer->Print(" AsyncService;\n");
nnobleebebb7e2014-12-10 16:31:01 -0800870
yang-g0bbc87f2016-01-15 09:22:50 -0800871 // Server side - Generic
872 for (int i = 0; i < service->method_count(); ++i) {
873 (*vars)["Idx"] = as_string(i);
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -0800874 PrintHeaderServerMethodGeneric(printer, service->method(i).get(), vars);
yang-g0bbc87f2016-01-15 09:22:50 -0800875 }
nnobleebebb7e2014-12-10 16:31:01 -0800876
Vijay Paia9c0d7f2016-08-29 16:42:04 -0700877 // Server side - Streamed Unary
vjpaiba6597f2016-06-22 15:49:48 -0700878 for (int i = 0; i < service->method_count(); ++i) {
879 (*vars)["Idx"] = as_string(i);
Vijay Paia9c0d7f2016-08-29 16:42:04 -0700880 PrintHeaderServerMethodStreamedUnary(printer, service->method(i).get(),
Vijay Paid4d5f4c2016-08-29 16:59:21 -0700881 vars);
vjpaiba6597f2016-06-22 15:49:48 -0700882 }
883
Vijay Paiff91dea2016-08-04 11:07:50 -0700884 printer->Print("typedef ");
885 for (int i = 0; i < service->method_count(); ++i) {
886 (*vars)["method_name"] = service->method(i).get()->name();
887 if (service->method(i)->NoStreaming()) {
Vijay Paia9c0d7f2016-08-29 16:42:04 -0700888 printer->Print(*vars, "WithStreamedUnaryMethod_$method_name$<");
Vijay Paiff91dea2016-08-04 11:07:50 -0700889 }
890 }
891 printer->Print("Service");
892 for (int i = 0; i < service->method_count(); ++i) {
893 if (service->method(i)->NoStreaming()) {
894 printer->Print(" >");
895 }
896 }
Vijay Paia9c0d7f2016-08-29 16:42:04 -0700897 printer->Print(" StreamedUnaryService;\n");
Vijay Paiff91dea2016-08-04 11:07:50 -0700898
Vijay Paidb5b1cb2016-10-03 09:18:46 -0700899 // Server side - controlled server-side streaming
900 for (int i = 0; i < service->method_count(); ++i) {
901 (*vars)["Idx"] = as_string(i);
902 PrintHeaderServerMethodSplitStreaming(printer, service->method(i).get(),
Vijay Pai96e03112016-10-13 14:09:16 -0700903 vars);
Vijay Paidb5b1cb2016-10-03 09:18:46 -0700904 }
905
906 printer->Print("typedef ");
907 for (int i = 0; i < service->method_count(); ++i) {
908 (*vars)["method_name"] = service->method(i).get()->name();
909 if (service->method(i)->ServerOnlyStreaming()) {
910 printer->Print(*vars, "WithSplitStreamingMethod_$method_name$<");
911 }
912 }
913 printer->Print("Service");
914 for (int i = 0; i < service->method_count(); ++i) {
915 if (service->method(i)->ServerOnlyStreaming()) {
916 printer->Print(" >");
917 }
918 }
919 printer->Print(" SplitStreamedService;\n");
920
921 // Server side - typedef for controlled both unary and server-side streaming
922 printer->Print("typedef ");
923 for (int i = 0; i < service->method_count(); ++i) {
924 (*vars)["method_name"] = service->method(i).get()->name();
925 if (service->method(i)->ServerOnlyStreaming()) {
926 printer->Print(*vars, "WithSplitStreamingMethod_$method_name$<");
927 }
928 if (service->method(i)->NoStreaming()) {
929 printer->Print(*vars, "WithStreamedUnaryMethod_$method_name$<");
930 }
931 }
932 printer->Print("Service");
933 for (int i = 0; i < service->method_count(); ++i) {
934 if (service->method(i)->NoStreaming() ||
Vijay Pai96e03112016-10-13 14:09:16 -0700935 service->method(i)->ServerOnlyStreaming()) {
Vijay Paidb5b1cb2016-10-03 09:18:46 -0700936 printer->Print(" >");
937 }
938 }
939 printer->Print(" StreamedService;\n");
940
nnobleebebb7e2014-12-10 16:31:01 -0800941 printer->Outdent();
942 printer->Print("};\n");
yang-g9efec8e2016-04-14 14:34:55 -0700943 printer->Print(service->GetTrailingComments().c_str());
nnobleebebb7e2014-12-10 16:31:01 -0800944}
945
Masood Malekghassemiac592452016-07-01 11:58:04 -0700946grpc::string GetHeaderServices(File *file, const Parameters &params) {
Nicolas Nobled446eb82015-03-12 17:22:33 -0700947 grpc::string output;
Jan Tattermusch5dcebd92015-05-27 15:30:59 -0700948 {
Craig Tillerce40de52015-06-05 07:14:58 -0700949 // Scope the output stream so it closes and finalizes output to the string.
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -0800950 auto printer = file->CreatePrinter(&output);
Jan Tattermusch5dcebd92015-05-27 15:30:59 -0700951 std::map<grpc::string, grpc::string> vars;
yang-gbef0d872016-01-13 15:27:33 -0800952 // Package string is empty or ends with a dot. It is used to fully qualify
953 // method names.
954 vars["Package"] = file->package();
955 if (!file->package().empty()) {
956 vars["Package"].append(".");
957 }
nnobleebebb7e2014-12-10 16:31:01 -0800958
Jan Tattermusch5dcebd92015-05-27 15:30:59 -0700959 if (!params.services_namespace.empty()) {
960 vars["services_namespace"] = params.services_namespace;
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -0800961 printer->Print(vars, "\nnamespace $services_namespace$ {\n\n");
Jan Tattermusch5dcebd92015-05-27 15:30:59 -0700962 }
963
964 for (int i = 0; i < file->service_count(); ++i) {
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -0800965 PrintHeaderService(printer.get(), file->service(i).get(), &vars);
966 printer->Print("\n");
Jan Tattermusch5dcebd92015-05-27 15:30:59 -0700967 }
968
969 if (!params.services_namespace.empty()) {
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -0800970 printer->Print(vars, "} // namespace $services_namespace$\n\n");
Jan Tattermusch5dcebd92015-05-27 15:30:59 -0700971 }
Nicolas "Pixel" Noble375a82b2015-03-24 02:33:18 +0100972 }
nnobleebebb7e2014-12-10 16:31:01 -0800973 return output;
974}
975
Wouter van Oortmerssenb695b9b2016-04-13 18:18:08 -0700976grpc::string GetHeaderEpilogue(File *file, const Parameters & /*params*/) {
Nicolas "Pixel" Noble0caebbf2015-04-09 23:08:51 +0200977 grpc::string output;
Jan Tattermusch5dcebd92015-05-27 15:30:59 -0700978 {
979 // Scope the output stream so it closes and finalizes output to the string.
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -0800980 auto printer = file->CreatePrinter(&output);
Jan Tattermusch5dcebd92015-05-27 15:30:59 -0700981 std::map<grpc::string, grpc::string> vars;
Nicolas "Pixel" Noble0caebbf2015-04-09 23:08:51 +0200982
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -0800983 vars["filename"] = file->filename();
984 vars["filename_identifier"] = FilenameIdentifier(file->filename());
Nicolas "Pixel" Noble0caebbf2015-04-09 23:08:51 +0200985
Jan Tattermusch5dcebd92015-05-27 15:30:59 -0700986 if (!file->package().empty()) {
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -0800987 std::vector<grpc::string> parts = file->package_parts();
Nicolas "Pixel" Noble0caebbf2015-04-09 23:08:51 +0200988
Jan Tattermusch5dcebd92015-05-27 15:30:59 -0700989 for (auto part = parts.rbegin(); part != parts.rend(); part++) {
990 vars["part"] = *part;
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -0800991 printer->Print(vars, "} // namespace $part$\n");
Jan Tattermusch5dcebd92015-05-27 15:30:59 -0700992 }
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -0800993 printer->Print(vars, "\n");
Yang Gao1dc1a432015-04-10 13:53:11 -0700994 }
Jan Tattermusch5dcebd92015-05-27 15:30:59 -0700995
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -0800996 printer->Print(vars, "\n");
997 printer->Print(vars, "#endif // GRPC_$filename_identifier$__INCLUDED\n");
yang-g9efec8e2016-04-14 14:34:55 -0700998
999 printer->Print(file->GetTrailingComments().c_str());
Nicolas "Pixel" Noble0caebbf2015-04-09 23:08:51 +02001000 }
Nicolas "Pixel" Noble0caebbf2015-04-09 23:08:51 +02001001 return output;
1002}
1003
Wouter van Oortmerssenb695b9b2016-04-13 18:18:08 -07001004grpc::string GetSourcePrologue(File *file, const Parameters & /*params*/) {
Nicolas "Pixel" Noble0caebbf2015-04-09 23:08:51 +02001005 grpc::string output;
Jan Tattermusch5dcebd92015-05-27 15:30:59 -07001006 {
1007 // Scope the output stream so it closes and finalizes output to the string.
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -08001008 auto printer = file->CreatePrinter(&output);
Jan Tattermusch5dcebd92015-05-27 15:30:59 -07001009 std::map<grpc::string, grpc::string> vars;
Nicolas "Pixel" Noble0caebbf2015-04-09 23:08:51 +02001010
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -08001011 vars["filename"] = file->filename();
1012 vars["filename_base"] = file->filename_without_ext();
Wouter van Oortmerssenb695b9b2016-04-13 18:18:08 -07001013 vars["message_header_ext"] = file->message_header_ext();
1014 vars["service_header_ext"] = file->service_header_ext();
Nicolas "Pixel" Noble0caebbf2015-04-09 23:08:51 +02001015
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -08001016 printer->Print(vars, "// Generated by the gRPC protobuf plugin.\n");
1017 printer->Print(vars,
Masood Malekghassemiac592452016-07-01 11:58:04 -07001018 "// If you make any local change, they will be lost.\n");
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -08001019 printer->Print(vars, "// source: $filename$\n\n");
yang-g9efec8e2016-04-14 14:34:55 -07001020
Wouter van Oortmerssenb695b9b2016-04-13 18:18:08 -07001021 printer->Print(vars, "#include \"$filename_base$$message_header_ext$\"\n");
1022 printer->Print(vars, "#include \"$filename_base$$service_header_ext$\"\n");
1023 printer->Print(vars, file->additional_headers().c_str());
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -08001024 printer->Print(vars, "\n");
Jan Tattermusch5dcebd92015-05-27 15:30:59 -07001025 }
Nicolas "Pixel" Noble0caebbf2015-04-09 23:08:51 +02001026 return output;
1027}
1028
Masood Malekghassemiac592452016-07-01 11:58:04 -07001029grpc::string GetSourceIncludes(File *file, const Parameters &params) {
Nicolas "Pixel" Noble0caebbf2015-04-09 23:08:51 +02001030 grpc::string output;
Jan Tattermusch5dcebd92015-05-27 15:30:59 -07001031 {
1032 // Scope the output stream so it closes and finalizes output to the string.
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -08001033 auto printer = file->CreatePrinter(&output);
Jan Tattermusch5dcebd92015-05-27 15:30:59 -07001034 std::map<grpc::string, grpc::string> vars;
Nicolas "Pixel" Noble0caebbf2015-04-09 23:08:51 +02001035
Nicolas "Pixel" Noble1ba1f632016-03-17 19:57:54 +01001036 static const char *headers_strs[] = {
Masood Malekghassemiac592452016-07-01 11:58:04 -07001037 "grpc++/impl/codegen/async_stream.h",
1038 "grpc++/impl/codegen/async_unary_call.h",
1039 "grpc++/impl/codegen/channel_interface.h",
1040 "grpc++/impl/codegen/client_unary_call.h",
1041 "grpc++/impl/codegen/method_handler_impl.h",
1042 "grpc++/impl/codegen/rpc_service_method.h",
1043 "grpc++/impl/codegen/service_type.h",
1044 "grpc++/impl/codegen/sync_stream.h"};
Nicolas "Pixel" Noble1ba1f632016-03-17 19:57:54 +01001045 std::vector<grpc::string> headers(headers_strs, array_end(headers_strs));
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -08001046 PrintIncludes(printer.get(), headers, params);
Nicolas "Pixel" Noble0caebbf2015-04-09 23:08:51 +02001047
Jan Tattermusch5dcebd92015-05-27 15:30:59 -07001048 if (!file->package().empty()) {
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -08001049 std::vector<grpc::string> parts = file->package_parts();
Nicolas "Pixel" Noble0caebbf2015-04-09 23:08:51 +02001050
Jan Tattermusch5dcebd92015-05-27 15:30:59 -07001051 for (auto part = parts.begin(); part != parts.end(); part++) {
1052 vars["part"] = *part;
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -08001053 printer->Print(vars, "namespace $part$ {\n");
Jan Tattermusch5dcebd92015-05-27 15:30:59 -07001054 }
Yang Gao1dc1a432015-04-10 13:53:11 -07001055 }
Jan Tattermusch5dcebd92015-05-27 15:30:59 -07001056
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -08001057 printer->Print(vars, "\n");
Nicolas "Pixel" Noble0caebbf2015-04-09 23:08:51 +02001058 }
Nicolas "Pixel" Noble0caebbf2015-04-09 23:08:51 +02001059 return output;
1060}
1061
Masood Malekghassemiac592452016-07-01 11:58:04 -07001062void PrintSourceClientMethod(Printer *printer, const Method *method,
Nicolas Nobled446eb82015-03-12 17:22:33 -07001063 std::map<grpc::string, grpc::string> *vars) {
nnobleebebb7e2014-12-10 16:31:01 -08001064 (*vars)["Method"] = method->name();
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -08001065 (*vars)["Request"] = method->input_type_name();
1066 (*vars)["Response"] = method->output_type_name();
1067 if (method->NoStreaming()) {
nnobleebebb7e2014-12-10 16:31:01 -08001068 printer->Print(*vars,
Nicolas "Pixel" Noble375a82b2015-03-24 02:33:18 +01001069 "::grpc::Status $ns$$Service$::Stub::$Method$("
nnobleebebb7e2014-12-10 16:31:01 -08001070 "::grpc::ClientContext* context, "
1071 "const $Request$& request, $Response$* response) {\n");
1072 printer->Print(*vars,
yang-gef003082015-08-20 11:40:51 -07001073 " return ::grpc::BlockingUnaryCall(channel_.get(), "
Craig Tillerbd6c6182015-04-10 17:08:15 -07001074 "rpcmethod_$Method$_, "
nnobleebebb7e2014-12-10 16:31:01 -08001075 "context, request, response);\n"
1076 "}\n\n");
Yang Gao5680ff42015-01-14 12:14:21 -08001077 printer->Print(
1078 *vars,
Yang Gaoc6924c82015-05-05 10:42:51 -07001079 "::grpc::ClientAsyncResponseReader< $Response$>* "
1080 "$ns$$Service$::Stub::Async$Method$Raw(::grpc::ClientContext* context, "
Craig Tillerfd1b49b2015-02-23 12:53:39 -08001081 "const $Request$& request, "
Craig Tiller3676b382015-05-06 13:01:05 -07001082 "::grpc::CompletionQueue* cq) {\n");
yangg5bcea0d2015-01-06 10:35:03 -08001083 printer->Print(*vars,
Yang Gaoc6924c82015-05-05 10:42:51 -07001084 " return new "
Craig Tillerfd1b49b2015-02-23 12:53:39 -08001085 "::grpc::ClientAsyncResponseReader< $Response$>("
yang-gef003082015-08-20 11:40:51 -07001086 "channel_.get(), cq, "
Craig Tillerbd6c6182015-04-10 17:08:15 -07001087 "rpcmethod_$Method$_, "
Craig Tiller5f871ac2015-05-08 13:05:51 -07001088 "context, request);\n"
Craig Tillerfd1b49b2015-02-23 12:53:39 -08001089 "}\n\n");
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -08001090 } else if (method->ClientOnlyStreaming()) {
Craig Tillerfd1b49b2015-02-23 12:53:39 -08001091 printer->Print(*vars,
Yang Gaoc6924c82015-05-05 10:42:51 -07001092 "::grpc::ClientWriter< $Request$>* "
1093 "$ns$$Service$::Stub::$Method$Raw("
Craig Tillerfd1b49b2015-02-23 12:53:39 -08001094 "::grpc::ClientContext* context, $Response$* response) {\n");
1095 printer->Print(*vars,
Yang Gaoc6924c82015-05-05 10:42:51 -07001096 " return new ::grpc::ClientWriter< $Request$>("
yang-gef003082015-08-20 11:40:51 -07001097 "channel_.get(), "
Craig Tillerbd6c6182015-04-10 17:08:15 -07001098 "rpcmethod_$Method$_, "
Yang Gaoc6924c82015-05-05 10:42:51 -07001099 "context, response);\n"
yangg5bcea0d2015-01-06 10:35:03 -08001100 "}\n\n");
Yang Gao068c85b2015-02-12 15:21:24 -08001101 printer->Print(*vars,
Yang Gaoc6924c82015-05-05 10:42:51 -07001102 "::grpc::ClientAsyncWriter< $Request$>* "
1103 "$ns$$Service$::Stub::Async$Method$Raw("
Craig Tillerfd1b49b2015-02-23 12:53:39 -08001104 "::grpc::ClientContext* context, $Response$* response, "
1105 "::grpc::CompletionQueue* cq, void* tag) {\n");
1106 printer->Print(*vars,
Yang Gaoc6924c82015-05-05 10:42:51 -07001107 " return new ::grpc::ClientAsyncWriter< $Request$>("
yang-gef003082015-08-20 11:40:51 -07001108 "channel_.get(), cq, "
Craig Tillerbd6c6182015-04-10 17:08:15 -07001109 "rpcmethod_$Method$_, "
Yang Gaoc6924c82015-05-05 10:42:51 -07001110 "context, response, tag);\n"
Yang Gao068c85b2015-02-12 15:21:24 -08001111 "}\n\n");
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -08001112 } else if (method->ServerOnlyStreaming()) {
nnobleebebb7e2014-12-10 16:31:01 -08001113 printer->Print(
1114 *vars,
Yang Gaoc6924c82015-05-05 10:42:51 -07001115 "::grpc::ClientReader< $Response$>* "
1116 "$ns$$Service$::Stub::$Method$Raw("
Yang Gao07d83042015-02-13 14:11:31 -08001117 "::grpc::ClientContext* context, const $Request$& request) {\n");
yangg5bcea0d2015-01-06 10:35:03 -08001118 printer->Print(*vars,
Yang Gaoc6924c82015-05-05 10:42:51 -07001119 " return new ::grpc::ClientReader< $Response$>("
yang-gef003082015-08-20 11:40:51 -07001120 "channel_.get(), "
Craig Tillerbd6c6182015-04-10 17:08:15 -07001121 "rpcmethod_$Method$_, "
Yang Gaoc6924c82015-05-05 10:42:51 -07001122 "context, request);\n"
yangg5bcea0d2015-01-06 10:35:03 -08001123 "}\n\n");
Yang Gao068c85b2015-02-12 15:21:24 -08001124 printer->Print(*vars,
Yang Gaoc6924c82015-05-05 10:42:51 -07001125 "::grpc::ClientAsyncReader< $Response$>* "
1126 "$ns$$Service$::Stub::Async$Method$Raw("
Craig Tillerfd1b49b2015-02-23 12:53:39 -08001127 "::grpc::ClientContext* context, const $Request$& request, "
1128 "::grpc::CompletionQueue* cq, void* tag) {\n");
1129 printer->Print(*vars,
Yang Gaoc6924c82015-05-05 10:42:51 -07001130 " return new ::grpc::ClientAsyncReader< $Response$>("
yang-gef003082015-08-20 11:40:51 -07001131 "channel_.get(), cq, "
Craig Tillerbd6c6182015-04-10 17:08:15 -07001132 "rpcmethod_$Method$_, "
Yang Gaoc6924c82015-05-05 10:42:51 -07001133 "context, request, tag);\n"
Yang Gao068c85b2015-02-12 15:21:24 -08001134 "}\n\n");
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -08001135 } else if (method->BidiStreaming()) {
nnobleebebb7e2014-12-10 16:31:01 -08001136 printer->Print(
1137 *vars,
Yang Gaoc6924c82015-05-05 10:42:51 -07001138 "::grpc::ClientReaderWriter< $Request$, $Response$>* "
1139 "$ns$$Service$::Stub::$Method$Raw(::grpc::ClientContext* context) {\n");
Craig Tillerfd1b49b2015-02-23 12:53:39 -08001140 printer->Print(*vars,
Yang Gaoc6924c82015-05-05 10:42:51 -07001141 " return new ::grpc::ClientReaderWriter< "
Craig Tillerfd1b49b2015-02-23 12:53:39 -08001142 "$Request$, $Response$>("
yang-gef003082015-08-20 11:40:51 -07001143 "channel_.get(), "
Craig Tillerbd6c6182015-04-10 17:08:15 -07001144 "rpcmethod_$Method$_, "
Yang Gaoc6924c82015-05-05 10:42:51 -07001145 "context);\n"
Craig Tillerfd1b49b2015-02-23 12:53:39 -08001146 "}\n\n");
Craig Tiller277d3cf2015-04-14 14:04:51 -07001147 printer->Print(
1148 *vars,
Yang Gaoc6924c82015-05-05 10:42:51 -07001149 "::grpc::ClientAsyncReaderWriter< $Request$, $Response$>* "
1150 "$ns$$Service$::Stub::Async$Method$Raw(::grpc::ClientContext* context, "
Craig Tiller277d3cf2015-04-14 14:04:51 -07001151 "::grpc::CompletionQueue* cq, void* tag) {\n");
Craig Tillerfd1b49b2015-02-23 12:53:39 -08001152 printer->Print(*vars,
Yang Gaoc6924c82015-05-05 10:42:51 -07001153 " return new "
Craig Tillerfd1b49b2015-02-23 12:53:39 -08001154 "::grpc::ClientAsyncReaderWriter< $Request$, $Response$>("
yang-gef003082015-08-20 11:40:51 -07001155 "channel_.get(), cq, "
Craig Tillerbd6c6182015-04-10 17:08:15 -07001156 "rpcmethod_$Method$_, "
Yang Gaoc6924c82015-05-05 10:42:51 -07001157 "context, tag);\n"
Craig Tillerfd1b49b2015-02-23 12:53:39 -08001158 "}\n\n");
nnobleebebb7e2014-12-10 16:31:01 -08001159 }
1160}
1161
Masood Malekghassemiac592452016-07-01 11:58:04 -07001162void PrintSourceServerMethod(Printer *printer, const Method *method,
Nicolas Nobled446eb82015-03-12 17:22:33 -07001163 std::map<grpc::string, grpc::string> *vars) {
nnobleebebb7e2014-12-10 16:31:01 -08001164 (*vars)["Method"] = method->name();
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -08001165 (*vars)["Request"] = method->input_type_name();
1166 (*vars)["Response"] = method->output_type_name();
1167 if (method->NoStreaming()) {
nnobleebebb7e2014-12-10 16:31:01 -08001168 printer->Print(*vars,
Nicolas "Pixel" Noble375a82b2015-03-24 02:33:18 +01001169 "::grpc::Status $ns$$Service$::Service::$Method$("
yangga4b6f5d2014-12-17 15:53:12 -08001170 "::grpc::ServerContext* context, "
nnobleebebb7e2014-12-10 16:31:01 -08001171 "const $Request$* request, $Response$* response) {\n");
Nicolas "Pixel" Nobleb14fbf72015-06-10 23:49:23 +02001172 printer->Print(" (void) context;\n");
1173 printer->Print(" (void) request;\n");
1174 printer->Print(" (void) response;\n");
nnobleebebb7e2014-12-10 16:31:01 -08001175 printer->Print(
1176 " return ::grpc::Status("
Yang Gaoc1a2c312015-06-16 10:59:46 -07001177 "::grpc::StatusCode::UNIMPLEMENTED, \"\");\n");
nnobleebebb7e2014-12-10 16:31:01 -08001178 printer->Print("}\n\n");
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -08001179 } else if (method->ClientOnlyStreaming()) {
nnobleebebb7e2014-12-10 16:31:01 -08001180 printer->Print(*vars,
Nicolas "Pixel" Noble375a82b2015-03-24 02:33:18 +01001181 "::grpc::Status $ns$$Service$::Service::$Method$("
yangga4b6f5d2014-12-17 15:53:12 -08001182 "::grpc::ServerContext* context, "
Yang Gao1ff11f62015-01-14 11:45:32 -08001183 "::grpc::ServerReader< $Request$>* reader, "
nnobleebebb7e2014-12-10 16:31:01 -08001184 "$Response$* response) {\n");
Nicolas "Pixel" Nobleb14fbf72015-06-10 23:49:23 +02001185 printer->Print(" (void) context;\n");
1186 printer->Print(" (void) reader;\n");
1187 printer->Print(" (void) response;\n");
nnobleebebb7e2014-12-10 16:31:01 -08001188 printer->Print(
1189 " return ::grpc::Status("
Yang Gaoc1a2c312015-06-16 10:59:46 -07001190 "::grpc::StatusCode::UNIMPLEMENTED, \"\");\n");
nnobleebebb7e2014-12-10 16:31:01 -08001191 printer->Print("}\n\n");
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -08001192 } else if (method->ServerOnlyStreaming()) {
nnobleebebb7e2014-12-10 16:31:01 -08001193 printer->Print(*vars,
Nicolas "Pixel" Noble375a82b2015-03-24 02:33:18 +01001194 "::grpc::Status $ns$$Service$::Service::$Method$("
yangga4b6f5d2014-12-17 15:53:12 -08001195 "::grpc::ServerContext* context, "
nnobleebebb7e2014-12-10 16:31:01 -08001196 "const $Request$* request, "
Yang Gao1ff11f62015-01-14 11:45:32 -08001197 "::grpc::ServerWriter< $Response$>* writer) {\n");
Nicolas "Pixel" Nobleb14fbf72015-06-10 23:49:23 +02001198 printer->Print(" (void) context;\n");
1199 printer->Print(" (void) request;\n");
1200 printer->Print(" (void) writer;\n");
nnobleebebb7e2014-12-10 16:31:01 -08001201 printer->Print(
1202 " return ::grpc::Status("
Yang Gaoc1a2c312015-06-16 10:59:46 -07001203 "::grpc::StatusCode::UNIMPLEMENTED, \"\");\n");
nnobleebebb7e2014-12-10 16:31:01 -08001204 printer->Print("}\n\n");
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -08001205 } else if (method->BidiStreaming()) {
nnobleebebb7e2014-12-10 16:31:01 -08001206 printer->Print(*vars,
Nicolas "Pixel" Noble375a82b2015-03-24 02:33:18 +01001207 "::grpc::Status $ns$$Service$::Service::$Method$("
yangga4b6f5d2014-12-17 15:53:12 -08001208 "::grpc::ServerContext* context, "
Yang Gao1ff11f62015-01-14 11:45:32 -08001209 "::grpc::ServerReaderWriter< $Response$, $Request$>* "
nnobleebebb7e2014-12-10 16:31:01 -08001210 "stream) {\n");
Nicolas "Pixel" Nobleb14fbf72015-06-10 23:49:23 +02001211 printer->Print(" (void) context;\n");
1212 printer->Print(" (void) stream;\n");
nnobleebebb7e2014-12-10 16:31:01 -08001213 printer->Print(
1214 " return ::grpc::Status("
Yang Gaoc1a2c312015-06-16 10:59:46 -07001215 "::grpc::StatusCode::UNIMPLEMENTED, \"\");\n");
nnobleebebb7e2014-12-10 16:31:01 -08001216 printer->Print("}\n\n");
1217 }
1218}
1219
Masood Malekghassemiac592452016-07-01 11:58:04 -07001220void PrintSourceService(Printer *printer, const Service *service,
Nicolas Nobled446eb82015-03-12 17:22:33 -07001221 std::map<grpc::string, grpc::string> *vars) {
nnobleebebb7e2014-12-10 16:31:01 -08001222 (*vars)["Service"] = service->name();
Craig Tiller8c8d0aa2015-02-12 11:38:36 -08001223
Craig Tiller277d3cf2015-04-14 14:04:51 -07001224 printer->Print(*vars,
1225 "static const char* $prefix$$Service$_method_names[] = {\n");
Craig Tiller8c8d0aa2015-02-12 11:38:36 -08001226 for (int i = 0; i < service->method_count(); ++i) {
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -08001227 (*vars)["Method"] = service->method(i).get()->name();
Craig Tiller8c8d0aa2015-02-12 11:38:36 -08001228 printer->Print(*vars, " \"/$Package$$Service$/$Method$\",\n");
1229 }
1230 printer->Print(*vars, "};\n\n");
1231
yang-g8c2be9f2015-08-19 16:28:09 -07001232 printer->Print(*vars,
1233 "std::unique_ptr< $ns$$Service$::Stub> $ns$$Service$::NewStub("
David Garcia Quintase50c6c22016-01-13 16:02:00 -08001234 "const std::shared_ptr< ::grpc::ChannelInterface>& channel, "
yang-g8c2be9f2015-08-19 16:28:09 -07001235 "const ::grpc::StubOptions& options) {\n"
1236 " std::unique_ptr< $ns$$Service$::Stub> stub(new "
1237 "$ns$$Service$::Stub(channel));\n"
1238 " return stub;\n"
1239 "}\n\n");
Craig Tiller277d3cf2015-04-14 14:04:51 -07001240 printer->Print(*vars,
1241 "$ns$$Service$::Stub::Stub(const std::shared_ptr< "
David Garcia Quintase50c6c22016-01-13 16:02:00 -08001242 "::grpc::ChannelInterface>& channel)\n");
Craig Tillerbd6c6182015-04-10 17:08:15 -07001243 printer->Indent();
yang-gef003082015-08-20 11:40:51 -07001244 printer->Print(": channel_(channel)");
Craig Tillerbd6c6182015-04-10 17:08:15 -07001245 for (int i = 0; i < service->method_count(); ++i) {
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -08001246 auto method = service->method(i);
Craig Tillerbd6c6182015-04-10 17:08:15 -07001247 (*vars)["Method"] = method->name();
1248 (*vars)["Idx"] = as_string(i);
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -08001249 if (method->NoStreaming()) {
Craig Tillerbd6c6182015-04-10 17:08:15 -07001250 (*vars)["StreamingType"] = "NORMAL_RPC";
Vijay Paicdc25352016-08-25 15:33:02 -07001251 // NOTE: There is no reason to consider streamed-unary as a separate
vjpai7142a912016-06-23 10:16:00 -07001252 // category here since this part is setting up the client-side stub
1253 // and this appears as a NORMAL_RPC from the client-side.
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -08001254 } else if (method->ClientOnlyStreaming()) {
Craig Tillerbd6c6182015-04-10 17:08:15 -07001255 (*vars)["StreamingType"] = "CLIENT_STREAMING";
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -08001256 } else if (method->ServerOnlyStreaming()) {
Craig Tillerbd6c6182015-04-10 17:08:15 -07001257 (*vars)["StreamingType"] = "SERVER_STREAMING";
1258 } else {
1259 (*vars)["StreamingType"] = "BIDI_STREAMING";
1260 }
yang-g431f8c22015-08-20 10:59:29 -07001261 printer->Print(*vars,
1262 ", rpcmethod_$Method$_("
1263 "$prefix$$Service$_method_names[$Idx$], "
1264 "::grpc::RpcMethod::$StreamingType$, "
1265 "channel"
1266 ")\n");
Craig Tillerbd6c6182015-04-10 17:08:15 -07001267 }
Craig Tiller3beef682015-04-14 13:55:03 -07001268 printer->Print("{}\n\n");
Craig Tillerbd6c6182015-04-10 17:08:15 -07001269 printer->Outdent();
1270
nnobleebebb7e2014-12-10 16:31:01 -08001271 for (int i = 0; i < service->method_count(); ++i) {
Craig Tiller8c8d0aa2015-02-12 11:38:36 -08001272 (*vars)["Idx"] = as_string(i);
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -08001273 PrintSourceClientMethod(printer, service->method(i).get(), vars);
nnobleebebb7e2014-12-10 16:31:01 -08001274 }
1275
yang-gbef0d872016-01-13 15:27:33 -08001276 printer->Print(*vars, "$ns$$Service$::Service::Service() {\n");
nnobleebebb7e2014-12-10 16:31:01 -08001277 printer->Indent();
yang-ge39cb742016-02-10 23:50:08 -08001278 printer->Print(*vars, "(void)$prefix$$Service$_method_names;\n");
nnobleebebb7e2014-12-10 16:31:01 -08001279 for (int i = 0; i < service->method_count(); ++i) {
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -08001280 auto method = service->method(i);
Craig Tiller8c8d0aa2015-02-12 11:38:36 -08001281 (*vars)["Idx"] = as_string(i);
nnobleebebb7e2014-12-10 16:31:01 -08001282 (*vars)["Method"] = method->name();
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -08001283 (*vars)["Request"] = method->input_type_name();
1284 (*vars)["Response"] = method->output_type_name();
1285 if (method->NoStreaming()) {
nnobleebebb7e2014-12-10 16:31:01 -08001286 printer->Print(
1287 *vars,
yang-gbef0d872016-01-13 15:27:33 -08001288 "AddMethod(new ::grpc::RpcServiceMethod(\n"
Nicolas "Pixel" Noble375a82b2015-03-24 02:33:18 +01001289 " $prefix$$Service$_method_names[$Idx$],\n"
yangg5bcea0d2015-01-06 10:35:03 -08001290 " ::grpc::RpcMethod::NORMAL_RPC,\n"
Craig Tiller277d3cf2015-04-14 14:04:51 -07001291 " new ::grpc::RpcMethodHandler< $ns$$Service$::Service, "
1292 "$Request$, "
nnobleebebb7e2014-12-10 16:31:01 -08001293 "$Response$>(\n"
Craig Tiller50a7a682015-06-04 12:53:40 -07001294 " std::mem_fn(&$ns$$Service$::Service::$Method$), this)));\n");
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -08001295 } else if (method->ClientOnlyStreaming()) {
nnobleebebb7e2014-12-10 16:31:01 -08001296 printer->Print(
1297 *vars,
yang-gbef0d872016-01-13 15:27:33 -08001298 "AddMethod(new ::grpc::RpcServiceMethod(\n"
Nicolas "Pixel" Noble375a82b2015-03-24 02:33:18 +01001299 " $prefix$$Service$_method_names[$Idx$],\n"
yangg5bcea0d2015-01-06 10:35:03 -08001300 " ::grpc::RpcMethod::CLIENT_STREAMING,\n"
Yang Gao1ff11f62015-01-14 11:45:32 -08001301 " new ::grpc::ClientStreamingHandler< "
Nicolas "Pixel" Noble375a82b2015-03-24 02:33:18 +01001302 "$ns$$Service$::Service, $Request$, $Response$>(\n"
Craig Tiller50a7a682015-06-04 12:53:40 -07001303 " std::mem_fn(&$ns$$Service$::Service::$Method$), this)));\n");
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -08001304 } else if (method->ServerOnlyStreaming()) {
nnobleebebb7e2014-12-10 16:31:01 -08001305 printer->Print(
1306 *vars,
yang-gbef0d872016-01-13 15:27:33 -08001307 "AddMethod(new ::grpc::RpcServiceMethod(\n"
Nicolas "Pixel" Noble375a82b2015-03-24 02:33:18 +01001308 " $prefix$$Service$_method_names[$Idx$],\n"
yangg5bcea0d2015-01-06 10:35:03 -08001309 " ::grpc::RpcMethod::SERVER_STREAMING,\n"
Yang Gao1ff11f62015-01-14 11:45:32 -08001310 " new ::grpc::ServerStreamingHandler< "
Nicolas "Pixel" Noble375a82b2015-03-24 02:33:18 +01001311 "$ns$$Service$::Service, $Request$, $Response$>(\n"
Craig Tiller50a7a682015-06-04 12:53:40 -07001312 " std::mem_fn(&$ns$$Service$::Service::$Method$), this)));\n");
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -08001313 } else if (method->BidiStreaming()) {
nnobleebebb7e2014-12-10 16:31:01 -08001314 printer->Print(
1315 *vars,
yang-gbef0d872016-01-13 15:27:33 -08001316 "AddMethod(new ::grpc::RpcServiceMethod(\n"
Nicolas "Pixel" Noble375a82b2015-03-24 02:33:18 +01001317 " $prefix$$Service$_method_names[$Idx$],\n"
yangg5bcea0d2015-01-06 10:35:03 -08001318 " ::grpc::RpcMethod::BIDI_STREAMING,\n"
Yang Gao1ff11f62015-01-14 11:45:32 -08001319 " new ::grpc::BidiStreamingHandler< "
Nicolas "Pixel" Noble375a82b2015-03-24 02:33:18 +01001320 "$ns$$Service$::Service, $Request$, $Response$>(\n"
Craig Tiller50a7a682015-06-04 12:53:40 -07001321 " std::mem_fn(&$ns$$Service$::Service::$Method$), this)));\n");
nnobleebebb7e2014-12-10 16:31:01 -08001322 }
1323 }
nnobleebebb7e2014-12-10 16:31:01 -08001324 printer->Outdent();
yang-gbef0d872016-01-13 15:27:33 -08001325 printer->Print(*vars, "}\n\n");
1326 printer->Print(*vars,
1327 "$ns$$Service$::Service::~Service() {\n"
1328 "}\n\n");
1329 for (int i = 0; i < service->method_count(); ++i) {
1330 (*vars)["Idx"] = as_string(i);
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -08001331 PrintSourceServerMethod(printer, service->method(i).get(), vars);
yang-gbef0d872016-01-13 15:27:33 -08001332 }
nnobleebebb7e2014-12-10 16:31:01 -08001333}
1334
Masood Malekghassemiac592452016-07-01 11:58:04 -07001335grpc::string GetSourceServices(File *file, const Parameters &params) {
Nicolas Nobled446eb82015-03-12 17:22:33 -07001336 grpc::string output;
Jan Tattermusch5dcebd92015-05-27 15:30:59 -07001337 {
1338 // Scope the output stream so it closes and finalizes output to the string.
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -08001339 auto printer = file->CreatePrinter(&output);
Jan Tattermusch5dcebd92015-05-27 15:30:59 -07001340 std::map<grpc::string, grpc::string> vars;
1341 // Package string is empty or ends with a dot. It is used to fully qualify
1342 // method names.
1343 vars["Package"] = file->package();
1344 if (!file->package().empty()) {
1345 vars["Package"].append(".");
1346 }
1347 if (!params.services_namespace.empty()) {
1348 vars["ns"] = params.services_namespace + "::";
1349 vars["prefix"] = params.services_namespace;
1350 } else {
1351 vars["ns"] = "";
1352 vars["prefix"] = "";
1353 }
nnobleebebb7e2014-12-10 16:31:01 -08001354
Jan Tattermusch5dcebd92015-05-27 15:30:59 -07001355 for (int i = 0; i < file->service_count(); ++i) {
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -08001356 PrintSourceService(printer.get(), file->service(i).get(), &vars);
1357 printer->Print("\n");
Jan Tattermusch5dcebd92015-05-27 15:30:59 -07001358 }
nnobleebebb7e2014-12-10 16:31:01 -08001359 }
1360 return output;
1361}
1362
Wouter van Oortmerssenb695b9b2016-04-13 18:18:08 -07001363grpc::string GetSourceEpilogue(File *file, const Parameters & /*params*/) {
Nicolas "Pixel" Noble0caebbf2015-04-09 23:08:51 +02001364 grpc::string temp;
1365
Yang Gao1dc1a432015-04-10 13:53:11 -07001366 if (!file->package().empty()) {
Wouter van Oortmerssenaf09a732016-03-09 17:03:21 -08001367 std::vector<grpc::string> parts = file->package_parts();
Nicolas "Pixel" Noble0caebbf2015-04-09 23:08:51 +02001368
Yang Gao1dc1a432015-04-10 13:53:11 -07001369 for (auto part = parts.begin(); part != parts.end(); part++) {
1370 temp.append("} // namespace ");
1371 temp.append(*part);
1372 temp.append("\n");
1373 }
Nicolas "Pixel" Noble0caebbf2015-04-09 23:08:51 +02001374 temp.append("\n");
1375 }
1376
Nicolas "Pixel" Noble0caebbf2015-04-09 23:08:51 +02001377 return temp;
1378}
1379
Craig Tiller190d3602015-02-18 09:23:38 -08001380} // namespace grpc_cpp_generator