OpenSDN source code
t_cpp_generator.cc
Go to the documentation of this file.
1 /*
2  * Licensed to the Apache Software Foundation (ASF) under one
3  * or more contributor license agreements. See the NOTICE file
4  * distributed with this work for additional information
5  * regarding copyright ownership. The ASF licenses this file
6  * to you under the Apache License, Version 2.0 (the
7  * "License"); you may not use this file except in compliance
8  * with the License. You may obtain a copy of the License at
9  *
10  * http://www.apache.org/licenses/LICENSE-2.0
11  *
12  * Unless required by applicable law or agreed to in writing,
13  * software distributed under the License is distributed on an
14  * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
15  * KIND, either express or implied. See the License for the
16  * specific language governing permissions and limitations
17  * under the License.
18  *
19  * Contains some contributions under the Thrift Software License.
20  * Please see doc/old-thrift-license.txt in the Thrift distribution for
21  * details.
22  */
23 
24 #include <cassert>
25 
26 #include <fstream>
27 #include <iostream>
28 #include <sstream>
29 #include <string>
30 #include <vector>
31 #ifdef SANDESH
32 #include <algorithm>
33 #endif
34 
35 #include <sys/stat.h>
36 #include <boost/tuple/tuple.hpp>
37 #include "platform.h"
38 #include "t_oop_generator.h"
39 using namespace std;
40 using boost::tuple;
41 using boost::tuples::make_tuple;
42 
47 class t_cpp_generator : public t_oop_generator {
48  public:
49  t_cpp_generator(
50  t_program* program,
51  const std::map<std::string, std::string>& parsed_options,
52  const std::string& option_string)
53  : t_oop_generator(program)
54  {
55  (void) option_string;
56  std::map<std::string, std::string>::const_iterator iter;
57 
58  iter = parsed_options.find("pure_enums");
59  gen_pure_enums_ = (iter != parsed_options.end());
60 
61  iter = parsed_options.find("dense");
62  gen_dense_ = (iter != parsed_options.end());
63 
64  iter = parsed_options.find("include_prefix");
65  use_include_prefix_ = (iter != parsed_options.end());
66 
67  iter = parsed_options.find("cob_style");
68  gen_cob_style_ = (iter != parsed_options.end());
69 
70  iter = parsed_options.find("no_client_completion");
71  gen_no_client_completion_ = (iter != parsed_options.end());
72 
73  iter = parsed_options.find("templates");
74  gen_templates_ = (iter != parsed_options.end());
75 
76  out_dir_base_ = "gen-cpp";
77  }
78 
83  void init_generator();
84  void close_generator();
85 
86  void generate_consts(std::vector<t_const*> consts);
87 
92  void generate_typedef(t_typedef* ttypedef);
93  void generate_enum(t_enum* tenum);
94  void generate_struct(t_struct* tstruct) {
95  generate_cpp_struct(tstruct, false);
96  }
97  void generate_xception(t_struct* txception) {
98  generate_cpp_struct(txception, true);
99  }
100 #ifdef SANDESH
101  void generate_sandesh(t_sandesh* tsandesh) {
102  generate_cpp_sandesh(tsandesh);
103  }
104  void generate_cpp_sandesh(t_sandesh* tsandesh);
105  void generate_sandesh_definition (std::ofstream& out, t_sandesh* tsandesh);
106  void generate_sandesh_fingerprint (std::ofstream& out, t_sandesh* tsandesh, bool is_definition);
107  void generate_sandesh_http_reader (std::ofstream& out, t_sandesh* tsandesh);
108  void generate_sandesh_reader (std::ofstream& out, t_sandesh* tsandesh);
109  void generate_sandesh_writer (std::ofstream& out, t_sandesh* tsandesh);
110  void generate_sandesh_creator (std::ofstream& out, t_sandesh* tsandesh);
111  void generate_sandesh_member_init_list(std::ofstream& out, t_sandesh* tsandesh, bool init_dval = false);
112  void generate_sandesh_base_init (ofstream& out, t_sandesh* tsandesh, bool init_dval);
113  void generate_sandesh_hints (std::ofstream& out, t_sandesh* tsandesh);
114  void generate_sandesh_request (std::ofstream& out, t_sandesh* tsandesh);
115  void generate_sandesh_default_ctor (std::ofstream& out, t_sandesh* tsandesh, bool is_request);
116  void generate_sandesh_loggers (std::ofstream& out, t_sandesh* tsandesh);
117  void generate_logger_field (std::ofstream& out, t_field *tfield, string prefix, bool log_value_only, bool no_name_log, bool for_sandesh = false);
118  void generate_logger_struct (std::ofstream& out, t_struct *tstruct, string prefix, string name);
119  void generate_logger_container (std::ofstream& out, t_type* ttype, string name, bool log_value_only);
120  void generate_logger_map_element (std::ofstream& out, t_map* tmap, string iter, bool log_value_only);
121  void generate_logger_set_element (std::ofstream& out, t_set* tset, string iter, bool log_value_only);
122  void generate_logger_list_element (std::ofstream& out, t_list* tlist, string iter, bool log_value_only);
123  void generate_sandesh_get_size (std::ofstream& out, t_sandesh* tsandesh);
124  void generate_get_size_field (std::ofstream& out, t_field *tfield);
125  void generate_get_size_struct (std::ofstream& out, t_struct *tstruct, string name);
126  void generate_get_size_container (std::ofstream& out, t_type* ttype, string name);
127  void generate_get_size_map_element (std::ofstream& out, t_map* tmap, string name);
128  void generate_get_size_list_element(std::ofstream& out, t_list* ttype, string name);
129  void generate_get_size_set_element (std::ofstream& out, t_set* ttype, string name);
130  void generate_sandesh_trace (std::ofstream& out, t_sandesh* tsandesh);
131  void generate_sandesh_context (std::ofstream& out, t_sandesh* tsandesh, string val);
132  void generate_sandesh_seqnum(std::ofstream& out, t_sandesh* tsandesh);
133  void generate_sandesh_versionsig(std::ofstream& out, t_sandesh* tsandesh);
134  void generate_sandesh_static_seqnum_def(std::ofstream& out, t_sandesh* tsandesh);
135  void generate_sandesh_static_versionsig_def(std::ofstream& out, t_sandesh* tsandesh);
136  void generate_sandesh_trace_seqnum_ctor(std::ofstream& out, t_sandesh* tsandesh);
137  void generate_static_const_string_definition(std::ofstream& out, t_sandesh* tsandesh);
138  std::string generate_sandesh_no_static_const_string_function(t_sandesh *tsandesh, bool signature, bool autogen_darg, bool trace = false, bool request = false, bool ctorcall = false);
139  void generate_static_const_string_definition(std::ofstream& out, std::string name,
140  const vector<t_field*>& fields);
141  std::string generate_sandesh_async_creator(t_sandesh *tsandesh, bool signature, bool expand_autogen, bool skip_autogen,
142  std::string prefix, std::string suffix, bool category_level_file_line_first,
143  bool autogen_category_level, bool drop_log_reason, bool use_sandesh_object = false);
144  void generate_sandesh_rate_limit_fn(ofstream &out, t_sandesh *tsandesh);
145  void generate_sandesh_async_send_fn(ofstream &out, t_sandesh *tsandesh,
146  bool generate_sandesh_object, bool generate_rate_limit, bool generate_system_log);
147  void generate_sandesh_async_send_macros(ofstream &out, t_sandesh *tsandesh,
148  bool generate_sandesh_object);
149  void generate_sandesh_static_log_fn(ofstream &out, t_sandesh *tsandesh,
150  bool generate_sandesh_object);
151  void generate_sandesh_async_create_fn(ofstream &out, t_sandesh *tsandesh);
152  void generate_sandesh_async_create_macro(ofstream &out, t_sandesh *tsandesh);
153  void generate_sandesh_session_log_unrolled_fn(ofstream &out, t_sandesh *tsandesh);
154  void generate_sandesh_session_adjust_session_end_point_objects_fn(ofstream &out,
155  t_sandesh *tsandesh);
156  void generate_sandesh_flow_send_fn(ofstream &out, t_sandesh *tsandesh);
157  void generate_sandesh_systemlog_creators(ofstream &out, t_sandesh *tsandesh);
158  void generate_sandesh_objectlog_creators(ofstream &out, t_sandesh *tsandesh);
159  void generate_sandesh_flow_creators(ofstream &out, t_sandesh *tsandesh);
160  void generate_sandesh_uve_creator(std::ofstream& out, t_sandesh* tsandesh);
161  std::string generate_sandesh_trace_creator(t_sandesh *tsandesh, bool signature, bool expand_autogen, bool skip_autogen,
162  std::string prefix, std::string suffix);
163  void generate_sandesh_updater(ofstream& out, t_sandesh* tsandesh);
164  void generate_isRatelimitPass(ofstream& out, t_sandesh* tsandesh);
165  void generate_sandesh_static_rate_limit_log_def(ofstream& out, t_sandesh* tsandesh);
166  void generate_sandesh_static_rate_limit_mutex_def(ofstream& out, t_sandesh* tsandesh);
167  void generate_sandesh_static_rate_limit_buffer_def(ofstream& out, t_sandesh* tsandesh);
168 
169  typedef enum {
170  MANDATORY = 0,
171  INLINE = 1,
172  HIDDEN = 2,
173  PERIODIC = 3,
174  HIDDEN_PER = 4,
175  CA_MAX = 5
176  } CacheAttribute;
177 
178  typedef enum {
179  RM_DIAL = 0,
180  RM_AGG = 1,
181  RM_DIFF = 2,
182  RM_MAX = 3
183  } RawMetric;
184 
185  struct DSInfo {
186  DSInfo(bool is_map, size_t period, CacheAttribute cat,
187  string rawtype, RawMetric rmtype,
188  string resulttype, string algo, string annotation,
189  string compattr, string subcompattr, string prealgo) :
190  is_map_(is_map), period_(period), cat_(cat),
191  rawtype_(rawtype), rmtype_(rmtype),
192  resulttype_(resulttype), algo_(algo),
193  annotation_(annotation),
194  compattr_(compattr) , subcompattr_(subcompattr), prealgo_(prealgo) {}
195  bool is_map_;
196  size_t period_;
197  CacheAttribute cat_;
198  string rawtype_;
199  RawMetric rmtype_;
200  string resulttype_;
201  string algo_;
202  string annotation_;
203  string compattr_;
204  string subcompattr_;
205  string prealgo_;
206  };
207  void derived_stats_info(t_struct* tstruct,
208  map<string,DSInfo>& dsmap,
209  // val is set of ds attributes
210  map<string,set<string> >& rawmap);
211 
212  void cache_attr_info(t_struct* tstruct, std::map<string, CacheAttribute>& attrs);
213 
214  void freq_info(t_struct* tstruct,
215  set<string>& inl, set<string>& never, set<string>& periodic);
216 #endif
217 
218  void generate_cpp_struct(t_struct* tstruct, bool is_exception);
219 
220  void generate_service(t_service* tservice);
221 
222  void print_const_value(std::ofstream& out, std::string name, t_type* type, t_const_value* value);
223  std::string render_const_value(std::ofstream& out, std::string name, t_type* type, t_const_value* value);
224 
225  void generate_struct_definition (std::ofstream& out, t_struct* tstruct, bool is_exception=false, bool pointers=false, bool read=true, bool write=true);
226  void generate_struct_fingerprint (std::ofstream& out, t_struct* tstruct, bool is_definition);
227  void generate_struct_reader (std::ofstream& out, t_struct* tstruct, bool pointers=false);
228  void generate_struct_writer (std::ofstream& out, t_struct* tstruct, bool pointers=false);
229  void generate_struct_result_writer (std::ofstream& out, t_struct* tstruct, bool pointers=false);
230 #ifdef SANDESH
231  void generate_static_const_string_definition(std::ofstream& out, t_struct* tstruct);
232  void generate_struct_logger (ofstream& out, const string& name,
233  const vector<t_field*>& fields);
234  void generate_struct_get_size (ofstream& out, const string& name,
235  const vector<t_field*>& fields);
236 #endif
237 
242  void generate_service_interface (t_service* tservice, string style);
243  void generate_service_interface_factory (t_service* tservice, string style);
244  void generate_service_null (t_service* tservice, string style);
245  void generate_service_multiface (t_service* tservice);
246  void generate_service_helpers (t_service* tservice);
247  void generate_service_client (t_service* tservice, string style);
248  void generate_service_processor (t_service* tservice, string style);
249  void generate_service_skeleton (t_service* tservice);
250  void generate_process_function (t_service* tservice, t_function* tfunction,
251  string style, bool specialized=false);
252  void generate_function_helpers (t_service* tservice, t_function* tfunction);
253  void generate_service_async_skeleton (t_service* tservice);
254 
259  void generate_deserialize_field (std::ofstream& out,
260  t_field* tfield,
261  std::string prefix="",
262  std::string suffix="");
263 
264  void generate_deserialize_struct (std::ofstream& out,
265  t_struct* tstruct,
266  std::string prefix="");
267 
268  void generate_deserialize_container (std::ofstream& out,
269  t_type* ttype,
270  std::string prefix="");
271 
272  void generate_deserialize_set_element (std::ofstream& out,
273  t_set* tset,
274  std::string prefix="");
275 
276  void generate_deserialize_map_element (std::ofstream& out,
277  t_map* tmap,
278  std::string prefix="");
279 
280  void generate_deserialize_list_element (std::ofstream& out,
281  t_list* tlist,
282  std::string prefix,
283  bool push_back,
284  std::string index);
285 
286  void generate_serialize_field (std::ofstream& out,
287  t_field* tfield,
288  std::string prefix="",
289  std::string suffix="");
290 
291  void generate_serialize_struct (std::ofstream& out,
292  t_struct* tstruct,
293  std::string prefix="");
294 
295  void generate_serialize_container (std::ofstream& out,
296  t_type* ttype,
297  std::string prefix="");
298 
299  void generate_serialize_map_element (std::ofstream& out,
300  t_map* tmap,
301  std::string iter);
302 
303  void generate_serialize_set_element (std::ofstream& out,
304  t_set* tmap,
305  std::string iter);
306 
307  void generate_serialize_list_element (std::ofstream& out,
308  t_list* tlist,
309  std::string iter);
310 
311  void generate_function_call (ostream& out,
312  t_function* tfunction,
313  string target,
314  string iface,
315  string arg_prefix);
316 
317 #ifdef SANDESH
318  void generate_serialize_sandesh (std::ofstream& out,
319  t_sandesh* tsandesh,
320  std::string prefix="");
321 #endif
322 
323  /*
324  * Helper rendering functions
325  */
326 
327  std::string namespace_prefix(std::string ns);
328  std::string namespace_open(std::string ns);
329  std::string namespace_close(std::string ns);
330  std::string type_name(t_type* ttype, bool in_typedef=false, bool arg=false);
331  std::string base_type_name(t_base_type::t_base tbase);
332 #ifdef SANDESH
333  std::string declare_field(t_field* tfield, bool init=false, bool pointer=false, bool constant=false, bool reference=false, bool constructor=false);
334 #else
335  std::string declare_field(t_field* tfield, bool init=false, bool pointer=false, bool constant=false, bool reference=false);
336 #endif
337  std::string function_signature(t_function* tfunction, std::string style, std::string prefix="", bool name_params=true);
338  std::string cob_function_signature(t_function* tfunction, std::string prefix="", bool name_params=true);
339  std::string argument_list(t_struct* tstruct, bool name_params=true, bool start_comma=false);
340  std::string type_to_enum(t_type* ttype);
341  std::string local_reflection_name(const char*, t_type* ttype, bool external=false);
342 
343  void generate_enum_constant_list(std::ofstream& f,
344  const vector<t_enum_value*>& constants,
345  const char* prefix,
346  const char* suffix,
347  bool include_values);
348 
349  // These handles checking gen_dense_ and checking for duplicates.
350  void generate_local_reflection(std::ofstream& out, t_type* ttype, bool is_definition);
351  void generate_local_reflection_pointer(std::ofstream& out, t_type* ttype);
352 
353  bool is_complex_type(t_type* ttype) {
354  ttype = get_true_type(ttype);
355 
356  return
357  ttype->is_container() ||
358  ttype->is_struct() ||
359  ttype->is_xception() ||
360 #ifdef SANDESH
361  ttype->is_sandesh() ||
362  (ttype->is_base_type() && (((t_base_type *)ttype)->get_base() == t_base_type::TYPE_XML)) ||
363 #endif
364  (ttype->is_base_type() && (((t_base_type*)ttype)->get_base() == t_base_type::TYPE_STRING));
365  }
366 
367  void set_use_include_prefix(bool use_include_prefix) {
368  use_include_prefix_ = use_include_prefix;
369  }
370 
371  private:
376  std::string get_include_prefix(const t_program& program) const;
377 
382  void generate_common_struct_reader_body(std::ofstream& out,
383  t_struct_common* tstruct,
384  bool pointers = false);
385 
390  void generate_common_struct_writer_body(std::ofstream& out,
391  t_struct_common* tstruct,
392  bool pointers = false);
393 
397  bool gen_pure_enums_;
398 
402  bool gen_dense_;
403 
407  bool gen_templates_;
408 
413  bool use_include_prefix_;
414 
418  bool gen_cob_style_;
419 
423  bool gen_no_client_completion_;
424 
429  std::string ns_open_;
430  std::string ns_close_;
431 
437  std::ofstream f_types_;
438  std::ofstream f_types_impl_;
439  std::ofstream f_types_tcc_;
440  std::ofstream f_header_;
441  std::ofstream f_service_;
442  std::ofstream f_service_tcc_;
443 #ifdef SANDESH
444  std::ofstream f_request_impl_;
445  std::ofstream f_html_template_;
446  struct sandesh_logger {
447  enum type {
448  BUFFER,
449  LOG,
450  FORCED_LOG,
451  };
452  };
453  void generate_sandesh_logger(std::ofstream& out, t_sandesh* tsandesh,
454  sandesh_logger::type ltype);
455  void generate_sandesh_static_logger(ofstream &out, t_sandesh *tsandesh,
456  bool generate_sandesh_object);
457  void generate_sandesh_static_drop_logger(ofstream &out, t_sandesh *tsandesh,
458  bool generate_sandesh_object);
459 #endif
460 
464  std::set<std::string> reflected_fingerprints_;
465 
466  // The ProcessorGenerator is used to generate parts of the code,
467  // so it needs access to many of our protected members and methods.
468  //
469  // TODO: The code really should be cleaned up so that helper methods for
470  // writing to the output files are separate from the generator classes
471  // themselves.
472  friend class ProcessorGenerator;
473 };
474 
479 void t_cpp_generator::init_generator() {
480  // Make output directory
481  MKDIR(get_out_dir().c_str());
482 
483  // Make output file
484  string f_types_name = get_out_dir()+program_name_+"_types.h";
485  f_types_.open(f_types_name.c_str());
486 
487  string f_types_impl_name = get_out_dir()+program_name_+"_types.cpp";
488  f_types_impl_.open(f_types_impl_name.c_str());
489 
490 #ifdef SANDESH
491  string f_request_impl_name = get_out_dir()+program_name_+"_request_skeleton.cpp";
492  f_request_impl_.open(f_request_impl_name.c_str());
493 
494  string f_html_template_name = get_out_dir()+program_name_+"_html_template.cpp";
495  f_html_template_.open(f_html_template_name.c_str());
496 #endif
497 
498  if (gen_templates_) {
499  // If we don't open the stream, it appears to just discard data,
500  // which is fine.
501  string f_types_tcc_name = get_out_dir()+program_name_+"_types.tcc";
502  f_types_tcc_.open(f_types_tcc_name.c_str());
503  }
504 
505  // Print header
506  f_types_ <<
507  autogen_comment();
508  f_types_impl_ <<
509  autogen_comment();
510  f_types_tcc_ <<
511  autogen_comment();
512 #ifdef SANDESH
513  f_request_impl_ <<
514  autogen_comment();
515  f_html_template_ <<
516  autogen_comment();
517 #endif
518 
519  // Start ifndef
520  f_types_ <<
521  "#ifndef " << program_name_ << "_TYPES_H" << endl <<
522  "#define " << program_name_ << "_TYPES_H" << endl <<
523  endl;
524  f_types_tcc_ <<
525  "#ifndef " << program_name_ << "_TYPES_TCC" << endl <<
526  "#define " << program_name_ << "_TYPES_TCC" << endl <<
527  endl;
528 
529  // Include base types
530  f_types_ <<
531  "#include <sandesh/Thrift.h>" << endl <<
532 #ifndef SANDESH
533  "#include <TApplicationException.h>" << endl <<
534 #else
535  "#include <tbb/atomic.h>" << endl <<
536  "#include <boost/shared_ptr.hpp>" << endl <<
537  "#include <sandesh/derived_stats.h>" << endl <<
538  "#include <sandesh/derived_stats_algo.h>" << endl <<
539  "#include <boost/pointer_cast.hpp>" << endl <<
540 #endif
541  "#include <base/trace.h>" << endl;
542  if (program_name_ != "sandesh") {
543  f_types_ << "#include <sandesh/sandesh_types.h>" << endl;
544  f_types_ << "#include <sandesh/sandesh_constants.h>" << endl;
545  if (program_name_ != "derived_stats_results" && program_name_ != "vns") {
546  f_types_ << "#include <sandesh/sandesh.h>" << endl;
547  }
548  }
549  f_types_ <<
550  "#include <sandesh/protocol/TProtocol.h>" << endl <<
551  "#include <sandesh/transport/TTransport.h>" << endl <<
552  endl;
553 
554  // Include other Thrift includes
555  const vector<t_program*>& includes = program_->get_includes();
556  for (size_t i = 0; i < includes.size(); ++i) {
557  f_types_ <<
558  "#include \"" << get_include_prefix(*(includes[i])) <<
559  includes[i]->get_name() << "_types.h\"" << endl;
560 
561  // XXX(simpkins): If gen_templates_ is enabled, we currently assume all
562  // included files were also generated with templates enabled.
563  f_types_tcc_ <<
564  "#include \"" << get_include_prefix(*(includes[i])) <<
565  includes[i]->get_name() << "_types.tcc\"" << endl;
566  }
567  f_types_ << endl;
568 
569  // Include custom headers
570  const vector<string>& cpp_includes = program_->get_cpp_includes();
571  for (size_t i = 0; i < cpp_includes.size(); ++i) {
572  if (cpp_includes[i][0] == '<') {
573  f_types_ <<
574  "#include " << cpp_includes[i] << endl;
575  } else {
576  f_types_ <<
577  "#include \"" << cpp_includes[i] << "\"" << endl;
578  }
579  }
580  f_types_ <<
581  endl;
582 
583  // Include the types file
584 #ifdef SANDESH
585  f_types_impl_ <<
586  "#include <boost/date_time/posix_time/posix_time.hpp>" << endl << endl <<
587  "#include <base/logging.h>" << endl << endl <<
588  "#include <sandesh/sandesh_uve.h>" << endl <<
589  "#include <sandesh/sandesh_http.h>" << endl <<
590  "#include <sandesh/sandesh_trace.h>" << endl <<
591  "#include <curl/curl.h>" << endl <<
592  "#include <boost/foreach.hpp>" << endl <<
593  "#include <boost/assign/list_of.hpp>" << endl <<
594  "#include <boost/make_shared.hpp>" << endl << endl;
595 #endif
596  f_types_impl_ <<
597  "#include \"" << get_include_prefix(*get_program()) << program_name_ <<
598  "_types.h\"" << endl <<
599  endl;
600  f_types_tcc_ <<
601  "#include \"" << get_include_prefix(*get_program()) << program_name_ <<
602  "_types.h\"" << endl <<
603  endl;
604 #ifdef SANDESH
605  f_request_impl_ <<
606  "#include <sandesh/sandesh_types.h>" << endl <<
607  "#include <sandesh/sandesh_constants.h>" << endl <<
608  "#include <sandesh/sandesh.h>" << endl <<
609  "#include \"" << get_include_prefix(*get_program()) << program_name_ <<
610  "_types.h\"" << endl <<
611  endl;
612  f_html_template_ <<
613  "#include <sandesh/sandesh_http.h>" << endl;
614  f_html_template_ <<
615  "static SandeshHttp::HtmlInfo h_info(" <<
616  program_name_ << "_xml," << program_name_ << "_xml_len);" << endl;
617  f_html_template_ <<
618  "static SandeshHttp sh(\"" << program_name_ << ".xml\", h_info);" << endl;
619  f_html_template_ << "int " << program_name_ << "_marker = 0;" << endl;
620 
621 #endif
622 
623 
624  // If we are generating local reflection metadata, we need to include
625  // the definition of TypeSpec.
626  if (gen_dense_) {
627  f_types_impl_ <<
628  "#include <TReflectionLocal.h>" << endl <<
629  endl;
630  }
631 
632  // Open namespace
633  ns_open_ = namespace_open(program_->get_namespace("cpp"));
634  ns_close_ = namespace_close(program_->get_namespace("cpp"));
635 
636  f_types_ <<
637  ns_open_ << endl <<
638  endl;
639 
640  f_types_impl_ <<
641  ns_open_ << endl <<
642  endl;
643 
644  f_types_tcc_ <<
645  ns_open_ << endl <<
646  endl;
647 
648 #ifdef SANDESH
649  f_request_impl_ <<
650  ns_open_ << endl <<
651  endl;
652 #endif
653 }
654 
658 void t_cpp_generator::close_generator() {
659  // Close namespace
660  f_types_ <<
661  ns_close_ << endl <<
662  endl;
663  f_types_impl_ <<
664  ns_close_ << endl;
665  f_types_tcc_ <<
666  ns_close_ << endl <<
667  endl;
668 #ifdef SANDESH
669  f_request_impl_ <<
670  ns_close_ << endl;
671 #endif
672 
673  // Include the types.tcc file from the types header file,
674  // so clients don't have to explicitly include the tcc file.
675  // TODO(simpkins): Make this a separate option.
676  if (gen_templates_) {
677  f_types_ <<
678  "#include \"" << get_include_prefix(*get_program()) << program_name_ <<
679  "_types.tcc\"" << endl <<
680  endl;
681  }
682 
683  // Close ifndef
684  f_types_ <<
685  "#endif" << endl;
686  f_types_tcc_ <<
687  "#endif" << endl;
688 
689  // Close output file
690  f_types_.close();
691  f_types_impl_.close();
692  f_types_tcc_.close();
693 #ifdef SANDESH
694  f_request_impl_.close();
695  f_html_template_.close();
696 #endif
697 }
698 
704 void t_cpp_generator::generate_typedef(t_typedef* ttypedef) {
705  f_types_ <<
706  indent() << "typedef " << type_name(ttypedef->get_type(), true) << " " << ttypedef->get_symbolic() << ";" << endl <<
707  endl;
708 }
709 
710 
711 void t_cpp_generator::generate_enum_constant_list(std::ofstream& f,
712  const vector<t_enum_value*>& constants,
713  const char* prefix,
714  const char* suffix,
715  bool include_values) {
716  f << " {" << endl;
717  indent_up();
718 
719  vector<t_enum_value*>::const_iterator c_iter;
720  bool first = true;
721  for (c_iter = constants.begin(); c_iter != constants.end(); ++c_iter) {
722  if (first) {
723  first = false;
724  } else {
725  f << "," << endl;
726  }
727  indent(f)
728  << prefix << (*c_iter)->get_name() << suffix;
729  if (include_values && (*c_iter)->has_value()) {
730  f << " = " << (*c_iter)->get_value();
731  }
732  }
733 
734  f << endl;
735  indent_down();
736  indent(f) << "};" << endl;
737 }
738 
745 void t_cpp_generator::generate_enum(t_enum* tenum) {
746  vector<t_enum_value*> constants = tenum->get_constants();
747 
748  std::string enum_name = tenum->get_name();
749  if (!gen_pure_enums_) {
750  enum_name = "type";
751  f_types_ <<
752  indent() << "struct " << tenum->get_name() << " {" << endl;
753  indent_up();
754  }
755  f_types_ <<
756  indent() << "enum " << enum_name;
757 
758  generate_enum_constant_list(f_types_, constants, "", "", true);
759 
760  if (!gen_pure_enums_) {
761  indent_down();
762  f_types_ << "};" << endl;
763  }
764 
765  f_types_ << endl;
766 
770  std::string prefix = "";
771  if (!gen_pure_enums_) {
772  prefix = tenum->get_name() + "::";
773  }
774 
775  f_types_impl_ <<
776  indent() << "int _k" << tenum->get_name() << "Values[] =";
777  generate_enum_constant_list(f_types_impl_, constants, prefix.c_str(), "", false);
778 
779  f_types_impl_ <<
780  indent() << "const char* _k" << tenum->get_name() << "Names[] =";
781  generate_enum_constant_list(f_types_impl_, constants, "\"", "\"", false);
782 
783  f_types_ <<
784  indent() << "extern const std::map<int, const char*> _" <<
785  tenum->get_name() << "_VALUES_TO_NAMES;" << endl << endl;
786 
787  f_types_impl_ <<
788  indent() << "const std::map<int, const char*> _" << tenum->get_name() <<
789  "_VALUES_TO_NAMES(::contrail::sandesh::TEnumIterator(" << constants.size() <<
790  ", _k" << tenum->get_name() << "Values" <<
791  ", _k" << tenum->get_name() << "Names), " <<
792  "::contrail::sandesh::TEnumIterator(-1, NULL, NULL));" << endl << endl;
793 
794  generate_local_reflection(f_types_, tenum, false);
795  generate_local_reflection(f_types_impl_, tenum, true);
796 }
797 
801 void t_cpp_generator::generate_consts(std::vector<t_const*> consts) {
802  string f_consts_name = get_out_dir()+program_name_+"_constants.h";
803  ofstream f_consts;
804  f_consts.open(f_consts_name.c_str());
805 
806  string f_consts_impl_name = get_out_dir()+program_name_+"_constants.cpp";
807  ofstream f_consts_impl;
808  f_consts_impl.open(f_consts_impl_name.c_str());
809 
810  // Print header
811  f_consts <<
812  autogen_comment();
813  f_consts_impl <<
814  autogen_comment();
815 
816  // Start ifndef
817  f_consts <<
818  "#ifndef " << program_name_ << "_CONSTANTS_H" << endl <<
819  "#define " << program_name_ << "_CONSTANTS_H" << endl <<
820  endl <<
821  "#include \"" << get_include_prefix(*get_program()) << program_name_ <<
822  "_types.h\"" << endl <<
823  endl <<
824  ns_open_ << endl <<
825  endl;
826 
827 #ifdef SANDESH
828  f_consts_impl <<
829  "#include <base/trace.h>" << endl <<
830  "#include <sandesh/sandesh_types.h>" << endl <<
831  "#include <sandesh/sandesh_constants.h>" << endl <<
832  "#include <sandesh/sandesh.h>" << endl <<
833  "#include <sandesh/sandesh_trace.h>" << endl << endl;
834 #endif
835  f_consts_impl <<
836  "#include \"" << get_include_prefix(*get_program()) << program_name_ <<
837  "_constants.h\"" << endl <<
838  endl <<
839  ns_open_ << endl <<
840  endl;
841 
842  f_consts <<
843  "class " << program_name_ << "Constants {" << endl <<
844  " public:" << endl <<
845  " " << program_name_ << "Constants();" << endl <<
846  endl;
847  indent_up();
848  vector<t_const*>::iterator c_iter;
849  for (c_iter = consts.begin(); c_iter != consts.end(); ++c_iter) {
850  string name = (*c_iter)->get_name();
851  t_type* type = (*c_iter)->get_type();
852  f_consts <<
853  indent() << type_name(type) << " " << name << ";" << endl;
854  }
855  indent_down();
856  f_consts <<
857  "};" << endl;
858 
859  f_consts_impl <<
860  "const " << program_name_ << "Constants g_" << program_name_ << "_constants;" << endl <<
861  endl <<
862  program_name_ << "Constants::" << program_name_ << "Constants() {" << endl;
863  indent_up();
864  for (c_iter = consts.begin(); c_iter != consts.end(); ++c_iter) {
865  print_const_value(f_consts_impl,
866  (*c_iter)->get_name(),
867  (*c_iter)->get_type(),
868  (*c_iter)->get_value());
869  }
870  indent_down();
871  indent(f_consts_impl) <<
872  "}" << endl;
873 
874  f_consts <<
875  endl <<
876  "extern const " << program_name_ << "Constants g_" << program_name_ << "_constants;" << endl <<
877  endl <<
878  ns_close_ << endl <<
879  endl <<
880  "#endif" << endl;
881  f_consts.close();
882 
883  f_consts_impl <<
884  endl <<
885  ns_close_ << endl <<
886  endl;
887 }
888 
894 void t_cpp_generator::print_const_value(ofstream& out, string name, t_type* type, t_const_value* value) {
895  type = get_true_type(type);
896  if (type->is_base_type()) {
897  string v2 = render_const_value(out, name, type, value);
898  indent(out) << name << " = " << v2 << ";" << endl <<
899  endl;
900  } else if (type->is_enum()) {
901  indent(out) << name << " = (" << type_name(type) << ")" << value->get_integer() << ";" << endl <<
902  endl;
903  } else if (type->is_struct() || type->is_xception()) {
904  const vector<t_field*>& fields = ((t_struct*)type)->get_members();
905  vector<t_field*>::const_iterator f_iter;
906  const map<t_const_value*, t_const_value*>& val = value->get_map();
907  map<t_const_value*, t_const_value*>::const_iterator v_iter;
908  for (v_iter = val.begin(); v_iter != val.end(); ++v_iter) {
909  t_type* field_type = NULL;
910  for (f_iter = fields.begin(); f_iter != fields.end(); ++f_iter) {
911  if ((*f_iter)->get_name() == v_iter->first->get_string()) {
912  field_type = (*f_iter)->get_type();
913  }
914  }
915  if (field_type == NULL) {
916  throw "type error: " + type->get_name() + " has no field " + v_iter->first->get_string();
917  }
918  string val = render_const_value(out, name, field_type, v_iter->second);
919  indent(out) << name << "." << v_iter->first->get_string() << " = " << val << ";" << endl;
920  indent(out) << name << ".__isset." << v_iter->first->get_string() << " = true;" << endl;
921  }
922  out << endl;
923  } else if (type->is_map()) {
924  t_type* ktype = ((t_map*)type)->get_key_type();
925  t_type* vtype = ((t_map*)type)->get_val_type();
926  const map<t_const_value*, t_const_value*>& val = value->get_map();
927  map<t_const_value*, t_const_value*>::const_iterator v_iter;
928  for (v_iter = val.begin(); v_iter != val.end(); ++v_iter) {
929  string key = render_const_value(out, name, ktype, v_iter->first);
930  string val = render_const_value(out, name, vtype, v_iter->second);
931  indent(out) << name << ".insert(std::make_pair(" << key << ", " << val << "));" << endl;
932  }
933  out << endl;
934  } else if (type->is_list()) {
935  t_type* etype = ((t_list*)type)->get_elem_type();
936  const vector<t_const_value*>& val = value->get_list();
937  vector<t_const_value*>::const_iterator v_iter;
938  for (v_iter = val.begin(); v_iter != val.end(); ++v_iter) {
939  string val = render_const_value(out, name, etype, *v_iter);
940  indent(out) << name << ".push_back(" << val << ");" << endl;
941  }
942  out << endl;
943  } else if (type->is_set()) {
944  t_type* etype = ((t_set*)type)->get_elem_type();
945  const vector<t_const_value*>& val = value->get_list();
946  vector<t_const_value*>::const_iterator v_iter;
947  for (v_iter = val.begin(); v_iter != val.end(); ++v_iter) {
948  string val = render_const_value(out, name, etype, *v_iter);
949  indent(out) << name << ".insert(" << val << ");" << endl;
950  }
951  out << endl;
952  } else {
953  throw "INVALID TYPE IN print_const_value: " + type->get_name();
954  }
955 }
956 
960 string t_cpp_generator::render_const_value(ofstream& out, string name, t_type* type, t_const_value* value) {
961  (void) name;
962  std::ostringstream render;
963 
964  if (type->is_base_type()) {
965  t_base_type::t_base tbase = ((t_base_type*)type)->get_base();
966  switch (tbase) {
967  case t_base_type::TYPE_STRING:
968 #ifdef SANDESH
969  case t_base_type::TYPE_STATIC_CONST_STRING:
970  case t_base_type::TYPE_XML:
971 #endif
972  render << '"' << get_escaped_string(value) << '"';
973  break;
974  case t_base_type::TYPE_BOOL:
975  render << ((value->get_integer() > 0) ? "true" : "false");
976  break;
977  case t_base_type::TYPE_BYTE:
978  case t_base_type::TYPE_I16:
979  case t_base_type::TYPE_I32:
980 #ifdef SANDESH
981  case t_base_type::TYPE_U16:
982  case t_base_type::TYPE_U32:
983 #endif
984  render << value->get_integer();
985  break;
986  case t_base_type::TYPE_I64:
987  render << value->get_integer() << "LL";
988  break;
989 #ifdef SANDESH
990  case t_base_type::TYPE_U64:
991  render << value->get_integer() << "ULL";
992  break;
993  case t_base_type::TYPE_UUID:
994  render << "boost::uuids::string_generator()(\"" << value->get_uuid() << "\")";
995  break;
996 #endif
997  case t_base_type::TYPE_DOUBLE:
998  if (value->get_type() == t_const_value::CV_INTEGER) {
999  render << value->get_integer();
1000  } else {
1001  render << value->get_double();
1002  }
1003  break;
1004 #ifdef SANDESH
1005  case t_base_type::TYPE_IPV4:
1006  render << value->get_integer();
1007  break;
1008  case t_base_type::TYPE_IPADDR:
1009  render << "boost::asio::ip::address::from_string(\"" <<
1010  value->get_string() << "\")";
1011  break;
1012 #endif
1013  default:
1014  throw "compiler error: no const of base type " + t_base_type::t_base_name(tbase);
1015  }
1016  } else if (type->is_enum()) {
1017  render << "(" << type_name(type) << ")" << value->get_integer();
1018  } else {
1019  string t = tmp("tmp");
1020  indent(out) << type_name(type) << " " << t << ";" << endl;
1021  print_const_value(out, t, type, value);
1022  render << t;
1023  }
1024 
1025  return render.str();
1026 }
1027 
1035 void t_cpp_generator::generate_cpp_struct(t_struct* tstruct, bool is_exception) {
1036  generate_struct_definition(f_types_, tstruct, is_exception);
1037  generate_struct_fingerprint(f_types_impl_, tstruct, true);
1038  generate_local_reflection(f_types_, tstruct, false);
1039  generate_local_reflection(f_types_impl_, tstruct, true);
1040  generate_local_reflection_pointer(f_types_impl_, tstruct);
1041 #ifdef SANDESH
1042  generate_static_const_string_definition(f_types_impl_, tstruct);
1043 #endif
1044 
1045  std::ofstream& out = (gen_templates_ ? f_types_tcc_ : f_types_impl_);
1046  generate_struct_reader(out, tstruct);
1047  generate_struct_writer(out, tstruct);
1048 #ifdef SANDESH
1049  generate_struct_logger(out, tstruct->get_name(), tstruct->get_members());
1050  generate_struct_get_size(out, tstruct->get_name(), tstruct->get_members());
1051 #endif
1052 }
1053 
1054 #ifdef SANDESH
1062 void t_cpp_generator::generate_cpp_sandesh(t_sandesh* tsandesh) {
1063  bool is_request =
1064  ((t_base_type *)tsandesh->get_type())->is_sandesh_request();
1065  bool is_trace =
1066  ((t_base_type *)tsandesh->get_type())->is_sandesh_trace() ||
1067  ((t_base_type *)tsandesh->get_type())->is_sandesh_trace_object();
1068  bool is_uve =
1069  ((t_base_type *)tsandesh->get_type())->is_sandesh_uve();
1070  bool is_alarm =
1071  ((t_base_type *)tsandesh->get_type())->is_sandesh_alarm();
1072  bool is_system =
1073  ((t_base_type *)tsandesh->get_type())->is_sandesh_system();
1074 
1075  generate_sandesh_definition(f_types_, tsandesh);
1076  generate_sandesh_fingerprint(f_types_impl_, tsandesh, true);
1077  std::ofstream& out = f_types_impl_;
1078  generate_static_const_string_definition(out, tsandesh);
1079  generate_sandesh_static_versionsig_def(out, tsandesh);
1080  generate_sandesh_creator(out, tsandesh);
1081  generate_sandesh_reader(out, tsandesh);
1082  generate_sandesh_writer(out, tsandesh);
1083  generate_sandesh_loggers(out, tsandesh);
1084  generate_sandesh_get_size(out, tsandesh);
1085 
1086  if (!is_trace) {
1087  generate_sandesh_static_seqnum_def(out, tsandesh);
1088  }
1089  if (is_request) {
1090  generate_sandesh_http_reader(out, tsandesh);
1091  // Generate a skeleton impl file
1092  generate_sandesh_request(f_request_impl_, tsandesh);
1093  }
1094  if (is_uve || is_alarm) {
1095  generate_sandesh_updater(out,tsandesh);
1096  }
1097  if (is_system) {
1098  generate_sandesh_static_rate_limit_log_def(out, tsandesh);
1099  generate_sandesh_static_rate_limit_mutex_def(out, tsandesh);
1100  generate_sandesh_static_rate_limit_buffer_def(out, tsandesh);
1101  }
1102 
1103 }
1104 
1111 void t_cpp_generator::generate_sandesh_request(ofstream& out,
1112  t_sandesh* tsandesh) {
1113  out << "void " << tsandesh->get_name() << "::" <<
1114  "HandleRequest() const {" << endl;
1115  indent_up();
1116 
1117  indent(out) << "// This autogenerated skeleton file illustrates the " <<
1118  "function needed to" << endl;
1119  indent(out) << "// implement " << tsandesh->get_name() << "." << endl;
1120  indent(out) << "// You should copy it to another filename to avoid " <<
1121  "overwriting it. Add the" << endl;
1122  indent(out) << "// HandleRequest() implementation to it, and include it " <<
1123  "in your module's" << endl;
1124  indent(out) << "// SConscript. Alternatively, add the implementation of " <<
1125  "HandleRequest()" << endl;
1126  indent(out) << "// in your module's sources." << endl;
1127  scope_down(out);
1128 }
1129 
1136 std::string t_cpp_generator::generate_sandesh_no_static_const_string_function(
1137  t_sandesh* tsandesh, bool signature, bool autogen_darg,
1138  bool trace, bool request, bool ctorcall) {
1139  string result = "";
1140 
1141  // Get members
1142  vector<t_field*>::const_iterator m_iter;
1143  const vector<t_field*>& members = tsandesh->get_members();
1144 
1145  bool init_function = false;
1146 
1147  if (trace) {
1148  result += "(";
1149  init_function = true;
1150  if (signature) {
1151  result += "SandeshTraceBufferPtr trace_buf, std::string file, int32_t line";
1152  } else {
1153  result += "trace_buf, file, line";
1154  }
1155  } else {
1156  if (signature) {
1157  if (!ctorcall) {
1158  result += "(";
1159  init_function = true;
1160  result += "uint32_t seqno";
1161  }
1162  } else {
1163  result += "(";
1164  init_function = true;
1165  if (ctorcall)
1166  result += "0";
1167  else
1168  result += "lseqnum_++";
1169  }
1170  }
1171 
1172  for (m_iter = members.begin(); m_iter != members.end(); ++m_iter) {
1173  t_type* t = get_true_type((*m_iter)->get_type());
1174  if (t->is_static_const_string()) {
1175  continue;
1176  }
1177  if ((*m_iter)->get_auto_generated() && trace) {
1178  continue;
1179  }
1180  if (((t_base_type *)tsandesh->get_type())->is_sandesh_object() &&
1181  ((*m_iter)->get_req() == t_field::T_OPTIONAL)) {
1182  continue;
1183  }
1184  if (!init_function) {
1185  result += "(";
1186  init_function = true;
1187  } else {
1188  result += ", ";
1189  }
1190 
1191  if (signature) {
1192  // Special handling for auto-generated members
1193  if (autogen_darg && (*m_iter)->get_auto_generated()) {
1194  result += declare_field(*m_iter, true, false, false, false, true);
1195  } else {
1196  bool use_const = !(t->is_base_type() || t->is_enum()) || t->is_string();
1197  result += declare_field(*m_iter, false, false, use_const, !t->is_base_type() || t->is_string(), true);
1198  }
1199  } else {
1200  result += (*m_iter)->get_name();
1201  }
1202  }
1203  if (!init_function) {
1204  result += "(";
1205  }
1206 
1207  if (request) {
1208  if (signature) {
1209  if (init_function) {
1210  result += ", ";
1211  }
1212  result += "const std::string& context, SandeshConnection * sconn = NULL";
1213  } else {
1214  if (init_function) {
1215  result += ", ";
1216  }
1217  result += "context, sconn";
1218  }
1219  }
1220 
1221  result += ")";
1222  return result;
1223 }
1224 
1225 std::string t_cpp_generator::generate_sandesh_async_creator(t_sandesh* tsandesh, bool signature,
1226  bool expand_autogen, bool skip_autogen, std::string prefix, std::string suffix,
1227  bool category_level_file_line_first, bool autogen_category_level, bool
1228  drop_log_reason, bool use_sandesh_object)
1229  {
1230  string result = "";
1231  string temp = "";
1232  string category_def = prefix + "category" + suffix;
1233  string level_def = prefix + "level" + suffix;
1234  string category_dec = "std::string " + category_def;
1235  string level_dec = "SandeshLevel::type " + level_def;
1236  string module_name = "\"\"";
1237  string drop_reason_dec;
1238  string drop_reason_def;
1239  bool is_flow = ((t_base_type *)tsandesh->get_type())->is_sandesh_flow();
1240  // Get members
1241  vector<t_field*>::const_iterator m_iter;
1242  const vector<t_field*>& members = tsandesh->get_members();
1243  bool init_function = false;
1244  if (drop_log_reason) {
1245  drop_reason_def = prefix + "drop_reason" + suffix + ", ";
1246  drop_reason_dec = "const std::string& " + drop_reason_def;
1247  }
1248 
1249  if (category_level_file_line_first) {
1250  if (signature) {
1251  result += "(" + drop_reason_dec + category_dec + ", " + level_dec;
1252  if (!is_flow) result += ", std::string file, int32_t line";
1253  } else {
1254  if (!autogen_category_level) {
1255  result += "(" + drop_reason_dec + category_def + ", " + level_def;
1256  if (!is_flow) result += ", __FILE__, __LINE__";
1257  } else {
1258  result += "(" + drop_reason_dec + module_name + ", SandeshLevel::SYS_INFO";
1259  if (!is_flow) result += ", __FILE__, __LINE__";
1260  }
1261  }
1262  init_function = true;
1263  } else {
1264  if (signature) {
1265  if (!autogen_category_level) {
1266  result += "(" + drop_reason_dec + category_dec + ", " + level_dec;
1267  init_function = true;
1268  }
1269  } else {
1270  if (!autogen_category_level) {
1271  result += "(" + drop_reason_def + category_def + ", " + level_def;
1272  init_function = true;
1273  }
1274  }
1275  }
1276  if(use_sandesh_object) {
1277  if (!init_function) {
1278  result += "(snh";
1279  init_function = true;
1280  } else {
1281  if(signature) {
1282  result += ", " + tsandesh->get_name() + " *snh";
1283  } else {
1284  result += ", snh";
1285  }
1286  }
1287  } else {
1288  for (m_iter = members.begin(); m_iter != members.end(); ++m_iter) {
1289  temp = "";
1290  t_type* t = get_true_type((*m_iter)->get_type());
1291  if (t->is_static_const_string()) {
1292  continue;
1293  }
1294  bool is_optional = (*m_iter)->get_req() == t_field::T_OPTIONAL;
1295  if(is_optional) {
1296  continue;
1297  }
1298  bool skip = (*m_iter)->get_auto_generated() &&
1299  ((*m_iter)->get_name() == "file" || (*m_iter)->get_name() == "line") &&
1300  category_level_file_line_first;
1301  if (!init_function) {
1302  result += "(";
1303  init_function = true;
1304  } else {
1305  if (!skip) {
1306  temp += ", ";
1307  }
1308  }
1309  if (signature) {
1310  result += temp;
1311  // Special signature processing for autogen members
1312  if ((*m_iter)->get_auto_generated()) {
1313  if (!skip) {
1314  result += declare_field(*m_iter, true, false, false, !t->is_base_type(), true);
1315  }
1316  } else {
1317  bool use_const = !(t->is_base_type() || t->is_enum()) || t->is_string();
1318  result += declare_field(*m_iter, false, false, use_const,
1319  !t->is_base_type() || t->is_string(), true);
1320  }
1321  } else {
1322  if (skip) {
1323  continue;
1324  }
1325  // Special processing for autogen members
1326  if ((*m_iter)->get_auto_generated()) {
1327  // Skip
1328  if (skip_autogen) {
1329  continue;
1330  }
1331  // Map to preprocessor macros
1332  if (expand_autogen) {
1333  if ((*m_iter)->get_name() == "file") {
1334  result += temp;
1335  result += "__FILE__";
1336  } else if ((*m_iter)->get_name() == "line") {
1337  result += temp;
1338  result += "__LINE__";
1339  }
1340  } else {
1341  result += temp;
1342  result += prefix + (*m_iter)->get_name() + suffix;
1343  }
1344  } else {
1345  result += temp;
1346  result += prefix + (*m_iter)->get_name() + suffix;
1347  }
1348  }
1349  }
1350  }
1351  if (!init_function) {
1352  result += "(";
1353  }
1354  result += ")";
1355  return result;
1356 }
1357 
1358 void t_cpp_generator::generate_sandesh_async_create_fn(ofstream &out,
1359  t_sandesh *tsandesh) {
1360  std::string creator_func_name = "Create";
1361  out << indent() << "static " << tsandesh->get_name() << "* "
1362  << creator_func_name << "(std::string file = \"\", int32_t line = 0) {"
1363  << endl;
1364  indent_up();
1365  out << indent() << tsandesh->get_name() <<
1366  " * snh = new " << tsandesh->get_name() << "(lseqnum_++, file, line);"
1367  << endl;
1368  out << indent() << "return snh;" << endl;
1369  indent_down();
1370  indent(out) << "}" << endl << endl;
1371 }
1372 
1373 void t_cpp_generator::generate_sandesh_async_create_macro(ofstream &out,
1374  t_sandesh *tsandesh) {
1375  std::string creator_func_name = "Create";
1376  // Generate creator macro
1377  string creator_name = tsandesh->get_name() + creator_func_name;
1378  string creator_name_usc = underscore(creator_name);
1379  string creator_name_uc = uppercase(creator_name_usc);
1380  out << indent() << "#define " << creator_name_uc <<
1381  "() \\" << endl;
1382  indent_up();
1383  out << indent() << tsandesh->get_name() << "::" << creator_func_name <<
1384  "(__FILE__, __LINE__)" << endl;
1385  indent_down();
1386  indent(out) << endl << endl;
1387 }
1388 
1389 void t_cpp_generator::generate_sandesh_async_send_fn(
1390  ofstream &out, t_sandesh *tsandesh, bool generate_sandesh_object,
1391  bool generate_rate_limit, bool generate_system_log) {
1392  std::string sender_func_name = "Send";
1393  // Generate sender
1394  out << indent() << "static void " << sender_func_name;
1395  out << generate_sandesh_async_creator(tsandesh, true, false, false, "", "",
1396  true, false, false, generate_sandesh_object);
1397  out << " {" << endl;
1398  indent_up();
1399  if (generate_sandesh_object) {
1400  out << indent() << "snh->set_level(level);" << endl;
1401  out << indent() << "snh->set_category(category);" << endl;
1402  }
1403  out << indent() << "if (HandleTest(level, category)) {" << endl;
1404  indent_up();
1405  out << indent() << "Log";
1406  if (generate_sandesh_object) {
1407  out << "(category, level, snh);" << endl;
1408  out << indent() << "snh->Release();" << endl;
1409  } else {
1410  out << generate_sandesh_async_creator(tsandesh, false, false, false, "",
1411  "", false, false, false) << "; " << endl;
1412  }
1413  out << indent() << "return;" << endl;
1414  scope_down(out);
1415  if (generate_system_log) {
1416  out << indent() << "if (IsSendingAllMessagesDisabled() || " <<
1417  "IsSendingSystemLogsDisabled()) {" << endl;
1418  } else {
1419  out << indent() << "if (IsSendingAllMessagesDisabled() || " <<
1420  "IsSendingObjectLogsDisabled()) {" << endl;
1421  }
1422  indent_up();
1423  out << indent() << "UpdateTxMsgFailStats(\"" << tsandesh->get_name() <<
1424  "\", 0, SandeshTxDropReason::SendingDisabled);" << endl;
1425  out << indent() << "Log";
1426  if (generate_sandesh_object) {
1427  out << "(category, level, snh);" << endl;
1428  out << indent() << "snh->Release();" << endl;
1429  } else {
1430  out << generate_sandesh_async_creator(tsandesh, false, false, false, "",
1431  "", false, false, false) << "; " << endl;
1432  }
1433  out << indent() << "return;" << endl;
1434  scope_down(out);
1435  if (generate_rate_limit) {
1436  out << indent() << "if (!IsRatelimitPass()) {" << endl;
1437  indent_up();
1438  out << indent() << "UpdateTxMsgFailStats(\"" << tsandesh->get_name() <<
1439  "\", 0, SandeshTxDropReason::RatelimitDrop);" << endl;
1440  out << indent() << "if (do_rate_limit_drop_log_) {" << endl;
1441  indent_up();
1442  out << indent() << "std::stringstream ratelimit_val;" << endl;
1443  out << indent() << " ratelimit_val << Sandesh::get_send_rate_limit();"
1444  << endl;
1445  out << indent() << "std::string drop_reason = \"SANDESH: Ratelimit"
1446  " Drop (\" + ratelimit_val.str() + std::string(\" messages"
1447  "/second): \") ;" << endl;
1448  out << indent() << "DropLog";
1449  if (generate_sandesh_object) {
1450  out << "(drop_reason, category, level, snh);" << endl;
1451  out << indent() << "snh->Release();" << endl;
1452  } else {
1453  out << generate_sandesh_async_creator(tsandesh, false, false,
1454  false, "", "", false, false, true) << "; " << endl;
1455  }
1456  out << indent() << "do_rate_limit_drop_log_ = false;" << endl;
1457  scope_down(out);
1458  out << indent() << "return;" << endl;
1459  scope_down(out);
1460  }
1461  out << indent() << "if (level >= SendingLevel()) {" << endl;
1462  indent_up();
1463  out << indent() << "UpdateTxMsgFailStats(\"" << tsandesh->get_name() <<
1464  "\", 0, SandeshTxDropReason::QueueLevel);" << endl;
1465  out << indent() << "std::string drop_reason = \"SANDESH: Queue Drop:"
1466  " \";" << endl;
1467  out << indent() << "DropLog";
1468  if (generate_sandesh_object) {
1469  out << "(drop_reason, category, level, snh);" << endl;
1470  out << indent() << "snh->Release();" << endl;
1471  } else {
1472  out << generate_sandesh_async_creator(tsandesh, false, false, false, "",
1473  "", false, false, true) << "; " << endl;
1474  }
1475  out << indent() << "return;" << endl;
1476  scope_down(out);
1477  if (!generate_sandesh_object) {
1478  out << indent() << tsandesh->get_name() <<
1479  " * snh = new " << tsandesh->get_name() <<
1480  generate_sandesh_no_static_const_string_function(tsandesh,
1481  false, false, false, false) << ";" << endl;
1482  out << indent() << "snh->set_level(level);" << endl;
1483  out << indent() << "snh->set_category(category);" << endl;
1484  }
1485  out << indent() << "snh->Dispatch();" << endl;
1486  indent_down();
1487  indent(out) << "}" << endl << endl;
1488 }
1489 
1490 void t_cpp_generator::generate_sandesh_async_send_macros(ofstream &out,
1491  t_sandesh *tsandesh, bool generate_sandesh_object) {
1492  std::string sender_func_name = "Send";
1493  std::string logger_func_name = "Log";
1494  // Generate creator macro
1495  string creator_name = tsandesh->get_name() + logger_func_name;
1496  if (generate_sandesh_object) {
1497  creator_name += "Sandesh";
1498  }
1499  string creator_name_usc = underscore(creator_name);
1500  string creator_name_uc = uppercase(creator_name_usc);
1501  out << indent() << "#define " << creator_name_uc;
1502  out << generate_sandesh_async_creator(tsandesh, false, false, true, "_",
1503  "", false, false, false, generate_sandesh_object);
1504  out << "\\" << endl;
1505  indent_up();
1506  out << indent() << tsandesh->get_name() << "::" << sender_func_name;
1507  out << generate_sandesh_async_creator(tsandesh, false, true, false,
1508  "(_", ")", true, false, false, generate_sandesh_object) << endl;
1509  indent_down();
1510  indent(out) << endl << endl;
1511 
1512  // Generate creator macro (legacy)
1513  creator_name = tsandesh->get_name() + sender_func_name;
1514  if (generate_sandesh_object) {
1515  creator_name += "Sandesh";
1516  }
1517  creator_name_usc = underscore(creator_name);
1518  creator_name_uc = uppercase(creator_name_usc);
1519  out << indent() << "#define " << creator_name_uc;
1520  out << generate_sandesh_async_creator(tsandesh, false, false, true, "_",
1521  "", false, true, false, generate_sandesh_object);
1522  out << "\\" << endl;
1523  indent_up();
1524  out << indent() << tsandesh->get_name() << "::" << sender_func_name;
1525  out << generate_sandesh_async_creator(tsandesh, false, true, false,
1526  "(_", ")", true, true, false, generate_sandesh_object) << endl;
1527  indent_down();
1528  indent(out) << endl << endl;
1529 }
1530 
1531 void t_cpp_generator::generate_sandesh_rate_limit_fn(ofstream &out,
1532  t_sandesh *tsandesh) {
1533  out << indent() << "static bool IsRatelimitPass() {" << endl;
1534  indent_up();
1535  generate_isRatelimitPass(out, tsandesh);
1536  indent_down();
1537  indent(out) << "}" << endl << endl;
1538 }
1539 
1540 void t_cpp_generator::generate_sandesh_systemlog_creators(ofstream &out,
1541  t_sandesh *tsandesh) {
1542  bool generate_sandesh_object = false;
1543  bool generate_rate_limit = true;
1544  bool generate_system_log = true;
1545  // Generate send function and macros
1546  generate_sandesh_async_send_fn(out, tsandesh, generate_sandesh_object,
1547  generate_rate_limit, generate_system_log);
1548  generate_sandesh_async_send_macros(out, tsandesh, generate_sandesh_object);
1549  // Generate DropLog
1550  generate_sandesh_static_drop_logger(out, tsandesh,
1551  generate_sandesh_object);
1552  // Generate Log
1553  generate_sandesh_static_logger(out, tsandesh, generate_sandesh_object);
1554  // Generate rate limit
1555  generate_sandesh_rate_limit_fn(out, tsandesh);
1556 }
1557 
1558 void t_cpp_generator::generate_sandesh_objectlog_creators(ofstream &out,
1559  t_sandesh *tsandesh) {
1560  // First generate without sandesh object semantics
1561  bool generate_sandesh_object = false;
1562  bool generate_rate_limit = false;
1563  bool generate_system_log = false;
1564  // Generate send function and macros
1565  generate_sandesh_async_send_fn(out, tsandesh, generate_sandesh_object,
1566  generate_rate_limit, generate_system_log);
1567  generate_sandesh_async_send_macros(out, tsandesh, generate_sandesh_object);
1568  // Generate DropLog
1569  generate_sandesh_static_drop_logger(out, tsandesh,
1570  generate_sandesh_object);
1571  // Generate Log
1572  generate_sandesh_static_logger(out, tsandesh, generate_sandesh_object);
1573  // Next generate with sandesh object semantics
1574  generate_sandesh_object = true;
1575  // Generate create function and macros
1576  generate_sandesh_async_create_fn(out, tsandesh);
1577  generate_sandesh_async_create_macro(out, tsandesh);
1578  // Generate send function and macros
1579  generate_sandesh_async_send_fn(out, tsandesh, generate_sandesh_object,
1580  generate_rate_limit, generate_system_log);
1581  generate_sandesh_async_send_macros(out, tsandesh, generate_sandesh_object);
1582  // Generate DropLog
1583  generate_sandesh_static_drop_logger(out, tsandesh,
1584  generate_sandesh_object);
1585  // Generate Log
1586  generate_sandesh_static_logger(out, tsandesh, generate_sandesh_object);
1587 }
1588 
1589 void t_cpp_generator::generate_sandesh_session_log_unrolled_fn(ofstream &out,
1590  t_sandesh *tsandesh) {
1591  //To be removed
1592  out << indent() << "static void LogUnrolled";
1593  out << generate_sandesh_async_creator(tsandesh, true, false, false, "", "",
1594  true, false, false, false);
1595  out << ";" << endl;
1596 }
1597 
1598 void t_cpp_generator::generate_sandesh_session_adjust_session_end_point_objects_fn(ofstream &out,
1599  t_sandesh *tsandesh) {
1600  out << "static void adjust_session_end_point_objects(std::vector"
1601  " <SessionEndpoint> & session_data);" << endl;
1602  out << endl;
1603 }
1604 
1605 void t_cpp_generator::generate_sandesh_flow_send_fn(ofstream &out,
1606  t_sandesh *tsandesh) {
1607  std::string sender_func_name = "Send";
1608  std::string logger_func_name = "Log";
1609  // Generate sender
1610  out << indent() << "static void " << sender_func_name;
1611  out << generate_sandesh_async_creator(tsandesh, true, false, false, "", "",
1612  true, false, false, false);
1613  out << " {" << endl;
1614  indent_up();
1615  out << indent() << "if (HandleTest(level, category)) {" << endl;
1616  indent_up();
1617  out << indent() << "return;" << endl;
1618  scope_down(out);
1619  out << indent() << "if (IsSendingAllMessagesDisabled() ||" <<
1620  " IsSendingFlowsDisabled()) {" << endl;
1621  indent_up();
1622  out << indent() << "UpdateTxMsgFailStats(\"" << tsandesh->get_name() <<
1623  "\", 0, SandeshTxDropReason::SendingDisabled);" << endl;
1624  out << indent() << "if (IsLoggingDroppedAllowed(SandeshType::FLOW))" <<
1625  " {" << endl;
1626  indent_up();
1627  out << indent() << "Log" <<
1628  generate_sandesh_async_creator(tsandesh, false,
1629  false, false, "", "", false, false, false) << ";" << endl;
1630  scope_down(out);
1631  out << indent() << "return;" << endl;
1632  scope_down(out);
1633 
1634  out << indent() <<
1635  "if (is_send_slo_to_logger_enabled() || is_send_sampled_to_logger_enabled()) { " <<
1636  endl;
1637  indent_up();
1638  const t_type *t = tsandesh->get_type();
1639  if (((t_base_type *)t)->is_sandesh_session()) {
1640  out << indent() << "LogUnrolled(category, level, session_data);" << endl;
1641  }
1642  scope_down(out);
1643  out << indent() << "if (level >= SendingLevel()) {" << endl;
1644  indent_up();
1645  out << indent() << "UpdateTxMsgFailStats(\"" << tsandesh->get_name() <<
1646  "\", 0, SandeshTxDropReason::QueueLevel);" << endl;
1647  out << indent() << "if (IsLoggingDroppedAllowed(SandeshType::FLOW))" <<
1648  " {" << endl;
1649  indent_up();
1650  out << indent() << "std::string drop_reason = \"SANDESH: Queue Drop:"
1651  " \";" << endl;
1652  out << indent() << "DropLog" <<
1653  generate_sandesh_async_creator(tsandesh, false,
1654  false, false, "", "", false, false, true) << ";" << endl;
1655  scope_down(out);
1656  out << indent() << "return;" << endl;
1657  scope_down(out);
1658  out << indent() << tsandesh->get_name() <<
1659  " * snh = new " << tsandesh->get_name() <<
1660  generate_sandesh_no_static_const_string_function(tsandesh,
1661  false, false, false, false) << ";" << endl;
1662  if (((t_base_type *)t)->is_sandesh_session()) {
1663  out << indent() << "if (!is_send_sampled_to_collector_enabled() && !is_send_slo_to_collector_enabled()) {" << endl;
1664  indent_up();
1665  out << indent() << "return;" << endl;
1666  scope_down(out);
1667  out << indent() << " if (is_send_sampled_to_collector_enabled() != is_send_slo_to_collector_enabled()) {" << endl;
1668  indent_up();
1669  out << indent() << "adjust_session_end_point_objects(snh->session_data);";
1670  scope_down(out);
1671  }
1672  out << endl;
1673  out << indent() << "snh->set_level(level);" << endl;
1674  out << indent() << "snh->set_category(category);" << endl;
1675  out << indent() << "snh->Dispatch();" << endl;
1676  indent_down();
1677  indent(out) << "}" << endl << endl;
1678 }
1679 
1680 void t_cpp_generator::generate_sandesh_flow_creators(ofstream &out,
1681  t_sandesh *tsandesh) {
1682  bool generate_sandesh_object = false;
1683  const t_type *t = tsandesh->get_type();
1684  // Generate LogUnrolled declaration only for Session message
1685  if (((t_base_type *)t)->is_sandesh_session()) {
1686  generate_sandesh_session_log_unrolled_fn(out, tsandesh);
1687  generate_sandesh_session_adjust_session_end_point_objects_fn(out, tsandesh);
1688  }
1689  // Generate send function and macros
1690  generate_sandesh_flow_send_fn(out, tsandesh);
1691  generate_sandesh_async_send_macros(out, tsandesh, generate_sandesh_object);
1692  // Generate DropLog
1693  generate_sandesh_static_drop_logger(out, tsandesh,
1694  generate_sandesh_object);
1695  // Generate Log
1696  generate_sandesh_static_logger(out, tsandesh, generate_sandesh_object);
1697 }
1698 
1699 std::string t_cpp_generator::generate_sandesh_trace_creator(t_sandesh *tsandesh,
1700  bool signature, bool expand_autogen, bool skip_autogen,
1701  std::string prefix, std::string suffix) {
1702  string result = "";
1703  string temp = "";
1704  // Get members
1705  vector<t_field*>::const_iterator m_iter;
1706  const vector<t_field*>& members = tsandesh->get_members();
1707 
1708  result += "(";
1709  if (signature) {
1710  result += "SandeshTraceBufferPtr tracebuf";
1711  if (!skip_autogen) {
1712  result += ", std::string file, int32_t line";
1713  }
1714  } else {
1715  result += prefix + "trace_buf" + suffix;
1716  if (!skip_autogen) {
1717  if (expand_autogen) {
1718  result += ", __FILE__, __LINE__";
1719  } else {
1720  result += ", " + prefix + "file" + suffix + ", " +
1721  prefix + "line" + suffix;
1722  }
1723  }
1724  }
1725 
1726  for (m_iter = members.begin(); m_iter != members.end(); ++m_iter) {
1727  temp = "";
1728  t_type* t = get_true_type((*m_iter)->get_type());
1729  if (t->is_static_const_string()) {
1730  continue;
1731  }
1732  if ((*m_iter)->get_auto_generated()) {
1733  continue;
1734  }
1735  temp += ", ";
1736  if (signature) {
1737  result += temp;
1738  bool use_const = !(t->is_base_type() || t->is_enum()) || t->is_string();
1739  result += declare_field(*m_iter, false, false, use_const,
1740  !t->is_base_type() || t->is_string(), true);
1741  } else {
1742  result += temp;
1743  result += prefix + (*m_iter)->get_name() + suffix;
1744  }
1745  }
1746  result += ")";
1747  return result;
1748 }
1749 
1757 void t_cpp_generator::generate_sandesh_member_init_list(ofstream& out,
1758  t_sandesh *tsandesh, bool init_dval) {
1759  // Get members
1760  vector<t_field*>::const_iterator m_iter;
1761  const vector<t_field*>& members = tsandesh->get_members();
1762 
1763  out << " : ";
1764  generate_sandesh_base_init(out, tsandesh, init_dval);
1765 
1766  for (m_iter = members.begin(); m_iter != members.end(); ++m_iter) {
1767  t_type* t = get_true_type((*m_iter)->get_type());
1768  // If initializing with default values ignore complex types
1769  if (init_dval && (!t->is_base_type() && !t->is_enum())) {
1770  continue;
1771  }
1772  if (t->is_static_const_string()) {
1773  continue;
1774  }
1775  if (((t_base_type *)tsandesh->get_type())->is_sandesh_object() &&
1776  ((*m_iter)->get_req() == t_field::T_OPTIONAL)) {
1777  continue;
1778  }
1779  string dval;
1780  if (t->is_enum()) {
1781  dval += "(" + type_name(t) + ")";
1782  }
1783  if (t->is_string() || t->is_xml()) {
1784  dval += "\"\"";
1785  } else if (t->is_uuid()) {
1786  dval += "boost::uuids::nil_uuid()";
1787  } else if (t->is_ipaddr()) {
1788  dval += "";
1789  } else {
1790  dval += "0";
1791  }
1792  t_const_value* cv = (*m_iter)->get_value();
1793  if (cv != NULL) {
1794  dval = render_const_value(out, (*m_iter)->get_name(), t, cv);
1795  } else {
1796  if (!init_dval) {
1797  dval = (*m_iter)->get_name();
1798  }
1799  }
1800  out << ", " << (*m_iter)->get_name() << "(" << dval << ")";
1801  }
1802 
1803 }
1804 
1811 void t_cpp_generator::generate_sandesh_seqnum(ofstream& out,
1812  t_sandesh* tsandesh) {
1813  if (((t_base_type *)tsandesh->get_type())->is_sandesh_uve() ||
1814  ((t_base_type *)tsandesh->get_type())->is_sandesh_alarm()) {
1815  indent(out) << "static tbb::atomic<uint32_t> lseqnum_;" << endl;
1816  } else {
1817  indent(out) << "static uint32_t lseqnum_;" << endl;
1818  }
1819 }
1820 
1821 
1822 string generate_sandesh_base_name(t_sandesh* tsandesh, bool type) {
1823  const t_type *t = tsandesh->get_type();
1824  if (((t_base_type *)t)->is_sandesh_request()) {
1825  if (type) {
1826  return "SandeshType::REQUEST";
1827  } else {
1828  return "SandeshRequest";
1829  }
1830  } else if (((t_base_type *)t)->is_sandesh_response()) {
1831  if (type) {
1832  return "SandeshType::RESPONSE";
1833  } else {
1834  return "SandeshResponse";
1835  }
1836  } else if (((t_base_type *)t)->is_sandesh_uve()) {
1837  if (type) {
1838  return "SandeshType::UVE";
1839  } else {
1840  return "SandeshUVE";
1841  }
1842  } else if (((t_base_type *)t)->is_sandesh_alarm()) {
1843  if (type) {
1844  return "SandeshType::ALARM";
1845  } else {
1846  return "SandeshAlarm";
1847  }
1848  } else if (((t_base_type *)t)->is_sandesh_system()) {
1849  if (type) {
1850  return "SandeshType::SYSTEM";
1851  } else {
1852  return "SandeshSystem";
1853  }
1854  } else if (((t_base_type *)t)->is_sandesh_buffer()) {
1855  if (type) {
1856  return "SandeshType::BUFFER";
1857  } else {
1858  return "SandeshBuffer";
1859  }
1860  } else if (((t_base_type *)t)->is_sandesh_trace() ||
1861  ((t_base_type *)t)->is_sandesh_trace_object()) {
1862  if (type) {
1863  return "SandeshType::TRACE";
1864  } else {
1865  return "SandeshTrace";
1866  }
1867  } else if (((t_base_type *)t)->is_sandesh_object()) {
1868  if (type) {
1869  return "SandeshType::OBJECT";
1870  } else {
1871  return "SandeshObject";
1872  }
1873  } else if (((t_base_type *)t)->is_sandesh_flow()) {
1874  if (((t_base_type *)t)->is_sandesh_session()) {
1875  if (type) {
1876  return "SandeshType::SESSION";
1877  } else {
1878  return "SandeshFlowSession";
1879  }
1880  }
1881  else {
1882  if (type) {
1883  return "SandeshType::FLOW";
1884  } else {
1885  return "SandeshFlow";
1886  }
1887  }
1888  } else {
1889  return "";
1890  }
1891 }
1892 
1899 void t_cpp_generator::generate_sandesh_versionsig(ofstream& out,
1900  t_sandesh* tsandesh) {
1901  indent(out) << "static uint32_t versionsig_;" << endl;
1902 }
1903 
1904 void t_cpp_generator::generate_sandesh_base_init(
1905  ofstream& out, t_sandesh* tsandesh, bool init_dval) {
1906  out << generate_sandesh_base_name(tsandesh, false);
1907 
1908  if (init_dval) {
1909  out << "(\"" << tsandesh->get_name() << "\",lseqnum_++)";
1910  } else {
1911  out << "(\"" << tsandesh->get_name() << "\",seqno)";
1912  }
1913 }
1914 
1915 void t_cpp_generator::generate_sandesh_trace_seqnum_ctor(ofstream& out,
1916  t_sandesh* tsandesh) {
1917  indent(out) << "set_seqnum(0);" << endl;
1918 }
1919 
1926 void t_cpp_generator::generate_sandesh_context(ofstream& out,
1927  t_sandesh* tsandesh,
1928  string val) {
1929  indent(out) << "set_context(" << val << ");" << endl;
1930 }
1931 
1932 
1939 void t_cpp_generator::generate_sandesh_hints(ofstream& out,
1940  t_sandesh* tsandesh) {
1941  bool has_key_annotation = tsandesh->has_key_annotation();
1942  string shints = "0";
1943 
1944  if (has_key_annotation) {
1945  shints += " | g_sandesh_constants.SANDESH_KEY_HINT";
1946  }
1947  indent(out) << "set_hints(" << shints << ");" << endl;
1948 }
1949 
1956 void t_cpp_generator::generate_sandesh_default_ctor(ofstream& out,
1957  t_sandesh* tsandesh, bool is_request) {
1958  // Get members
1959  vector<t_field*>::const_iterator m_iter;
1960  const vector<t_field*>& members = tsandesh->get_members();
1961  // Generate default constructor
1962  indent(out) << tsandesh->get_name() << "()";
1963  generate_sandesh_member_init_list(out, tsandesh, true);
1964  out << " {" << endl;
1965  indent_up();
1966  // TODO(dreiss): When everything else in Thrift is perfect,
1967  // do more of these in the initializer list.
1968  for (m_iter = members.begin(); m_iter != members.end(); ++m_iter) {
1969  t_type* t = get_true_type((*m_iter)->get_type());
1970  if (!t->is_base_type()) {
1971  t_const_value* cv = (*m_iter)->get_value();
1972  if (cv != NULL) {
1973  print_const_value(out, (*m_iter)->get_name(), t, cv);
1974  }
1975  }
1976  }
1977  generate_sandesh_hints(out, tsandesh);
1978  if (is_request) {
1979  generate_sandesh_context(out, tsandesh, "\"\"");
1980  }
1981  scope_down(out);
1982 }
1983 
1984 
1991 void t_cpp_generator::generate_sandesh_definition(ofstream& out,
1992  t_sandesh* tsandesh) {
1993  const t_type *t = tsandesh->get_type();
1994  assert(t->is_base_type());
1995  bool is_trace = ((t_base_type *)t)->is_sandesh_trace() ||
1996  ((t_base_type *)t)->is_sandesh_trace_object();
1997  bool is_request = ((t_base_type *)t)->is_sandesh_request();
1998  bool is_response = ((t_base_type *)t)->is_sandesh_response();
1999  bool is_uve = ((t_base_type *)t)->is_sandesh_uve();
2000  bool is_alarm = ((t_base_type *)t)->is_sandesh_alarm();
2001  bool is_buffer = ((t_base_type *)t)->is_sandesh_buffer();
2002  bool is_system = ((t_base_type *)t)->is_sandesh_system();
2003  bool is_object = ((t_base_type *)t)->is_sandesh_object();
2004  bool is_flow = ((t_base_type *)t)->is_sandesh_flow();
2005  bool is_session = ((t_base_type *)t)->is_sandesh_session();
2006  string extends;
2007  if (is_request) {
2008  extends = " : public SandeshRequest";
2009  } else if (is_response) {
2010  extends = " : public SandeshResponse";
2011  } else if (is_uve) {
2012  extends = " : public SandeshUVE";
2013  } else if (is_alarm) {
2014  extends = " : public SandeshAlarm";
2015  } else if (is_system) {
2016  extends = " : public SandeshSystem";
2017  } else if (is_buffer) {
2018  extends = " : public SandeshBuffer";
2019  } else if (is_trace) {
2020  extends = " : public SandeshTrace";
2021  } else if (is_object) {
2022  extends = " : public SandeshObject";
2023  } else if (is_flow) {
2024  if (is_session) {
2025  extends = " : public SandeshFlowSession";
2026  } else {
2027  extends = " : public SandeshFlow";
2028  }
2029  }
2030 
2031  // Get members
2032  vector<t_field*>::const_iterator m_iter;
2033  const vector<t_field*>& members = tsandesh->get_members();
2034 
2035  // Write the isset structure declaration outside the class. This makes
2036  // the generated code amenable to processing by SWIG.
2037  // We only declare the struct if it gets used in the class.
2038 
2039  // Isset struct has boolean fields, but only for non-required fields.
2040  bool has_nonrequired_fields = false;
2041  for (m_iter = members.begin(); m_iter != members.end(); ++m_iter) {
2042  if ((*m_iter)->get_req() != t_field::T_REQUIRED)
2043  has_nonrequired_fields = true;
2044  }
2045 
2046  if (has_nonrequired_fields) {
2047 
2048  out <<
2049  indent() << "typedef struct _" << tsandesh->get_name() << "__isset {" << endl;
2050  indent_up();
2051 
2052  indent(out) <<
2053  "_" << tsandesh->get_name() << "__isset() ";
2054  bool first = true;
2055  for (m_iter = members.begin(); m_iter != members.end(); ++m_iter) {
2056  if ((*m_iter)->get_req() == t_field::T_REQUIRED) {
2057  continue;
2058  }
2059  if (first) {
2060  first = false;
2061  out <<
2062  ": " << (*m_iter)->get_name() << "(false)";
2063  } else {
2064  out <<
2065  ", " << (*m_iter)->get_name() << "(false)";
2066  }
2067  }
2068  out << " {}" << endl;
2069 
2070  for (m_iter = members.begin(); m_iter != members.end(); ++m_iter) {
2071  if ((*m_iter)->get_req() != t_field::T_REQUIRED) {
2072  indent(out) <<
2073  "bool " << (*m_iter)->get_name() << ";" << endl;
2074  }
2075  }
2076 
2077  indent_down();
2078  indent(out) <<
2079  "} _" << tsandesh->get_name() << "__isset;" << endl;
2080  }
2081 
2082  out << endl;
2083 
2084  // Open sandesh def
2085  out << indent() << "class " << tsandesh->get_name() << extends << " {" << endl;
2086 
2087  // Public members
2088  out << indent() << " public:" << endl;
2089  indent_up();
2090  generate_sandesh_fingerprint(out, tsandesh, false);
2091 
2092  // Create a getter and setter function for all fields except
2093  // static const string
2094  for (m_iter = members.begin(); m_iter != members.end(); ++m_iter) {
2095  if (((*m_iter)->get_type())->is_static_const_string()) {
2096  continue;
2097  }
2098  out << endl << indent() <<
2099  "void set_" << (*m_iter)->get_name() <<
2100  "(" << type_name((*m_iter)->get_type(), false, true);
2101  out << " val) {" << endl << indent() <<
2102  indent() << (*m_iter)->get_name() << " = val;" << endl;
2103 
2104  // assume all fields are required except optional fields.
2105  // for optional fields change __isset.name to true
2106  bool is_optional = (*m_iter)->get_req() == t_field::T_OPTIONAL;
2107  if (is_optional) {
2108  out << indent() << indent() << "__isset." <<
2109  (*m_iter)->get_name() << " = true;" << endl;
2110  }
2111  out << indent() << "}" << endl;
2112 
2113  out << endl << indent() <<
2114  type_name((*m_iter)->get_type(), false, true)
2115  << " get_" << (*m_iter)->get_name() <<
2116  "() const {" << endl;
2117  out << indent() << indent() << "return " <<
2118  (*m_iter)->get_name() << ";" << endl;
2119  out << indent() << "}" << endl;
2120  }
2121  out << endl;
2122 
2123  // Generate an equality testing operator. Make it inline since the compiler
2124  // will do a better job than we would when deciding whether to inline it.
2125  out <<
2126  indent() << "bool operator == (const " << tsandesh->get_name() << " & " <<
2127  (members.size() > 0 ? "rhs" : "/* rhs */") << ") const" << endl;
2128  scope_up(out);
2129  for (m_iter = members.begin(); m_iter != members.end(); ++m_iter) {
2130  // Most existing Thrift code does not use isset or optional/required,
2131  // so we treat "default" fields as required.
2132  if ((*m_iter)->get_req() != t_field::T_OPTIONAL) {
2133  out <<
2134  indent() << "if (!(" << (*m_iter)->get_name()
2135  << " == rhs." << (*m_iter)->get_name() << "))" << endl <<
2136  indent() << " return false;" << endl;
2137  } else {
2138  out <<
2139  indent() << "if (__isset." << (*m_iter)->get_name()
2140  << " != rhs.__isset." << (*m_iter)->get_name() << ")" << endl <<
2141  indent() << " return false;" << endl <<
2142  indent() << "else if (__isset." << (*m_iter)->get_name() << " && !("
2143  << (*m_iter)->get_name() << " == rhs." << (*m_iter)->get_name()
2144  << "))" << endl <<
2145  indent() << " return false;" << endl;
2146  }
2147  }
2148  indent(out) << "return true;" << endl;
2149  scope_down(out);
2150  out <<
2151  indent() << "bool operator != (const " << tsandesh->get_name() << " &rhs) const {" << endl <<
2152  indent() << " return !(*this == rhs);" << endl <<
2153  indent() << "}" << endl << endl;
2154 
2155  // Destructor
2156  if (tsandesh->annotations_.find("final") == tsandesh->annotations_.end()) {
2157  out << endl << indent() << "virtual ~" <<
2158  tsandesh->get_name() << "() {}" << endl << endl;
2159  }
2160 
2161  // Add the __isset data member if we need it, using the definition from above
2162  if (has_nonrequired_fields) {
2163  out << endl << indent() << "_" << tsandesh->get_name() <<
2164  "__isset __isset;" << endl;
2165  }
2166 
2167  out << indent() << "virtual std::string ModuleName() const { " <<
2168  "return \"" << program_name_ << "\"; }" << endl << endl;
2169 
2170  // Is this a Sandesh Request ?
2171  if (is_request) {
2172  // Generate default constructor
2173  generate_sandesh_default_ctor(out, tsandesh, true);
2174  // Request registration
2175  indent(out) << "SANDESH_REGISTER_DEC_TYPE(" << tsandesh->get_name() <<
2176  ");" << endl;
2177  out << indent() << "virtual void HandleRequest() const;" << endl;
2178  indent(out) << "virtual bool RequestFromHttp(" <<
2179  "const std::string& ctx, const std::string& snh_query);" << endl;
2180 
2181  // Generate creator
2182  out << indent() << "static void Request" <<
2183  generate_sandesh_no_static_const_string_function(tsandesh, true, true, false, is_request, true) <<
2184  " {" << endl;
2185  indent_up();
2186  out << indent() << tsandesh->get_name() <<
2187  " * snh = new " << tsandesh->get_name() <<
2188  generate_sandesh_no_static_const_string_function(tsandesh, false, false, false, is_request) <<
2189  ";" << endl;
2190  out << indent() << "snh->Dispatch(sconn);" << endl;
2191  indent_down();
2192  indent(out) << "}" << endl << endl;
2193  } else if (is_response) {
2194  // Sandesh response
2195  // Generate default constructor
2196  generate_sandesh_default_ctor(out, tsandesh, false);
2197  // Generate creator
2198  string creator_func_name = "Response";
2199  out << indent() << "void " << creator_func_name <<
2200  "() { Dispatch(); }" << endl;
2201  } else if (is_uve || is_alarm) {
2202  const vector<t_field*>& fields = tsandesh->get_members();
2203  vector<t_field*>::const_iterator f_iter = fields.begin();
2204  assert((*f_iter)->get_name() == "data");
2205 
2206  bool is_proxy = false;
2207  std::map<std::string, std::string>::iterator ait;
2208  ait = ((*f_iter)->get_type())->annotations_.find("timeout");
2209  if (ait != ((*f_iter)->get_type())->annotations_.end()) {
2210  is_proxy = true;
2211  }
2212 
2213  if (is_proxy) {
2214  indent(out) << "static void Send(const " << type_name((*f_iter)->get_type()) <<
2215  "& data, std::string table = \"\", uint64_t mono_usec=0, int partition=-1);" << endl;
2216  indent(out) << "static void Send(const " << type_name((*f_iter)->get_type()) <<
2217  "& data, SandeshLevel::type Xlevel = SandeshLevel::SYS_NOTICE, " <<
2218  "std::string table = \"\", uint64_t mono_usec=0, int partition=-1);" << endl;
2219  std::map<std::string, std::string>::iterator pit;
2220  pit = ((*f_iter)->get_type())->annotations_.find("period");
2221  indent(out) << "static const uint64_t kProxyPeriod_us = " <<
2222  pit->second.c_str() << "000000;" << endl;
2223 
2224  } else {
2225  indent(out) << "static void Send(const " << type_name((*f_iter)->get_type()) <<
2226  "& data, std::string table = \"\", uint64_t mono_usec=0);" << endl;
2227  indent(out) << "static void Send(const " << type_name((*f_iter)->get_type()) <<
2228  "& data, SandeshLevel::type Xlevel, " <<
2229  "std::string table = \"\", uint64_t mono_usec=0);" << endl;
2230  }
2231 
2232  indent(out) << "static void Send(const " << type_name((*f_iter)->get_type()) <<
2233  "& cdata, SandeshUVE::SendType stype, uint32_t seqno," <<
2234  " uint32_t cycle, std::string ctx = \"\");" << endl;
2235  indent(out) << "static void Send(const " << type_name((*f_iter)->get_type()) <<
2236  "& cdata, SandeshLevel::type Xlevel, SandeshUVE::SendType stype, " <<
2237  "uint32_t seqno, uint32_t cycle, std::string ctx = \"\");" << endl;
2238  } else if (is_system) {
2239  generate_sandesh_systemlog_creators(out, tsandesh);
2240  } else if (is_flow) {
2241  generate_sandesh_flow_creators(out, tsandesh);
2242  } else if (is_object) {
2243  generate_sandesh_objectlog_creators(out, tsandesh);
2244  } else if (is_trace) {
2245  // Sandesh trace
2246  out << indent() << "virtual void SendTrace(" <<
2247  "const std::string& tcontext, bool more) {" << endl;
2248  indent_up();
2249  indent(out) << "set_context(tcontext);" << endl;
2250  indent(out) << "set_more(more);" << endl;
2251  out << indent() << tsandesh->get_name() <<
2252  " * snh = new " << tsandesh->get_name() << "(*this);" << endl;
2253  indent(out) << "snh->Dispatch();" << endl;
2254  indent_down();
2255  indent(out) << "}" << endl << endl;
2256  // Generate Trace function and Macro
2257  string creator_func_name = "TraceMsg";
2258  out << endl << indent() << "static void " << creator_func_name <<
2259  generate_sandesh_trace_creator(tsandesh, true, false, false, "", "") <<
2260  ";" << endl << endl;
2261  string creator_macro_name = tsandesh->get_name() + "Trace";
2262  string creator_name_usc = underscore(creator_macro_name);
2263  string creator_name_uc = uppercase(creator_name_usc);
2264  out << indent() << "#define " << creator_name_uc <<
2265  generate_sandesh_trace_creator(tsandesh, false, false, true, "_", "") <<
2266  "\\" << endl;
2267  indent_up();
2268  out << indent() << tsandesh->get_name() << "::" << creator_func_name <<
2269  generate_sandesh_trace_creator(tsandesh, false, true, false, "(_", ")") <<
2270  endl;
2271  indent_down();
2272  } else if (is_buffer) {
2273  // Sandesh kernel
2274  // Generate default constructor
2275  generate_sandesh_default_ctor(out, tsandesh, false);
2276  // Request registration
2277  indent(out) << "SANDESH_REGISTER_DEC_TYPE(" << tsandesh->get_name() <<
2278  ");" << endl << endl;
2279  }
2280 
2281  //Generate versionsig return function
2282  out << indent() << "virtual const int32_t versionsig() const { return versionsig_;}" << endl;
2283  out << indent() << "static const int32_t sversionsig() { return versionsig_;}" << endl;
2284 
2285  if (!is_trace) {
2286  out << indent() << "static int32_t lseqnum() { return lseqnum_;}" << endl;
2287  }
2288 
2289  if (is_uve || is_alarm) {
2290  const vector<t_field*>& fields = tsandesh->get_members();
2291  vector<t_field*>::const_iterator f_iter = fields.begin();
2292  assert((*f_iter)->get_name() == "data");
2293  string dtype = type_name((*f_iter)->get_type());
2294  out << indent() <<
2295  "static std::map<std::string, std::string> _DSConf(void);" << endl;
2296  out << indent() << "static void _InitDerivedStats(" << dtype <<
2297  " & _data, const map<string,string> & _dsconf);" << endl;
2298  out << indent() << "static bool UpdateUVE(" << dtype <<
2299  " & _data, " << dtype <<
2300  " & tdata, uint64_t mono_usec, SandeshLevel::type Xlevel);" << endl;
2301  out << indent() << "bool LoadUVE(SendType stype, uint32_t cycle);" << endl;
2302  }
2303 
2304  out << indent() << "std::string ToString() const;" << endl;
2305 
2306  out << indent() << "size_t GetSize() const;" << endl;
2307 
2308  // Private members
2309  out << endl;
2310  out << "private:" << endl << endl;
2311  if (((t_base_type *)t)->is_sandesh_buffer()) {
2312  indent(out) << "virtual void Process(SandeshContext *context);" << endl << endl;
2313  } else if (!((t_base_type *)t)->is_sandesh_response()) {
2314  // Explicit constructor
2315  indent(out) << "explicit " << tsandesh->get_name();
2316  out << generate_sandesh_no_static_const_string_function(tsandesh, true, false, false, is_request);
2317  generate_sandesh_member_init_list(out, tsandesh, false);
2318  out << " {" << endl;
2319  indent_up();
2320  // TODO(dreiss): When everything else in Thrift is perfect,
2321  // do more of these in the initializer list.
2322  for (m_iter = members.begin(); m_iter != members.end(); ++m_iter) {
2323  t_type* t = get_true_type((*m_iter)->get_type());
2324  if (!t->is_base_type()) {
2325  t_const_value* cv = (*m_iter)->get_value();
2326  if (cv != NULL) {
2327  print_const_value(out, (*m_iter)->get_name(), t, cv);
2328  }
2329  }
2330  }
2331  if (is_trace) {
2332  generate_sandesh_trace_seqnum_ctor(out, tsandesh);
2333  }
2334  if (is_request) {
2335  generate_sandesh_context(out, tsandesh, "context");
2336  out << indent() << "if (context == \"ctrl\") " <<
2337  "set_hints(g_sandesh_constants.SANDESH_CONTROL_HINT);" << endl;
2338  out << indent() << "else set_hints(0);" << endl;
2339  out << indent() << "(void)sconn;" << endl;
2340 
2341  } else {
2342  generate_sandesh_hints(out, tsandesh);
2343  }
2344  scope_down(out);
2345  }
2346 
2347  // Create emplty constructor since objectlogs can have optional fields
2348  if (((t_base_type *)t)->is_sandesh_object()) {
2349  out << endl << indent() << "explicit " << tsandesh->get_name()
2350  << "(uint32_t seqno," " std::string file, int32_t line) : ";
2351  generate_sandesh_base_init(out, tsandesh, false);
2352  out << ", file(file), line(line) {" << endl;
2353  indent_up();
2354  generate_sandesh_hints(out, tsandesh);
2355  scope_down(out);
2356  }
2357 
2358  // Declare all fields
2359  for (m_iter = members.begin(); m_iter != members.end(); ++m_iter) {
2360  indent(out) << declare_field(*m_iter, false, false, false) << endl;
2361  }
2362 
2363  if (!is_trace) {
2364  generate_sandesh_seqnum(out, tsandesh);
2365  }
2366 
2367  generate_sandesh_versionsig(out, tsandesh);
2368 
2369  out << indent() << "static const char *name_;" << endl;
2370 
2371  out << indent() << "void Log() const;" << endl;
2372 
2373  out << indent() << "void ForcedLog() const;" << endl;
2374 
2375  out << indent() << "int32_t Read(" <<
2376  "boost::shared_ptr<contrail::sandesh::protocol::TProtocol> iprot);" << endl;
2377 
2378  out << indent() << "int32_t Write(" <<
2379  "boost::shared_ptr<contrail::sandesh::protocol::TProtocol> oprot) const;" << endl;
2380 
2381  if (((t_base_type *)t)->is_sandesh_system()) {
2382  out << indent() << "static bool do_rate_limit_drop_log_;" << endl;
2383 
2384  out << indent() << "static boost::circular_buffer<time_t>"
2385  " rate_limit_buffer_;" << endl;
2386 
2387  out << indent() << "static tbb::mutex rate_limit_mutex_;" << endl;
2388  }
2389 
2390  out << endl;
2391  indent_down();
2392  indent(out) << "};" << endl << endl;
2393 }
2394 
2395 void t_cpp_generator::generate_sandesh_fingerprint(ofstream& out,
2396  t_sandesh* tsandesh,
2397  bool is_definition) {
2398  string stat, nspace, comment;
2399  if (is_definition) {
2400  stat = "";
2401  nspace = tsandesh->get_name() + "::";
2402  comment = " ";
2403  } else {
2404  stat = "static ";
2405  nspace = "";
2406  comment = "; // ";
2407  }
2408 
2409  if (tsandesh->has_fingerprint()) {
2410  out <<
2411  indent() << stat << "const char* " << nspace
2412  << "ascii_fingerprint" << comment << "= \"" <<
2413  tsandesh->get_ascii_fingerprint() << "\";" << endl <<
2414  indent() << stat << "const uint8_t " << nspace <<
2415  "binary_fingerprint[" << t_type::fingerprint_len << "]" << comment << "= {";
2416  const char* comma = "";
2417  for (int i = 0; i < t_type::fingerprint_len; i++) {
2418  out << comma << "0x" << t_struct::byte_to_hex(tsandesh->get_binary_fingerprint()[i]);
2419  comma = ",";
2420  }
2421  out << "};" << endl << endl;
2422  }
2423 }
2424 
2425 void t_cpp_generator::cache_attr_info(t_struct* tstruct,
2426  std::map<string, CacheAttribute> &attrs) {
2427  attrs.clear();
2428  const vector<t_field*>& members = tstruct->get_members();
2429  vector<t_field*>::const_iterator m_iter;
2430  for (m_iter = members.begin(); m_iter != members.end(); ++m_iter) {
2431  std::map<std::string, std::string>::iterator jt;
2432  jt = (*m_iter)->annotations_.find("hidden");
2433  if ((*m_iter)->get_req()!=t_field::T_OPTIONAL) {
2434  // Mandatory fields cannot have DerivedStats,
2435  // or be of hidden/periodic type
2436  assert(jt == (*m_iter)->annotations_.end());
2437  assert((*m_iter)->annotations_.find("stats") == (*m_iter)->annotations_.end());
2438  assert((*m_iter)->annotations_.find("mstats") == (*m_iter)->annotations_.end());
2439  attrs.insert(std::make_pair((*m_iter)->get_name(),
2440  MANDATORY));
2441  continue;
2442  }
2443  // DervivedStats attributes are not handled here
2444  if (((*m_iter)->annotations_.find("stats") != (*m_iter)->annotations_.end()) ||
2445  ((*m_iter)->annotations_.find("mstats") != (*m_iter)->annotations_.end())) {
2446  continue;
2447  }
2448  // All raw attributes are Non-periodic
2449  if (jt!=(*m_iter)->annotations_.end()) {
2450  attrs.insert(std::make_pair((*m_iter)->get_name(), HIDDEN));
2451  } else {
2452  attrs.insert(std::make_pair((*m_iter)->get_name(), INLINE));
2453  }
2454  }
2455 }
2456 
2457 void t_cpp_generator::derived_stats_info(t_struct* tstruct,
2458  // val is rawtype,resulttype,algo,annotation,compattr,subcompattr
2459  map<string,DSInfo> & dsmap,
2460  // val is set of ds attributes
2461  map<string,set<string> >& rawmap) {
2462  const vector<t_field*>& members = tstruct->get_members();
2463  vector<t_field*>::const_iterator m_iter;
2464  for (m_iter = members.begin(); m_iter != members.end(); ++m_iter) {
2465  std::map<std::string, std::string>::iterator jt,mt;
2466  jt = (*m_iter)->annotations_.find("stats");
2467  mt = (*m_iter)->annotations_.find("mstats");
2468  if ((jt != (*m_iter)->annotations_.end()) ||
2469  (mt != (*m_iter)->annotations_.end())) {
2470  assert((*m_iter)->get_req() == t_field::T_OPTIONAL);
2471 
2472  t_type* retype = get_true_type((*m_iter)->get_type());
2473  string restype;
2474 
2475  bool is_ds_map = false;
2476  // for map derived-stats
2477  if (mt!= (*m_iter)->annotations_.end()) {
2478  // out of mstats and stats, only one should be specified
2479  assert(jt == (*m_iter)->annotations_.end());
2480  is_ds_map = true;
2481  assert(retype->is_map());
2482  t_type* vtype = ((t_map*)retype)->get_val_type();
2483  restype = type_name(get_true_type(vtype));
2484  } else {
2485  restype = type_name(get_true_type((*m_iter)->get_type()));
2486  }
2487 
2488  const string &tstr = ((jt == (*m_iter)->annotations_.end()) ?
2489  mt->second : jt->second);
2490  size_t pos = tstr.find(':');
2491 
2492  string residual = tstr.substr(pos+1, string::npos);
2493  size_t rpos = residual.find(':');
2494  string algo, anno;
2495  if (rpos == string::npos) {
2496  // No Derived Stats arguments are present
2497  algo = residual;
2498  anno = string();
2499  } else {
2500  algo = residual.substr(0,rpos);
2501  anno = residual.substr(rpos+1, string::npos);
2502  }
2503 
2504  string rawfullattr;
2505  string rawperiodattr = tstr.substr(0,pos);
2506  size_t ppos = rawperiodattr.find("-");
2507  int period = -1;
2508  string prealgo;
2509  if (ppos != string::npos) {
2510  string periodpre = rawperiodattr.substr(0,ppos);
2511  size_t prepos = periodpre.find(".");
2512  if (prepos != string::npos) {
2513  period = atoi(periodpre.substr(0,prepos).c_str());
2514  prealgo = periodpre.substr(prepos+1, string::npos);
2515  } else {
2516  period = atoi(rawperiodattr.substr(0,ppos).c_str());
2517  }
2518 
2519  rawfullattr = rawperiodattr.substr(ppos+1, string::npos);
2520  } else {
2521  rawfullattr = rawperiodattr;
2522  }
2523 
2524  size_t cpos = rawfullattr.find('.');
2525  string rawattr;
2526  string compattr("");
2527  string subcompattr("");
2528 
2529  // Does the raw attribute have a composite underneath it?
2530  if (cpos != string::npos) {
2531  rawattr = rawfullattr.substr(0,cpos);
2532  string rescomp = rawfullattr.substr(cpos+1, string::npos);
2533  size_t upos = rescomp.find('.');
2534  if (upos != string::npos) {
2535  compattr = rescomp.substr(0,upos);
2536  subcompattr = rescomp.substr(upos+1, string::npos);
2537  } else {
2538  compattr = rescomp;
2539  }
2540  } else {
2541  rawattr = rawfullattr;
2542  }
2543 
2544  string rawtype;
2545  RawMetric rmt = RM_DIAL;
2546  vector<t_field*>::const_iterator s_iter;
2547  for (s_iter = members.begin(); s_iter != members.end(); ++s_iter) {
2548  if (rawattr.compare((*s_iter)->get_name())==0) {
2549  map<string,string>::const_iterator cit =
2550  (*s_iter)->annotations_.find("metric");
2551  if (cit != (*s_iter)->annotations_.end()) {
2552  if (cit->second.compare("agg") == 0) rmt = RM_AGG;
2553  else if (cit->second.compare("diff") == 0) rmt = RM_DIFF;
2554  }
2555  t_type* ratype = get_true_type((*s_iter)->get_type());
2556  t_type* vtype = ratype;
2557  if (is_ds_map) {
2558  assert(ratype->is_map());
2559  vtype = ((t_map*)ratype)->get_val_type();
2560  }
2561  rawtype = type_name(vtype);
2562  if (!compattr.empty()) {
2563  // The raw attribute is a composite/subcomposite
2564  // Find info for the composite attribute.
2565  assert(vtype->is_struct());
2566  rawtype = string("");
2567  const vector<t_field*>& cmembers = ((t_struct*)vtype)->get_members();
2568  vector<t_field*>::const_iterator c_iter;
2569  for (c_iter = cmembers.begin(); c_iter != cmembers.end(); ++c_iter) {
2570  if (compattr.compare((*c_iter)->get_name())==0) {
2571  t_type* catype = get_true_type((*c_iter)->get_type());
2572  if (!subcompattr.empty()) {
2573  // The raw attribute is a subcomposite
2574  assert(catype->is_struct());
2575  const vector<t_field*>& umem =
2576  ((t_struct*)catype)->get_members();
2577  vector<t_field*>::const_iterator u_iter;
2578  for (u_iter = umem.begin(); u_iter != umem.end(); ++u_iter) {
2579  if (subcompattr.compare((*u_iter)->get_name())==0) {
2580  t_type* uatype = get_true_type((*u_iter)->get_type());
2581  rawtype = type_name(uatype);
2582  break;
2583  }
2584  }
2585 
2586  } else {
2587  rawtype = type_name(catype);
2588  }
2589  break;
2590  }
2591  }
2592  }
2593  break;
2594  }
2595  }
2596  assert(!rawtype.empty());
2597 
2598  bool is_hidden =
2599  (*m_iter)->annotations_.find("hidden") != (*m_iter)->annotations_.end();
2600 
2601  CacheAttribute cat = INLINE;
2602  if (tstruct->annotations_.find("period") != tstruct->annotations_.end()) {
2603  // By default, DS attributes in periodic structs have period 1
2604  if (period==-1) period=1;
2605  if (is_hidden) {
2606  if (period!=0) cat = HIDDEN_PER;
2607  else cat = HIDDEN;
2608  } else {
2609  if (period!=0) cat = PERIODIC;
2610  // else this DS stays INLINE
2611  }
2612  } else {
2613  // DS attributes in non-periodic structs must have period 0
2614  if (period==-1) period=0;
2615  if (is_hidden) {
2616  cat = HIDDEN;
2617  }
2618  // else this DS stays INLINE
2619  assert(period==0);
2620  }
2621 
2622  DSInfo dsi(is_ds_map, period, cat, rawtype, rmt,
2623  restype, algo, anno, compattr, subcompattr, prealgo);
2624 
2625  // map of derived stats
2626  dsmap.insert(make_pair((*m_iter)->get_name(), dsi));
2627 
2628  // map of raw attributes
2629  map<string,set<string> >::iterator r_iter = rawmap.find(rawattr);
2630  if (r_iter != rawmap.end()) {
2631  r_iter->second.insert((*m_iter)->get_name());
2632  } else {
2633  set<string> dss;
2634  dss.insert((*m_iter)->get_name());
2635  rawmap.insert(make_pair(rawattr,dss));
2636  }
2637  }
2638  }
2639 }
2640 #endif
2647 void t_cpp_generator::generate_struct_definition(ofstream& out,
2648  t_struct* tstruct,
2649  bool is_exception,
2650  bool pointers,
2651  bool read,
2652  bool write) {
2653  string extends = "";
2654  if (is_exception) {
2655  extends = " : public ::contrail::sandesh::TException";
2656  }
2657 
2658  // Get members
2659  vector<t_field*>::const_iterator m_iter;
2660  const vector<t_field*>& members = tstruct->get_members();
2661 
2662  // Write the isset structure declaration outside the class. This makes
2663  // the generated code amenable to processing by SWIG.
2664  // We only declare the struct if it gets used in the class.
2665 
2666  // Isset struct has boolean fields, but only for non-required fields.
2667  bool has_nonrequired_fields = false;
2668  for (m_iter = members.begin(); m_iter != members.end(); ++m_iter) {
2669  if ((*m_iter)->get_req() != t_field::T_REQUIRED)
2670  has_nonrequired_fields = true;
2671  }
2672 
2673  bool del_support = false;
2674  bool proxy_support = false;
2675  if (has_nonrequired_fields && (!pointers || read)) {
2676 
2677  out <<
2678  indent() << "typedef struct _" << tstruct->get_name() << "__isset {" << endl;
2679  indent_up();
2680 
2681  indent(out) <<
2682  "_" << tstruct->get_name() << "__isset() ";
2683  bool first = true;
2684  for (m_iter = members.begin(); m_iter != members.end(); ++m_iter) {
2685  if ((*m_iter)->get_req() == t_field::T_REQUIRED) {
2686  continue;
2687  }
2688  t_type* t = get_true_type((*m_iter)->get_type());
2689  if (t->is_base_type() &&
2690  ((*m_iter)->get_name().compare("deleted") == 0)) {
2691  t_base_type::t_base tbase = ((t_base_type*)t)->get_base();
2692  if (tbase == t_base_type::TYPE_BOOL) del_support = true;
2693  }
2694  if (t->is_base_type() &&
2695  ((*m_iter)->get_name().compare("proxy") == 0)) {
2696  t_base_type::t_base tbase = ((t_base_type*)t)->get_base();
2697  if (tbase == t_base_type::TYPE_STRING) proxy_support = true;
2698  }
2699 
2700  if (first) {
2701  first = false;
2702  out <<
2703  ": " << (*m_iter)->get_name() << "(false)";
2704  } else {
2705  out <<
2706  ", " << (*m_iter)->get_name() << "(false)";
2707  }
2708  }
2709  out << " {}" << endl;
2710 
2711  for (m_iter = members.begin(); m_iter != members.end(); ++m_iter) {
2712  if ((*m_iter)->get_req() != t_field::T_REQUIRED) {
2713  indent(out) <<
2714  "bool " << (*m_iter)->get_name() << ";" << endl;
2715  }
2716  }
2717 
2718  indent_down();
2719  indent(out) <<
2720  "} _" << tstruct->get_name() << "__isset;" << endl << endl;
2721 
2722  }
2723 
2724  out << endl;
2725 
2726  // Open struct def
2727  out <<
2728  indent() << "class " << tstruct->get_name() << extends << " {" << endl <<
2729  indent() << " public:" << endl <<
2730  endl;
2731  indent_up();
2732 
2733  // Put the fingerprint up top for all to see.
2734  generate_struct_fingerprint(out, tstruct, false);
2735 
2736 #ifdef SANDESH
2737  bool is_table = false;
2738 
2739  std::map<string, CacheAttribute> cache_attrs;
2740  cache_attr_info(tstruct, cache_attrs);
2741 
2742  map<string,DSInfo> dsinfo;
2743  map<string,set<string> > rawmap;
2744  derived_stats_info(tstruct, dsinfo, rawmap);
2745  map<string,DSInfo >::iterator ds_iter;
2746 #endif
2747 
2748  if (!pointers) {
2749  // Default constructor
2750 #ifdef SANDESH
2751  for (m_iter = members.begin(); m_iter != members.end(); ++m_iter) {
2752  if ((*m_iter)->get_name() == "name") {
2753  std::map<std::string, std::string>::iterator it;
2754  it = (*m_iter)->annotations_.find("key");
2755  if (it != (*m_iter)->annotations_.end()) {
2756  is_table = true;
2757  indent(out) << tstruct->get_name() << "(std::string __tbl = \"" <<
2758  it->second << "\")";
2759  break;
2760  }
2761  }
2762  }
2763  if (!is_table) {
2764  indent(out) <<
2765  tstruct->get_name() << "()";
2766  }
2767 #else
2768  indent(out) <<
2769  tstruct->get_name() << "()";
2770 #endif
2771 
2772  bool init_ctor = false;
2773 
2774  for (m_iter = members.begin(); m_iter != members.end(); ++m_iter) {
2775  t_type* t = get_true_type((*m_iter)->get_type());
2776 #ifdef SANDESH
2777  if (t->is_static_const_string()) {
2778  continue;
2779  }
2780 #endif
2781  if (t->is_base_type() || t->is_enum()) {
2782  string dval;
2783  if (t->is_enum()) {
2784  dval += "(" + type_name(t) + ")";
2785  }
2786  if (t->is_string() || t->is_xml()) {
2787  dval += "\"\"";
2788  } else if (t->is_uuid()) {
2789  dval += "boost::uuids::nil_uuid()";
2790  } else if (t->is_ipaddr()) {
2791  dval += "";
2792  } else {
2793  dval += "0";
2794  }
2795  t_const_value* cv = (*m_iter)->get_value();
2796  if (cv != NULL) {
2797  dval = render_const_value(out, (*m_iter)->get_name(), t, cv);
2798  }
2799  if (!init_ctor) {
2800  init_ctor = true;
2801  out << " : ";
2802  out << (*m_iter)->get_name() << "(" << dval << ")";
2803  } else {
2804  out << ", " << (*m_iter)->get_name() << "(" << dval << ")";
2805  }
2806  }
2807  }
2808 
2809 #ifdef SANDESH
2810  if (is_table) {
2811  out << ", table_(__tbl)";
2812  }
2813 #endif
2814  out << " {" << endl;
2815  indent_up();
2816  // TODO(dreiss): When everything else in Thrift is perfect,
2817  // do more of these in the initializer list.
2818  for (m_iter = members.begin(); m_iter != members.end(); ++m_iter) {
2819  t_type* t = get_true_type((*m_iter)->get_type());
2820 
2821  if (!t->is_base_type()) {
2822  t_const_value* cv = (*m_iter)->get_value();
2823  if (cv != NULL) {
2824  print_const_value(out, (*m_iter)->get_name(), t, cv);
2825  }
2826  }
2827  }
2828  scope_down(out);
2829  }
2830 
2831  if (tstruct->annotations_.find("final") == tstruct->annotations_.end()) {
2832  out <<
2833  endl <<
2834  indent() << "virtual ~" << tstruct->get_name() << "() throw() {}" << endl << endl;
2835  }
2836 
2837  // Pointer to this structure's reflection local typespec.
2838  if (gen_dense_) {
2839  indent(out) <<
2840  "static ::contrail::sandesh::reflection::local::TypeSpec* local_reflection;" <<
2841  endl << endl;
2842  }
2843 
2844  // Declare all fields
2845  for (m_iter = members.begin(); m_iter != members.end(); ++m_iter) {
2846  indent(out) <<
2847  declare_field(*m_iter, false, pointers && !(*m_iter)->get_type()->is_xception(), !read) << endl;
2848  }
2849 #ifdef SANDESH
2850  for (ds_iter = dsinfo.begin(); ds_iter != dsinfo.end(); ++ds_iter) {
2851 
2852  // val is rawtype,resulttype,algo,annotation,compattr,subcompattr
2853  if ((ds_iter->second.cat_ == PERIODIC) ||
2854  (ds_iter->second.cat_ == HIDDEN_PER)) {
2855  if (ds_iter->second.prealgo_.empty()) {
2856  indent(out) << "boost::shared_ptr< ::contrail::sandesh::DerivedStatsPeriodicIf< ::contrail::sandesh::" <<
2857  ds_iter->second.algo_ << ", " <<
2858  ds_iter->second.rawtype_ << ", " <<
2859  ds_iter->second.resulttype_.substr(0, ds_iter->second.resulttype_.size() - 3) << ", " <<
2860  ds_iter->second.resulttype_ <<
2861  "> > __dsobj_" << ds_iter->first << ";" << endl;
2862  } else {
2863  indent(out) << "boost::shared_ptr< ::contrail::sandesh::DerivedStatsPeriodicAnomalyIf< ::contrail::sandesh::" <<
2864  ds_iter->second.algo_ << ", " <<
2865  ds_iter->second.rawtype_ << ", ::contrail::sandesh::" <<
2866  ds_iter->second.prealgo_ << ", " <<
2867  ds_iter->second.resulttype_ <<
2868  "> > __dsobj_" << ds_iter->first << ";" << endl;
2869  }
2870  } else {
2871  indent(out) << "boost::shared_ptr< ::contrail::sandesh::DerivedStatsIf< ::contrail::sandesh::" <<
2872  ds_iter->second.algo_ << ", " <<
2873  ds_iter->second.rawtype_ << ", " << ds_iter->second.resulttype_ <<
2874  "> > __dsobj_" << ds_iter->first << ";" << endl;
2875  }
2876 
2877  }
2878  if (is_table) {
2879  indent(out) << "std::string table_;" << endl;
2880  }
2881 
2882 #endif
2883 
2884  // Add the __isset data member if we need it, using the definition from above
2885  if (has_nonrequired_fields && (!pointers || read)) {
2886  out <<
2887  endl <<
2888  indent() << "_" << tstruct->get_name() << "__isset __isset;" << endl;
2889  }
2890 
2891  // Create a setter and getter function for each field except static const
2892  // string
2893  vector<string> dskeys;
2894  for (m_iter = members.begin(); m_iter != members.end(); ++m_iter) {
2895  if (pointers) {
2896  continue;
2897  }
2898 #ifdef SANDESH
2899  if (((*m_iter)->get_type())->is_static_const_string()) {
2900  continue;
2901  }
2902  std::map<std::string, std::string>::iterator jt;
2903  jt = (*m_iter)->annotations_.find("aggtype");
2904  if (jt != (*m_iter)->annotations_.end()) {
2905  if (jt->second.compare("listkey")==0) {
2906  dskeys.push_back((*m_iter)->get_name());
2907  }
2908  }
2909 #endif
2910  out <<
2911  endl <<
2912 #ifdef SANDESH
2913  indent() << (dsinfo.find((*m_iter)->get_name()) == dsinfo.end() ?
2914  "void set_" : "void __set_") << (*m_iter)->get_name() <<
2915 #else
2916  indent() << "void __set_" << (*m_iter)->get_name() <<
2917 #endif
2918  "(" << type_name((*m_iter)->get_type(), false, true);
2919  out << " val) {" << endl << indent() <<
2920  indent() << (*m_iter)->get_name() << " = val;" << endl;
2921 
2922  // assume all fields are required except optional fields.
2923  // for optional fields change __isset.name to true
2924  bool is_optional = (*m_iter)->get_req() == t_field::T_OPTIONAL;
2925  if (is_optional) {
2926  out <<
2927  indent() <<
2928  indent() << "__isset." << (*m_iter)->get_name() << " = true;" << endl;
2929  }
2930  out <<
2931  indent()<< "}" << endl;
2932 
2933 #ifdef SANDESH
2934  out << endl << indent() << type_name((*m_iter)->get_type(), false, true)
2935  << " get_";
2936  out << (*m_iter)->get_name() << "() const {" << endl;
2937  out << indent() << indent() << "return " << (*m_iter)->get_name() << ";" << endl;
2938  out << indent() << "}" << endl;
2939 #endif
2940  }
2941  out << endl;
2942 
2943  if (!pointers) {
2944  out <<
2945  indent() << "bool operator == (long " <<
2946  (members.size() > 0 ? "rhs" : "/* rhs */") << ") const" << endl;
2947  scope_up(out);
2948  for (m_iter = members.begin(); m_iter != members.end(); ++m_iter) {
2949  // Most existing Thrift code does not use isset or optional/required,
2950  // so we treat "default" fields as required.
2951  t_type* type = get_true_type((*m_iter)->get_type());
2952  if (type->is_base_type()) {
2953  t_base_type::t_base tbase = ((t_base_type*)type)->get_base();
2954  switch (tbase) {
2955  case t_base_type::TYPE_BYTE:
2956  case t_base_type::TYPE_I16:
2957  case t_base_type::TYPE_I32:
2958  case t_base_type::TYPE_I64:
2959  case t_base_type::TYPE_U16:
2960  case t_base_type::TYPE_U32:
2961  case t_base_type::TYPE_U64:
2962  case t_base_type::TYPE_BOOL:
2963  if ((*m_iter)->get_req() != t_field::T_OPTIONAL) {
2964  out <<
2965  indent() << "if (!((long)" << (*m_iter)->get_name()
2966  << " == rhs))" << endl <<
2967  indent() << " return false;" << endl;
2968  } else {
2969  out <<
2970  indent() << "if (__isset." << (*m_iter)->get_name()
2971  << " && !((long)"
2972  << (*m_iter)->get_name() << " == rhs))" << endl <<
2973  indent() << " return false;" << endl;
2974  }
2975  break;
2976  default:
2977  out << indent() << "return false;" << endl;
2978  continue;
2979  }
2980  }
2981  }
2982  indent(out) << "return true;" << endl;
2983  scope_down(out);
2984  // Generate an equality testing operator. Make it inline since the compiler
2985  // will do a better job than we would when deciding whether to inline it.
2986  out <<
2987  indent() << "bool operator == (const " << tstruct->get_name() << " & " <<
2988  (members.size() > 0 ? "rhs" : "/* rhs */") << ") const" << endl;
2989  scope_up(out);
2990  for (m_iter = members.begin(); m_iter != members.end(); ++m_iter) {
2991  // Most existing Thrift code does not use isset or optional/required,
2992  // so we treat "default" fields as required.
2993  if ((*m_iter)->get_req() != t_field::T_OPTIONAL) {
2994  out <<
2995  indent() << "if (!(" << (*m_iter)->get_name()
2996  << " == rhs." << (*m_iter)->get_name() << "))" << endl <<
2997  indent() << " return false;" << endl;
2998  } else {
2999  out <<
3000  indent() << "if (__isset." << (*m_iter)->get_name()
3001  << " != rhs.__isset." << (*m_iter)->get_name() << ")" << endl <<
3002  indent() << " return false;" << endl <<
3003  indent() << "else if (__isset." << (*m_iter)->get_name() << " && !("
3004  << (*m_iter)->get_name() << " == rhs." << (*m_iter)->get_name()
3005  << "))" << endl <<
3006  indent() << " return false;" << endl;
3007  }
3008  }
3009  indent(out) << "return true;" << endl;
3010  scope_down(out);
3011  out <<
3012  indent() << "bool operator != (const " << tstruct->get_name() << " &rhs) const {" << endl <<
3013  indent() << " return !(*this == rhs);" << endl <<
3014  indent() << "}" << endl << endl;
3015 
3016  // Generate the declaration of a less-than operator. This must be
3017  // implemented by the application developer if they wish to use it. (They
3018  // will get a link error if they try to use it without an implementation.)
3019  out <<
3020  indent() << "bool operator < (const "
3021  << tstruct->get_name() << " & ) const;" << endl << endl;
3022  }
3023 
3024  if (read) {
3025  if (gen_templates_) {
3026  out <<
3027  indent() << "template <class Protocol_>" << endl <<
3028  indent() << "uint32_t read(Protocol_* iprot);" << endl;
3029  } else {
3030  out <<
3031  indent() << "int32_t read(" <<
3032  "boost::shared_ptr<contrail::sandesh::protocol::TProtocol> iprot);" << endl;
3033  }
3034  }
3035  if (write) {
3036  if (gen_templates_) {
3037  out <<
3038  indent() << "template <class Protocol_>" << endl <<
3039  indent() << "uint32_t write(Protocol_* oprot) const;" << endl;
3040  } else {
3041  out <<
3042  indent() << "int32_t write(" <<
3043  "boost::shared_ptr<contrail::sandesh::protocol::TProtocol> oprot) const;" << endl;
3044  }
3045  }
3046 #ifdef SANDESH
3047  out << indent() << "std::string log() const;" << endl;
3048  if (!pointers) {
3049  out << indent() <<
3050  tstruct->get_name() << " operator+(const " << tstruct->get_name() <<
3051  "& right) const" << endl;
3052  scope_up(out);
3053  out << indent() << tstruct->get_name() << " result;" << endl;
3054  for (m_iter = members.begin(); m_iter != members.end(); ++m_iter) {
3055  std::string nn = (*m_iter)->get_name();
3056  ds_iter = dsinfo.find((*m_iter)->get_name());
3057  std::string sn;
3058  if (ds_iter == dsinfo.end()) sn = std::string("set_");
3059  else sn = std::string("__set_");
3060 
3061  t_type* type = get_true_type((*m_iter)->get_type());
3062  if (type->is_base_type()) {
3063  t_base_type::t_base tbase = ((t_base_type*)type)->get_base();
3064  switch (tbase) {
3065  case t_base_type::TYPE_BYTE:
3066  case t_base_type::TYPE_I16:
3067  case t_base_type::TYPE_I32:
3068  case t_base_type::TYPE_I64:
3069  case t_base_type::TYPE_DOUBLE:
3070  case t_base_type::TYPE_U16:
3071  case t_base_type::TYPE_U32:
3072  case t_base_type::TYPE_U64:
3073  if ((*m_iter)->get_req() != t_field::T_OPTIONAL) {
3074  out <<
3075  indent() << "result." << sn << nn << "(get_" << nn <<
3076  "() + right.get_" << nn << "());" << endl << endl;
3077  } else {
3078  out <<
3079  indent() << "if ((__isset." << nn << ") && (right.__isset." <<
3080  nn << ")) result." << sn << nn << "(get_" << nn <<
3081  "() + right.get_" << nn << "());" << endl <<
3082  indent() << "else if (__isset." << nn << ") result." << sn <<
3083  nn << "(get_" << nn << "());" << endl <<
3084  indent() << "else if (right.__isset." << nn << ") result." << sn <<
3085  nn << "(right.get_" << nn << "());" << endl << endl;
3086  }
3087  default:
3088  continue;
3089  }
3090  }
3091  }
3092  indent(out) << "return result;" << endl;
3093  scope_down(out);
3094 
3095  out << indent() <<
3096  tstruct->get_name() << " operator/(int div) const" << endl;
3097  scope_up(out);
3098  out << indent() << tstruct->get_name() << " result;" << endl;
3099  for (m_iter = members.begin(); m_iter != members.end(); ++m_iter) {
3100  std::string nn = (*m_iter)->get_name();
3101  ds_iter = dsinfo.find((*m_iter)->get_name());
3102  std::string sn;
3103  if (ds_iter == dsinfo.end()) sn = std::string("set_");
3104  else sn = std::string("__set_");
3105 
3106  t_type* type = get_true_type((*m_iter)->get_type());
3107  if (type->is_base_type()) {
3108  t_base_type::t_base tbase = ((t_base_type*)type)->get_base();
3109  switch (tbase) {
3110  case t_base_type::TYPE_BYTE:
3111  case t_base_type::TYPE_I16:
3112  case t_base_type::TYPE_I32:
3113  case t_base_type::TYPE_I64:
3114  case t_base_type::TYPE_DOUBLE:
3115  case t_base_type::TYPE_U16:
3116  case t_base_type::TYPE_U32:
3117  case t_base_type::TYPE_U64:
3118  if ((*m_iter)->get_req() != t_field::T_OPTIONAL) {
3119  out <<
3120  indent() << "result." << sn << nn << "(get_" << nn <<
3121  "() / div);" << endl << endl;
3122  } else {
3123  out <<
3124  indent() << "if (__isset." << nn << ") result." << sn << nn << "(get_" <<
3125  nn << "() / div);" << endl << endl;
3126  }
3127  default:
3128  continue;
3129  }
3130  }
3131  }
3132  indent(out) << "return result;" << endl;
3133  scope_down(out);
3134 
3135  out << indent() <<
3136  tstruct->get_name() << " operator-(const " << tstruct->get_name() <<
3137  "& right) const" << endl;
3138  scope_up(out);
3139  out << indent() << tstruct->get_name() << " result;" << endl;
3140  for (m_iter = members.begin(); m_iter != members.end(); ++m_iter) {
3141  std::string nn = (*m_iter)->get_name();
3142  ds_iter = dsinfo.find((*m_iter)->get_name());
3143  std::string sn;
3144  if (ds_iter == dsinfo.end()) sn = std::string("set_");
3145  else sn = std::string("__set_");
3146  t_type* type = get_true_type((*m_iter)->get_type());
3147  if (type->is_base_type()) {
3148  t_base_type::t_base tbase = ((t_base_type*)type)->get_base();
3149  switch (tbase) {
3150  case t_base_type::TYPE_BYTE:
3151  case t_base_type::TYPE_I16:
3152  case t_base_type::TYPE_I32:
3153  case t_base_type::TYPE_I64:
3154  case t_base_type::TYPE_DOUBLE:
3155  case t_base_type::TYPE_U16:
3156  case t_base_type::TYPE_U32:
3157  case t_base_type::TYPE_U64:
3158  if ((*m_iter)->get_req() != t_field::T_OPTIONAL) {
3159  out <<
3160  indent() << "result." << sn << nn << "(get_" << nn <<
3161  "() - right.get_" << nn << "());" << endl << endl;
3162  } else {
3163  out <<
3164  indent() << "if ((__isset." << nn << ") && (right.__isset." <<
3165  nn << ")) {" << endl <<
3166  indent() << " result." << sn << nn << "(get_" << nn <<
3167  "() - right.get_" << nn << "()); }" << endl <<
3168  indent() << "else if (__isset." << nn << ") result." << sn <<
3169  nn << "(get_" << nn << "());" << endl;
3170  }
3171  default:
3172  continue;
3173  }
3174  }
3175  }
3176  indent(out) << "return result;" << endl;
3177  scope_down(out);
3178  }
3179  out << indent() << "size_t GetSize() const;" << endl;
3180  out << indent() << "std::string __listkey(void) {" << endl;
3181  out << indent() << " std::string __result;" << endl;
3182  bool first = true;
3183  vector<string>::iterator kit;
3184  for (kit = dskeys.begin(); kit != dskeys.end(); kit++) {
3185  if (!first) {
3186  out << indent() << " __result.append(\":\");" << endl;
3187  }
3188  out << indent() << " __result += " << *kit << ";" << endl;
3189  first = false;
3190  }
3191  out << indent() << " return __result;" << endl;
3192  out << indent() << "}" << endl;
3193 #endif
3194  out << endl;
3195 
3196  indent_down();
3197  indent(out) <<
3198  "};" << endl <<
3199  endl;
3200 
3201  if (del_support) {
3202  out <<
3203  indent() << "template <>" << endl <<
3204  indent() << "struct SandeshStructDeleteTrait<" <<
3205  tstruct->get_name() << "> {" << endl <<
3206  indent() << " static bool get(const " <<
3207  tstruct->get_name() << "& s) { return s.get_deleted(); }" <<
3208  endl <<
3209  indent() << "};" << endl;
3210  }
3211  if (proxy_support) {
3212  out <<
3213  indent() << "template <>" << endl <<
3214  indent() << "struct SandeshStructProxyTrait<" <<
3215  tstruct->get_name() << "> {" << endl <<
3216  indent() << " static std::string get(const " <<
3217  tstruct->get_name() << "& s) { return s.get_proxy(); }" <<
3218  endl <<
3219  indent() << "};" << endl;
3220  }
3221 }
3222 
3229 void t_cpp_generator::generate_struct_fingerprint(ofstream& out,
3230  t_struct* tstruct,
3231  bool is_definition) {
3232  string stat, nspace, comment;
3233  if (is_definition) {
3234  stat = "";
3235  nspace = tstruct->get_name() + "::";
3236  comment = " ";
3237  } else {
3238  stat = "static ";
3239  nspace = "";
3240  comment = "; // ";
3241  }
3242 
3243  if (tstruct->has_fingerprint()) {
3244  out <<
3245  indent() << stat << "const char* " << nspace
3246  << "ascii_fingerprint" << comment << "= \"" <<
3247  tstruct->get_ascii_fingerprint() << "\";" << endl <<
3248  indent() << stat << "const uint8_t " << nspace <<
3249  "binary_fingerprint[" << t_type::fingerprint_len << "]" << comment << "= {";
3250  const char* comma = "";
3251  for (int i = 0; i < t_type::fingerprint_len; i++) {
3252  out << comma << "0x" << t_struct::byte_to_hex(tstruct->get_binary_fingerprint()[i]);
3253  comma = ",";
3254  }
3255  out << "};" << endl << endl;
3256  }
3257 }
3258 
3262 void t_cpp_generator::generate_local_reflection(std::ofstream& out,
3263  t_type* ttype,
3264  bool is_definition) {
3265  if (!gen_dense_) {
3266  return;
3267  }
3268  ttype = get_true_type(ttype);
3269  assert(ttype->has_fingerprint());
3270  string key = ttype->get_ascii_fingerprint() + (is_definition ? "-defn" : "-decl");
3271  // Note that we have generated this fingerprint. If we already did, bail out.
3272  if (!reflected_fingerprints_.insert(key).second) {
3273  return;
3274  }
3275  // Let each program handle its own structures.
3276  if (ttype->get_program() != NULL && ttype->get_program() != program_) {
3277  return;
3278  }
3279 
3280  // Do dependencies.
3281  if (ttype->is_list()) {
3282  generate_local_reflection(out, ((t_list*)ttype)->get_elem_type(), is_definition);
3283  } else if (ttype->is_set()) {
3284  generate_local_reflection(out, ((t_set*)ttype)->get_elem_type(), is_definition);
3285  } else if (ttype->is_map()) {
3286  generate_local_reflection(out, ((t_map*)ttype)->get_key_type(), is_definition);
3287  generate_local_reflection(out, ((t_map*)ttype)->get_val_type(), is_definition);
3288  } else if (ttype->is_struct() || ttype->is_xception()) {
3289  // Hacky hacky. For efficiency and convenience, we need a dummy "T_STOP"
3290  // type at the end of our typespec array. Unfortunately, there is no
3291  // T_STOP type, so we use the global void type, and special case it when
3292  // generating its typespec.
3293 
3294  const vector<t_field*>& members = ((t_struct*)ttype)->get_sorted_members();
3295  vector<t_field*>::const_iterator m_iter;
3296  for (m_iter = members.begin(); m_iter != members.end(); ++m_iter) {
3297  generate_local_reflection(out, (**m_iter).get_type(), is_definition);
3298  }
3299  generate_local_reflection(out, g_type_void, is_definition);
3300 
3301  // For definitions of structures, do the arrays of metas and field specs also.
3302  if (is_definition) {
3303  out <<
3304  indent() << "::contrail::sandesh::reflection::local::FieldMeta" << endl <<
3305  indent() << local_reflection_name("metas", ttype) <<"[] = {" << endl;
3306  indent_up();
3307  for (m_iter = members.begin(); m_iter != members.end(); ++m_iter) {
3308  indent(out) << "{ " << (*m_iter)->get_key() << ", " <<
3309  (((*m_iter)->get_req() == t_field::T_OPTIONAL) ? "true" : "false") <<
3310  " }," << endl;
3311  }
3312  // Zero for the T_STOP marker.
3313  indent(out) << "{ 0, false }" << endl << "};" << endl;
3314  indent_down();
3315 
3316  out <<
3317  indent() << "::contrail::sandesh::reflection::local::TypeSpec*" << endl <<
3318  indent() << local_reflection_name("specs", ttype) <<"[] = {" << endl;
3319  indent_up();
3320  for (m_iter = members.begin(); m_iter != members.end(); ++m_iter) {
3321  indent(out) << "&" <<
3322  local_reflection_name("typespec", (*m_iter)->get_type(), true) << "," << endl;
3323  }
3324  indent(out) << "&" <<
3325  local_reflection_name("typespec", g_type_void) << "," << endl;
3326  indent_down();
3327  indent(out) << "};" << endl;
3328  }
3329  }
3330 
3331  out <<
3332  indent() << "// " << ttype->get_fingerprint_material() << endl <<
3333  indent() << (is_definition ? "" : "extern ") <<
3334  "::contrail::sandesh::reflection::local::TypeSpec" << endl <<
3335  local_reflection_name("typespec", ttype) <<
3336  (is_definition ? "(" : ";") << endl;
3337 
3338  if (!is_definition) {
3339  out << endl;
3340  return;
3341  }
3342 
3343  indent_up();
3344 
3345  if (ttype->is_void()) {
3346  indent(out) << "::contrail::sandesh::protocol::T_STOP";
3347  } else {
3348  indent(out) << type_to_enum(ttype);
3349  }
3350 
3351  if (ttype->is_struct()) {
3352  out << "," << endl <<
3353  indent() << type_name(ttype) << "::binary_fingerprint," << endl <<
3354  indent() << local_reflection_name("metas", ttype) << "," << endl <<
3355  indent() << local_reflection_name("specs", ttype);
3356  } else if (ttype->is_list()) {
3357  out << "," << endl <<
3358  indent() << "&" << local_reflection_name("typespec", ((t_list*)ttype)->get_elem_type(), true) << "," << endl <<
3359  indent() << "NULL";
3360  } else if (ttype->is_set()) {
3361  out << "," << endl <<
3362  indent() << "&" << local_reflection_name("typespec", ((t_set*)ttype)->get_elem_type(), true) << "," << endl <<
3363  indent() << "NULL";
3364  } else if (ttype->is_map()) {
3365  out << "," << endl <<
3366  indent() << "&" << local_reflection_name("typespec", ((t_map*)ttype)->get_key_type(), true) << "," << endl <<
3367  indent() << "&" << local_reflection_name("typespec", ((t_map*)ttype)->get_val_type(), true);
3368  }
3369 
3370  out << ");" << endl << endl;
3371 
3372  indent_down();
3373 }
3374 
3379 void t_cpp_generator::generate_local_reflection_pointer(std::ofstream& out,
3380  t_type* ttype) {
3381  if (!gen_dense_) {
3382  return;
3383  }
3384  indent(out) <<
3385  "::contrail::sandesh::reflection::local::TypeSpec* " <<
3386  ttype->get_name() << "::local_reflection = " << endl <<
3387  indent() << " &" << local_reflection_name("typespec", ttype) << ";" <<
3388  endl << endl;
3389 }
3390 
3397 void t_cpp_generator::generate_struct_reader(ofstream& out,
3398  t_struct* tstruct,
3399  bool pointers) {
3400  if (gen_templates_) {
3401  out <<
3402  indent() << "template <class Protocol_>" << endl <<
3403  indent() << "int32_t " << tstruct->get_name() <<
3404  "::read(Protocol_* iprot) {" << endl;
3405  } else {
3406  indent(out) <<
3407  "int32_t " << tstruct->get_name() <<
3408  "::read(boost::shared_ptr<contrail::sandesh::protocol::TProtocol> iprot) {" << endl;
3409  }
3410 
3411  generate_common_struct_reader_body(out, tstruct, pointers);
3412 
3413  indent(out) <<
3414  "}" << endl << endl;
3415 }
3416 
3423 void t_cpp_generator::generate_struct_writer(ofstream& out,
3424  t_struct* tstruct,
3425  bool pointers) {
3426  if (gen_templates_) {
3427  out <<
3428  indent() << "template <class Protocol_>" << endl <<
3429  indent() << "int32_t " << tstruct->get_name() <<
3430  "::write(Protocol_* oprot) const {" << endl;
3431  } else {
3432  indent(out) <<
3433  "int32_t " << tstruct->get_name() <<
3434  "::write(boost::shared_ptr<contrail::sandesh::protocol::TProtocol> oprot) const {" << endl;
3435  }
3436 
3437  generate_common_struct_writer_body(out, tstruct, pointers);
3438 
3439  indent(out) <<
3440  "}" << endl <<
3441  endl;
3442 }
3443 
3444 #ifdef SANDESH
3451 void t_cpp_generator::generate_struct_get_size(ofstream& out, const string& name,
3452  const vector<t_field*>& fields) {
3453  //Generate GetSize function to return size of sandesh
3454  indent(out) << "size_t " << name <<
3455  "::GetSize() const {" << endl;
3456  indent_up();
3457  indent(out) << "size_t size = 0;" << endl;
3458  vector<t_field*>::const_iterator f_iter;
3459  for (f_iter = fields.begin(); f_iter != fields.end(); ++f_iter) {
3460  generate_get_size_field(out, *f_iter);
3461  }
3462  indent(out) << "return size;" << endl;
3463  indent_down();
3464  indent(out) << "}" << endl << endl;
3465 }
3466 
3473 void t_cpp_generator::generate_struct_logger(ofstream& out, const string& name,
3474  const vector<t_field*>& fields) {
3475  indent(out) << "std::string " << name <<
3476  "::log() const {" << endl;
3477  indent_up();
3478  out << indent() << "std::stringstream Xbuf;" << endl;
3479  vector<t_field*>::const_iterator f_iter;
3480  bool init = false;
3481  for (f_iter = fields.begin(); f_iter != fields.end(); ++f_iter) {
3482  string prefix = "\" \" << ";
3483  if (!init) {
3484  init = true;
3485  prefix = "";
3486  }
3487  generate_logger_field(out, *f_iter, prefix, false, false);
3488  }
3489  if (!init) {
3490  out << indent() << "return std::string();" << endl;
3491  } else {
3492  out << indent() << "return Xbuf.str();" << endl;
3493  }
3494  indent_down();
3495  indent(out) << "}" << endl << endl;
3496 }
3497 
3504 void t_cpp_generator::generate_static_const_string_definition(ofstream& out,
3505  string name,
3506  const vector<t_field*>& fields) {
3507  vector<t_field*>::const_iterator f_iter;
3508  for (f_iter = fields.begin(); f_iter != fields.end(); ++f_iter) {
3509  if (!((*f_iter)->get_type())->is_static_const_string()) {
3510  continue;
3511  }
3512  t_const_value* cv = (*f_iter)->get_value();
3513  assert(cv != NULL);
3514  out << base_type_name(t_base_type::TYPE_STRING) << " " << name <<
3515  "::";
3516  print_const_value(out, (*f_iter)->get_name(), (*f_iter)->get_type(), cv);
3517  }
3518 }
3519 
3526 void t_cpp_generator::generate_static_const_string_definition(ofstream& out,
3527  t_sandesh* tsandesh) {
3528  generate_static_const_string_definition(out, tsandesh->get_name(),
3529  tsandesh->get_sorted_members());
3530 }
3537 void t_cpp_generator::generate_static_const_string_definition(ofstream& out,
3538  t_struct* tstruct) {
3539  generate_static_const_string_definition(out, tstruct->get_name(),
3540  tstruct->get_sorted_members());
3541 }
3542 
3543 void t_cpp_generator::generate_sandesh_static_seqnum_def(ofstream& out,
3544  t_sandesh* tsandesh) {
3545  if (((t_base_type *)tsandesh->get_type())->is_sandesh_uve() ||
3546  ((t_base_type *)tsandesh->get_type())->is_sandesh_alarm()) {
3547  out << "tbb::atomic<uint32_t> " << tsandesh->get_name() << "::lseqnum_;" << endl;
3548  } else {
3549  out << "uint32_t " << tsandesh->get_name() << "::lseqnum_ = 1;" << endl;
3550  }
3551 }
3552 
3553 void t_cpp_generator::generate_sandesh_static_versionsig_def(ofstream& out,
3554  t_sandesh* tsandesh) {
3555  out << "uint32_t " << tsandesh->get_name() << "::versionsig_ = "
3556  << tsandesh->get_4byte_fingerprint() << "U;" << endl << endl;
3557 }
3558 
3559 void t_cpp_generator::generate_sandesh_static_rate_limit_buffer_def(
3560  ofstream& out, t_sandesh* tsandesh) {
3561  out << "boost::circular_buffer<time_t> " << tsandesh->get_name() <<
3562  "::rate_limit_buffer_(Sandesh::get_send_rate_limit());" << endl << endl;
3563 }
3564 
3565 void t_cpp_generator::generate_sandesh_static_rate_limit_mutex_def(
3566  ofstream& out,t_sandesh* tsandesh) {
3567  out << "tbb::mutex " << tsandesh->get_name() << "::rate_limit_mutex_; "
3568  << endl << endl;
3569 }
3570 
3571 void t_cpp_generator::generate_sandesh_static_rate_limit_log_def(
3572  ofstream& out, t_sandesh* tsandesh) {
3573  out << "bool " << tsandesh->get_name() << "::do_rate_limit_drop_log_ = true;"
3574  << endl << endl;
3575 }
3576 
3577 
3584 void t_cpp_generator::generate_sandesh_http_reader(ofstream& out,
3585  t_sandesh* tsandesh) {
3586  indent(out) << "#include <boost/tokenizer.hpp>" << endl;
3587  indent(out) << "#include <base/util.h>" << endl << endl;
3588  indent(out) << "bool " << tsandesh->get_name() << "::RequestFromHttp" <<
3589  "(const std::string& ctx, const std::string& snh_query) {" << endl;
3590 
3591  indent_up();
3592 
3593  const vector<t_field*>& fields = tsandesh->get_members();
3594  vector<t_field*>::const_iterator f_iter;
3595 
3596  indent(out) << program_name_ << "_marker = 0;" << endl;
3597  // Declare stack tmp variables
3598  out << endl <<
3599  indent() << "using std::string;" << endl <<
3600  indent() << "boost::char_separator<char> sep(\"&\");" << endl <<
3601  indent() << "boost::char_separator<char> varsep(\"=\");" << endl <<
3602  endl <<
3603  indent() << "CURL * cr = curl_easy_init();" <<
3604  indent() << "boost::tokenizer<boost::char_separator<char> >" <<
3605  "tokens(snh_query, sep);" << endl <<
3606  indent() << "boost::tokenizer<boost::char_separator<char> >" <<
3607  "::iterator it1 = tokens.begin();" << endl << endl;
3608 
3609  indent(out) << "string entity_name;" << endl;
3610  indent(out) << "int fields_vector_size =" << fields.size() << ";" << endl;
3611 
3612 
3613  // Required variables aren't in __isset, so we need tmp vars to check them.
3614  indent(out) << "for(int i =0; i < fields_vector_size; i++)" << endl;
3615  scope_up(out);
3616  indent(out) << "if (it1 != tokens.end())" << endl;
3617  scope_up(out);
3618  indent(out) << "string tok = (*it1++).c_str();" << endl ;
3619  indent(out) << "boost::tokenizer<boost::char_separator<char> >" <<
3620  " var(tok, varsep);" << endl;
3621  indent(out) << "boost::tokenizer<boost::char_separator<char> >" <<
3622  "::iterator it2 = var.begin();" << endl << endl;
3623 
3624  indent(out) << "string tok_new = (*it2).c_str();" << endl;
3625  indent(out) << "if (it2 != var.end()&& ++it2 != var.end()) " << endl;
3626  scope_up(out);
3627 
3628 
3629  indent(out) << "char *unescaped = NULL;" << endl;
3630  indent(out) << "unescaped = (char*) (malloc(sizeof(char)*1024));" <<endl;
3631 
3632  indent(out) << "std::string tmpstr;" << endl;
3633 
3634  for (f_iter = fields.begin(); f_iter != fields.end(); ++f_iter)
3635  {
3636  t_type *ftype = (*f_iter)->get_type();
3637  if (ftype->is_base_type())
3638  {
3639  t_base_type *btype = static_cast<t_base_type *>(ftype);
3640  if (btype->is_string() || btype->is_xml())
3641  {
3642  indent(out) << "entity_name=\"" << (*f_iter)->get_name() << "\";" << endl;
3643  indent(out) << "if((tok_new.substr(0,(entity_name.length()))).compare(entity_name) == 0)" << endl;
3644  scope_up(out);
3645  indent(out) << "unescaped = curl_easy_unescape(cr, (*it2).c_str(), 0, NULL);" << endl;
3646  indent(out) << "tmpstr=unescaped;" << endl;
3647  indent(out) << "set_" << (*f_iter)->get_name() << "(boost::lexical_cast<std::string>(tmpstr));" << endl;
3648  indent(out) << "curl_free(unescaped);" << endl;
3649  // delete[] unescaped;
3650  indent(out) << "continue;" << endl;
3651  scope_down(out);
3652  }
3653  else if (btype->is_uuid())
3654  {
3655  indent(out) << "entity_name=\"" << (*f_iter)->get_name() << "\";" << endl;
3656  indent(out) << "if((tok_new.substr(0,(entity_name.length()))).compare(entity_name) == 0)" << endl;
3657  scope_up(out);
3658  indent(out) << "std::stringstream ss;" << endl;
3659  indent(out) << "ss << *it2;" << endl;
3660  indent(out) << type_name((*f_iter)->get_type()) << " tmpval;" << endl;
3661  indent(out) << "ss >> tmpval;" << endl;
3662  indent(out) << "set_" << (*f_iter)->get_name() << "(tmpval);" << endl;
3663  indent(out) << "continue;" << endl;
3664  scope_down(out);
3665  }
3666  else if (btype->is_ipaddr())
3667  {
3668  indent(out) << "entity_name=\"" << (*f_iter)->get_name() << "\";" << endl;
3669  indent(out) << "if((tok_new.substr(0,(entity_name.length()))).compare(entity_name) == 0)" << endl;
3670  scope_up(out);
3671  indent(out) << "boost::system::error_code ec;" << endl;
3672  indent(out) << type_name((*f_iter)->get_type()) << " tmpval;" << endl;
3673  indent(out) << "tmpval = boost::asio::ip::address::from_string((*it2), ec);" << endl;
3674  indent(out) << "set_" << (*f_iter)->get_name() << "(tmpval);" << endl;
3675  indent(out) << "continue;" << endl;
3676  scope_down(out);
3677  }
3678  else
3679  {
3680  indent(out) << "entity_name=\"" << (*f_iter)->get_name() << "\";" << endl;
3681  indent(out) << "if((tok_new.substr(0,(entity_name.length()))).compare(entity_name) == 0)" << endl;
3682  scope_up(out);
3683  assert(btype->is_integer());
3684  indent(out) << type_name((*f_iter)->get_type()) << " tmpval;" << endl;
3685  indent(out) << "stringToInteger((*it2), tmpval);" << endl;
3686  indent(out) << "set_" << (*f_iter)->get_name() << "(tmpval);" << endl;
3687  indent(out) << "continue;" << endl;
3688  scope_down(out);
3689  }
3690  }
3691  else
3692  {
3693  // Ignore this field
3694  indent(out) << "++it2;" << endl;
3695  }
3696  }
3697  vector<t_field*>::const_iterator f_iter2 = fields.begin();
3698  if (f_iter2 != fields.end())
3699  {
3700  t_type *ftype = (*f_iter2)->get_type();
3701  if (ftype->is_base_type())
3702  {
3703  t_base_type *btype = static_cast<t_base_type *>(ftype);
3704  if (btype->is_string() || btype->is_xml())
3705  {
3706  indent(out) << "entity_name=\"" << (*f_iter2)->get_name() << "\";" << endl;
3707  indent(out) << "unescaped = curl_easy_unescape(cr, (*it2).c_str(), 0, NULL);" << endl;
3708  indent(out) << "tmpstr=unescaped;" << endl;
3709  indent(out) << "set_" << (*f_iter2)->get_name() << "(boost::lexical_cast<std::string>(tmpstr));" << endl;
3710  indent(out) << "curl_free(unescaped);" << endl;
3711  indent(out) << "continue;" << endl;
3712 
3713  }
3714  else if (btype->is_uuid())
3715  {
3716  indent(out) << "entity_name=\"" << (*f_iter2)->get_name() << "\";" << endl;
3717  indent(out) << "std::stringstream ss;" << endl;
3718  indent(out) << "ss << *it2;" << endl;
3719  indent(out) << type_name((*f_iter2)->get_type()) << " tmpval;" << endl;
3720  indent(out) << "ss >> tmpval;" << endl;
3721  indent(out) << "set_" << (*f_iter2)->get_name() << "(tmpval);" << endl;
3722  indent(out) << "continue;" << endl;
3723  }
3724  else if (btype->is_ipaddr())
3725  {
3726  indent(out) << "entity_name=\"" << (*f_iter2)->get_name() << "\";" << endl;
3727  indent(out) << "boost::system::error_code ec;" << endl;
3728  indent(out) << type_name((*f_iter2)->get_type()) << " tmpval;" << endl;
3729  indent(out) << "tmpval = boost::asio::ip::address::from_string((*it2), ec);" << endl;
3730  indent(out) << "set_" << (*f_iter2)->get_name() << "(tmpval);" << endl;
3731  indent(out) << "continue;" << endl;
3732  }
3733  else
3734  {
3735  indent(out) << "entity_name=\"" << (*f_iter2)->get_name() << "\";" << endl;
3736  assert(btype->is_integer());
3737  indent(out) << type_name((*f_iter2)->get_type()) << " tmpval;" << endl;
3738  indent(out) << "stringToInteger((*it2), tmpval);" << endl;
3739  indent(out) << "set_" << (*f_iter2)->get_name() << "(tmpval);" << endl;
3740  indent(out) << "continue;" << endl;
3741  }
3742  }
3743  }
3744  indent(out) << "free(unescaped);" << endl;
3745  scope_down(out);
3746  scope_down(out);
3747  scope_down(out);
3748 
3749 
3750  indent(out) << "set_context(ctx);" << endl;
3751  indent(out) << "curl_easy_cleanup(cr);" << endl;
3752  indent(out) << "return true;" << endl;
3753  indent_down();
3754  indent(out) << "}" << endl << endl;
3755 }
3756 
3757 
3764 void t_cpp_generator::generate_sandesh_updater(ofstream& out,
3765  t_sandesh* tsandesh) {
3766 
3767  const vector<t_field*>& fields = tsandesh->get_members();
3768  vector<t_field*>::const_iterator f_iter = fields.begin();
3769  assert((*f_iter)->get_name() == "data");
3770 
3771  bool periodic_ustruct = false;
3772 
3773  std::map<std::string, std::string>::iterator ait;
3774  ait = ((*f_iter)->get_type())->annotations_.find("period");
3775  if (ait != ((*f_iter)->get_type())->annotations_.end()) periodic_ustruct = true;
3776 
3777  string dtype = type_name((*f_iter)->get_type());
3778  t_type* t = get_true_type((*f_iter)->get_type());
3779  assert(t->is_struct());
3780 
3781  t_struct* ts = (t_struct*)t;
3782 
3783  map<string,DSInfo> dsinfo;
3784  map<string,set<string> > rawmap;
3785  derived_stats_info(ts, dsinfo, rawmap);
3786 
3787  std::map<string, CacheAttribute> cache_attrs;
3788  cache_attr_info(ts, cache_attrs);
3789 
3790  const vector<t_field*>& sfields = ts->get_members();
3791  vector<t_field*>::const_iterator s_iter;
3792 
3793  indent(out) << "bool " << tsandesh->get_name() <<
3794  "::LoadUVE(SendType stype, uint32_t cycle) {" << endl;
3795  indent_up();
3796  indent(out) << "(void)cycle;" << endl;
3797  indent(out) << "bool is_periodic_attr = false;" << endl << endl;
3798 
3799  for (s_iter = sfields.begin(); s_iter != sfields.end(); ++s_iter) {
3800  string snm = (*s_iter)->get_name();
3801  if (snm == "deleted") continue;
3802 
3803  // For non-derived stats, setting of attribute would have also set
3804  // the isset flag
3805  map<string,DSInfo>::const_iterator ds_iter = dsinfo.find(snm);
3806  if (ds_iter == dsinfo.end()) {
3807 
3808  CacheAttribute cat = (cache_attrs.find(snm))->second;
3809  if (cat == MANDATORY) continue;
3810 
3811  indent(out) << "if (data.__isset." << snm << ") {" << endl;
3812  indent_up();
3813  if (cat == HIDDEN) {
3814  indent(out) << "if (stype != ST_INTROSPECT) data.__isset." <<
3815  snm << " = false;" << endl;
3816  } else if (cat == INLINE) {
3817  indent(out) << "if (stype == ST_PERIODIC) " <<
3818  "data.__isset." << snm << " = false;" << endl;
3819  } else assert(0);
3820  indent_down();
3821  indent(out) << "}" << endl << endl;
3822  } else {
3823  indent(out) << "if (data.__dsobj_" << snm << "->IsResult()) {" << endl;
3824  indent_up();
3825  CacheAttribute cat = ds_iter->second.cat_;
3826  if (cat == HIDDEN) {
3827  indent(out) << "if (stype != ST_INTROSPECT) data.__isset." <<
3828  snm << " = false;" << endl;
3829  } else if (cat == HIDDEN_PER) {
3830  indent(out) << "if (stype == ST_PERIODIC) {" << endl;
3831  indent_up();
3832  indent(out) << "if (cycle % " << ds_iter->second.period_ <<
3833  " == 0) data.__dsobj_" << snm << "->Flush(data." <<
3834  snm << ");" << endl;
3835  indent(out) << "data.__isset." << snm << " = false;" << endl;
3836  indent_down();
3837  indent(out) << "} else if (stype == ST_SYNC) data.__isset." <<
3838  snm << " = false;" << endl;
3839  } else if (cat == INLINE) {
3840  indent(out) << "if (stype == ST_PERIODIC) " <<
3841  "data.__isset." << snm << " = false;" << endl;
3842  } else if (cat == PERIODIC) {
3843  indent(out) << "if (stype == ST_PERIODIC) {" << endl;
3844  indent_up();
3845 
3846  indent(out) << "if (cycle % " << ds_iter->second.period_ <<
3847  " == 0) {" << endl;
3848  indent_up();
3849 
3850  indent(out) << "data.__dsobj_" << snm << "->Flush(data." <<
3851  snm << ");" << endl;
3852  indent(out) << "data.__dsobj_" << snm << "->FillResult(data." <<
3853  snm << ", data.__isset." << snm << ", true);" << endl;
3854 
3855  indent_down();
3856  indent(out) << "} else data.__isset." << snm << " = false;" << endl;
3857 
3858  indent_down();
3859  indent(out) << "} else if (stype == ST_SYNC) data.__isset." <<
3860  snm << " = false;" << endl;
3861  } else assert(0);
3862 
3863  indent(out) << "else data.__dsobj_" << snm << "->FillResult(data." <<
3864  snm << ", data.__isset." << snm << ", true);" <<endl;
3865 
3866  if (cat == PERIODIC) {
3867  indent(out) << "is_periodic_attr |= data.__isset." << snm << ";" << endl;
3868  }
3869  indent_down();
3870  indent(out) << "}" << endl << endl;
3871  }
3872  }
3873  indent(out) << "if (stype == ST_PERIODIC) return ((is_periodic_attr) || (data.get_deleted()));" << endl;
3874  indent(out) << "else return true;" << endl;
3875  indent_down();
3876  indent(out) << "}" << endl << endl;
3877 
3878  indent(out) << "std::map<std::string, std::string> " << tsandesh->get_name() <<
3879  "::_DSConf(void) {" << endl;
3880  indent_up();
3881 
3882  bool is_dsconf = false;
3883  for (map<string,DSInfo>::const_iterator ds_iter = dsinfo.begin();
3884  ds_iter != dsinfo.end(); ++ds_iter) {
3885  if (!is_dsconf)
3886  indent(out) << "std::map<std::string, std::string> _dsconf = " <<
3887  "boost::assign::map_list_of" << endl;
3888  indent(out) << "(\"" << ds_iter->first << "\", \"" <<
3889  ds_iter->second.annotation_ << "\")" << endl;
3890  is_dsconf = true;
3891  }
3892  if (is_dsconf) indent(out) << ";" << endl;
3893  else indent(out) << "std::map<std::string, std::string> _dsconf;" << endl;
3894 
3895  indent(out) << "return _dsconf;" << endl;
3896  indent_down();
3897  indent(out) << "}" << endl << endl;
3898 
3899  indent(out) << "void " << tsandesh->get_name() <<
3900  "::_InitDerivedStats(" << dtype <<
3901  " & _data, const map<string,string> & _dsconf) {" << endl;
3902  indent_up();
3903 
3904  for (map<string,DSInfo>::const_iterator ds_iter = dsinfo.begin();
3905  ds_iter != dsinfo.end(); ++ds_iter) {
3906 
3907  indent(out) << "{" << endl;
3908  indent_up();
3909  indent(out) << "map<string,string>::const_iterator _dci = _dsconf.find(\"" <<
3910  ds_iter->first << "\");" << endl;
3911  if (ds_iter->second.rmtype_ == RM_AGG) {
3912  indent(out) << "bool is_agg = true;" << endl;
3913  } else {
3914  indent(out) << "bool is_agg = false;" << endl;
3915  }
3916  indent(out) << "assert(_dci != _dsconf.end());" << endl;
3917 
3918  indent(out) << "_data.__dsobj_" << ds_iter->first << ".reset();" << endl;
3919 
3920  if ((ds_iter->second.cat_ == PERIODIC) ||
3921  (ds_iter->second.cat_ == HIDDEN_PER)) {
3922 
3923  if (ds_iter->second.prealgo_.empty()) {
3924  indent(out) << "_data.__dsobj_" << ds_iter->first << " = boost::make_shared<" <<
3925  " ::contrail::sandesh::DerivedStatsPeriodicIf< ::contrail::sandesh::" <<
3926  ds_iter->second.algo_ << ", " <<
3927  ds_iter->second.rawtype_ << ", " <<
3928  ds_iter->second.resulttype_.substr(0, ds_iter->second.resulttype_.size() - 3) << ", " <<
3929  ds_iter->second.resulttype_ << "> >(_dci->second, is_agg);" << endl;
3930  } else {
3931  indent(out) << "_data.__dsobj_" << ds_iter->first << " = boost::make_shared<" <<
3932  " ::contrail::sandesh::DerivedStatsPeriodicAnomalyIf< ::contrail::sandesh::" <<
3933  ds_iter->second.algo_ << ", " <<
3934  ds_iter->second.rawtype_ << ", ::contrail::sandesh::" <<
3935  ds_iter->second.prealgo_ << ", " <<
3936  ds_iter->second.resulttype_ << "> >(_dci->second, is_agg);" << endl;
3937  }
3938  } else {
3939  indent(out) << "_data.__dsobj_" << ds_iter->first << " = boost::make_shared<" <<
3940  " ::contrail::sandesh::DerivedStatsIf< ::contrail::sandesh::" <<
3941  ds_iter->second.algo_ << ", " <<
3942  ds_iter->second.rawtype_ << ", " << ds_iter->second.resulttype_ <<
3943  "> >(_dci->second, is_agg);" << endl;
3944  }
3945  indent_down();
3946  indent(out) << "}" << endl;
3947  }
3948  indent_down();
3949  indent(out) << "}" << endl << endl;
3950 
3951  indent(out) << "bool " << tsandesh->get_name() <<
3952  "::UpdateUVE(" << dtype <<
3953  " & _data, " << dtype <<
3954  " & tdata, uint64_t mono_usec, SandeshLevel::type Xlevel) {" << endl;
3955 
3956  indent_up();
3957 
3958  indent(out) << "bool send = false;" << endl;
3959 
3960  for (s_iter = sfields.begin(); s_iter != sfields.end(); ++s_iter) {
3961  string snm = (*s_iter)->get_name();
3962 
3963  // Only set those attributes that are NOT derived stats results
3964  if (dsinfo.find(snm) == dsinfo.end()) {
3965  CacheAttribute cat = (cache_attrs.find(snm))->second;
3966  if ((*s_iter)->get_req() == t_field::T_OPTIONAL) {
3967  // don't send non-inline attributes
3968  if (cat != INLINE) {
3969  indent(out) << "if (_data.__isset." << snm << ") { _data.__isset." <<
3970  snm << " = false;" << endl;
3971  } else {
3972  indent(out) << "if (_data.__isset." << snm << ") { send = true;" << endl;
3973  }
3974  } else {
3975  if ((snm.compare("name") == 0) || (snm.compare("proxy") == 0)) {
3976  indent(out) << "{" << endl;
3977  } else {
3978  indent(out) << "{ send = true;" << endl;
3979  }
3980  }
3981  indent_up();
3982  indent(out) << "tdata.set_" << snm << "(_data.get_" <<
3983  snm << "());" << endl;
3984  map<string,set<string> >::const_iterator r_iter = rawmap.find(snm);
3985  // Update all derivied stats of this raw attribute
3986  if (r_iter != rawmap.end()) {
3987  t_type* ratype = (*s_iter)->get_type();
3988  if (ratype->is_map()) {
3989  t_type* vtype = ((t_map*)ratype)->get_val_type();
3990  indent(out) << "typedef " << type_name((*s_iter)->get_type()) <<
3991  " _T_" << snm << ";" << endl;
3992  indent(out) << "std::map<string,bool> _delmap_" << snm
3993  << ";" << endl;
3994  indent(out) << "BOOST_FOREACH(const _T_" << snm <<
3995  "::value_type& _tp, _data.get_" << snm << "())" << endl;
3996  indent(out) << " _delmap_" << snm <<
3997  ".insert(make_pair(_tp.first, SandeshStructDeleteTrait<" <<
3998  type_name(vtype) << ">::get(_tp.second)));" << endl;
3999  }
4000  set<string>::const_iterator d_iter;
4001  for (d_iter = r_iter->second.begin(); d_iter != r_iter->second.end(); ++d_iter) {
4002  map<string,DSInfo>::const_iterator c_iter = dsinfo.find(*d_iter);
4003  CacheAttribute dat = c_iter->second.cat_;
4004 
4005  if (c_iter->second.compattr_.empty()) {
4006  if (!c_iter->second.is_map_) {
4007  indent(out) << "tdata.__dsobj_" << *d_iter <<
4008  "->Update(_data.get_" << snm << "(), mono_usec);" << endl;
4009  } else {
4010  indent(out) << "tdata.__dsobj_" << *d_iter <<
4011  "->Update(_data.get_" << snm << "(), _delmap_" <<
4012  snm << ", mono_usec);" << endl;
4013  }
4014  } else {
4015  string getexpr = string("get_") + c_iter->second.compattr_ +
4016  string("()");
4017  if (!c_iter->second.subcompattr_.empty()) {
4018  getexpr += string(".get_") + c_iter->second.subcompattr_ +
4019  string("()");
4020  }
4021  if (!c_iter->second.is_map_) {
4022  indent(out) << "tdata.__dsobj_" << *d_iter << "->Update(_data.get_" <<
4023  snm << "()." << getexpr << ", mono_usec);" << endl;
4024  } else {
4025  indent(out) << "std::map<string," << c_iter->second.rawtype_ << "> temp_" <<
4026  *d_iter << ";" << endl;
4027  indent(out) << "BOOST_FOREACH(const _T_" << snm <<
4028  "::value_type& _tp, _data.get_" << snm << "()) {" << endl;
4029  indent_up();
4030  indent(out) << "temp_" << *d_iter <<
4031  ".insert(make_pair(_tp.first, _tp.second." <<
4032  getexpr << "));" << endl;
4033  indent_down();
4034  indent(out) << "}" << endl;
4035  indent(out) << "tdata.__dsobj_" << *d_iter << "->Update(temp_" <<
4036  *d_iter << ", _delmap_" << snm << ", mono_usec);" << endl;
4037  indent(out) << "temp_" << *d_iter << ".clear();" << endl;
4038  }
4039  }
4040  // If the DS is inline or mandatory
4041  if (dat == INLINE) {
4042  indent(out) << "tdata.__dsobj_" << *d_iter << "->FillResult(_data." <<
4043  *d_iter << ", _data.__isset." << *d_iter << ");" << endl;
4044  indent(out) << "if (_data.__isset." << *d_iter <<") send = true;" << endl;
4045  }
4046  }
4047  }
4048 
4049  indent_down();
4050  indent(out) << "}" << endl << endl;
4051  }
4052  }
4053  if (!periodic_ustruct) {
4054  indent(out) << "send = true;" << endl;
4055  }
4056  indent(out) << "return send;" << endl;
4057  indent_down();
4058  indent(out) << "}" << endl << endl;
4059 
4060  f_iter++;
4061  assert(f_iter == fields.end());
4062 }
4063 
4070 void t_cpp_generator::generate_sandesh_reader(ofstream& out,
4071  t_sandesh* tsandesh) {
4072  indent(out) <<
4073  "int32_t " << tsandesh->get_name() <<
4074  "::Read(boost::shared_ptr<contrail::sandesh::protocol::TProtocol> iprot) {"
4075  << endl;
4076 
4077  generate_common_struct_reader_body(out, tsandesh, false);
4078 
4079  indent(out) <<
4080  "}" << endl << endl;
4081 }
4082 
4089 void t_cpp_generator::generate_sandesh_writer(ofstream& out,
4090  t_sandesh* tsandesh) {
4091  indent(out) <<
4092  "int32_t " << tsandesh->get_name() <<
4093  "::Write(boost::shared_ptr<contrail::sandesh::protocol::TProtocol> oprot) const {" <<
4094  endl;
4095 
4096  generate_common_struct_writer_body(out, tsandesh, false);
4097 
4098  indent(out) <<
4099  "}" << endl <<
4100  endl;
4101 }
4102 
4110 void t_cpp_generator::generate_sandesh_uve_creator(
4111  std::ofstream& out, t_sandesh* tsandesh) {
4112  const vector<t_field*>& fields = tsandesh->get_members();
4113  vector<t_field*>::const_iterator f_iter = fields.begin();
4114  assert((*f_iter)->get_name() == "data");
4115 
4116  bool is_proxy = false;
4117  std::map<std::string, std::string>::iterator ait;
4118  ait = ((*f_iter)->get_type())->annotations_.find("period");
4119 
4120  std::string sname = tsandesh->get_name();
4121  indent(out) << "SANDESH_UVE_DEF(" << sname << "," <<
4122  type_name((*f_iter)->get_type());
4123 
4124  if (ait == ((*f_iter)->get_type())->annotations_.end()) {
4125  indent(out) << ", 0, 0);" << endl;
4126  } else {
4127  std::map<std::string, std::string>::iterator tmit;
4128  tmit = ((*f_iter)->get_type())->annotations_.find("timeout");
4129  if (tmit == ((*f_iter)->get_type())->annotations_.end()) {
4130  indent(out) << ", " << atoi(ait->second.c_str()) << ", 0);" << endl;
4131  } else {
4132  indent(out) << ", " << atoi(ait->second.c_str()) << ", " <<
4133  atoi(tmit->second.c_str()) << ");" << endl;
4134  is_proxy = true;
4135  }
4136  }
4137 
4138  indent(out) << "void " << sname <<
4139  "::Send(const " << type_name((*f_iter)->get_type()) <<
4140  "& cdata, SandeshLevel::type Xlevel, SandeshUVE::SendType stype," <<
4141  " uint32_t seqno, uint32_t cycle, std::string ctx) {" << endl;
4142  indent_up();
4143  indent(out) << sname << " *snh = new " << sname << "(seqno, cdata);" << endl;
4144  indent(out) << "snh->set_level(Xlevel);" << endl;
4145  indent(out) << "if (snh->LoadUVE(stype, cycle)) {" << endl;
4146  indent_up();
4147  indent(out) << "snh->set_context(ctx); snh->set_more(!ctx.empty());" << endl;
4148  indent(out) << "if (stype == SandeshUVE::ST_SYNC) snh->set_hints(snh->hints() | " <<
4149  "g_sandesh_constants.SANDESH_SYNC_HINT);" << endl;
4150  indent(out) << "snh->Dispatch();" << endl;
4151  indent_down();
4152  indent(out) << "} else snh->Release();" << endl;
4153  indent_down();
4154  indent(out) << "}" << endl << endl;
4155  // Generate legacy
4156  indent(out) << "void " << sname <<
4157  "::Send(const " << type_name((*f_iter)->get_type()) <<
4158  "& cdata, SandeshUVE::SendType stype, uint32_t seqno," <<
4159  " uint32_t cycle, std::string ctx) {" << endl;
4160  indent_up();
4161  indent(out) << sname << " *snh = new " << sname << "(seqno, cdata);" << endl;
4162  indent(out) << "snh->set_level(SandeshLevel::SYS_NOTICE);" << endl;
4163  indent(out) << "if (snh->LoadUVE(stype, cycle)) {" << endl;
4164  indent_up();
4165  indent(out) << "snh->set_context(ctx); snh->set_more(!ctx.empty());" << endl;
4166  indent(out) << "if (stype == SandeshUVE::ST_SYNC) snh->set_hints(snh->hints() | " <<
4167  "g_sandesh_constants.SANDESH_SYNC_HINT);" << endl;
4168  indent(out) << "snh->Dispatch();" << endl;
4169  indent_down();
4170  indent(out) << "} else snh->Release();" << endl;
4171  indent_down();
4172  indent(out) << "}" << endl << endl;
4173 
4174  if (is_proxy) {
4175  indent(out) << "void " << sname <<
4176  "::Send(const " << type_name((*f_iter)->get_type()) <<
4177  "& data, SandeshLevel::type Xlevel, std::string table, " <<
4178  "uint64_t mono_usec, int partition) {" << endl;
4179  indent_up();
4180  } else {
4181  indent(out) << "void " << sname <<
4182  "::Send(const " << type_name((*f_iter)->get_type()) <<
4183  "& data, SandeshLevel::type Xlevel, std::string table, " <<
4184  "uint64_t mono_usec) {" << endl;
4185  indent_up();
4186  indent(out) << "int partition = -1;" << endl;
4187  }
4188 
4189  indent(out) << type_name((*f_iter)->get_type()) <<
4190  " & cdata = const_cast<" << type_name((*f_iter)->get_type()) <<
4191  " &>(data);" << endl;
4192  indent(out) << "uint32_t msg_seqno = lseqnum_.fetch_and_increment() + 1;" << endl;
4193  indent(out) << "if (!table.empty()) cdata.table_ = table;" << endl;
4194  indent(out) << "if (uvemap" << sname <<
4195  ".UpdateUVE(cdata, msg_seqno, mono_usec, partition, Xlevel)) {" << endl;
4196  indent_up();
4197  indent(out) << sname << " *snh = new " << sname << "(msg_seqno, cdata);" << endl;
4198  indent(out) << "snh->set_level(Xlevel);" << endl;
4199  indent(out) << "snh->Dispatch();" << endl;
4200  indent_down();
4201  indent(out) << "}" << endl;
4202 
4203  indent_down();
4204  indent(out) << "}" << endl << endl;
4205  // Generate legacy
4206  if (is_proxy) {
4207  indent(out) << "void " << sname <<
4208  "::Send(const " << type_name((*f_iter)->get_type()) <<
4209  "& data, std::string table, " <<
4210  "uint64_t mono_usec, int partition) {" << endl;
4211  indent_up();
4212  } else {
4213  indent(out) << "void " << sname <<
4214  "::Send(const " << type_name((*f_iter)->get_type()) <<
4215  "& data, std::string table, " <<
4216  "uint64_t mono_usec) {" << endl;
4217  indent_up();
4218  indent(out) << "int partition = -1;" << endl;
4219  }
4220 
4221  indent(out) << type_name((*f_iter)->get_type()) <<
4222  " & cdata = const_cast<" << type_name((*f_iter)->get_type()) <<
4223  " &>(data);" << endl;
4224  indent(out) << "uint32_t msg_seqno = lseqnum_.fetch_and_increment() + 1;" << endl;
4225  indent(out) << "if (!table.empty()) cdata.table_ = table;" << endl;
4226  indent(out) << "if (uvemap" << sname <<
4227  ".UpdateUVE(cdata, msg_seqno, mono_usec, partition, SandeshLevel::SYS_NOTICE)) {" << endl;
4228  indent_up();
4229  indent(out) << sname << " *snh = new " << sname << "(msg_seqno, cdata);" << endl;
4230  indent(out) << "snh->set_level(SandeshLevel::SYS_NOTICE);" << endl;
4231  indent(out) << "snh->Dispatch();" << endl;
4232  indent_down();
4233  indent(out) << "}" << endl;
4234 
4235  indent_down();
4236  indent(out) << "}" << endl << endl;
4237 }
4238 
4245 void t_cpp_generator::generate_sandesh_creator(ofstream& out,
4246  t_sandesh* tsandesh) {
4247  const t_type *t = tsandesh->get_type();
4248  assert(t->is_base_type());
4249 
4250  // Creator function name
4251  if (((t_base_type *)t)->is_sandesh_request()) {
4252  indent(out) << "extern int " << program_name_ << "_marker;" << endl;
4253  // Request registration
4254  indent(out) << "SANDESH_REGISTER_DEF_TYPE(" << tsandesh->get_name() <<
4255  ");" << endl << endl;
4256  return;
4257  } else if (((t_base_type *)t)->is_sandesh_trace() ||
4258  ((t_base_type *)t)->is_sandesh_trace_object()) {
4259  generate_sandesh_trace(out, tsandesh);
4260  return;
4261  } else if (((t_base_type *)t)->is_sandesh_buffer()) {
4262  // Buffer registration
4263  indent(out) << "SANDESH_REGISTER_DEF_TYPE(" << tsandesh->get_name() <<
4264  ");" << endl << endl;
4265  return;
4266  } else if (((t_base_type *)t)->is_sandesh_uve() ||
4267  ((t_base_type *)t)->is_sandesh_alarm()) {
4268  generate_sandesh_uve_creator(out, tsandesh);
4269  return;
4270  }
4271  return;
4272 }
4273 
4277 void t_cpp_generator::generate_get_size_field(ofstream& out, t_field *tfield) {
4278  t_type* type = get_true_type(tfield->get_type());
4279  string name = tfield->get_name();
4280  // Handle optional elements
4281  if (tfield->get_req() == t_field::T_OPTIONAL) {
4282  out << indent() << "if (__isset." << name << ") {" <<
4283  endl;
4284  indent_up();
4285  }
4286  if (type->is_struct()) {
4287  generate_get_size_struct(out, (t_struct *)type, name);
4288  } else if (type->is_container()) {
4289  generate_get_size_container(out, type, name);
4290  } else if (type->is_string() || type->is_xml() ||
4291  type->is_static_const_string()) {
4292  out << indent() << "size += " << name << ".length();" << endl;
4293  } else {
4294  out << indent() << "size += sizeof(" + name +");" << endl;
4295  }
4296  // Handle optional elements
4297  if (tfield->get_req() == t_field::T_OPTIONAL) {
4298  indent_down();
4299  out << indent() << "}" << endl;
4300  }
4301 }
4302 
4306 void t_cpp_generator::generate_logger_field(ofstream& out,
4307  t_field *tfield,
4308  string prefix,
4309  bool log_value_only,
4310  bool no_name_log,
4311  bool for_sandesh) {
4312  t_type* type = get_true_type(tfield->get_type());
4313  string name = tfield->get_name();
4314  // Handle optional elements
4315  if (tfield->get_req() == t_field::T_OPTIONAL) {
4316  if (for_sandesh) {
4317  return;
4318  }
4319  out << indent() << "if (__isset." << name << ") {" <<
4320  endl;
4321  indent_up();
4322  }
4323  if (type->is_struct()) {
4324  if (no_name_log) {
4325  out << indent() << "Xbuf << " << prefix << "\"[ \";" << endl;
4326  } else {
4327  out << indent() << "Xbuf << " << prefix << "\"" << name << "\"" << " << \"= [ \";" << endl;
4328  }
4329  generate_logger_struct(out, (t_struct *)type, prefix, name);
4330  out << indent() << "Xbuf << " << "\" ]\";" << endl;
4331  } else if (type->is_container()) {
4332  if (no_name_log) {
4333  out << indent() << "Xbuf << " << prefix << "\"[ \";" << endl;
4334  } else {
4335  out << indent() << "Xbuf << " << prefix << "\"" << name << "\"" << " << \"= [ \";" << endl;
4336  }
4337  generate_logger_container(out, type, name, log_value_only);
4338  out << indent() << "Xbuf << " << "\" ]\";" << endl;
4339  } else if (type->is_base_type() || type->is_enum()) {
4340  t_base_type *tbase = static_cast<t_base_type *>(type);
4341  if (log_value_only) {
4342  if (tbase->is_integer()) {
4343  out << indent() << "Xbuf << " << prefix <<
4344  "integerToString(" << name <<
4345  ");" << endl;
4346  } else {
4347  out << indent() << "Xbuf << " << prefix <<
4348  name << ";" << endl;
4349  }
4350  } else {
4351  if (tbase->is_integer()) {
4352  out << indent() << "Xbuf << " << prefix <<
4353  "\"" << name << "\"" <<
4354  " << \" = \" << integerToString(" << name <<
4355  ");" << endl;
4356  } else {
4357  if (tbase->get_name() == "ipv4") {
4358  scope_up(out);
4359  out << indent() << "struct in_addr ip_addr_"<< name <<";" << endl;
4360  out << indent() << "ip_addr_" << name << ".s_addr = htonl(" << name << ");" << endl;
4361  out << indent() << "Xbuf << " << prefix <<
4362  "\"" << name << "\"" <<
4363  " << \" = \" << inet_ntoa(ip_addr_" << name << ");" << endl;
4364  scope_down(out);
4365  }
4366  else {
4367  out << indent() << "Xbuf << " << prefix <<
4368  "\"" << name << "\"" <<
4369  " << \" = \" << " << name <<
4370  ";" << endl;
4371  }
4372  }
4373  }
4374  }
4375  // Handle optional elements
4376  if (tfield->get_req() == t_field::T_OPTIONAL) {
4377  indent_down();
4378  out << indent() << "}" << endl;
4379  }
4380 }
4381 
4385 void t_cpp_generator::generate_get_size_container(ofstream& out,
4386  t_type* ttype,
4387  string name) {
4388  scope_up(out);
4389  string iter = tmp("_iter");
4390  out << indent() << type_name(ttype) << "::const_iterator " << iter
4391  << ";" << endl;
4392  out << indent() << "for (" << iter << " = " << name << ".begin(); "
4393  << iter << " != " << name << ".end(); ++" << iter << ")" << endl;
4394  scope_up(out);
4395  if (ttype->is_map()) {
4396  generate_get_size_map_element(out, (t_map *)ttype, iter);
4397  } else if(ttype->is_set()) {
4398  generate_get_size_set_element(out, (t_set *)ttype, iter);
4399  } else if(ttype->is_list()) {
4400  generate_get_size_list_element(out, (t_list *)ttype, iter);
4401  }
4402  scope_down(out);
4403  scope_down(out);
4404 }
4405 
4409 void t_cpp_generator::generate_get_size_map_element(ofstream& out,
4410  t_map* tmap,
4411  string iter) {
4412  t_field kfield(tmap->get_key_type(), iter + "->first");
4413  generate_get_size_field(out, &kfield);
4414  t_field vfield(tmap->get_val_type(), iter + "->second");
4415  generate_get_size_field(out, &vfield);
4416 }
4417 
4421 void t_cpp_generator::generate_get_size_list_element(ofstream& out,
4422  t_list* tlist,
4423  string iter) {
4424  t_field efield(tlist->get_elem_type(), "(*" + iter + ")");
4425  generate_get_size_field(out, &efield);
4426 }
4427 
4431 void t_cpp_generator::generate_get_size_set_element(ofstream& out,
4432  t_set* tset,
4433  string iter) {
4434  t_field efield(tset->get_elem_type(), "(*" + iter + ")");
4435  generate_get_size_field(out, &efield);
4436 }
4437 
4441 void t_cpp_generator::generate_logger_container(ofstream& out,
4442  t_type* ttype,
4443  string name,
4444  bool log_value_only) {
4445  scope_up(out);
4446  string iter = tmp("_iter");
4447  string prefix = "\" \" << ";
4448  out << indent() << "Xbuf << " << prefix << "\"[\";" << endl;
4449  out <<
4450  indent() << type_name(ttype) << "::const_iterator " << iter << ";" << endl <<
4451  indent() << "for (" << iter << " = " << name << ".begin(); " << iter << " != " << name << ".end(); ++" << iter << ")" << endl;
4452  scope_up(out);
4453  if (ttype->is_map()) {
4454  generate_logger_map_element(out, (t_map*)ttype, iter, log_value_only);
4455  } else if (ttype->is_set()) {
4456  generate_logger_set_element(out, (t_set*)ttype, iter, log_value_only);
4457  } else if (ttype->is_list()) {
4458  generate_logger_list_element(out, (t_list*)ttype, iter, log_value_only);
4459  }
4460  out << indent() << "Xbuf << " << "\", \";" << endl;
4461  scope_down(out);
4462  out << indent() << "Xbuf << " << prefix << "\"]\";" << endl;
4463  scope_down(out);
4464 }
4465 
4469 void t_cpp_generator::generate_logger_map_element(ofstream& out,
4470  t_map* tmap,
4471  string iter,
4472  bool log_value_only) {
4473  string prefix = "\" \" << ";
4474  t_field kfield(tmap->get_key_type(), iter + "->first");
4475  generate_logger_field(out, &kfield, prefix, log_value_only, true);
4476  t_field vfield(tmap->get_val_type(), iter + "->second");
4477  generate_logger_field(out, &vfield, prefix, log_value_only, true);
4478 }
4479 
4483 void t_cpp_generator::generate_logger_set_element(ofstream& out,
4484  t_set* tset,
4485  string iter,
4486  bool log_value_only) {
4487  string prefix = "\" \" << ";
4488  t_field efield(tset->get_elem_type(), "(*" + iter + ")");
4489  generate_logger_field(out, &efield, prefix, log_value_only, true);
4490 }
4491 
4495 void t_cpp_generator::generate_logger_list_element(ofstream& out,
4496  t_list* tlist,
4497  string iter,
4498  bool log_value_only) {
4499  string prefix = "\" \" << ";
4500  t_field efield(tlist->get_elem_type(), "(*" + iter + ")");
4501  generate_logger_field(out, &efield, prefix, log_value_only, true);
4502 }
4503 
4507 void t_cpp_generator::generate_get_size_struct(ofstream& out,
4508  t_struct *tstruct,
4509  string name) {
4510  (void) tstruct;
4511  out << indent() << "size += " << name << ".GetSize();" << endl;
4512 }
4513 
4517 void t_cpp_generator::generate_logger_struct(ofstream& out,
4518  t_struct *tstruct,
4519  string prefix,
4520  string name) {
4521  (void) tstruct;
4522  out << indent() << "Xbuf << " << prefix << name << ".log();" << endl;
4523 }
4524 
4531 void t_cpp_generator::generate_sandesh_static_drop_logger(ofstream &out,
4532  t_sandesh *tsandesh, bool generate_sandesh_object) {
4533  // Generate DropLog
4534  if (generate_sandesh_object) {
4535  out << indent() << "static void DropLog(const std::string& " <<
4536  "drop_reason, std::string category, SandeshLevel::type level, " <<
4537  tsandesh->get_name() << " *snh) {" << endl;
4538  } else {
4539  out << indent() << "static void DropLog" <<
4540  generate_sandesh_async_creator(tsandesh, true, false, false, "", "",
4541  false, false, true) << " {" << endl;
4542  }
4543  indent_up();
4544  out << indent() << "log4cplus::LogLevel Xlog4level(" <<
4545  "SandeshLevelTolog4Level(SandeshLevel::SYS_ERR));" << endl;
4546  out << indent() <<
4547  "log4cplus::Logger Xlogger = Sandesh::logger();" << endl;
4548  out << indent() <<
4549  "if (!Xlogger.isEnabledFor(Xlog4level)) {" << endl;
4550  indent_up();
4551  out << indent() << "return;" << endl;
4552  scope_down(out);
4553  out << indent() << "log4cplus::tostringstream Xbuf;" << endl;
4554  // Only systemlog, objectlog, flowlog, and UVE have valid level
4555  // and category
4556  if (tsandesh->is_level_category_supported()) {
4557  if (generate_sandesh_object) {
4558  out << indent() << "Xbuf << drop_reason << snh->ToString();"
4559  << endl;
4560  } else {
4561  out << indent() << "Xbuf << drop_reason << category << " <<
4562  "\" [\" << LevelToString(level) << \"]: " <<
4563  tsandesh->get_name() << ": \";" << endl;
4564  }
4565  } else {
4566  assert(!generate_sandesh_object);
4567  out << indent() << "Xbuf << drop_reason << \"" <<
4568  tsandesh->get_name() << ": \";" << endl;
4569  }
4570 
4571  bool is_system =
4572  ((t_base_type *)tsandesh->get_type())->is_sandesh_system();
4573  bool is_trace =
4574  ((t_base_type *)tsandesh->get_type())->is_sandesh_trace();
4575  bool log_value_only = is_system || is_trace;
4576  bool init = false;
4577  string prefix;
4578  const vector<t_field*>& fields = tsandesh->get_members();
4579  vector<t_field*>::const_iterator f_iter;
4580  for (f_iter = fields.begin(); f_iter != fields.end(); ++f_iter) {
4581  if (init) {
4582  prefix = "\" \" << ";
4583  } else {
4584  init = true;
4585  }
4586  if (!generate_sandesh_object) {
4587  generate_logger_field(out, *f_iter, prefix, log_value_only, false,
4588  true);
4589  }
4590  }
4591  out << indent() << "Xlogger.forcedLog(Xlog4level, Xbuf.str());" <<
4592  endl;
4593  indent_down();
4594  indent(out) << "}" << endl << endl;
4595 }
4596 
4597 void t_cpp_generator::generate_sandesh_static_logger(ofstream &out,
4598  t_sandesh *tsandesh, bool generate_sandesh_object) {
4599  // Generate Log
4600  if (generate_sandesh_object) {
4601  out << indent() << "static void Log(std::string category, " <<
4602  "SandeshLevel::type level, " << tsandesh->get_name() <<
4603  " *snh) {" << endl;
4604  } else {
4605  out << indent() << "static void Log" <<
4606  generate_sandesh_async_creator(tsandesh, true, false, false, "", "",
4607  false, false, false) << " {" << endl;
4608  }
4609  indent_up();
4610  out << indent() <<
4611  "if (!IsLevelCategoryLoggingAllowed(" <<
4612  generate_sandesh_base_name(tsandesh, true) <<
4613  ", level, category)) {" << endl;
4614  indent_up();
4615  out << indent() << "return;" << endl;
4616  scope_down(out);
4617  out << indent() << "log4cplus::LogLevel Xlog4level(" <<
4618  "SandeshLevelTolog4Level(level));" << endl;
4619  out << indent() <<
4620  "log4cplus::Logger Xlogger = Sandesh::logger();" << endl;
4621  out << indent() <<
4622  "if (!Xlogger.isEnabledFor(Xlog4level)) {" << endl;
4623  indent_up();
4624  out << indent() << "return;" << endl;
4625  scope_down(out);
4626  out << indent() << "log4cplus::tostringstream Xbuf;" << endl;
4627  // Only systemlog, objectlog, flowlog, and UVE have valid level
4628  // and category
4629  if (tsandesh->is_level_category_supported()) {
4630  if (generate_sandesh_object) {
4631  out << indent() << "Xbuf << snh->ToString();" << endl;
4632  } else {
4633  out << indent() << "Xbuf << category << " <<
4634  "\" [\" << LevelToString(level) << \"]: " <<
4635  tsandesh->get_name() << ": \";" << endl;
4636  }
4637  } else {
4638  assert(!generate_sandesh_object);
4639  out << indent() << "Xbuf << \"" <<
4640  tsandesh->get_name() << ": \";" << endl;
4641  }
4642 
4643  bool is_system =
4644  ((t_base_type *)tsandesh->get_type())->is_sandesh_system();
4645  bool is_trace =
4646  ((t_base_type *)tsandesh->get_type())->is_sandesh_trace();
4647  bool log_value_only = is_system || is_trace;
4648  bool init = false;
4649  string prefix;
4650  const vector<t_field*>& fields = tsandesh->get_members();
4651  vector<t_field*>::const_iterator f_iter;
4652  for (f_iter = fields.begin(); f_iter != fields.end(); ++f_iter) {
4653  if (init) {
4654  prefix = "\" \" << ";
4655  } else {
4656  init = true;
4657  }
4658  if (!generate_sandesh_object) {
4659  generate_logger_field(out, *f_iter, prefix, log_value_only, false,
4660  true);
4661  }
4662  }
4663  out << indent() << "Xlogger.forcedLog(Xlog4level, Xbuf.str());" <<
4664  endl;
4665  indent_down();
4666  indent(out) << "}" << endl << endl;
4667 }
4668 
4676 void t_cpp_generator::generate_sandesh_logger(ofstream &out,
4677  t_sandesh *tsandesh,
4678  sandesh_logger::type ltype) {
4679  if (ltype == sandesh_logger::BUFFER) {
4680  indent(out) << "std::string " << tsandesh->get_name() <<
4681  "::ToString() const {" << endl;
4682  indent_up();
4683  } else if (ltype == sandesh_logger::LOG) {
4684  indent(out) << "void " << tsandesh->get_name() <<
4685  "::Log() const {" << endl;
4686  indent_up();
4687  } else if (ltype == sandesh_logger::FORCED_LOG) {
4688  indent(out) << "void " << tsandesh->get_name() <<
4689  "::ForcedLog() const {" << endl;
4690  indent_up();
4691  } else {
4692  assert(0);
4693  }
4694  const vector<t_field*>& fields = tsandesh->get_members();
4695  vector<t_field*>::const_iterator f_iter;
4696  bool init = false;
4697  bool log_value_only =
4698  ((t_base_type *)tsandesh->get_type())->is_sandesh_system() ||
4699  ((t_base_type *)tsandesh->get_type())->is_sandesh_trace();
4700  for (f_iter = fields.begin(); f_iter != fields.end(); ++f_iter) {
4701  string prefix = "\" \" << ";
4702  // Handle init
4703  if (!init) {
4704  init = true;
4705  string category_str = "Sandesh::category()";
4706  string level_str = "Sandesh::level()";
4707  if (ltype == sandesh_logger::LOG) {
4708  out << indent() << "if (!IsLevelCategoryLoggingAllowed(" <<
4709  generate_sandesh_base_name(tsandesh, true) << ", " <<
4710  level_str << ", " << category_str << ")) {" << endl;
4711  indent_up();
4712  out << indent() << "return;" << endl;
4713  scope_down(out);
4714  out << indent() << "log4cplus::LogLevel Xlog4level(" <<
4715  "SandeshLevelTolog4Level(" << level_str << "));" << endl;
4716  out << indent() <<
4717  "log4cplus::Logger Xlogger = Sandesh::logger();" << endl;
4718  out << indent() <<
4719  "if (!Xlogger.isEnabledFor(Xlog4level)) {" << endl;
4720  indent_up();
4721  out << indent() << "return;" << endl;
4722  scope_down(out);
4723  out << indent() << "log4cplus::tostringstream Xbuf;" << endl;
4724  } else if (ltype == sandesh_logger::FORCED_LOG) {
4725  out << indent() << "log4cplus::LogLevel Xlog4level(" <<
4726  "SandeshLevelTolog4Level(" << level_str << "));" << endl;
4727  out << indent() <<
4728  "log4cplus::Logger Xlogger = Sandesh::logger();" << endl;
4729  out << indent() << "log4cplus::tostringstream Xbuf;" << endl;
4730  } else if (ltype == sandesh_logger::BUFFER) {
4731  out << indent() << "std::stringstream Xbuf;" << endl;
4732  // Timestamp
4733  out << indent() <<
4734  "Xbuf << integerToString(timestamp()) << \" \";" <<
4735  endl;
4736  }
4737  prefix = "";
4738  // Only systemlog, objectlog, and flowlog have valid level and category
4739  if (tsandesh->is_level_category_supported()) {
4740  out << indent() << "Xbuf << " << category_str <<
4741  " << \" [\" << LevelToString(" << level_str << ") << \"]: "
4742  << tsandesh->get_name() << ": \";" << endl;
4743  } else {
4744  out << indent() << "Xbuf << \"" << tsandesh->get_name() <<
4745  ": \";" << endl;
4746  }
4747  }
4748  generate_logger_field(out, *f_iter, prefix, log_value_only, false);
4749  }
4750  if (init) {
4751  if (ltype == sandesh_logger::BUFFER) {
4752  out << indent() << "return Xbuf.str();" << endl;
4753  } else {
4754  out << indent() << "Xlogger.forcedLog(Xlog4level, Xbuf.str());" <<
4755  endl;
4756  }
4757  } else {
4758  if (ltype == sandesh_logger::BUFFER) {
4759  out << indent() << "return std::string();" << endl;
4760  }
4761  }
4762  indent_down();
4763  indent(out) << "}" << endl << endl;
4764 }
4765 
4772 void t_cpp_generator::generate_sandesh_get_size(ofstream& out,
4773  t_sandesh* tsandesh) {
4774  //Generate GetSize function to return size of sandesh
4775  indent(out) << "size_t " << tsandesh->get_name() <<
4776  "::GetSize() const {" << endl;
4777  indent_up();
4778  indent(out) << "size_t size = 0;" << endl;
4779  const vector<t_field*>& fields = tsandesh->get_members();
4780  vector<t_field*>::const_iterator f_iter;
4781  for (f_iter = fields.begin(); f_iter != fields.end(); ++f_iter) {
4782  generate_get_size_field(out, *f_iter);
4783  }
4784  indent(out) << "return size;" << endl;
4785  indent_down();
4786  indent(out) << "}" << endl << endl;
4787 }
4788 
4795 void t_cpp_generator::generate_sandesh_loggers(ofstream& out,
4796  t_sandesh* tsandesh) {
4797  // Generate Log printing to log4cplus
4798  generate_sandesh_logger(out, tsandesh, sandesh_logger::LOG);
4799  // Generate forcedLog printing to log4cplus
4800  generate_sandesh_logger(out, tsandesh, sandesh_logger::FORCED_LOG);
4801  // Generate Log printing to buffer
4802  generate_sandesh_logger(out, tsandesh, sandesh_logger::BUFFER);
4803 }
4804 
4811 void t_cpp_generator::generate_sandesh_trace(ofstream& out,
4812  t_sandesh* tsandesh) {
4813  std::string creator_name = "trace_sandesh";
4814  indent(out) << "void " << tsandesh->get_name() << "::TraceMsg" <<
4815  generate_sandesh_no_static_const_string_function(tsandesh, true, false, true) <<
4816  " {" << endl;
4817  indent_up();
4818  out << indent() << "TraceSandeshType *trace = " <<
4819  "TraceSandeshType::GetInstance();" << endl;
4820  out << indent() << "if (trace != NULL && trace->IsTraceOn() && trace_buf->IsTraceOn()) {" << endl;
4821  indent_up();
4822  out << indent() << tsandesh->get_name() << " *" << creator_name << " = " <<
4823  "new " << tsandesh->get_name() <<
4824  generate_sandesh_no_static_const_string_function(tsandesh, false, false, false, false, true) <<
4825  ";" << endl;
4826  out << indent() << creator_name << "->set_category(trace_buf->Name());" << endl;
4827  out << indent() << "uint32_t seqnum(trace_buf->GetNextSeqNum());" << endl;
4828  out << indent() << creator_name << "->set_seqnum(seqnum);" << endl;
4829  out << indent() << "trace_buf->TraceWrite(" << creator_name << ");" << endl;
4830  out << indent() << "if ((IsLocalLoggingEnabled() && IsTracePrintEnabled()) || IsUnitTest()) " << creator_name << "->Log();" << endl;
4831  indent_down();
4832  out << indent() << "}" <<endl;
4833  indent_down();
4834  indent(out) <<
4835  "}" << endl << endl;
4836 }
4837 
4838 /*
4839  * Generate IsRatelimitPass function
4840  *
4841  */
4842 void t_cpp_generator::generate_isRatelimitPass(ofstream& out,
4843  t_sandesh* tsandesh) {
4844  out << indent() << "uint32_t send_rate_limit = Sandesh::get_send_rate_limit();" << endl;
4845  out << indent() << "if (send_rate_limit == 0) {" << endl;
4846  indent_up();
4847  out << indent() << "return false;" << endl;
4848  indent_down();
4849  out << indent() << "}" << endl;
4850  out << indent() << "tbb::mutex::scoped_lock lock(rate_limit_mutex_);" << endl;
4851  out << indent() << "if (rate_limit_buffer_.capacity() !="
4852  " send_rate_limit) {" << endl;
4853  indent_up();
4854  out << indent() << "//Resize the buffer to the "
4855  "buffer_threshold_" << endl;
4856  out << indent() << "rate_limit_buffer_.rresize(send_rate_limit);"
4857  << endl;
4858  out << indent() << "rate_limit_buffer_.set_capacity("
4859  "send_rate_limit);" << endl;
4860  indent_down();
4861  out << indent() << "}" << endl;
4862  out << indent() << "time_t current_time = time(0);" << endl;
4863  out << indent() << "if (rate_limit_buffer_.capacity() == rate_limit_buffer_"
4864  ".size()) {" << endl;
4865  indent_up();
4866  out << indent() << "if (*rate_limit_buffer_.begin() == current_time) {" << endl;
4867  indent_up();
4868  out << indent() << "//update tx and call droplog" << endl;
4869  out << indent() << "//Dont have to log more than once" << endl;
4870  out << indent() << "return false;" << endl;
4871  indent_down();
4872  out << indent() << "}" << endl;
4873  indent_down();
4874  out << indent() << "}" << endl;
4875  out << indent() << "//Should log failure after a sucessful write" << endl;
4876  out << indent() << "do_rate_limit_drop_log_ = true;" << endl;
4877  out << indent() << "rate_limit_buffer_.push_back(current_time);" << endl;
4878  out << indent() << "return true;" << endl;
4879 }
4880 
4881 
4882 #endif
4883 
4892 void t_cpp_generator::generate_struct_result_writer(ofstream& out,
4893  t_struct* tstruct,
4894  bool pointers) {
4895  string name = tstruct->get_name();
4896  const vector<t_field*>& fields = tstruct->get_sorted_members();
4897  vector<t_field*>::const_iterator f_iter;
4898 
4899  if (gen_templates_) {
4900  out <<
4901  indent() << "template <class Protocol_>" << endl <<
4902  indent() << "uint32_t " << tstruct->get_name() <<
4903  "::write(Protocol_* oprot) const {" << endl;
4904  } else {
4905  indent(out) <<
4906  "uint32_t " << tstruct->get_name() <<
4907  "::write(::contrail::sandesh::protocol::TProtocol* oprot) const {" << endl;
4908  }
4909  indent_up();
4910 
4911  out <<
4912  endl <<
4913  indent() << "uint32_t xfer = 0;" << endl <<
4914  endl;
4915 
4916  indent(out) <<
4917  "xfer += oprot->writeStructBegin(\"" << name << "\");" << endl;
4918 
4919  bool first = true;
4920  for (f_iter = fields.begin(); f_iter != fields.end(); ++f_iter) {
4921  if (first) {
4922  first = false;
4923  out <<
4924  endl <<
4925  indent() << "if ";
4926  } else {
4927  out <<
4928  " else if ";
4929  }
4930 
4931  out << "(this->__isset." << (*f_iter)->get_name() << ") {" << endl;
4932 
4933  indent_up();
4934 
4935  // Write field header
4936  out <<
4937  indent() << "xfer += oprot->writeFieldBegin(" <<
4938  "\"" << (*f_iter)->get_name() << "\", " <<
4939  type_to_enum((*f_iter)->get_type()) << ", " <<
4940  (*f_iter)->get_key() << ");" << endl;
4941  // Write field contents
4942  if (pointers) {
4943  generate_serialize_field(out, *f_iter, "(*(this->", "))");
4944  } else {
4945  generate_serialize_field(out, *f_iter, "this->");
4946  }
4947  // Write field closer
4948  indent(out) << "xfer += oprot->writeFieldEnd();" << endl;
4949 
4950  indent_down();
4951  indent(out) << "}";
4952  }
4953 
4954  // Write the struct map
4955  out <<
4956  endl <<
4957  indent() << "xfer += oprot->writeFieldStop();" << endl <<
4958  indent() << "xfer += oprot->writeStructEnd();" << endl <<
4959  indent() << "return xfer;" << endl;
4960 
4961  indent_down();
4962  indent(out) <<
4963  "}" << endl <<
4964  endl;
4965 }
4966 
4975 void t_cpp_generator::generate_service(t_service* tservice) {
4976  string svcname = tservice->get_name();
4977 
4978  // Make output files
4979  string f_header_name = get_out_dir()+svcname+".h";
4980  f_header_.open(f_header_name.c_str());
4981 
4982  // Print header file includes
4983  f_header_ <<
4984  autogen_comment();
4985  f_header_ <<
4986  "#ifndef " << svcname << "_H" << endl <<
4987  "#define " << svcname << "_H" << endl <<
4988  endl;
4989  if (gen_cob_style_) {
4990  f_header_ <<
4991  "#include <transport/TBufferTransports.h>" << endl << // TMemoryBuffer
4992  "#include <tr1/functional>" << endl <<
4993  "namespace apache { namespace thrift { namespace async {" << endl <<
4994  "class TAsyncChannel;" << endl <<
4995  "}}}" << endl;
4996  }
4997  f_header_ <<
4998  "#include <TProcessor.h>" << endl;
4999  if (gen_cob_style_) {
5000  f_header_ <<
5001  "#include <async/TAsyncProcessor.h>" << endl;
5002  }
5003  f_header_ <<
5004  "#include \"" << get_include_prefix(*get_program()) << program_name_ <<
5005  "_types.h\"" << endl;
5006 
5007  t_service* extends_service = tservice->get_extends();
5008  if (extends_service != NULL) {
5009  f_header_ <<
5010  "#include \"" << get_include_prefix(*(extends_service->get_program())) <<
5011  extends_service->get_name() << ".h\"" << endl;
5012  }
5013 
5014  f_header_ <<
5015  endl <<
5016  ns_open_ << endl <<
5017  endl;
5018 
5019  // Service implementation file includes
5020  string f_service_name = get_out_dir()+svcname+".cpp";
5021  f_service_.open(f_service_name.c_str());
5022  f_service_ <<
5023  autogen_comment();
5024  f_service_ <<
5025  "#include \"" << get_include_prefix(*get_program()) << svcname << ".h\"" << endl;
5026  if (gen_cob_style_) {
5027  f_service_ <<
5028  "#include \"async/TAsyncChannel.h\"" << endl;
5029  }
5030  if (gen_templates_) {
5031  f_service_ <<
5032  "#include \"" << get_include_prefix(*get_program()) << svcname <<
5033  ".tcc\"" << endl;
5034 
5035  string f_service_tcc_name = get_out_dir()+svcname+".tcc";
5036  f_service_tcc_.open(f_service_tcc_name.c_str());
5037  f_service_tcc_ <<
5038  autogen_comment();
5039  f_service_tcc_ <<
5040  "#include \"" << get_include_prefix(*get_program()) << svcname <<
5041  ".h\"" << endl;
5042 
5043  f_service_tcc_ <<
5044  "#ifndef " << svcname << "_TCC" << endl <<
5045  "#define " << svcname << "_TCC" << endl <<
5046  endl;
5047 
5048  if (gen_cob_style_) {
5049  f_service_tcc_ <<
5050  "#include \"async/TAsyncChannel.h\"" << endl;
5051  }
5052  }
5053 
5054  f_service_ <<
5055  endl << ns_open_ << endl << endl;
5056  f_service_tcc_ <<
5057  endl << ns_open_ << endl << endl;
5058 
5059  // Generate all the components
5060  generate_service_interface(tservice, "");
5061  generate_service_interface_factory(tservice, "");
5062  generate_service_null(tservice, "");
5063  generate_service_helpers(tservice);
5064  generate_service_client(tservice, "");
5065  generate_service_processor(tservice, "");
5066  generate_service_multiface(tservice);
5067  generate_service_skeleton(tservice);
5068 
5069  // Generate all the cob components
5070  if (gen_cob_style_) {
5071  generate_service_interface(tservice, "CobCl");
5072  generate_service_interface(tservice, "CobSv");
5073  generate_service_interface_factory(tservice, "CobSv");
5074  generate_service_null(tservice, "CobSv");
5075  generate_service_client(tservice, "Cob");
5076  generate_service_processor(tservice, "Cob");
5077  generate_service_async_skeleton(tservice);
5078  }
5079 
5080  // Close the namespace
5081  f_service_ <<
5082  ns_close_ << endl <<
5083  endl;
5084  f_service_tcc_ <<
5085  ns_close_ << endl <<
5086  endl;
5087  f_header_ <<
5088  ns_close_ << endl <<
5089  endl;
5090 
5091  // TODO(simpkins): Make this a separate option
5092  if (gen_templates_) {
5093  f_header_ <<
5094  "#include \"" << get_include_prefix(*get_program()) << svcname <<
5095  ".tcc\"" << endl <<
5096  "#include \"" << get_include_prefix(*get_program()) << program_name_ <<
5097  "_types.tcc\"" << endl <<
5098  endl;
5099  }
5100 
5101  f_header_ <<
5102  "#endif" << endl;
5103  f_service_tcc_ <<
5104  "#endif" << endl;
5105 
5106  // Close the files
5107  f_service_tcc_.close();
5108  f_service_.close();
5109  f_header_.close();
5110 }
5111 
5118 void t_cpp_generator::generate_service_helpers(t_service* tservice) {
5119  vector<t_function*> functions = tservice->get_functions();
5120  vector<t_function*>::iterator f_iter;
5121  std::ofstream& out = (gen_templates_ ? f_service_tcc_ : f_service_);
5122 
5123  for (f_iter = functions.begin(); f_iter != functions.end(); ++f_iter) {
5124  t_struct* ts = (*f_iter)->get_arglist();
5125  string name_orig = ts->get_name();
5126 
5127  // TODO(dreiss): Why is this stuff not in generate_function_helpers?
5128  ts->set_name(tservice->get_name() + "_" + (*f_iter)->get_name() + "_args");
5129  generate_struct_definition(f_header_, ts, false);
5130  generate_struct_reader(out, ts);
5131  generate_struct_writer(out, ts);
5132  ts->set_name(tservice->get_name() + "_" + (*f_iter)->get_name() + "_pargs");
5133  generate_struct_definition(f_header_, ts, false, true, false, true);
5134  generate_struct_writer(out, ts, true);
5135  ts->set_name(name_orig);
5136 
5137  generate_function_helpers(tservice, *f_iter);
5138  }
5139 }
5140 
5146 void t_cpp_generator::generate_service_interface(t_service* tservice, string style) {
5147 
5148  string service_if_name = service_name_ + style + "If";
5149  if (style == "CobCl") {
5150  // Forward declare the client.
5151  string client_name = service_name_ + "CobClient";
5152  if (gen_templates_) {
5153  client_name += "T";
5154  service_if_name += "T";
5155  indent(f_header_) <<
5156  "template <class Protocol_>" << endl;
5157  }
5158  indent(f_header_) << "class " << client_name << ";" <<
5159  endl << endl;
5160  }
5161 
5162  string extends = "";
5163  if (tservice->get_extends() != NULL) {
5164  extends = " : virtual public " + type_name(tservice->get_extends()) +
5165  style + "If";
5166  if (style == "CobCl" && gen_templates_) {
5167  // TODO(simpkins): If gen_templates_ is enabled, we currently assume all
5168  // parent services were also generated with templates enabled.
5169  extends += "T<Protocol_>";
5170  }
5171  }
5172 
5173  if (style == "CobCl" && gen_templates_) {
5174  f_header_ << "template <class Protocol_>" << endl;
5175  }
5176  f_header_ <<
5177  "class " << service_if_name << extends << " {" << endl <<
5178  " public:" << endl;
5179  indent_up();
5180  f_header_ <<
5181  indent() << "virtual ~" << service_if_name << "() {}" << endl;
5182 
5183  vector<t_function*> functions = tservice->get_functions();
5184  vector<t_function*>::iterator f_iter;
5185  for (f_iter = functions.begin(); f_iter != functions.end(); ++f_iter) {
5186  f_header_ <<
5187  indent() << "virtual " << function_signature(*f_iter, style) << " = 0;" << endl;
5188  }
5189  indent_down();
5190  f_header_ <<
5191  "};" << endl << endl;
5192 
5193  if (style == "CobCl" && gen_templates_) {
5194  // generate a backwards-compatible typedef for clients that do not
5195  // know about the new template-style code
5196  f_header_ <<
5197  "typedef " << service_if_name <<
5198  "< ::contrail::sandesh::protocol::TProtocol> " <<
5199  service_name_ << style << "If;" <<
5200  endl << endl;
5201  }
5202 }
5203 
5209 void t_cpp_generator::generate_service_interface_factory(t_service* tservice,
5210  string style) {
5211  string service_if_name = service_name_ + style + "If";
5212 
5213  // Figure out the name of the upper-most parent class.
5214  // Getting everything to work out properly with inheritance is annoying.
5215  // Here's what we're doing for now:
5216  //
5217  // - All handlers implement getHandler(), but subclasses use covariant return
5218  // types to return their specific service interface class type. We have to
5219  // use raw pointers because of this; shared_ptr<> can't be used for
5220  // covariant return types.
5221  //
5222  // - Since we're not using shared_ptr<>, we also provide a releaseHandler()
5223  // function that must be called to release a pointer to a handler obtained
5224  // via getHandler().
5225  //
5226  // releaseHandler() always accepts a pointer to the upper-most parent class
5227  // type. This is necessary since the parent versions of releaseHandler()
5228  // may accept any of the parent types, not just the most specific subclass
5229  // type. Implementations can use dynamic_cast to cast the pointer to the
5230  // subclass type if desired.
5231  t_service* base_service = tservice;
5232  while (base_service->get_extends() != NULL) {
5233  base_service = base_service->get_extends();
5234  }
5235  string base_if_name = type_name(base_service) + style + "If";
5236 
5237  // Generate the abstract factory class
5238  string factory_name = service_if_name + "Factory";
5239  string extends;
5240  if (tservice->get_extends() != NULL) {
5241  extends = " : virtual public " + type_name(tservice->get_extends()) +
5242  style + "IfFactory";
5243  }
5244 
5245  f_header_ <<
5246  "class " << factory_name << extends << " {" << endl <<
5247  " public:" << endl;
5248  indent_up();
5249  f_header_ <<
5250  indent() << "typedef " << service_if_name << " Handler;" << endl <<
5251  endl <<
5252  indent() << "virtual ~" << factory_name << "() {}" << endl <<
5253  endl <<
5254  indent() << "virtual " << service_if_name << "* getHandler(" <<
5255  "const ::contrail::sandesh::TConnectionInfo& connInfo) = 0;" <<
5256  endl <<
5257  indent() << "virtual void releaseHandler(" << base_if_name <<
5258  "* /* handler */) = 0;" << endl;
5259 
5260  indent_down();
5261  f_header_ <<
5262  "};" << endl << endl;
5263 
5264  // Generate the singleton factory class
5265  string singleton_factory_name = service_if_name + "SingletonFactory";
5266  f_header_ <<
5267  "class " << singleton_factory_name <<
5268  " : virtual public " << factory_name << " {" << endl <<
5269  " public:" << endl;
5270  indent_up();
5271  f_header_ <<
5272  indent() << singleton_factory_name << "(const boost::shared_ptr<" <<
5273  service_if_name << ">& iface) : iface_(iface) {}" << endl <<
5274  indent() << "virtual ~" << singleton_factory_name << "() {}" << endl <<
5275  endl <<
5276  indent() << "virtual " << service_if_name << "* getHandler(" <<
5277  "const ::contrail::sandesh::TConnectionInfo&) {" << endl <<
5278  indent() << " return iface_.get();" << endl <<
5279  indent() << "}" << endl <<
5280  indent() << "virtual void releaseHandler(" << base_if_name <<
5281  "* /* handler */) {}" << endl;
5282 
5283  f_header_ <<
5284  endl <<
5285  " protected:" << endl <<
5286  indent() << "boost::shared_ptr<" << service_if_name << "> iface_;" << endl;
5287 
5288  indent_down();
5289  f_header_ <<
5290  "};" << endl << endl;
5291 }
5292 
5298 void t_cpp_generator::generate_service_null(t_service* tservice, string style) {
5299  string extends = "";
5300  if (tservice->get_extends() != NULL) {
5301  extends = " , virtual public " + type_name(tservice->get_extends()) + style + "Null";
5302  }
5303  f_header_ <<
5304  "class " << service_name_ << style << "Null : virtual public " << service_name_ << style << "If" << extends << " {" << endl <<
5305  " public:" << endl;
5306  indent_up();
5307  f_header_ <<
5308  indent() << "virtual ~" << service_name_ << style << "Null() {}" << endl;
5309  vector<t_function*> functions = tservice->get_functions();
5310  vector<t_function*>::iterator f_iter;
5311  for (f_iter = functions.begin(); f_iter != functions.end(); ++f_iter) {
5312  f_header_ <<
5313  indent() << function_signature(*f_iter, style, "", false) << " {" << endl;
5314  indent_up();
5315 
5316  t_type* returntype = (*f_iter)->get_returntype();
5317  t_field returnfield(returntype, "_return");
5318 
5319  if (style == "") {
5320  if (returntype->is_void() || is_complex_type(returntype)) {
5321  f_header_ << indent() << "return;" << endl;
5322  } else {
5323  f_header_ <<
5324  indent() << declare_field(&returnfield, true) << endl <<
5325  indent() << "return _return;" << endl;
5326  }
5327  } else if (style == "CobSv") {
5328  if (returntype->is_void()) {
5329  f_header_ << indent() << "return cob();" << endl;
5330  } else {
5331  t_field returnfield(returntype, "_return");
5332  f_header_ <<
5333  indent() << declare_field(&returnfield, true) << endl <<
5334  indent() << "return cob(_return);" << endl;
5335  }
5336 
5337  } else {
5338  throw "UNKNOWN STYLE";
5339  }
5340 
5341  indent_down();
5342  f_header_ <<
5343  indent() << "}" << endl;
5344  }
5345  indent_down();
5346  f_header_ <<
5347  "};" << endl << endl;
5348 }
5349 
5350 void t_cpp_generator::generate_function_call(ostream& out, t_function* tfunction, string target, string iface, string arg_prefix) {
5351  bool first = true;
5352  t_type* ret_type = get_true_type(tfunction->get_returntype());
5353  out << indent();
5354  if (!tfunction->is_oneway() && !ret_type->is_void()) {
5355  if (is_complex_type(ret_type)) {
5356  first = false;
5357  out << iface << "->" << tfunction->get_name() << "(" << target;
5358  } else {
5359  out << target << " = " << iface << "->" << tfunction->get_name() << "(";
5360  }
5361  } else {
5362  out << iface << "->" << tfunction->get_name() << "(";
5363  }
5364  const std::vector<t_field*>& fields = tfunction->get_arglist()->get_members();
5365  vector<t_field*>::const_iterator f_iter;
5366  for (f_iter = fields.begin(); f_iter != fields.end(); ++f_iter) {
5367  if (first) {
5368  first = false;
5369  } else {
5370  out << ", ";
5371  }
5372  out << arg_prefix << (*f_iter)->get_name();
5373  }
5374  out << ");" << endl;
5375 }
5376 
5377 void t_cpp_generator::generate_service_async_skeleton(t_service* tservice) {
5378  string svcname = tservice->get_name();
5379 
5380  // Service implementation file includes
5381  string f_skeleton_name = get_out_dir()+svcname+"_async_server.skeleton.cpp";
5382 
5383  string ns = namespace_prefix(tservice->get_program()->get_namespace("cpp"));
5384 
5385  ofstream f_skeleton;
5386  f_skeleton.open(f_skeleton_name.c_str());
5387  f_skeleton <<
5388  "// This autogenerated skeleton file illustrates one way to adapt a synchronous" << endl <<
5389  "// interface into an asynchronous interface. You should copy it to another" << endl <<
5390  "// filename to avoid overwriting it and rewrite as asynchronous any functions" << endl <<
5391  "// that would otherwise introduce unwanted latency." << endl <<
5392  endl <<
5393  "#include \"" << get_include_prefix(*get_program()) << svcname << ".h\"" << endl <<
5394  "#include <protocol/TBinaryProtocol.h>" << endl <<
5395  "#include <async/TEventServer.h>" << endl <<
5396  endl <<
5397  "using namespace ::contrail::sandesh;" << endl <<
5398  "using namespace ::contrail::sandesh::protocol;" << endl <<
5399  "using namespace ::contrail::sandesh::transport;" << endl <<
5400  "using namespace ::contrail::sandesh::async;" << endl <<
5401  endl <<
5402  "using boost::shared_ptr;" << endl <<
5403  endl;
5404 
5405  if (!ns.empty()) {
5406  f_skeleton <<
5407  "using namespace " << string(ns, 0, ns.size()-2) << ";" << endl <<
5408  endl;
5409  }
5410 
5411  f_skeleton <<
5412  "class " << svcname << "AsyncHandler : " <<
5413  "public " << svcname << "CobSvIf {" << endl <<
5414  " public:" << endl;
5415  indent_up();
5416  f_skeleton <<
5417  indent() << svcname << "AsyncHandler() {" << endl <<
5418  indent() << " syncHandler_ = std::unique_ptr<" << svcname <<
5419  "Handler>(new " << svcname << "Handler);" << endl <<
5420  indent() << " // Your initialization goes here" << endl <<
5421  indent() << "}" << endl;
5422  f_skeleton <<
5423  indent() << "virtual ~" << service_name_ << "AsyncHandler();" << endl;
5424 
5425  vector<t_function*> functions = tservice->get_functions();
5426  vector<t_function*>::iterator f_iter;
5427  for (f_iter = functions.begin(); f_iter != functions.end(); ++f_iter) {
5428  f_skeleton <<
5429  endl <<
5430  indent() << function_signature(*f_iter, "CobSv", "", true) << " {" << endl;
5431  indent_up();
5432 
5433  t_type* returntype = (*f_iter)->get_returntype();
5434  t_field returnfield(returntype, "_return");
5435 
5436  string target = returntype->is_void() ? "" : "_return";
5437  if (!returntype->is_void()) {
5438  f_skeleton <<
5439  indent() << declare_field(&returnfield, true) << endl;
5440  }
5441  generate_function_call(f_skeleton, *f_iter, target, "syncHandler_", "");
5442  f_skeleton << indent() << "return cob(" << target << ");" << endl;
5443 
5444  scope_down(f_skeleton);
5445  }
5446  f_skeleton << endl <<
5447  " protected:" << endl <<
5448  indent() << "std::unique_ptr<" << svcname << "Handler> syncHandler_;" << endl;
5449  indent_down();
5450  f_skeleton <<
5451  "};" << endl << endl;
5452 }
5453 
5461 void t_cpp_generator::generate_service_multiface(t_service* tservice) {
5462  // Generate the dispatch methods
5463  vector<t_function*> functions = tservice->get_functions();
5464  vector<t_function*>::iterator f_iter;
5465 
5466  string extends = "";
5467  string extends_multiface = "";
5468  if (tservice->get_extends() != NULL) {
5469  extends = type_name(tservice->get_extends());
5470  extends_multiface = ", public " + extends + "Multiface";
5471  }
5472 
5473  string list_type = string("std::vector<boost::shared_ptr<") + service_name_ + "If> >";
5474 
5475  // Generate the header portion
5476  f_header_ <<
5477  "class " << service_name_ << "Multiface : " <<
5478  "virtual public " << service_name_ << "If" <<
5479  extends_multiface << " {" << endl <<
5480  " public:" << endl;
5481  indent_up();
5482  f_header_ <<
5483  indent() << service_name_ << "Multiface(" << list_type << "& ifaces) : ifaces_(ifaces) {" << endl;
5484  if (!extends.empty()) {
5485  f_header_ <<
5486  indent() << " std::vector<boost::shared_ptr<" + service_name_ + "If> >::iterator iter;" << endl <<
5487  indent() << " for (iter = ifaces.begin(); iter != ifaces.end(); ++iter) {" << endl <<
5488  indent() << " " << extends << "Multiface::add(*iter);" << endl <<
5489  indent() << " }" << endl;
5490  }
5491  f_header_ <<
5492  indent() << "}" << endl <<
5493  indent() << "virtual ~" << service_name_ << "Multiface() {}" << endl;
5494  indent_down();
5495 
5496  // Protected data members
5497  f_header_ <<
5498  " protected:" << endl;
5499  indent_up();
5500  f_header_ <<
5501  indent() << list_type << " ifaces_;" << endl <<
5502  indent() << service_name_ << "Multiface() {}" << endl <<
5503  indent() << "void add(boost::shared_ptr<" << service_name_ << "If> iface) {" << endl;
5504  if (!extends.empty()) {
5505  f_header_ <<
5506  indent() << " " << extends << "Multiface::add(iface);" << endl;
5507  }
5508  f_header_ <<
5509  indent() << " ifaces_.push_back(iface);" << endl <<
5510  indent() << "}" << endl;
5511  indent_down();
5512 
5513  f_header_ <<
5514  indent() << " public:" << endl;
5515  indent_up();
5516 
5517  for (f_iter = functions.begin(); f_iter != functions.end(); ++f_iter) {
5518  t_struct* arglist = (*f_iter)->get_arglist();
5519  const vector<t_field*>& args = arglist->get_members();
5520  vector<t_field*>::const_iterator a_iter;
5521 
5522  string call = string("ifaces_[i]->") + (*f_iter)->get_name() + "(";
5523  bool first = true;
5524  if (is_complex_type((*f_iter)->get_returntype())) {
5525  call += "_return";
5526  first = false;
5527  }
5528  for (a_iter = args.begin(); a_iter != args.end(); ++a_iter) {
5529  if (first) {
5530  first = false;
5531  } else {
5532  call += ", ";
5533  }
5534  call += (*a_iter)->get_name();
5535  }
5536  call += ")";
5537 
5538  f_header_ <<
5539  indent() << function_signature(*f_iter, "") << " {" << endl;
5540  indent_up();
5541  f_header_ <<
5542  indent() << "size_t sz = ifaces_.size();" << endl <<
5543  indent() << "for (size_t i = 0; i < sz; ++i) {" << endl;
5544  if (!(*f_iter)->get_returntype()->is_void()) {
5545  f_header_ <<
5546  indent() << " if (i == sz - 1) {" << endl;
5547  if (is_complex_type((*f_iter)->get_returntype())) {
5548  f_header_ <<
5549  indent() << " " << call << ";" << endl <<
5550  indent() << " return;" << endl;
5551  } else {
5552  f_header_ <<
5553  indent() << " return " << call << ";" << endl;
5554  }
5555  f_header_ <<
5556  indent() << " } else {" << endl <<
5557  indent() << " " << call << ";" << endl <<
5558  indent() << " }" << endl;
5559  } else {
5560  f_header_ <<
5561  indent() << " " << call << ";" << endl;
5562  }
5563 
5564  f_header_ <<
5565  indent() << "}" << endl;
5566 
5567  indent_down();
5568  f_header_ <<
5569  indent() << "}" << endl <<
5570  endl;
5571  }
5572 
5573  indent_down();
5574  f_header_ <<
5575  indent() << "};" << endl <<
5576  endl;
5577 }
5578 
5584 void t_cpp_generator::generate_service_client(t_service* tservice, string style) {
5585  string ifstyle;
5586  if (style == "Cob") {
5587  ifstyle = "CobCl";
5588  }
5589 
5590  std::ofstream& out = (gen_templates_ ? f_service_tcc_ : f_service_);
5591  string template_header, template_suffix, short_suffix, protocol_type, _this;
5592  string const prot_factory_type =
5593  "::contrail::sandesh::protocol::TProtocolFactory";
5594  if (gen_templates_) {
5595  template_header = "template <class Protocol_>\n";
5596  short_suffix = "T";
5597  template_suffix = "T<Protocol_>";
5598  protocol_type = "Protocol_";
5599  _this = "this->";
5600  } else {
5601  protocol_type = "::contrail::sandesh::protocol::TProtocol";
5602  }
5603  string prot_ptr = "boost::shared_ptr< " + protocol_type + ">";
5604  string client_suffix = "Client" + template_suffix;
5605  string if_suffix = "If";
5606  if (style == "Cob") {
5607  if_suffix += template_suffix;
5608  }
5609 
5610  string extends = "";
5611  string extends_client = "";
5612  if (tservice->get_extends() != NULL) {
5613  // TODO(simpkins): If gen_templates_ is enabled, we currently assume all
5614  // parent services were also generated with templates enabled.
5615  extends = type_name(tservice->get_extends());
5616  extends_client = ", public " + extends + style + client_suffix;
5617  }
5618 
5619  // Generate the header portion
5620  f_header_ <<
5621  template_header <<
5622  "class " << service_name_ << style << "Client" << short_suffix << " : " <<
5623  "virtual public " << service_name_ << ifstyle << if_suffix <<
5624  extends_client << " {" << endl <<
5625  " public:" << endl;
5626 
5627  indent_up();
5628  if (style != "Cob") {
5629  f_header_ <<
5630  indent() << service_name_ << style << "Client" << short_suffix <<
5631  "(" << prot_ptr << " prot) :" <<
5632  endl;
5633  if (extends.empty()) {
5634  f_header_ <<
5635  indent() << " piprot_(prot)," << endl <<
5636  indent() << " poprot_(prot) {" << endl <<
5637  indent() << " iprot_ = prot.get();" << endl <<
5638  indent() << " oprot_ = prot.get();" << endl <<
5639  indent() << "}" << endl;
5640  } else {
5641  f_header_ <<
5642  indent() << " " << extends << style << client_suffix <<
5643  "(prot, prot) {}" << endl;
5644  }
5645 
5646  f_header_ <<
5647  indent() << service_name_ << style << "Client" << short_suffix <<
5648  "(" << prot_ptr << " iprot, " << prot_ptr << " oprot) :" << endl;
5649  if (extends.empty()) {
5650  f_header_ <<
5651  indent() << " piprot_(iprot)," << endl <<
5652  indent() << " poprot_(oprot) {" << endl <<
5653  indent() << " iprot_ = iprot.get();" << endl <<
5654  indent() << " oprot_ = oprot.get();" << endl <<
5655  indent() << "}" << endl;
5656  } else {
5657  f_header_ <<
5658  indent() << " " << extends << style << client_suffix <<
5659  "(iprot, oprot) {}" << endl;
5660  }
5661 
5662  // Generate getters for the protocols.
5663  // Note that these are not currently templated for simplicity.
5664  // TODO(simpkins): should they be templated?
5665  f_header_ <<
5666  indent() << "boost::shared_ptr< ::contrail::sandesh::protocol::TProtocol> getInputProtocol() {" << endl <<
5667  indent() << " return " << _this << "piprot_;" << endl <<
5668  indent() << "}" << endl;
5669 
5670  f_header_ <<
5671  indent() << "boost::shared_ptr< ::contrail::sandesh::protocol::TProtocol> getOutputProtocol() {" << endl <<
5672  indent() << " return " << _this << "poprot_;" << endl <<
5673  indent() << "}" << endl;
5674 
5675  } else /* if (style == "Cob") */ {
5676  f_header_ <<
5677  indent() << service_name_ << style << "Client" << short_suffix << "(" <<
5678  "boost::shared_ptr< ::contrail::sandesh::async::TAsyncChannel> channel, " <<
5679  "::contrail::sandesh::protocol::TProtocolFactory* protocolFactory) :" <<
5680  endl;
5681  if (extends.empty()) {
5682  f_header_ <<
5683  indent() << " channel_(channel)," << endl <<
5684  indent() << " itrans_(new ::contrail::sandesh::transport::TMemoryBuffer())," << endl <<
5685  indent() << " otrans_(new ::contrail::sandesh::transport::TMemoryBuffer())," << endl;
5686  if (gen_templates_) {
5687  // TProtocolFactory classes return generic TProtocol pointers.
5688  // We have to dynamic cast to the Protocol_ type we are expecting.
5689  f_header_ <<
5690  indent() << " piprot_(boost::dynamic_pointer_cast<Protocol_>(" <<
5691  "protocolFactory->getProtocol(itrans_)))," << endl <<
5692  indent() << " poprot_(boost::dynamic_pointer_cast<Protocol_>(" <<
5693  "protocolFactory->getProtocol(otrans_))) {" << endl;
5694  // Throw a TException if either dynamic cast failed.
5695  f_header_ <<
5696  indent() << " if (!piprot_ || !poprot_) {" << endl <<
5697  indent() << " throw ::contrail::sandesh::TException(\"" <<
5698  "TProtocolFactory returned unexpected protocol type in " <<
5699  service_name_ << style << "Client" << short_suffix <<
5700  " constructor\");" << endl <<
5701  indent() << " }" << endl;
5702  } else {
5703  f_header_ <<
5704  indent() << " piprot_(protocolFactory->getProtocol(itrans_))," <<
5705  endl <<
5706  indent() << " poprot_(protocolFactory->getProtocol(otrans_)) {" <<
5707  endl;
5708  }
5709  f_header_ <<
5710  indent() << " iprot_ = piprot_.get();" << endl <<
5711  indent() << " oprot_ = poprot_.get();" << endl <<
5712  indent() << "}" << endl;
5713  } else {
5714  f_header_ <<
5715  indent() << " " << extends << style << client_suffix <<
5716  "(channel, protocolFactory) {}" << endl;
5717  }
5718  }
5719 
5720  if (style == "Cob") {
5721  f_header_ <<
5722  indent() << "boost::shared_ptr< ::contrail::sandesh::async::TAsyncChannel> getChannel() {" << endl <<
5723  indent() << " return " << _this << "channel_;" << endl <<
5724  indent() << "}" << endl;
5725  if (!gen_no_client_completion_) {
5726  f_header_ <<
5727  indent() << "virtual void completed__(bool /* success */) {}" << endl;
5728  }
5729  }
5730 
5731  vector<t_function*> functions = tservice->get_functions();
5732  vector<t_function*>::const_iterator f_iter;
5733  for (f_iter = functions.begin(); f_iter != functions.end(); ++f_iter) {
5734  indent(f_header_) << function_signature(*f_iter, ifstyle) << ";" << endl;
5735  // TODO(dreiss): Use private inheritance to avoid generating thise in cob-style.
5736  t_function send_function(g_type_void,
5737  string("send_") + (*f_iter)->get_name(),
5738  (*f_iter)->get_arglist());
5739  indent(f_header_) << function_signature(&send_function, "") << ";" << endl;
5740  if (!(*f_iter)->is_oneway()) {
5741  t_struct noargs(program_);
5742  t_function recv_function((*f_iter)->get_returntype(),
5743  string("recv_") + (*f_iter)->get_name(),
5744  &noargs);
5745  indent(f_header_) << function_signature(&recv_function, "") << ";" << endl;
5746  }
5747  }
5748  indent_down();
5749 
5750  if (extends.empty()) {
5751  f_header_ <<
5752  " protected:" << endl;
5753  indent_up();
5754 
5755  if (style == "Cob") {
5756  f_header_ <<
5757  indent() << "boost::shared_ptr< ::contrail::sandesh::async::TAsyncChannel> channel_;" << endl <<
5758  indent() << "boost::shared_ptr< ::contrail::sandesh::transport::TMemoryBuffer> itrans_;" << endl <<
5759  indent() << "boost::shared_ptr< ::contrail::sandesh::transport::TMemoryBuffer> otrans_;" << endl;
5760  }
5761  f_header_ <<
5762  indent() << prot_ptr << " piprot_;" << endl <<
5763  indent() << prot_ptr << " poprot_;" << endl <<
5764  indent() << protocol_type << "* iprot_;" << endl <<
5765  indent() << protocol_type << "* oprot_;" << endl;
5766 
5767  indent_down();
5768  }
5769 
5770  f_header_ <<
5771  "};" << endl <<
5772  endl;
5773 
5774  if (gen_templates_) {
5775  // Output a backwards compatibility typedef using
5776  // TProtocol as the template parameter.
5777  f_header_ <<
5778  "typedef " << service_name_ << style <<
5779  "ClientT< ::contrail::sandesh::protocol::TProtocol> " <<
5780  service_name_ << style << "Client;" << endl <<
5781  endl;
5782  }
5783 
5784  string scope = service_name_ + style + client_suffix + "::";
5785 
5786  // Generate client method implementations
5787  for (f_iter = functions.begin(); f_iter != functions.end(); ++f_iter) {
5788  string funname = (*f_iter)->get_name();
5789 
5790  // Open function
5791  if (gen_templates_) {
5792  indent(out) << template_header;
5793  }
5794  indent(out) <<
5795  function_signature(*f_iter, ifstyle, scope) << endl;
5796  scope_up(out);
5797  indent(out) <<
5798  "send_" << funname << "(";
5799 
5800  // Get the struct of function call params
5801  t_struct* arg_struct = (*f_iter)->get_arglist();
5802 
5803  // Declare the function arguments
5804  const vector<t_field*>& fields = arg_struct->get_members();
5805  vector<t_field*>::const_iterator fld_iter;
5806  bool first = true;
5807  for (fld_iter = fields.begin(); fld_iter != fields.end(); ++fld_iter) {
5808  if (first) {
5809  first = false;
5810  } else {
5811  out << ", ";
5812  }
5813  out << (*fld_iter)->get_name();
5814  }
5815  out << ");" << endl;
5816 
5817  if (style != "Cob") {
5818  if (!(*f_iter)->is_oneway()) {
5819  out << indent();
5820  if (!(*f_iter)->get_returntype()->is_void()) {
5821  if (is_complex_type((*f_iter)->get_returntype())) {
5822  out << "recv_" << funname << "(_return);" << endl;
5823  } else {
5824  out << "return recv_" << funname << "();" << endl;
5825  }
5826  } else {
5827  out <<
5828  "recv_" << funname << "();" << endl;
5829  }
5830  }
5831  } else {
5832  if (!(*f_iter)->is_oneway()) {
5833  out <<
5834  indent() << _this << "channel_->sendAndRecvMessage(" <<
5835  "std::tr1::bind(cob, this), " << _this << "otrans_.get(), " <<
5836  _this << "itrans_.get());" << endl;
5837  } else {
5838  out <<
5839  indent() << _this << "channel_->sendMessage(" <<
5840  "std::tr1::bind(cob, this), " << _this << "otrans_.get());" << endl;
5841  }
5842  }
5843  scope_down(out);
5844  out << endl;
5845 
5846  //if (style != "Cob") // TODO(dreiss): Libify the client and don't generate this for cob-style
5847  if (true) {
5848  // Function for sending
5849  t_function send_function(g_type_void,
5850  string("send_") + (*f_iter)->get_name(),
5851  (*f_iter)->get_arglist());
5852 
5853  // Open the send function
5854  if (gen_templates_) {
5855  indent(out) << template_header;
5856  }
5857  indent(out) <<
5858  function_signature(&send_function, "", scope) << endl;
5859  scope_up(out);
5860 
5861  // Function arguments and results
5862  string argsname = tservice->get_name() + "_" + (*f_iter)->get_name() + "_pargs";
5863  string resultname = tservice->get_name() + "_" + (*f_iter)->get_name() + "_presult";
5864 
5865  // Serialize the request
5866  out <<
5867  indent() << "int32_t cseqid = 0;" << endl <<
5868  indent() << _this << "oprot_->writeMessageBegin(\"" <<
5869  (*f_iter)->get_name() <<
5870  "\", ::contrail::sandesh::protocol::T_CALL, cseqid);" << endl <<
5871  endl <<
5872  indent() << argsname << " args;" << endl;
5873 
5874  for (fld_iter = fields.begin(); fld_iter != fields.end(); ++fld_iter) {
5875  out <<
5876  indent() << "args." << (*fld_iter)->get_name() << " = &" << (*fld_iter)->get_name() << ";" << endl;
5877  }
5878 
5879  out <<
5880  indent() << "args.write(" << _this << "oprot_);" << endl <<
5881  endl <<
5882  indent() << _this << "oprot_->writeMessageEnd();" << endl <<
5883  indent() << _this << "oprot_->getTransport()->writeEnd();" << endl <<
5884  indent() << _this << "oprot_->getTransport()->flush();" << endl;
5885 
5886  scope_down(out);
5887  out << endl;
5888 
5889  // Generate recv function only if not an oneway function
5890  if (!(*f_iter)->is_oneway()) {
5891  t_struct noargs(program_);
5892  t_function recv_function((*f_iter)->get_returntype(),
5893  string("recv_") + (*f_iter)->get_name(),
5894  &noargs);
5895  // Open the recv function
5896  if (gen_templates_) {
5897  indent(out) << template_header;
5898  }
5899  indent(out) <<
5900  function_signature(&recv_function, "", scope) << endl;
5901  scope_up(out);
5902 
5903  out <<
5904  endl <<
5905  indent() << "int32_t rseqid = 0;" << endl <<
5906  indent() << "std::string fname;" << endl <<
5907  indent() << "::contrail::sandesh::protocol::TMessageType mtype;" << endl;
5908  if (style == "Cob" && !gen_no_client_completion_) {
5909  out <<
5910  indent() << "bool completed = false;" << endl << endl <<
5911  indent() << "try {";
5912  indent_up();
5913  }
5914  out << endl <<
5915  indent() << _this << "iprot_->readMessageBegin(fname, mtype, rseqid);" << endl <<
5916  indent() << "if (mtype == ::contrail::sandesh::protocol::T_EXCEPTION) {" << endl <<
5917  indent() << " ::contrail::sandesh::TApplicationException x;" << endl <<
5918  indent() << " x.read(" << _this << "iprot_);" << endl <<
5919  indent() << " " << _this << "iprot_->readMessageEnd();" << endl <<
5920  indent() << " " << _this << "iprot_->getTransport()->readEnd();" <<
5921  endl;
5922  if (style == "Cob" && !gen_no_client_completion_) {
5923  out <<
5924  indent() << " completed = true;" << endl <<
5925  indent() << " completed__(true);" << endl;
5926  }
5927  out <<
5928  indent() << " throw x;" << endl <<
5929  indent() << "}" << endl <<
5930  indent() << "if (mtype != ::contrail::sandesh::protocol::T_REPLY) {" << endl <<
5931  indent() << " " << _this << "iprot_->skip(" <<
5932  "::contrail::sandesh::protocol::T_STRUCT);" << endl <<
5933  indent() << " " << _this << "iprot_->readMessageEnd();" << endl <<
5934  indent() << " " << _this << "iprot_->getTransport()->readEnd();" <<
5935  endl;
5936  if (style == "Cob" && !gen_no_client_completion_) {
5937  out <<
5938  indent() << " completed = true;" << endl <<
5939  indent() << " completed__(false);" << endl;
5940  }
5941  out <<
5942  indent() << "}" << endl <<
5943  indent() << "if (fname.compare(\"" << (*f_iter)->get_name() << "\") != 0) {" << endl <<
5944  indent() << " " << _this << "iprot_->skip(" <<
5945  "::contrail::sandesh::protocol::T_STRUCT);" << endl <<
5946  indent() << " " << _this << "iprot_->readMessageEnd();" << endl <<
5947  indent() << " " << _this << "iprot_->getTransport()->readEnd();" <<
5948  endl;
5949  if (style == "Cob" && !gen_no_client_completion_) {
5950  out <<
5951  indent() << " completed = true;" << endl <<
5952  indent() << " completed__(false);" << endl;
5953  }
5954  out <<
5955  indent() << "}" << endl;
5956 
5957  if (!(*f_iter)->get_returntype()->is_void() &&
5958  !is_complex_type((*f_iter)->get_returntype())) {
5959  t_field returnfield((*f_iter)->get_returntype(), "_return");
5960  out <<
5961  indent() << declare_field(&returnfield) << endl;
5962  }
5963 
5964  out <<
5965  indent() << resultname << " result;" << endl;
5966 
5967  if (!(*f_iter)->get_returntype()->is_void()) {
5968  out <<
5969  indent() << "result.success = &_return;" << endl;
5970  }
5971 
5972  out <<
5973  indent() << "result.read(" << _this << "iprot_);" << endl <<
5974  indent() << _this << "iprot_->readMessageEnd();" << endl <<
5975  indent() << _this << "iprot_->getTransport()->readEnd();" << endl <<
5976  endl;
5977 
5978  // Careful, only look for _result if not a void function
5979  if (!(*f_iter)->get_returntype()->is_void()) {
5980  if (is_complex_type((*f_iter)->get_returntype())) {
5981  out <<
5982  indent() << "if (result.__isset.success) {" << endl <<
5983  indent() << " // _return pointer has now been filled" << endl;
5984  if (style == "Cob" && !gen_no_client_completion_) {
5985  out <<
5986  indent() << " completed = true;" << endl <<
5987  indent() << " completed__(true);" << endl;
5988  }
5989  out <<
5990  indent() << " return;" << endl <<
5991  indent() << "}" << endl;
5992  } else {
5993  out <<
5994  indent() << "if (result.__isset.success) {" << endl;
5995  if (style == "Cob" && !gen_no_client_completion_) {
5996  out <<
5997  indent() << " completed = true;" << endl <<
5998  indent() << " completed__(true);" << endl;
5999  }
6000  out <<
6001  indent() << " return _return;" << endl <<
6002  indent() << "}" << endl;
6003  }
6004  }
6005 
6006  t_struct* xs = (*f_iter)->get_xceptions();
6007  const std::vector<t_field*>& xceptions = xs->get_members();
6008  vector<t_field*>::const_iterator x_iter;
6009  for (x_iter = xceptions.begin(); x_iter != xceptions.end(); ++x_iter) {
6010  out <<
6011  indent() << "if (result.__isset." << (*x_iter)->get_name() << ") {" << endl;
6012  if (style == "Cob" && !gen_no_client_completion_) {
6013  out <<
6014  indent() << " completed = true;" << endl <<
6015  indent() << " completed__(true);" << endl;
6016  }
6017  out <<
6018  indent() << " throw result." << (*x_iter)->get_name() << ";" << endl <<
6019  indent() << "}" << endl;
6020  }
6021 
6022  // We only get here if we are a void function
6023  if ((*f_iter)->get_returntype()->is_void()) {
6024  if (style == "Cob" && !gen_no_client_completion_) {
6025  out <<
6026  indent() << "completed = true;" << endl <<
6027  indent() << "completed__(true);" << endl;
6028  }
6029  indent(out) <<
6030  "return;" << endl;
6031  } else {
6032  if (style == "Cob" && !gen_no_client_completion_) {
6033  out <<
6034  indent() << "completed = true;" << endl <<
6035  indent() << "completed__(true);" << endl;
6036  }
6037  out <<
6038  indent() << "throw ::contrail::sandesh::TApplicationException(::contrail::sandesh::TApplicationException::MISSING_RESULT, \"" << (*f_iter)->get_name() << " failed: unknown result\");" << endl;
6039  }
6040  if (style == "Cob" && !gen_no_client_completion_) {
6041  indent_down();
6042  out <<
6043  indent() << "} catch (...) {" << endl <<
6044  indent() << " if (!completed) {" << endl <<
6045  indent() << " completed__(false);" << endl <<
6046  indent() << " }" << endl <<
6047  indent() << " throw;" << endl <<
6048  indent() << "}" << endl;
6049  }
6050  // Close function
6051  scope_down(out);
6052  out << endl;
6053  }
6054  }
6055  }
6056 }
6057 
6058 class ProcessorGenerator {
6059  public:
6060  ProcessorGenerator(t_cpp_generator* generator, t_service* service,
6061  const string& style);
6062 
6063  void run() {
6064  generate_class_definition();
6065  // Generate the process() function
6066  generate_process();
6067  // Generate the process_fn() function
6068  generate_process_fn();
6069  // Generate all of the process subfunctions
6070  generate_process_functions();
6071 
6072  generate_factory();
6073  }
6074 
6075  void generate_class_definition();
6076  void generate_process();
6077  void generate_process_fn();
6078  void generate_process_functions();
6079  void generate_factory();
6080 
6081  protected:
6082  std::string type_name(t_type* ttype, bool in_typedef=false, bool arg=false) {
6083  return generator_->type_name(ttype, in_typedef, arg);
6084  }
6085 
6086  std::string indent() {
6087  return generator_->indent();
6088  }
6089  std::ostream& indent(std::ostream &os) {
6090  return generator_->indent(os);
6091  }
6092 
6093  void indent_up() {
6094  generator_->indent_up();
6095  }
6096  void indent_down() {
6097  generator_->indent_down();
6098  }
6099 
6100  t_cpp_generator* generator_;
6101  t_service* service_;
6102  std::ofstream& f_header_;
6103  std::ofstream& f_out_;
6104  string service_name_;
6105  string style_;
6106  string pstyle_;
6107  string class_name_;
6108  string if_name_;
6109  string factory_class_name_;
6110  string finish_cob_;
6111  string finish_cob_decl_;
6112  string ret_type_;
6113  string call_context_;
6114  string cob_arg_;
6115  string call_context_arg_;
6116  string call_context_decl_;
6117  string template_header_;
6118  string template_suffix_;
6119  string typename_str_;
6120  string class_suffix_;
6121  string extends_;
6122  string extends_processor_;
6123 };
6124 
6125 ProcessorGenerator::ProcessorGenerator(t_cpp_generator* generator,
6126  t_service* service,
6127  const string& style)
6128  : generator_(generator),
6129  service_(service),
6130  f_header_(generator->f_header_),
6131  f_out_(generator->gen_templates_ ?
6132  generator->f_service_tcc_ : generator->f_service_),
6133  service_name_(generator->service_name_),
6134  style_(style) {
6135  if (style_ == "Cob") {
6136  pstyle_ = "Async";
6137  class_name_ = service_name_ + pstyle_ + "Processor";
6138  if_name_ = service_name_ + "CobSvIf";
6139 
6140  finish_cob_ = "std::tr1::function<void(bool ok)> cob, ";
6141  finish_cob_decl_ = "std::tr1::function<void(bool ok)>, ";
6142  cob_arg_ = "cob, ";
6143  ret_type_ = "void ";
6144  } else {
6145  class_name_ = service_name_ + "Processor";
6146  if_name_ = service_name_ + "If";
6147 
6148  ret_type_ = "bool ";
6149  // TODO(edhall) callContext should eventually be added to TAsyncProcessor
6150  call_context_ = ", void* callContext";
6151  call_context_arg_ = ", callContext";
6152  call_context_decl_ = ", void*";
6153  }
6154 
6155  factory_class_name_ = class_name_ + "Factory";
6156 
6157  if (generator->gen_templates_) {
6158  template_header_ = "template <class Protocol_>\n";
6159  template_suffix_ = "<Protocol_>";
6160  typename_str_ = "typename ";
6161  class_name_ += "T";
6162  factory_class_name_ += "T";
6163  }
6164 
6165  if (service_->get_extends() != NULL) {
6166  extends_ = type_name(service_->get_extends()) + pstyle_ + "Processor";
6167  if (generator_->gen_templates_) {
6168  // TODO(simpkins): If gen_templates_ is enabled, we currently assume all
6169  // parent services were also generated with templates enabled.
6170  extends_ += "T<Protocol_>";
6171  }
6172  extends_processor_ = extends_;
6173  } else {
6174  extends_processor_ = "::contrail::sandesh::T" + pstyle_ + "Processor";
6175  }
6176 }
6177 
6178 void ProcessorGenerator::generate_class_definition() {
6179  // Generate the dispatch methods
6180  vector<t_function*> functions = service_->get_functions();
6181  vector<t_function*>::iterator f_iter;
6182 
6183  // Generate the header portion
6184  f_header_ <<
6185  template_header_ <<
6186  "class " << class_name_ <<
6187  " : public " << extends_processor_ << " {" << endl;
6188 
6189  // Protected data members
6190  f_header_ <<
6191  " protected:" << endl;
6192  indent_up();
6193  f_header_ <<
6194  indent() << "boost::shared_ptr<" << if_name_ << "> iface_;" << endl;
6195  f_header_ <<
6196  indent() << "virtual " << ret_type_ << "process_fn(" << finish_cob_ << "contrail::sandesh::protocol::TProtocol* iprot, contrail::sandesh::protocol::TProtocol* oprot, std::string& fname, int32_t seqid" << call_context_ << ");" << endl;
6197  indent_down();
6198 
6199  // Process function declarations
6200  f_header_ <<
6201  " private:" << endl;
6202  indent_up();
6203  f_header_ <<
6204  indent() << "std::map<std::string, void (" <<
6205  class_name_ << "::*)(" << finish_cob_decl_ <<
6206  "int32_t, contrail::sandesh::protocol::TProtocol*, " <<
6207  "contrail::sandesh::protocol::TProtocol*" << call_context_decl_ <<
6208  ")> processMap_;" << endl;
6209  for (f_iter = functions.begin(); f_iter != functions.end(); ++f_iter) {
6210  indent(f_header_) <<
6211  "void process_" << (*f_iter)->get_name() << "(" << finish_cob_ << "int32_t seqid, contrail::sandesh::protocol::TProtocol* iprot, contrail::sandesh::protocol::TProtocol* oprot" << call_context_ << ");" << endl;
6212  if (generator_->gen_templates_) {
6213  indent(f_header_) <<
6214  "void process_" << (*f_iter)->get_name() << "(" << finish_cob_ <<
6215  "int32_t seqid, Protocol_* iprot, Protocol_* oprot" <<
6216  call_context_ << ");" << endl;
6217  }
6218  if (style_ == "Cob") {
6219  // XXX Factor this out, even if it is a pain.
6220  string ret_arg = ((*f_iter)->get_returntype()->is_void()
6221  ? ""
6222  : ", const " + type_name((*f_iter)->get_returntype()) + "& _return");
6223  f_header_ <<
6224  indent() << "void return_" << (*f_iter)->get_name() <<
6225  "(std::tr1::function<void(bool ok)> cob, int32_t seqid, " <<
6226  "::contrail::sandesh::protocol::TProtocol* oprot, " <<
6227  "void* ctx" << ret_arg << ");" << endl;
6228  if (generator_->gen_templates_) {
6229  f_header_ <<
6230  indent() << "void return_" << (*f_iter)->get_name() <<
6231  "(std::tr1::function<void(bool ok)> cob, int32_t seqid, " <<
6232  "Protocol_* oprot, void* ctx" << ret_arg << ");" << endl;
6233  }
6234  // XXX Don't declare throw if it doesn't exist
6235  f_header_ <<
6236  indent() << "void throw_" << (*f_iter)->get_name() <<
6237  "(std::tr1::function<void(bool ok)> cob, int32_t seqid, " <<
6238  "contrail::sandesh::protocol::TProtocol* oprot, void* ctx, " <<
6239  "contrail::sandesh::TDelayedException* _throw);" << endl;
6240  if (generator_->gen_templates_) {
6241  f_header_ <<
6242  indent() << "void throw_" << (*f_iter)->get_name() <<
6243  "(std::tr1::function<void(bool ok)> cob, int32_t seqid, " <<
6244  "Protocol_* oprot, void* ctx, " <<
6245  "::contrail::sandesh::TDelayedException* _throw);" << endl;
6246  }
6247  }
6248  }
6249  indent_down();
6250 
6251  indent_up();
6252  string declare_map = "";
6253  indent_up();
6254 
6255  for (f_iter = functions.begin(); f_iter != functions.end(); ++f_iter) {
6256  declare_map += indent();
6257  declare_map += "processMap_[\"";
6258  declare_map += (*f_iter)->get_name();
6259  declare_map += "\"] = &";
6260  declare_map += class_name_;
6261  declare_map += "::process_";
6262  declare_map += (*f_iter)->get_name();
6263  declare_map += ";\n";
6264  }
6265  indent_down();
6266 
6267  f_header_ <<
6268  " public:" << endl <<
6269  indent() << class_name_ <<
6270  "(boost::shared_ptr<" << if_name_ << "> iface) :" << endl;
6271  if (extends_.empty()) {
6272  f_header_ <<
6273  indent() << " iface_(iface) {" << endl;
6274  } else {
6275  f_header_ <<
6276  indent() << " " << extends_ << "(iface)," << endl <<
6277  indent() << " iface_(iface) {" << endl;
6278  }
6279  f_header_ <<
6280  declare_map <<
6281  indent() << "}" << endl <<
6282  endl <<
6283  indent() << "virtual " << ret_type_ << "process(" << finish_cob_ << "boost::shared_ptr<contrail::sandesh::protocol::TProtocol> piprot, boost::shared_ptr<contrail::sandesh::protocol::TProtocol> poprot" << call_context_ << ");" << endl <<
6284  indent() << "virtual ~" << class_name_ <<
6285  "() {}" << endl;
6286  indent_down();
6287  f_header_ <<
6288  "};" << endl << endl;
6289 
6290  if (generator_->gen_templates_) {
6291  // Generate a backwards compatible typedef, for callers who don't know
6292  // about the new template-style code.
6293  //
6294  // We can't use TProtocol as the template parameter, since ProcessorT
6295  // provides overloaded versions of most methods, one of which accepts
6296  // TProtocol pointers, and one which accepts Protocol_ pointers. This
6297  // results in a compile error if instantiated with Protocol_ == TProtocol.
6298  // Therefore, we define TDummyProtocol solely so we can use it as the
6299  // template parameter here.
6300  f_header_ <<
6301  "typedef " << class_name_ <<
6302  "<contrail::sandesh::protocol::TDummyProtocol> " <<
6303  service_name_ << pstyle_ << "Processor;" << endl << endl;
6304  }
6305 }
6306 
6307 void ProcessorGenerator::generate_process() {
6308  f_out_ <<
6309  template_header_ <<
6310  ret_type_ << class_name_ <<
6311  template_suffix_ << "::process(" << finish_cob_ <<
6312  "boost::shared_ptr<contrail::sandesh::protocol::TProtocol> piprot, " <<
6313  "boost::shared_ptr<contrail::sandesh::protocol::TProtocol> poprot" <<
6314  call_context_ << ") {" << endl;
6315  indent_up();
6316 
6317  f_out_ <<
6318  endl <<
6319  indent() << "::contrail::sandesh::protocol::TProtocol* iprot = piprot.get();" << endl <<
6320  indent() << "::contrail::sandesh::protocol::TProtocol* oprot = poprot.get();" << endl <<
6321  indent() << "std::string fname;" << endl <<
6322  indent() << "::contrail::sandesh::protocol::TMessageType mtype;" << endl <<
6323  indent() << "int32_t seqid;" << endl <<
6324  endl <<
6325  indent() << "iprot->readMessageBegin(fname, mtype, seqid);" << endl <<
6326  endl <<
6327  indent() << "if (mtype != ::contrail::sandesh::protocol::T_CALL && mtype != ::contrail::sandesh::protocol::T_ONEWAY) {" << endl <<
6328  indent() << " iprot->skip(::contrail::sandesh::protocol::T_STRUCT);" << endl <<
6329  indent() << " iprot->readMessageEnd();" << endl <<
6330  indent() << " iprot->getTransport()->readEnd();" << endl <<
6331  indent() << " ::contrail::sandesh::TApplicationException x(::contrail::sandesh::TApplicationException::INVALID_MESSAGE_TYPE);" << endl <<
6332  indent() << " oprot->writeMessageBegin(fname, ::contrail::sandesh::protocol::T_EXCEPTION, seqid);" << endl <<
6333  indent() << " x.write(oprot);" << endl <<
6334  indent() << " oprot->writeMessageEnd();" << endl <<
6335  indent() << " oprot->getTransport()->writeEnd();" << endl <<
6336  indent() << " oprot->getTransport()->flush();" << endl <<
6337  indent() << (style_ == "Cob" ? " return cob(true);" : " return true;") << endl <<
6338  indent() << "}" << endl <<
6339  endl <<
6340  indent() << "return process_fn(" << (style_ == "Cob" ? "cob, " : "")
6341  << "iprot, oprot, fname, seqid" << call_context_arg_ << ");" <<
6342  endl;
6343 
6344  indent_down();
6345  f_out_ <<
6346  indent() << "}" << endl <<
6347  endl;
6348 }
6349 
6350 void ProcessorGenerator::generate_process_fn() {
6351  f_out_ <<
6352  template_header_ <<
6353  ret_type_ << class_name_ <<
6354  template_suffix_ << "::process_fn(" << finish_cob_ <<
6355  "contrail::sandesh::protocol::TProtocol* iprot, " <<
6356  "contrail::sandesh::protocol::TProtocol* oprot, " <<
6357  "std::string& fname, int32_t seqid" << call_context_ << ") {" << endl;
6358  indent_up();
6359 
6360  // HOT: member function pointer map
6361  f_out_ <<
6362  indent() << typename_str_ << "std::map<std::string, void (" <<
6363  class_name_ << "::*)(" << finish_cob_decl_ <<
6364  "int32_t, contrail::sandesh::protocol::TProtocol*, " <<
6365  "contrail::sandesh::protocol::TProtocol*" << call_context_decl_ << ")>::iterator pfn;" << endl <<
6366  indent() << "pfn = processMap_.find(fname);" << endl <<
6367  indent() << "if (pfn == processMap_.end()) {" << endl;
6368  if (extends_.empty()) {
6369  f_out_ <<
6370  indent() << " iprot->skip(contrail::sandesh::protocol::T_STRUCT);" << endl <<
6371  indent() << " iprot->readMessageEnd();" << endl <<
6372  indent() << " iprot->getTransport()->readEnd();" << endl <<
6373  indent() << " ::contrail::sandesh::TApplicationException x(::contrail::sandesh::TApplicationException::UNKNOWN_METHOD, \"Invalid method name: '\"+fname+\"'\");" << endl <<
6374  indent() << " oprot->writeMessageBegin(fname, ::contrail::sandesh::protocol::T_EXCEPTION, seqid);" << endl <<
6375  indent() << " x.write(oprot);" << endl <<
6376  indent() << " oprot->writeMessageEnd();" << endl <<
6377  indent() << " oprot->getTransport()->writeEnd();" << endl <<
6378  indent() << " oprot->getTransport()->flush();" << endl <<
6379  indent() << (style_ == "Cob" ? " return cob(true);" : " return true;") << endl;
6380  } else {
6381  f_out_ <<
6382  indent() << " return "
6383  << extends_ << "::process_fn("
6384  << (style_ == "Cob" ? "cob, " : "")
6385  << "iprot, oprot, fname, seqid" << call_context_arg_ << ");" << endl;
6386  }
6387  f_out_ <<
6388  indent() << "}" << endl <<
6389  indent() << "(this->*(pfn->second))(" << cob_arg_ << "seqid, iprot, oprot" << call_context_arg_ << ");" << endl;
6390 
6391  // TODO(dreiss): return pfn ret?
6392  if (style_ == "Cob") {
6393  f_out_ <<
6394  indent() << "return;" << endl;
6395  } else {
6396  f_out_ <<
6397  indent() << "return true;" << endl;
6398  }
6399 
6400  indent_down();
6401  f_out_ <<
6402  "}" << endl <<
6403  endl;
6404 }
6405 
6406 void ProcessorGenerator::generate_process_functions() {
6407  vector<t_function*> functions = service_->get_functions();
6408  vector<t_function*>::iterator f_iter;
6409  for (f_iter = functions.begin(); f_iter != functions.end(); ++f_iter) {
6410  if (generator_->gen_templates_) {
6411  generator_->generate_process_function(service_, *f_iter, style_, false);
6412  generator_->generate_process_function(service_, *f_iter, style_, true);
6413  } else {
6414  generator_->generate_process_function(service_, *f_iter, style_, false);
6415  }
6416  }
6417 }
6418 
6419 void ProcessorGenerator::generate_factory() {
6420  string if_factory_name = if_name_ + "Factory";
6421 
6422  // Generate the factory class definition
6423  f_header_ <<
6424  template_header_ <<
6425  "class " << factory_class_name_ <<
6426  " : public ::contrail::sandesh::" <<
6427  (style_ == "Cob" ? "async::TAsyncProcessorFactory" : "TProcessorFactory") <<
6428  " {" << endl <<
6429  " public:" << endl;
6430  indent_up();
6431 
6432  f_header_ <<
6433  indent() << factory_class_name_ << "(const ::boost::shared_ptr< " <<
6434  if_factory_name << " >& handlerFactory) :" << endl <<
6435  indent() << " handlerFactory_(handlerFactory) {}" << endl <<
6436  endl <<
6437  indent() << "::boost::shared_ptr< ::contrail::sandesh::" <<
6438  (style_ == "Cob" ? "async::TAsyncProcessor" : "TProcessor") << " > " <<
6439  "getProcessor(const ::contrail::sandesh::TConnectionInfo& connInfo);" <<
6440  endl;
6441 
6442  f_header_ <<
6443  endl <<
6444  " protected:" << endl <<
6445  indent() << "::boost::shared_ptr< " << if_factory_name <<
6446  " > handlerFactory_;" << endl;
6447 
6448  indent_down();
6449  f_header_ <<
6450  "};" << endl << endl;
6451 
6452  // If we are generating templates, output a typedef for the plain
6453  // factory name.
6454  if (generator_->gen_templates_) {
6455  f_header_ <<
6456  "typedef " << factory_class_name_ <<
6457  "< ::contrail::sandesh::protocol::TDummyProtocol > " <<
6458  service_name_ << pstyle_ << "ProcessorFactory;" << endl << endl;
6459  }
6460 
6461  // Generate the getProcessor() method
6462  f_out_ <<
6463  template_header_ <<
6464  indent() << "::boost::shared_ptr< ::contrail::sandesh::" <<
6465  (style_ == "Cob" ? "async::TAsyncProcessor" : "TProcessor") << " > " <<
6466  factory_class_name_ << template_suffix_ << "::getProcessor(" <<
6467  "const ::contrail::sandesh::TConnectionInfo& connInfo) {" << endl;
6468  indent_up();
6469 
6470  f_out_ <<
6471  indent() << "::contrail::sandesh::ReleaseHandler< " << if_factory_name <<
6472  " > cleanup(handlerFactory_);" << endl <<
6473  indent() << "::boost::shared_ptr< " << if_name_ << " > handler(" <<
6474  "handlerFactory_->getHandler(connInfo), cleanup);" << endl <<
6475  indent() << "::boost::shared_ptr< ::contrail::sandesh::" <<
6476  (style_ == "Cob" ? "async::TAsyncProcessor" : "TProcessor") << " > " <<
6477  "processor(new " << class_name_ << template_suffix_ <<
6478  "(handler));" << endl <<
6479  indent() << "return processor;" << endl;
6480 
6481  indent_down();
6482  f_out_ <<
6483  indent() << "}" << endl;
6484 }
6485 
6491 void t_cpp_generator::generate_service_processor(t_service* tservice,
6492  string style) {
6493  ProcessorGenerator generator(this, tservice, style);
6494  generator.run();
6495 }
6496 
6502 void t_cpp_generator::generate_function_helpers(t_service* tservice,
6503  t_function* tfunction) {
6504  if (tfunction->is_oneway()) {
6505  return;
6506  }
6507 
6508  std::ofstream& out = (gen_templates_ ? f_service_tcc_ : f_service_);
6509 
6510  t_struct result(program_, tservice->get_name() + "_" + tfunction->get_name() + "_result");
6511  t_field success(tfunction->get_returntype(), "success", 0);
6512  if (!tfunction->get_returntype()->is_void()) {
6513  result.append(&success);
6514  }
6515 
6516  t_struct* xs = tfunction->get_xceptions();
6517  const vector<t_field*>& fields = xs->get_members();
6518  vector<t_field*>::const_iterator f_iter;
6519  for (f_iter = fields.begin(); f_iter != fields.end(); ++f_iter) {
6520  result.append(*f_iter);
6521  }
6522 
6523  generate_struct_definition(f_header_, &result, false);
6524  generate_struct_reader(out, &result);
6525  generate_struct_result_writer(out, &result);
6526 
6527  result.set_name(tservice->get_name() + "_" + tfunction->get_name() + "_presult");
6528  generate_struct_definition(f_header_, &result, false, true, true, gen_cob_style_);
6529  generate_struct_reader(out, &result, true);
6530  if (gen_cob_style_) {
6531  generate_struct_writer(out, &result, true);
6532  }
6533 
6534 }
6535 
6541 void t_cpp_generator::generate_process_function(t_service* tservice,
6542  t_function* tfunction,
6543  string style,
6544  bool specialized) {
6545  t_struct* arg_struct = tfunction->get_arglist();
6546  const std::vector<t_field*>& fields = arg_struct->get_members();
6547  vector<t_field*>::const_iterator f_iter;
6548 
6549  t_struct* xs = tfunction->get_xceptions();
6550  const std::vector<t_field*>& xceptions = xs->get_members();
6551  vector<t_field*>::const_iterator x_iter;
6552  string service_func_name = tservice->get_name() + "." +
6553  tfunction->get_name();
6554 
6555  std::ofstream& out = (gen_templates_ ? f_service_tcc_ : f_service_);
6556 
6557  string prot_type =
6558  (specialized ? "Protocol_" : "::contrail::sandesh::protocol::TProtocol");
6559  string class_suffix;
6560  if (gen_templates_) {
6561  class_suffix = "T<Protocol_>";
6562  }
6563 
6564  // I tried to do this as one function. I really did. But it was too hard.
6565  if (style != "Cob") {
6566  // Open function
6567  if (gen_templates_) {
6568  out <<
6569  indent() << "template <class Protocol_>" << endl;
6570  }
6571  const bool unnamed_oprot_seqid = tfunction->is_oneway() &&
6572  !(gen_templates_ && !specialized);
6573  out <<
6574  "void " << tservice->get_name() << "Processor" << class_suffix << "::" <<
6575  "process_" << tfunction->get_name() << "(" <<
6576  "int32_t" << (unnamed_oprot_seqid ? ", " : " seqid, ") <<
6577  prot_type << "* iprot, " <<
6578  prot_type << "*" << (unnamed_oprot_seqid ? ", " : " oprot, ") <<
6579  "void* callContext)" << endl;
6580  scope_up(out);
6581 
6582  if (gen_templates_ && !specialized) {
6583  // If these are instances of Protocol_, instead of any old TProtocol,
6584  // use the specialized process function instead.
6585  out <<
6586  indent() << "Protocol_* _iprot = dynamic_cast<Protocol_*>(iprot);" <<
6587  endl <<
6588  indent() << "Protocol_* _oprot = dynamic_cast<Protocol_*>(oprot);" <<
6589  endl <<
6590  indent() << "if (_iprot && _oprot) {" << endl <<
6591  indent() << " return process_" << tfunction->get_name() <<
6592  "(seqid, _iprot, _oprot, callContext);" << endl <<
6593  indent() << "}" << endl <<
6594  indent() << "T_GENERIC_PROTOCOL(this, iprot, _iprot);" << endl <<
6595  indent() << "T_GENERIC_PROTOCOL(this, oprot, _oprot);" << endl << endl;
6596  }
6597 
6598  string argsname = tservice->get_name() + "_" + tfunction->get_name() +
6599  "_args";
6600  string resultname = tservice->get_name() + "_" + tfunction->get_name() +
6601  "_result";
6602 
6603  out <<
6604  indent() << "void* ctx = NULL;" << endl <<
6605  indent() << "if (this->eventHandler_.get() != NULL) {" << endl <<
6606  indent() << " ctx = this->eventHandler_->getContext(\"" <<
6607  service_func_name << "\", callContext);" << endl <<
6608  indent() << "}" << endl <<
6609  indent() << "contrail::sandesh::TProcessorContextFreer freer(" <<
6610  "this->eventHandler_.get(), ctx, \"" << service_func_name << "\");" <<
6611  endl << endl <<
6612  indent() << "if (this->eventHandler_.get() != NULL) {" << endl <<
6613  indent() << " this->eventHandler_->preRead(ctx, \"" <<
6614  service_func_name << "\");" << endl <<
6615  indent() << "}" << endl << endl <<
6616  indent() << argsname << " args;" << endl <<
6617  indent() << "args.read(iprot);" << endl <<
6618  indent() << "iprot->readMessageEnd();" << endl <<
6619  indent() << "uint32_t bytes = iprot->getTransport()->readEnd();" <<
6620  endl << endl <<
6621  indent() << "if (this->eventHandler_.get() != NULL) {" << endl <<
6622  indent() << " this->eventHandler_->postRead(ctx, \"" <<
6623  service_func_name << "\", bytes);" << endl <<
6624  indent() << "}" << endl <<
6625  endl;
6626 
6627  // Declare result
6628  if (!tfunction->is_oneway()) {
6629  out <<
6630  indent() << resultname << " result;" << endl;
6631  }
6632 
6633  // Try block for functions with exceptions
6634  out <<
6635  indent() << "try {" << endl;
6636  indent_up();
6637 
6638  // Generate the function call
6639  bool first = true;
6640  out << indent();
6641  if (!tfunction->is_oneway() && !tfunction->get_returntype()->is_void()) {
6642  if (is_complex_type(tfunction->get_returntype())) {
6643  first = false;
6644  out << "iface_->" << tfunction->get_name() << "(result.success";
6645  } else {
6646  out << "result.success = iface_->" << tfunction->get_name() << "(";
6647  }
6648  } else {
6649  out <<
6650  "iface_->" << tfunction->get_name() << "(";
6651  }
6652  for (f_iter = fields.begin(); f_iter != fields.end(); ++f_iter) {
6653  if (first) {
6654  first = false;
6655  } else {
6656  out << ", ";
6657  }
6658  out << "args." << (*f_iter)->get_name();
6659  }
6660  out << ");" << endl;
6661 
6662  // Set isset on success field
6663  if (!tfunction->is_oneway() && !tfunction->get_returntype()->is_void()) {
6664  out <<
6665  indent() << "result.__isset.success = true;" << endl;
6666  }
6667 
6668  indent_down();
6669  out << indent() << "}";
6670 
6671  if (!tfunction->is_oneway()) {
6672  for (x_iter = xceptions.begin(); x_iter != xceptions.end(); ++x_iter) {
6673  out << " catch (" << type_name((*x_iter)->get_type()) << " &" <<
6674  (*x_iter)->get_name() << ") {" << endl;
6675  if (!tfunction->is_oneway()) {
6676  indent_up();
6677  out <<
6678  indent() << "result." << (*x_iter)->get_name() << " = " <<
6679  (*x_iter)->get_name() << ";" << endl <<
6680  indent() << "result.__isset." << (*x_iter)->get_name() <<
6681  " = true;" << endl;
6682  indent_down();
6683  out << indent() << "}";
6684  } else {
6685  out << "}";
6686  }
6687  }
6688  }
6689 
6690  out << " catch (const std::exception& e) {" << endl;
6691 
6692  indent_up();
6693  out <<
6694  indent() << "if (this->eventHandler_.get() != NULL) {" << endl <<
6695  indent() << " this->eventHandler_->handlerError(ctx, \"" <<
6696  service_func_name << "\");" << endl <<
6697  indent() << "}" << endl;
6698 
6699  if (!tfunction->is_oneway()) {
6700  out <<
6701  endl <<
6702  indent() << "contrail::sandesh::TApplicationException x(e.what());" <<
6703  endl <<
6704  indent() << "oprot->writeMessageBegin(\"" << tfunction->get_name() <<
6705  "\", contrail::sandesh::protocol::T_EXCEPTION, seqid);" << endl <<
6706  indent() << "x.write(oprot);" << endl <<
6707  indent() << "oprot->writeMessageEnd();" << endl <<
6708  indent() << "oprot->getTransport()->writeEnd();" << endl <<
6709  indent() << "oprot->getTransport()->flush();" << endl;
6710  }
6711  out << indent() << "return;" << endl;
6712  indent_down();
6713  out << indent() << "}" << endl << endl;
6714 
6715  // Shortcut out here for oneway functions
6716  if (tfunction->is_oneway()) {
6717  out <<
6718  indent() << "if (this->eventHandler_.get() != NULL) {" << endl <<
6719  indent() << " this->eventHandler_->asyncComplete(ctx, \"" <<
6720  service_func_name << "\");" << endl <<
6721  indent() << "}" << endl << endl <<
6722  indent() << "return;" << endl;
6723  indent_down();
6724  out << "}" << endl <<
6725  endl;
6726  return;
6727  }
6728 
6729  // Serialize the result into a struct
6730  out <<
6731  indent() << "if (this->eventHandler_.get() != NULL) {" << endl <<
6732  indent() << " this->eventHandler_->preWrite(ctx, \"" <<
6733  service_func_name << "\");" << endl <<
6734  indent() << "}" << endl << endl <<
6735  indent() << "oprot->writeMessageBegin(\"" << tfunction->get_name() <<
6736  "\", contrail::sandesh::protocol::T_REPLY, seqid);" << endl <<
6737  indent() << "result.write(oprot);" << endl <<
6738  indent() << "oprot->writeMessageEnd();" << endl <<
6739  indent() << "bytes = oprot->getTransport()->writeEnd();" << endl <<
6740  indent() << "oprot->getTransport()->flush();" << endl << endl <<
6741  indent() << "if (this->eventHandler_.get() != NULL) {" << endl <<
6742  indent() << " this->eventHandler_->postWrite(ctx, \"" <<
6743  service_func_name << "\", bytes);" << endl <<
6744  indent() << "}" << endl;
6745 
6746  // Close function
6747  scope_down(out);
6748  out << endl;
6749  }
6750 
6751  // Cob style.
6752  else {
6753  // Processor entry point.
6754  // TODO(edhall) update for callContext when TEventServer is ready
6755  if (gen_templates_) {
6756  out <<
6757  indent() << "template <class Protocol_>" << endl;
6758  }
6759  out <<
6760  "void " << tservice->get_name() << "AsyncProcessor" << class_suffix <<
6761  "::process_" << tfunction->get_name() <<
6762  "(std::tr1::function<void(bool ok)> cob, int32_t seqid, " <<
6763  prot_type << "* iprot, " << prot_type << "* oprot)" << endl;
6764  scope_up(out);
6765 
6766  // TODO(simpkins): we could try to consoldate this
6767  // with the non-cob code above
6768  if (gen_templates_ && !specialized) {
6769  // If these are instances of Protocol_, instead of any old TProtocol,
6770  // use the specialized process function instead.
6771  out <<
6772  indent() << "Protocol_* _iprot = dynamic_cast<Protocol_*>(iprot);" <<
6773  endl <<
6774  indent() << "Protocol_* _oprot = dynamic_cast<Protocol_*>(oprot);" <<
6775  endl <<
6776  indent() << "if (_iprot && _oprot) {" << endl <<
6777  indent() << " return process_" << tfunction->get_name() <<
6778  "(cob, seqid, _iprot, _oprot);" << endl <<
6779  indent() << "}" << endl <<
6780  indent() << "T_GENERIC_PROTOCOL(this, iprot, _iprot);" << endl <<
6781  indent() << "T_GENERIC_PROTOCOL(this, oprot, _oprot);" << endl << endl;
6782  }
6783 
6784  if (tfunction->is_oneway()) {
6785  out <<
6786  indent() << "(void) seqid;" << endl <<
6787  indent() << "(void) oprot;" << endl;
6788  }
6789 
6790  out <<
6791  indent() << tservice->get_name() + "_" + tfunction->get_name() <<
6792  "_args args;" << endl <<
6793  indent() << "void* ctx = NULL;" << endl <<
6794  indent() << "if (this->eventHandler_.get() != NULL) {" << endl <<
6795  indent() << " ctx = this->eventHandler_->getContext(\"" <<
6796  service_func_name << "\", NULL);" << endl <<
6797  indent() << "}" << endl <<
6798  indent() << "contrail::sandesh::TProcessorContextFreer freer(" <<
6799  "this->eventHandler_.get(), ctx, \"" << service_func_name << "\");" <<
6800  endl << endl <<
6801  indent() << "try {" << endl;
6802  indent_up();
6803  out <<
6804  indent() << "if (this->eventHandler_.get() != NULL) {" << endl <<
6805  indent() << " this->eventHandler_->preRead(ctx, \"" <<
6806  service_func_name << "\");" << endl <<
6807  indent() << "}" << endl <<
6808  indent() << "args.read(iprot);" << endl <<
6809  indent() << "iprot->readMessageEnd();" << endl <<
6810  indent() << "uint32_t bytes = iprot->getTransport()->readEnd();" <<
6811  endl <<
6812  indent() << "if (this->eventHandler_.get() != NULL) {" << endl <<
6813  indent() << " this->eventHandler_->postRead(ctx, \"" <<
6814  service_func_name << "\", bytes);" << endl <<
6815  indent() << "}" << endl;
6816  scope_down(out);
6817 
6818  // TODO(dreiss): Handle TExceptions? Expose to server?
6819  out <<
6820  indent() << "catch (const std::exception& exn) {" << endl <<
6821  indent() << " if (this->eventHandler_.get() != NULL) {" << endl <<
6822  indent() << " this->eventHandler_->handlerError(ctx, \"" <<
6823  service_func_name << "\");" << endl <<
6824  indent() << " }" << endl <<
6825  indent() << " return cob(false);" << endl <<
6826  indent() << "}" << endl;
6827 
6828  if (tfunction->is_oneway()) {
6829  out <<
6830  indent() << "if (this->eventHandler_.get() != NULL) {" << endl <<
6831  indent() << " this->eventHandler_->asyncComplete(ctx, \"" <<
6832  service_func_name << "\");" << endl <<
6833  indent() << "}" << endl;
6834  }
6835  // TODO(dreiss): Figure out a strategy for exceptions in async handlers.
6836  out <<
6837  indent() << "freer.unregister();" << endl;
6838  if (tfunction->is_oneway()) {
6839  // No return. Just hand off our cob.
6840  // TODO(dreiss): Call the cob immediately?
6841  out <<
6842  indent() << "iface_->" << tfunction->get_name() << "(" <<
6843  "std::tr1::bind(cob, true)" << endl;
6844  indent_up(); indent_up();
6845  } else {
6846  string ret_arg, ret_placeholder;
6847  if (!tfunction->get_returntype()->is_void()) {
6848  ret_arg = ", const " + type_name(tfunction->get_returntype()) +
6849  "& _return";
6850  ret_placeholder = ", std::tr1::placeholders::_1";
6851  }
6852 
6853  // When gen_templates_ is true, the return_ and throw_ functions are
6854  // overloaded. We have to declare pointers to them so that the compiler
6855  // can resolve the correct overloaded version.
6856  out <<
6857  indent() << "void (" << tservice->get_name() << "AsyncProcessor" <<
6858  class_suffix << "::*return_fn)(std::tr1::function<void(bool ok)> " <<
6859  "cob, int32_t seqid, " << prot_type << "* oprot, void* ctx" <<
6860  ret_arg << ") =" << endl;
6861  out <<
6862  indent() << " &" << tservice->get_name() << "AsyncProcessor" <<
6863  class_suffix << "::return_" << tfunction->get_name() << ";" << endl;
6864  if (!xceptions.empty()) {
6865  out <<
6866  indent() << "void (" << tservice->get_name() << "AsyncProcessor" <<
6867  class_suffix << "::*throw_fn)(std::tr1::function<void(bool ok)> " <<
6868  "cob, int32_t seqid, " << prot_type << "* oprot, void* ctx, " <<
6869  "::contrail::sandesh::TDelayedException* _throw) =" << endl;
6870  out <<
6871  indent() << " &" << tservice->get_name() << "AsyncProcessor" <<
6872  class_suffix << "::throw_" << tfunction->get_name() << ";" << endl;
6873  }
6874 
6875  out <<
6876  indent() << "iface_->" << tfunction->get_name() << "(" << endl;
6877  indent_up(); indent_up();
6878  out <<
6879  indent() << "std::tr1::bind(return_fn, this, cob, seqid, oprot, ctx" <<
6880  ret_placeholder << ")";
6881  if (!xceptions.empty()) {
6882  out
6883  << ',' << endl <<
6884  indent() << "std::tr1::bind(throw_fn, this, cob, seqid, oprot, " <<
6885  "ctx, std::tr1::placeholders::_1)";
6886  }
6887  }
6888 
6889  // XXX Whitespace cleanup.
6890  for (f_iter = fields.begin(); f_iter != fields.end(); ++f_iter) {
6891  out
6892  << ',' << endl <<
6893  indent() << "args." << (*f_iter)->get_name();
6894  }
6895  out << ");" << endl;
6896  indent_down(); indent_down();
6897  scope_down(out);
6898  out << endl;
6899 
6900  // Normal return.
6901  if (!tfunction->is_oneway()) {
6902  string ret_arg_decl, ret_arg_name;
6903  if (!tfunction->get_returntype()->is_void()) {
6904  ret_arg_decl = ", const " + type_name(tfunction->get_returntype()) +
6905  "& _return";
6906  ret_arg_name = ", _return";
6907  }
6908  if (gen_templates_) {
6909  out <<
6910  indent() << "template <class Protocol_>" << endl;
6911  }
6912  out <<
6913  "void " << tservice->get_name() << "AsyncProcessor" << class_suffix <<
6914  "::return_" << tfunction->get_name() <<
6915  "(std::tr1::function<void(bool ok)> cob, int32_t seqid, " <<
6916  prot_type << "* oprot, void* ctx" << ret_arg_decl << ')' << endl;
6917  scope_up(out);
6918 
6919  if (gen_templates_ && !specialized) {
6920  // If oprot is a Protocol_ instance,
6921  // use the specialized return function instead.
6922  out <<
6923  indent() << "Protocol_* _oprot = dynamic_cast<Protocol_*>(oprot);" <<
6924  endl <<
6925  indent() << "if (_oprot) {" << endl <<
6926  indent() << " return return_" << tfunction->get_name() <<
6927  "(cob, seqid, _oprot, ctx" << ret_arg_name << ");" << endl <<
6928  indent() << "}" << endl <<
6929  indent() << "T_GENERIC_PROTOCOL(this, oprot, _oprot);" <<
6930  endl << endl;
6931  }
6932 
6933  out <<
6934  indent() << tservice->get_name() << "_" << tfunction->get_name() <<
6935  "_presult result;" << endl;
6936  if (!tfunction->get_returntype()->is_void()) {
6937  // The const_cast here is unfortunate, but it would be a pain to avoid,
6938  // and we only do a write with this struct, which is const-safe.
6939  out <<
6940  indent() << "result.success = const_cast<" <<
6941  type_name(tfunction->get_returntype()) << "*>(&_return);" <<
6942  endl <<
6943  indent() << "result.__isset.success = true;" << endl;
6944  }
6945  // Serialize the result into a struct
6946  out <<
6947  endl <<
6948  indent() << "if (this->eventHandler_.get() != NULL) {" << endl <<
6949  indent() << " ctx = this->eventHandler_->getContext(\"" <<
6950  service_func_name << "\", NULL);" << endl <<
6951  indent() << "}" << endl <<
6952  indent() << "contrail::sandesh::TProcessorContextFreer freer(" <<
6953  "this->eventHandler_.get(), ctx, \"" << service_func_name <<
6954  "\");" << endl << endl <<
6955  indent() << "if (this->eventHandler_.get() != NULL) {" << endl <<
6956  indent() << " this->eventHandler_->preWrite(ctx, \"" <<
6957  service_func_name << "\");" << endl <<
6958  indent() << "}" << endl << endl <<
6959  indent() << "oprot->writeMessageBegin(\"" << tfunction->get_name() <<
6960  "\", contrail::sandesh::protocol::T_REPLY, seqid);" << endl <<
6961  indent() << "result.write(oprot);" << endl <<
6962  indent() << "oprot->writeMessageEnd();" << endl <<
6963  indent() << "uint32_t bytes = oprot->getTransport()->writeEnd();" <<
6964  endl <<
6965  indent() << "oprot->getTransport()->flush();" << endl <<
6966  indent() << "if (this->eventHandler_.get() != NULL) {" << endl <<
6967  indent() << " this->eventHandler_->postWrite(ctx, \"" <<
6968  service_func_name << "\", bytes);" << endl <<
6969  indent() << "}" << endl <<
6970  indent() << "return cob(true);" << endl;
6971  scope_down(out);
6972  out << endl;
6973  }
6974 
6975  // Exception return.
6976  if (!tfunction->is_oneway() && !xceptions.empty()) {
6977  if (gen_templates_) {
6978  out <<
6979  indent() << "template <class Protocol_>" << endl;
6980  }
6981  out <<
6982  "void " << tservice->get_name() << "AsyncProcessor" << class_suffix <<
6983  "::throw_" << tfunction->get_name() <<
6984  "(std::tr1::function<void(bool ok)> cob, int32_t seqid, " <<
6985  prot_type << "* oprot, void* ctx, " <<
6986  "::contrail::sandesh::TDelayedException* _throw)" << endl;
6987  scope_up(out);
6988 
6989  if (gen_templates_ && !specialized) {
6990  // If oprot is a Protocol_ instance,
6991  // use the specialized throw function instead.
6992  out <<
6993  indent() << "Protocol_* _oprot = dynamic_cast<Protocol_*>(oprot);" <<
6994  endl <<
6995  indent() << "if (_oprot) {" << endl <<
6996  indent() << " return throw_" << tfunction->get_name() <<
6997  "(cob, seqid, _oprot, ctx, _throw);" << endl <<
6998  indent() << "}" << endl <<
6999  indent() << "T_GENERIC_PROTOCOL(this, oprot, _oprot);" <<
7000  endl << endl;
7001  }
7002 
7003  // Get the event handler context
7004  out <<
7005  endl <<
7006  indent() << "if (this->eventHandler_.get() != NULL) {" << endl <<
7007  indent() << " ctx = this->eventHandler_->getContext(\"" <<
7008  service_func_name << "\", NULL);" << endl <<
7009  indent() << "}" << endl <<
7010  indent() << "contrail::sandesh::TProcessorContextFreer freer(" <<
7011  "this->eventHandler_.get(), ctx, \"" << service_func_name << "\");" <<
7012  endl << endl;
7013 
7014  // Throw the TDelayedException, and catch the result
7015  out <<
7016  indent() << tservice->get_name() << "_" << tfunction->get_name() <<
7017  "_result result;" << endl << endl <<
7018  indent() << "try {" << endl;
7019  indent_up();
7020  out <<
7021  indent() << "_throw->throw_it();" << endl <<
7022  indent() << "return cob(false);" << endl; // Is this possible? TBD.
7023  indent_down();
7024  out <<
7025  indent() << '}';
7026  for (x_iter = xceptions.begin(); x_iter != xceptions.end(); ++x_iter) {
7027  out << " catch (" << type_name((*x_iter)->get_type()) << " &" <<
7028  (*x_iter)->get_name() << ") {" << endl;
7029  indent_up();
7030  out <<
7031  indent() << "result." << (*x_iter)->get_name() << " = " <<
7032  (*x_iter)->get_name() << ";" << endl <<
7033  indent() << "result.__isset." << (*x_iter)->get_name() <<
7034  " = true;" << endl;
7035  scope_down(out);
7036  }
7037 
7038  // Handle the case where an undeclared exception is thrown
7039  out << " catch (std::exception& e) {" << endl;
7040  indent_up();
7041  out <<
7042  indent() << "if (this->eventHandler_.get() != NULL) {" << endl <<
7043  indent() << " this->eventHandler_->handlerError(ctx, \"" <<
7044  service_func_name << "\");" << endl <<
7045  indent() << "}" << endl <<
7046  endl <<
7047  indent() << "contrail::sandesh::TApplicationException x(e.what());" <<
7048  endl <<
7049  indent() << "oprot->writeMessageBegin(\"" << tfunction->get_name() <<
7050  "\", contrail::sandesh::protocol::T_EXCEPTION, seqid);" << endl <<
7051  indent() << "x.write(oprot);" << endl <<
7052  indent() << "oprot->writeMessageEnd();" << endl <<
7053  indent() << "oprot->getTransport()->writeEnd();" << endl <<
7054  indent() << "oprot->getTransport()->flush();" << endl <<
7055  // We pass true to the cob here, since we did successfully write a
7056  // response, even though it is an exception response.
7057  // It looks like the argument is currently ignored, anyway.
7058  indent() << "return cob(true);" << endl;
7059  scope_down(out);
7060 
7061  // Serialize the result into a struct
7062  out <<
7063  indent() << "if (this->eventHandler_.get() != NULL) {" << endl <<
7064  indent() << " this->eventHandler_->preWrite(ctx, \"" <<
7065  service_func_name << "\");" << endl <<
7066  indent() << "}" << endl << endl <<
7067  indent() << "oprot->writeMessageBegin(\"" << tfunction->get_name() <<
7068  "\", contrail::sandesh::protocol::T_REPLY, seqid);" << endl <<
7069  indent() << "result.write(oprot);" << endl <<
7070  indent() << "oprot->writeMessageEnd();" << endl <<
7071  indent() << "uint32_t bytes = oprot->getTransport()->writeEnd();" <<
7072  endl <<
7073  indent() << "oprot->getTransport()->flush();" << endl <<
7074  indent() << "if (this->eventHandler_.get() != NULL) {" << endl <<
7075  indent() << " this->eventHandler_->postWrite(ctx, \"" <<
7076  service_func_name << "\", bytes);" << endl <<
7077  indent() << "}" << endl <<
7078  indent() << "return cob(true);" << endl;
7079  scope_down(out);
7080  out << endl;
7081  } // for each function
7082  } // cob style
7083 }
7084 
7090 void t_cpp_generator::generate_service_skeleton(t_service* tservice) {
7091  string svcname = tservice->get_name();
7092 
7093  // Service implementation file includes
7094  string f_skeleton_name = get_out_dir()+svcname+"_server.skeleton.cpp";
7095 
7096  string ns = namespace_prefix(tservice->get_program()->get_namespace("cpp"));
7097 
7098  ofstream f_skeleton;
7099  f_skeleton.open(f_skeleton_name.c_str());
7100  f_skeleton <<
7101  "// This autogenerated skeleton file illustrates how to build a server." << endl <<
7102  "// You should copy it to another filename to avoid overwriting it." << endl <<
7103  endl <<
7104  "#include \"" << get_include_prefix(*get_program()) << svcname << ".h\"" << endl <<
7105  "#include <protocol/TBinaryProtocol.h>" << endl <<
7106  "#include <server/TSimpleServer.h>" << endl <<
7107  "#include <transport/TServerSocket.h>" << endl <<
7108  "#include <transport/TBufferTransports.h>" << endl <<
7109  endl <<
7110  "using namespace ::contrail::sandesh;" << endl <<
7111  "using namespace ::contrail::sandesh::protocol;" << endl <<
7112  "using namespace ::contrail::sandesh::transport;" << endl <<
7113  "using namespace ::contrail::sandesh::server;" << endl <<
7114  endl <<
7115  "using boost::shared_ptr;" << endl <<
7116  endl;
7117 
7118  if (!ns.empty()) {
7119  f_skeleton <<
7120  "using namespace " << string(ns, 0, ns.size()-2) << ";" << endl <<
7121  endl;
7122  }
7123 
7124  f_skeleton <<
7125  "class " << svcname << "Handler : virtual public " << svcname << "If {" << endl <<
7126  " public:" << endl;
7127  indent_up();
7128  f_skeleton <<
7129  indent() << svcname << "Handler() {" << endl <<
7130  indent() << " // Your initialization goes here" << endl <<
7131  indent() << "}" << endl <<
7132  endl;
7133 
7134  vector<t_function*> functions = tservice->get_functions();
7135  vector<t_function*>::iterator f_iter;
7136  for (f_iter = functions.begin(); f_iter != functions.end(); ++f_iter) {
7137  f_skeleton <<
7138  indent() << function_signature(*f_iter, "") << " {" << endl <<
7139  indent() << " // Your implementation goes here" << endl <<
7140  indent() << " printf(\"" << (*f_iter)->get_name() << "\\n\");" << endl <<
7141  indent() << "}" << endl <<
7142  endl;
7143  }
7144 
7145  indent_down();
7146  f_skeleton <<
7147  "};" << endl <<
7148  endl;
7149 
7150  f_skeleton <<
7151  indent() << "int main(int argc, char **argv) {" << endl;
7152  indent_up();
7153  f_skeleton <<
7154  indent() << "int port = 9090;" << endl <<
7155  indent() << "shared_ptr<" << svcname << "Handler> handler(new " << svcname << "Handler());" << endl <<
7156  indent() << "shared_ptr<TProcessor> processor(new " << svcname << "Processor(handler));" << endl <<
7157  indent() << "shared_ptr<TServerTransport> serverTransport(new TServerSocket(port));" << endl <<
7158  indent() << "shared_ptr<TTransportFactory> transportFactory(new TBufferedTransportFactory());" << endl <<
7159  indent() << "shared_ptr<TProtocolFactory> protocolFactory(new TBinaryProtocolFactory());" << endl <<
7160  endl <<
7161  indent() << "TSimpleServer server(processor, serverTransport, transportFactory, protocolFactory);" << endl <<
7162  indent() << "server.serve();" << endl <<
7163  indent() << "return 0;" << endl;
7164  indent_down();
7165  f_skeleton <<
7166  "}" << endl <<
7167  endl;
7168 
7169  // Close the files
7170  f_skeleton.close();
7171 }
7172 
7176 void t_cpp_generator::generate_deserialize_field(ofstream& out,
7177  t_field* tfield,
7178  string prefix,
7179  string suffix) {
7180  t_type* type = get_true_type(tfield->get_type());
7181 
7182  if (type->is_void()) {
7183  throw "CANNOT GENERATE DESERIALIZE CODE FOR void TYPE: " +
7184  prefix + tfield->get_name();
7185  }
7186 
7187  string name = prefix + tfield->get_name() + suffix;
7188 
7189  if (type->is_struct() || type->is_xception()) {
7190  generate_deserialize_struct(out, (t_struct*)type, name);
7191  } else if (type->is_container()) {
7192  generate_deserialize_container(out, type, name);
7193  } else if (type->is_base_type()) {
7194  indent(out) <<
7195  "if ((ret = iprot->";
7196  t_base_type::t_base tbase = ((t_base_type*)type)->get_base();
7197  switch (tbase) {
7198  case t_base_type::TYPE_VOID:
7199  throw "compiler error: cannot serialize void field in a struct: " + name;
7200  break;
7201  case t_base_type::TYPE_STRING:
7202 #ifdef SANDESH
7203  case t_base_type::TYPE_STATIC_CONST_STRING:
7204 #endif
7205  if (((t_base_type*)type)->is_binary()) {
7206  out << "readBinary(" << name << ")) < 0) {" << endl;
7207  }
7208  else {
7209  out << "readString(" << name << ")) < 0) {" << endl;
7210  }
7211  break;
7212 #ifdef SANDESH
7213  case t_base_type::TYPE_XML:
7214  out << "readXML(" << name << ")) < 0) {" << endl;
7215  break;
7216  case t_base_type::TYPE_UUID:
7217  out << "readUUID(" << name << ")) < 0) {" << endl;
7218  break;
7219 #endif
7220  case t_base_type::TYPE_BOOL:
7221  out << "readBool(" << name << ")) < 0) {" << endl;
7222  break;
7223  case t_base_type::TYPE_BYTE:
7224  out << "readByte(" << name << ")) < 0) {" << endl;
7225  break;
7226  case t_base_type::TYPE_I16:
7227  out << "readI16(" << name << ")) < 0) {" << endl;
7228  break;
7229  case t_base_type::TYPE_I32:
7230  out << "readI32(" << name << ")) < 0) {" << endl;
7231  break;
7232  case t_base_type::TYPE_I64:
7233  out << "readI64(" << name << ")) < 0) {" << endl;
7234  break;
7235 #ifdef SANDESH
7236  case t_base_type::TYPE_U16:
7237  out << "readU16(" << name << ")) < 0) {" << endl;
7238  break;
7239  case t_base_type::TYPE_U32:
7240  out << "readU32(" << name << ")) < 0) {" << endl;
7241  break;
7242  case t_base_type::TYPE_U64:
7243  out << "readU64(" << name << ")) < 0) {" << endl;
7244  break;
7245  case t_base_type::TYPE_IPV4:
7246  out << "readIPV4(" << name << ")) < 0) {" << endl;
7247  break;
7248  case t_base_type::TYPE_IPADDR:
7249  out << "readIPADDR(" << name << ")) < 0) {" << endl;
7250  break;
7251 #endif
7252  case t_base_type::TYPE_DOUBLE:
7253  out << "readDouble(" << name << ")) < 0) {" << endl;
7254  break;
7255  default:
7256  throw "compiler error: no C++ reader for base type " + t_base_type::t_base_name(tbase) + name;
7257  }
7258  indent_up();
7259  out << indent() << "return ret;" << endl;
7260  scope_down(out);
7261  out << indent() << "xfer += ret;" << endl;
7262  out <<
7263  endl;
7264  } else if (type->is_enum()) {
7265  string t = tmp("ecast");
7266  out <<
7267  indent() << "int32_t " << t << ";" << endl <<
7268  indent() << "if ((ret = iprot->readI32(" << t << ")) < 0) {" << endl <<
7269  indent() << " return ret;" << endl <<
7270  indent() << "}" << endl <<
7271  indent() << "xfer += ret;" << endl <<
7272  indent() << name << " = (" << type_name(type) << ")" << t << ";" << endl;
7273  } else {
7274  printf("DO NOT KNOW HOW TO DESERIALIZE FIELD '%s' TYPE '%s'\n",
7275  tfield->get_name().c_str(), type_name(type).c_str());
7276  }
7277 }
7278 
7285 void t_cpp_generator::generate_deserialize_struct(ofstream& out,
7286  t_struct* tstruct,
7287  string prefix) {
7288  (void) tstruct;
7289  indent(out) <<
7290  "if ((ret = " << prefix << ".read(iprot)) < 0) {" << endl;
7291  indent_up();
7292  indent(out) << "return ret;" << endl;
7293  scope_down(out);
7294  indent(out) << "xfer += ret;" << endl;
7295 }
7296 
7297 void t_cpp_generator::generate_deserialize_container(ofstream& out,
7298  t_type* ttype,
7299  string prefix) {
7300  scope_up(out);
7301 
7302  string size = tmp("_size");
7303  string ktype = tmp("_ktype");
7304  string vtype = tmp("_vtype");
7305  string etype = tmp("_etype");
7306 
7307  t_container* tcontainer = (t_container*)ttype;
7308  bool use_push = tcontainer->has_cpp_name();
7309 
7310  indent(out) <<
7311  prefix << ".clear();" << endl <<
7312  indent() << "uint32_t " << size << ";" << endl;
7313 
7314  // Declare variables, read header
7315  if (ttype->is_map()) {
7316  out <<
7317  indent() << "::contrail::sandesh::protocol::TType " << ktype << ";" << endl <<
7318  indent() << "::contrail::sandesh::protocol::TType " << vtype << ";" << endl <<
7319  indent() << "if ((ret = iprot->readMapBegin(" <<
7320  ktype << ", " << vtype << ", " << size << ")) < 0) {" << endl <<
7321  indent() << " return ret;" << endl <<
7322  indent() << "}" << endl <<
7323  indent() << "xfer += ret;" << endl;
7324  } else if (ttype->is_set()) {
7325  out <<
7326  indent() << "::contrail::sandesh::protocol::TType " << etype << ";" << endl <<
7327  indent() << "if ((ret = iprot->readSetBegin(" <<
7328  etype << ", " << size << ")) < 0) {" << endl <<
7329  indent() << " return ret;" << endl <<
7330  indent() << "}" << endl <<
7331  indent() << "xfer += ret;" << endl;
7332  } else if (ttype->is_list()) {
7333  out <<
7334  indent() << "::contrail::sandesh::protocol::TType " << etype << ";" << endl <<
7335  indent() << "if ((ret = iprot->readListBegin(" <<
7336  etype << ", " << size << ")) < 0) {" << endl <<
7337  indent() << " return ret;" << endl <<
7338  indent() << "}" << endl <<
7339  indent() << "xfer += ret;" << endl;
7340  if (!use_push) {
7341  indent(out) << prefix << ".resize(" << size << ");" << endl;
7342  }
7343  }
7344 
7345 
7346  // For loop iterates over elements
7347  string i = tmp("_i");
7348  out <<
7349  indent() << "uint32_t " << i << ";" << endl <<
7350  indent() << "for (" << i << " = 0; " << i << " < " << size << "; ++" << i << ")" << endl;
7351 
7352  scope_up(out);
7353 
7354  if (ttype->is_map()) {
7355  generate_deserialize_map_element(out, (t_map*)ttype, prefix);
7356  } else if (ttype->is_set()) {
7357 #ifdef SANDESH
7358  t_type *etype = ((t_set*)ttype)->get_elem_type();
7359  if (etype->is_base_type() || etype->is_enum()) {
7360  indent(out) << "if ((ret = iprot->readContainerElementBegin()) < 0) {" << endl;
7361  indent_up();
7362  indent(out) << "return ret;" << endl;
7363  scope_down(out);
7364  indent(out) << "xfer += ret;" << endl;
7365  }
7366 #endif
7367  generate_deserialize_set_element(out, (t_set*)ttype, prefix);
7368 #ifdef SANDESH
7369  if (etype->is_base_type() || etype->is_enum()) {
7370  indent(out) << "if ((ret = iprot->readContainerElementEnd()) < 0) {" << endl;
7371  indent_up();
7372  indent(out) << "return ret;" << endl;
7373  scope_down(out);
7374  indent(out) << "xfer += ret;" << endl;
7375  }
7376 #endif
7377  } else if (ttype->is_list()) {
7378 #ifdef SANDESH
7379  t_type *etype = ((t_list*)ttype)->get_elem_type();
7380  if (etype->is_base_type() || etype->is_enum()) {
7381  indent(out) << "if ((ret = iprot->readContainerElementBegin()) < 0) {" << endl;
7382  indent_up();
7383  indent(out) << "return ret;" << endl;
7384  scope_down(out);
7385  indent(out) << "xfer += ret;" << endl;
7386  }
7387 #endif
7388  generate_deserialize_list_element(out, (t_list*)ttype, prefix, use_push, i);
7389 #ifdef SANDESH
7390  if (etype->is_base_type() || etype->is_enum()) {
7391  indent(out) << "if ((ret = iprot->readContainerElementEnd()) < 0) {" << endl;
7392  indent_up();
7393  indent(out) << "return ret;" << endl;
7394  scope_down(out);
7395  indent(out) << "xfer += ret;" << endl;
7396  }
7397 #endif
7398  }
7399 
7400  scope_down(out);
7401 
7402  // Read container end
7403  if (ttype->is_map()) {
7404 #ifdef SANDESH
7405  indent(out) << "if ((ret = iprot->readMapEnd()) < 0) {" << endl;
7406  indent_up();
7407  indent(out) << "return ret;" << endl;
7408  scope_down(out);
7409  indent(out) << "xfer += ret;" << endl;
7410 #else
7411  indent(out) << "iprot->readMapEnd();" << endl;
7412 #endif
7413  } else if (ttype->is_set()) {
7414 #ifdef SANDESH
7415  indent(out) << "if ((ret = iprot->readSetEnd()) < 0) {" << endl;
7416  indent_up();
7417  indent(out) << "return ret;" << endl;
7418  scope_down(out);
7419  indent(out) << "xfer += ret;" << endl;
7420 #else
7421  indent(out) << "iprot->readSetEnd();" << endl;
7422 #endif
7423  } else if (ttype->is_list()) {
7424 #ifdef SANDESH
7425  indent(out) << "if ((ret = iprot->readListEnd()) < 0) {" << endl;
7426  indent_up();
7427  indent(out) << "return ret;" << endl;
7428  scope_down(out);
7429  indent(out) << "xfer += ret;" << endl;
7430 #else
7431  indent(out) << "iprot->readListEnd();" << endl;
7432 #endif
7433  }
7434 
7435  scope_down(out);
7436 }
7437 
7438 
7442 void t_cpp_generator::generate_deserialize_map_element(ofstream& out,
7443  t_map* tmap,
7444  string prefix) {
7445  string key = tmp("_key");
7446  string val = tmp("_val");
7447  t_field fkey(tmap->get_key_type(), key);
7448  t_field fval(tmap->get_val_type(), val);
7449 
7450  out <<
7451  indent() << declare_field(&fkey) << endl;
7452 
7453 #ifdef SANDESH
7454  t_type* kttype = tmap->get_key_type();
7455  if (kttype->is_base_type() || kttype->is_enum()) {
7456  indent(out) << "if ((ret = iprot->readContainerElementBegin()) < 0) {" << endl;
7457  indent_up();
7458  indent(out) << "return ret;" << endl;
7459  scope_down(out);
7460  indent(out) << "xfer += ret;" << endl;
7461  }
7462 #endif
7463  generate_deserialize_field(out, &fkey);
7464 #ifdef SANDESH
7465  if (kttype->is_base_type() || kttype->is_enum()) {
7466  indent(out) << "if ((ret = iprot->readContainerElementEnd()) < 0) {" << endl;
7467  indent_up();
7468  indent(out) << "return ret;" << endl;
7469  scope_down(out);
7470  indent(out) << "xfer += ret;" << endl;
7471  }
7472 #endif
7473  indent(out) <<
7474  declare_field(&fval, false, false, false, true) << " = " <<
7475  prefix << "[" << key << "];" << endl;
7476 
7477 #ifdef SANDESH
7478  t_type* vttype = tmap->get_val_type();
7479  if (vttype->is_base_type() || vttype->is_enum()) {
7480  indent(out) << "if ((ret = iprot->readContainerElementBegin()) < 0) {" << endl;
7481  indent_up();
7482  indent(out) << "return ret;" << endl;
7483  scope_down(out);
7484  indent(out) << "xfer += ret;" << endl;
7485  }
7486 #endif
7487  generate_deserialize_field(out, &fval);
7488 #ifdef SANDESH
7489  if (vttype->is_base_type() || vttype->is_enum()) {
7490  indent(out) << "if ((ret = iprot->readContainerElementEnd()) < 0) {" << endl;
7491  indent_up();
7492  indent(out) << "return ret;" << endl;
7493  scope_down(out);
7494  indent(out) << "xfer += ret;" << endl;
7495  }
7496 #endif
7497 }
7498 
7499 void t_cpp_generator::generate_deserialize_set_element(ofstream& out,
7500  t_set* tset,
7501  string prefix) {
7502  string elem = tmp("_elem");
7503  t_field felem(tset->get_elem_type(), elem);
7504 
7505  indent(out) <<
7506  declare_field(&felem) << endl;
7507 
7508  generate_deserialize_field(out, &felem);
7509 
7510  indent(out) <<
7511  prefix << ".insert(" << elem << ");" << endl;
7512 }
7513 
7514 void t_cpp_generator::generate_deserialize_list_element(ofstream& out,
7515  t_list* tlist,
7516  string prefix,
7517  bool use_push,
7518  string index) {
7519  if (use_push) {
7520  string elem = tmp("_elem");
7521  t_field felem(tlist->get_elem_type(), elem);
7522  indent(out) << declare_field(&felem) << endl;
7523  generate_deserialize_field(out, &felem);
7524  indent(out) << prefix << ".push_back(" << elem << ");" << endl;
7525  } else {
7526  t_field felem(tlist->get_elem_type(), prefix + "[" + index + "]");
7527  generate_deserialize_field(out, &felem);
7528  }
7529 }
7530 
7531 
7538 void t_cpp_generator::generate_serialize_field(ofstream& out,
7539  t_field* tfield,
7540  string prefix,
7541  string suffix) {
7542  t_type* type = get_true_type(tfield->get_type());
7543 
7544  string name = prefix + tfield->get_name() + suffix;
7545 
7546  // Do nothing for void types
7547  if (type->is_void()) {
7548  throw "CANNOT GENERATE SERIALIZE CODE FOR void TYPE: " + name;
7549  }
7550 
7551 
7552 
7553  if (type->is_struct() || type->is_xception()) {
7554  generate_serialize_struct(out,
7555  (t_struct*)type,
7556  name);
7557  } else if (type->is_container()) {
7558  generate_serialize_container(out, type, name);
7559  } else if (type->is_base_type() || type->is_enum()) {
7560 
7561  indent(out) <<
7562  "if ((ret = oprot->";
7563 
7564  if (type->is_base_type()) {
7565  t_base_type::t_base tbase = ((t_base_type*)type)->get_base();
7566  switch (tbase) {
7567  case t_base_type::TYPE_VOID:
7568  throw
7569  "compiler error: cannot serialize void field in a struct: " + name;
7570  break;
7571  case t_base_type::TYPE_STRING:
7572 #ifdef SANDESH
7573  case t_base_type::TYPE_STATIC_CONST_STRING:
7574 #endif
7575  if (((t_base_type*)type)->is_binary()) {
7576  out << "writeBinary(" << name << ")) < 0) {" << endl;
7577  }
7578  else {
7579  out << "writeString(" << name << ")) < 0) {" << endl;
7580  }
7581  break;
7582 #ifdef SANDESH
7583  case t_base_type::TYPE_XML:
7584  out << "writeXML(" << name << ")) < 0) {" << endl;
7585  break;
7586  case t_base_type::TYPE_UUID:
7587  out << "writeUUID(" << name << ")) < 0) {" << endl;
7588  break;
7589 #endif
7590  case t_base_type::TYPE_BOOL:
7591  out << "writeBool(" << name << ")) < 0) {" << endl;
7592  break;
7593  case t_base_type::TYPE_BYTE:
7594  out << "writeByte(" << name << ")) < 0) {" << endl;
7595  break;
7596  case t_base_type::TYPE_I16:
7597  out << "writeI16(" << name << ")) < 0) {" << endl;
7598  break;
7599  case t_base_type::TYPE_I32:
7600  out << "writeI32(" << name << ")) < 0) {" << endl;
7601  break;
7602  case t_base_type::TYPE_I64:
7603  out << "writeI64(" << name << ")) < 0) {" << endl;
7604  break;
7605 #ifdef SANDESH
7606  case t_base_type::TYPE_U16:
7607  out << "writeU16(" << name << ")) < 0) {" << endl;
7608  break;
7609  case t_base_type::TYPE_U32:
7610  out << "writeU32(" << name << ")) < 0) {" << endl;
7611  break;
7612  case t_base_type::TYPE_U64:
7613  out << "writeU64(" << name << ")) < 0) {" << endl;
7614  break;
7615  case t_base_type::TYPE_IPV4:
7616  out << "writeIPV4(" << name << ")) < 0) {" << endl;
7617  break;
7618  case t_base_type::TYPE_IPADDR:
7619  out << "writeIPADDR(" << name << ")) < 0) {" << endl;
7620  break;
7621 #endif
7622  case t_base_type::TYPE_DOUBLE:
7623  out << "writeDouble(" << name << ")) < 0) {" << endl;
7624  break;
7625  default:
7626  throw "compiler error: no C++ writer for base type " + t_base_type::t_base_name(tbase) + name;
7627  }
7628  } else if (type->is_enum()) {
7629  out << "writeI32((int32_t)" << name << ")) < 0) {" << endl;
7630  }
7631  indent_up();
7632  out << indent() << "return ret;" << endl;
7633  scope_down(out);
7634  out << indent() << "xfer += ret;" << endl;
7635  out << endl;
7636 #ifdef SANDESH
7637  } else if (type->is_sandesh()) {
7638  generate_serialize_sandesh(out, (t_sandesh*)type, name);
7639 #endif
7640  } else {
7641  printf("DO NOT KNOW HOW TO SERIALIZE FIELD '%s' TYPE '%s'\n",
7642  name.c_str(),
7643  type_name(type).c_str());
7644  }
7645 }
7646 
7647 #ifdef SANDESH
7654 void t_cpp_generator::generate_serialize_sandesh(ofstream& out,
7655  t_sandesh* tsandesh,
7656  string prefix) {
7657  (void) tsandesh;
7658  indent(out) <<
7659  "if ((ret = " << prefix << ".write(oprot)) < 0) {" << endl;
7660  indent_up();
7661  indent(out) << "return ret;" << endl;
7662  scope_down(out);
7663  indent(out) << "xfer += ret;" << endl;
7664 }
7665 #endif
7666 
7673 void t_cpp_generator::generate_serialize_struct(ofstream& out,
7674  t_struct* tstruct,
7675  string prefix) {
7676  (void) tstruct;
7677  indent(out) <<
7678  "if ((ret = " << prefix << ".write(oprot)) < 0) {" << endl;
7679  indent_up();
7680  indent(out) << "return ret;" << endl;
7681  scope_down(out);
7682  indent(out) << "xfer += ret;" << endl;
7683 }
7684 
7685 void t_cpp_generator::generate_serialize_container(ofstream& out,
7686  t_type* ttype,
7687  string prefix) {
7688  scope_up(out);
7689 
7690  if (ttype->is_map()) {
7691  indent(out) <<
7692  "if ((ret = oprot->writeMapBegin(" <<
7693  type_to_enum(((t_map*)ttype)->get_key_type()) << ", " <<
7694  type_to_enum(((t_map*)ttype)->get_val_type()) << ", " <<
7695  "static_cast<uint32_t>(" << prefix << ".size()))) < 0) {" << endl;
7696  indent_up();
7697  indent(out) << "return ret;" << endl;
7698  scope_down(out);
7699  indent(out) << "xfer += ret;" << endl;
7700  } else if (ttype->is_set()) {
7701  indent(out) <<
7702  "if ((ret = oprot->writeSetBegin(" <<
7703  type_to_enum(((t_set*)ttype)->get_elem_type()) << ", " <<
7704  "static_cast<uint32_t>(" << prefix << ".size()))) < 0) {" << endl;
7705  indent_up();
7706  indent(out) << "return ret;" << endl;
7707  scope_down(out);
7708  indent(out) << "xfer += ret;" << endl;
7709  } else if (ttype->is_list()) {
7710  indent(out) <<
7711  "if ((ret = oprot->writeListBegin(" <<
7712  type_to_enum(((t_list*)ttype)->get_elem_type()) << ", " <<
7713  "static_cast<uint32_t>(" << prefix << ".size()))) < 0) {" << endl;
7714  indent_up();
7715  indent(out) << "return ret;" << endl;
7716  scope_down(out);
7717  indent(out) << "xfer += ret;" << endl;
7718  }
7719 
7720  string iter = tmp("_iter");
7721  out <<
7722  indent() << type_name(ttype) << "::const_iterator " << iter << ";" << endl <<
7723  indent() << "for (" << iter << " = " << prefix << ".begin(); " << iter << " != " << prefix << ".end(); ++" << iter << ")" << endl;
7724  scope_up(out);
7725  if (ttype->is_map()) {
7726  generate_serialize_map_element(out, (t_map*)ttype, iter);
7727  } else if (ttype->is_set()) {
7728 #ifdef SANDESH
7729  t_type *etype = ((t_set*)ttype)->get_elem_type();
7730  if (etype->is_base_type() || etype->is_enum()) {
7731  indent(out) << "if ((ret = oprot->writeContainerElementBegin()) < 0) {" << endl;
7732  indent_up();
7733  indent(out) << "return ret;" << endl;
7734  scope_down(out);
7735  indent(out) << "xfer += ret;" << endl;
7736  }
7737 #endif
7738  generate_serialize_set_element(out, (t_set*)ttype, iter);
7739 #ifdef SANDESH
7740  if (etype->is_base_type() || etype->is_enum()) {
7741  indent(out) << "if ((ret = oprot->writeContainerElementEnd()) < 0) {" << endl;
7742  indent_up();
7743  indent(out) << "return ret;" << endl;
7744  scope_down(out);
7745  indent(out) << "xfer += ret;" << endl;
7746  }
7747 #endif
7748  } else if (ttype->is_list()) {
7749 #ifdef SANDESH
7750  t_type *etype = ((t_list*)ttype)->get_elem_type();
7751  if (etype->is_base_type() || etype->is_enum()) {
7752  indent(out) << "if ((ret = oprot->writeContainerElementBegin()) < 0) {" << endl;
7753  indent_up();
7754  indent(out) << "return ret;" << endl;
7755  scope_down(out);
7756  indent(out) << "xfer += ret;" << endl;
7757  }
7758 #endif
7759  generate_serialize_list_element(out, (t_list*)ttype, iter);
7760 #ifdef SANDESH
7761  if (etype->is_base_type() || etype->is_enum()) {
7762  indent(out) << "if ((ret = oprot->writeContainerElementEnd()) < 0) {" << endl;
7763  indent_up();
7764  indent(out) << "return ret;" << endl;
7765  scope_down(out);
7766  indent(out) << "xfer += ret;" << endl;
7767  }
7768 #endif
7769  }
7770  scope_down(out);
7771 
7772  if (ttype->is_map()) {
7773  indent(out) <<
7774  "if ((ret = oprot->writeMapEnd()) < 0) {" << endl;
7775  indent_up();
7776  indent(out) << "return ret;" << endl;
7777  scope_down(out);
7778  indent(out) << "xfer += ret;" << endl;
7779  } else if (ttype->is_set()) {
7780  indent(out) <<
7781  "if ((ret = oprot->writeSetEnd()) < 0) {" << endl;
7782  indent_up();
7783  indent(out) << "return ret;" << endl;
7784  scope_down(out);
7785  indent(out) << "xfer += ret;" << endl;
7786  } else if (ttype->is_list()) {
7787  indent(out) <<
7788  "if ((ret = oprot->writeListEnd()) < 0) {" << endl;
7789  indent_up();
7790  indent(out) << "return ret;" << endl;
7791  scope_down(out);
7792  indent(out) << "xfer += ret;" << endl;
7793  }
7794 
7795  scope_down(out);
7796 }
7797 
7802 void t_cpp_generator::generate_serialize_map_element(ofstream& out,
7803  t_map* tmap,
7804  string iter) {
7805 #ifdef SANDESH
7806  t_type* kttype = tmap->get_key_type();
7807  if (kttype->is_base_type() || kttype->is_enum()) {
7808  indent(out) << "if ((ret = oprot->writeContainerElementBegin()) < 0) {" << endl;
7809  indent_up();
7810  indent(out) << "return ret;" << endl;
7811  scope_down(out);
7812  indent(out) << "xfer += ret;" << endl;
7813  }
7814 #endif
7815  t_field kfield(tmap->get_key_type(), iter + "->first");
7816  generate_serialize_field(out, &kfield, "");
7817 #ifdef SANDESH
7818  if (kttype->is_base_type() || kttype->is_enum()) {
7819  indent(out) << "if ((ret = oprot->writeContainerElementEnd()) < 0) {" << endl;
7820  indent_up();
7821  indent(out) << "return ret;" << endl;
7822  scope_down(out);
7823  indent(out) << "xfer += ret;" << endl;
7824  }
7825 #endif
7826 #ifdef SANDESH
7827  t_type* vttype = tmap->get_val_type();
7828  if (vttype->is_base_type() || vttype->is_enum()) {
7829  indent(out) << "if ((ret = oprot->writeContainerElementBegin()) < 0) {" << endl;
7830  indent_up();
7831  indent(out) << "return ret;" << endl;
7832  scope_down(out);
7833  indent(out) << "xfer += ret;" << endl;
7834  }
7835 #endif
7836  t_field vfield(tmap->get_val_type(), iter + "->second");
7837  generate_serialize_field(out, &vfield, "");
7838 #ifdef SANDESH
7839  if (vttype->is_base_type() || vttype->is_enum()) {
7840  indent(out) << "if ((ret = oprot->writeContainerElementEnd()) < 0) {" << endl;
7841  indent_up();
7842  indent(out) << "return ret;" << endl;
7843  scope_down(out);
7844  indent(out) << "xfer += ret;" << endl;
7845  }
7846 #endif
7847 }
7848 
7852 void t_cpp_generator::generate_serialize_set_element(ofstream& out,
7853  t_set* tset,
7854  string iter) {
7855  t_field efield(tset->get_elem_type(), "(*" + iter + ")");
7856  generate_serialize_field(out, &efield, "");
7857 }
7858 
7862 void t_cpp_generator::generate_serialize_list_element(ofstream& out,
7863  t_list* tlist,
7864  string iter) {
7865  t_field efield(tlist->get_elem_type(), "(*" + iter + ")");
7866  generate_serialize_field(out, &efield, "");
7867 }
7868 
7875 string t_cpp_generator::namespace_prefix(string ns) {
7876  // Always start with "::", to avoid possible name collisions with
7877  // other names in one of the current namespaces.
7878  //
7879  // We also need a leading space, in case the name is used inside of a
7880  // template parameter. "MyTemplate<::foo::Bar>" is not valid C++,
7881  // since "<:" is an alternative token for "[".
7882  string result = " ::";
7883 
7884  if (ns.size() == 0) {
7885  return result;
7886  }
7887  string::size_type loc;
7888  while ((loc = ns.find(".")) != string::npos) {
7889  result += ns.substr(0, loc);
7890  result += "::";
7891  ns = ns.substr(loc+1);
7892  }
7893  if (ns.size() > 0) {
7894  result += ns + "::";
7895  }
7896  return result;
7897 }
7898 
7905 string t_cpp_generator::namespace_open(string ns) {
7906  if (ns.size() == 0) {
7907  return "";
7908  }
7909  string result = "";
7910  string separator = "";
7911  string::size_type loc;
7912  while ((loc = ns.find(".")) != string::npos) {
7913  result += separator;
7914  result += "namespace ";
7915  result += ns.substr(0, loc);
7916  result += " {";
7917  separator = " ";
7918  ns = ns.substr(loc+1);
7919  }
7920  if (ns.size() > 0) {
7921  result += separator + "namespace " + ns + " {";
7922  }
7923  return result;
7924 }
7925 
7932 string t_cpp_generator::namespace_close(string ns) {
7933  if (ns.size() == 0) {
7934  return "";
7935  }
7936  string result = "}";
7937  string::size_type loc;
7938  while ((loc = ns.find(".")) != string::npos) {
7939  result += "}";
7940  ns = ns.substr(loc+1);
7941  }
7942  result += " // namespace";
7943  return result;
7944 }
7945 
7952 string t_cpp_generator::type_name(t_type* ttype, bool in_typedef, bool arg) {
7953  if (ttype->is_base_type()) {
7954  string bname = base_type_name(((t_base_type*)ttype)->get_base());
7955  if (!arg) {
7956  return bname;
7957  }
7958 
7959  if (((t_base_type*)ttype)->get_base() == t_base_type::TYPE_STRING
7960 #ifdef SANDESH
7961  || ((t_base_type*)ttype)->get_base() == t_base_type::TYPE_XML
7962  || ((t_base_type*)ttype)->get_base() == t_base_type::TYPE_UUID
7963  || ((t_base_type*)ttype)->get_base() == t_base_type::TYPE_IPADDR
7964 #endif
7965  ) {
7966  return "const " + bname + "&";
7967  } else {
7968  return "const " + bname;
7969  }
7970  }
7971 
7972  // Check for a custom overloaded C++ name
7973  if (ttype->is_container()) {
7974  string cname;
7975 
7976  t_container* tcontainer = (t_container*) ttype;
7977  if (tcontainer->has_cpp_name()) {
7978  cname = tcontainer->get_cpp_name();
7979  } else if (ttype->is_map()) {
7980  t_map* tmap = (t_map*) ttype;
7981  cname = "std::map<" +
7982  type_name(tmap->get_key_type(), in_typedef) + ", " +
7983  type_name(tmap->get_val_type(), in_typedef) + "> ";
7984  } else if (ttype->is_set()) {
7985  t_set* tset = (t_set*) ttype;
7986  cname = "std::set<" + type_name(tset->get_elem_type(), in_typedef) + "> ";
7987  } else if (ttype->is_list()) {
7988  t_list* tlist = (t_list*) ttype;
7989  cname = "std::vector<" + type_name(tlist->get_elem_type(), in_typedef) + "> ";
7990  }
7991 
7992  if (arg) {
7993  return "const " + cname + "&";
7994  } else {
7995  return cname;
7996  }
7997  }
7998 
7999  string class_prefix;
8000  if (in_typedef && (ttype->is_struct() || ttype->is_xception())) {
8001  class_prefix = "class ";
8002  }
8003 
8004  // Check if it needs to be namespaced
8005  string pname;
8006  t_program* program = ttype->get_program();
8007  if (program != NULL && program != program_) {
8008  pname =
8009  class_prefix +
8010  namespace_prefix(program->get_namespace("cpp")) +
8011  ttype->get_name();
8012  } else {
8013  pname = class_prefix + ttype->get_name();
8014  }
8015 
8016  if (ttype->is_enum() && !gen_pure_enums_) {
8017  pname += "::type";
8018  }
8019 
8020  if (arg) {
8021  if (is_complex_type(ttype)) {
8022  return "const " + pname + "&";
8023  } else {
8024  return "const " + pname;
8025  }
8026  } else {
8027  return pname;
8028  }
8029 }
8030 
8037 string t_cpp_generator::base_type_name(t_base_type::t_base tbase) {
8038  switch (tbase) {
8039  case t_base_type::TYPE_VOID:
8040  return "void";
8041 #ifdef SANDESH
8042  case t_base_type::TYPE_STATIC_CONST_STRING:
8043  return "static std::string";
8044 #endif
8045  case t_base_type::TYPE_STRING:
8046 #ifdef SANDESH
8047  case t_base_type::TYPE_XML:
8048 #endif
8049  return "std::string";
8050  case t_base_type::TYPE_BOOL:
8051  return "bool";
8052  case t_base_type::TYPE_BYTE:
8053  return "int8_t";
8054  case t_base_type::TYPE_I16:
8055  return "int16_t";
8056  case t_base_type::TYPE_I32:
8057  return "int32_t";
8058  case t_base_type::TYPE_I64:
8059  return "int64_t";
8060 #ifdef SANDESH
8061  case t_base_type::TYPE_U16:
8062  return "uint16_t";
8063  case t_base_type::TYPE_U32:
8064  return "uint32_t";
8065  case t_base_type::TYPE_U64:
8066  return "uint64_t";
8067  case t_base_type::TYPE_IPV4:
8068  return "uint32_t";
8069  case t_base_type::TYPE_IPADDR:
8070  return "boost::asio::ip::address";
8071  case t_base_type::TYPE_UUID:
8072  return "boost::uuids::uuid";
8073 #endif
8074  case t_base_type::TYPE_DOUBLE:
8075  return "double";
8076  default:
8077  throw "compiler error: no C++ base type name for base type " + t_base_type::t_base_name(tbase);
8078  }
8079 }
8080 
8087 #ifdef SANDESH
8088 string t_cpp_generator::declare_field(t_field* tfield, bool init, bool pointer, bool constant, bool reference, bool constructor) {
8089 #else
8090 string t_cpp_generator::declare_field(t_field* tfield, bool init, bool pointer, bool constant, bool reference) {
8091 #endif
8092  // TODO(mcslee): do we ever need to initialize the field?
8093  string result = "";
8094  if (constant) {
8095  result += "const ";
8096  }
8097  result += type_name(tfield->get_type());
8098  if (pointer) {
8099  result += "*";
8100  }
8101  if (reference) {
8102  result += "&";
8103  }
8104  result += " " + tfield->get_name();
8105  if (init) {
8106  t_type* type = get_true_type(tfield->get_type());
8107 
8108  if (type->is_base_type()) {
8109  t_base_type::t_base tbase = ((t_base_type*)type)->get_base();
8110  switch (tbase) {
8111  case t_base_type::TYPE_VOID:
8112  break;
8113 #ifdef SANDESH
8114  case t_base_type::TYPE_UUID:
8115  result += " = boost::uuids::nil_uuid()";
8116  break;
8117  case t_base_type::TYPE_STATIC_CONST_STRING:
8118  case t_base_type::TYPE_XML:
8119 #endif
8120  case t_base_type::TYPE_STRING:
8121  result += " = \"\"";
8122  break;
8123  case t_base_type::TYPE_BOOL:
8124  result += " = false";
8125  break;
8126  case t_base_type::TYPE_BYTE:
8127  case t_base_type::TYPE_I16:
8128  case t_base_type::TYPE_I32:
8129  case t_base_type::TYPE_I64:
8130 #ifdef SANDESH
8131  case t_base_type::TYPE_U16:
8132  case t_base_type::TYPE_U32:
8133  case t_base_type::TYPE_U64:
8134  case t_base_type::TYPE_IPV4:
8135 #endif
8136  result += " = 0";
8137  break;
8138  case t_base_type::TYPE_DOUBLE:
8139  result += " = (double)0";
8140  break;
8141 #ifdef SANDESH
8142  case t_base_type::TYPE_IPADDR:
8143  result += " = boost::asio::ip::address()";
8144  break;
8145 #endif
8146  default:
8147  throw "compiler error: no C++ initializer for base type " + t_base_type::t_base_name(tbase);
8148  }
8149  } else if (type->is_enum()) {
8150  result += " = (" + type_name(type) + ")0";
8151  }
8152  }
8153  if (!reference
8154 #ifdef SANDESH
8155  && !constructor
8156 #endif
8157  ) {
8158  result += ";";
8159  }
8160  return result;
8161 }
8162 
8169 string t_cpp_generator::function_signature(t_function* tfunction,
8170  string style,
8171  string prefix,
8172  bool name_params) {
8173  t_type* ttype = tfunction->get_returntype();
8174  t_struct* arglist = tfunction->get_arglist();
8175  bool has_xceptions = !tfunction->get_xceptions()->get_members().empty();
8176 #ifdef SANDESH
8177  bool is_static = tfunction->is_static();
8178  bool is_virtual = tfunction->is_virtual();
8179  string func_begin_modifier = "", func_end_modifier = "";
8180 #endif
8181 
8182 #ifdef SANDESH
8183  if (is_static) {
8184  func_begin_modifier = "static ";
8185  } else if (is_virtual) {
8186  func_begin_modifier = "virtual ";
8187  func_end_modifier = " = 0";
8188  }
8189 #endif
8190 
8191  if (style == "") {
8192  if (is_complex_type(ttype)) {
8193  return
8194 #ifdef SANDESH
8195  func_begin_modifier +
8196 #endif
8197  "void " + prefix + tfunction->get_name() +
8198  "(" + type_name(ttype) + (name_params ? "& _return" : "& /* _return */") +
8199  argument_list(arglist, name_params, true) + ")"
8200 #ifdef SANDESH
8201  + func_end_modifier
8202 #endif
8203  ;
8204  } else {
8205  return
8206 #ifdef SANDESH
8207  func_begin_modifier +
8208 #endif
8209  type_name(ttype) + " " + prefix + tfunction->get_name() +
8210  "(" + argument_list(arglist, name_params) + ")"
8211 #ifdef SANDESH
8212  + func_end_modifier
8213 #endif
8214  ;
8215  }
8216  } else if (style.substr(0,3) == "Cob") {
8217  string cob_type;
8218  string exn_cob;
8219  if (style == "CobCl") {
8220  cob_type = "(" + service_name_ + "CobClient";
8221  if (gen_templates_) {
8222  cob_type += "T<Protocol_>";
8223  }
8224  cob_type += "* client)";
8225  } else if (style =="CobSv") {
8226  cob_type = (ttype->is_void()
8227  ? "()"
8228  : ("(" + type_name(ttype) + " const& _return)"));
8229  if (has_xceptions) {
8230  exn_cob = ", std::tr1::function<void(::contrail::sandesh::TDelayedException* _throw)> /* exn_cob */";
8231  }
8232  } else {
8233  throw "UNKNOWN STYLE";
8234  }
8235 
8236  return
8237  "void " + prefix + tfunction->get_name() +
8238  "(std::tr1::function<void" + cob_type + "> cob" + exn_cob +
8239  argument_list(arglist, name_params, true) + ")";
8240  } else {
8241  throw "UNKNOWN STYLE";
8242  }
8243 }
8244 
8251 string t_cpp_generator::argument_list(t_struct* tstruct, bool name_params, bool start_comma) {
8252  string result = "";
8253 
8254  const vector<t_field*>& fields = tstruct->get_members();
8255  vector<t_field*>::const_iterator f_iter;
8256  bool first = !start_comma;
8257  for (f_iter = fields.begin(); f_iter != fields.end(); ++f_iter) {
8258  if (first) {
8259  first = false;
8260  } else {
8261  result += ", ";
8262  }
8263  result += type_name((*f_iter)->get_type(), false, true) + " " +
8264  (name_params ? (*f_iter)->get_name() : "/* " + (*f_iter)->get_name() + " */");
8265  }
8266  return result;
8267 }
8268 
8275 string t_cpp_generator::type_to_enum(t_type* type) {
8276  type = get_true_type(type);
8277 
8278  if (type->is_base_type()) {
8279  t_base_type::t_base tbase = ((t_base_type*)type)->get_base();
8280  switch (tbase) {
8281  case t_base_type::TYPE_VOID:
8282  throw "NO T_VOID CONSTRUCT";
8283  case t_base_type::TYPE_STRING:
8284  return "::contrail::sandesh::protocol::T_STRING";
8285  case t_base_type::TYPE_BOOL:
8286  return "::contrail::sandesh::protocol::T_BOOL";
8287  case t_base_type::TYPE_BYTE:
8288  return "::contrail::sandesh::protocol::T_BYTE";
8289  case t_base_type::TYPE_I16:
8290  return "::contrail::sandesh::protocol::T_I16";
8291  case t_base_type::TYPE_I32:
8292  return "::contrail::sandesh::protocol::T_I32";
8293  case t_base_type::TYPE_I64:
8294  return "::contrail::sandesh::protocol::T_I64";
8295  case t_base_type::TYPE_DOUBLE:
8296  return "::contrail::sandesh::protocol::T_DOUBLE";
8297 #ifdef SANDESH
8298  case t_base_type::TYPE_XML:
8299  return "::contrail::sandesh::protocol::T_XML";
8300  case t_base_type::TYPE_UUID:
8301  return "::contrail::sandesh::protocol::T_UUID";
8302  case t_base_type::TYPE_U16:
8303  return "::contrail::sandesh::protocol::T_U16";
8304  case t_base_type::TYPE_U32:
8305  return "::contrail::sandesh::protocol::T_U32";
8306  case t_base_type::TYPE_U64:
8307  return "::contrail::sandesh::protocol::T_U64";
8308  case t_base_type::TYPE_IPV4:
8309  return "::contrail::sandesh::protocol::T_IPV4";
8310  case t_base_type::TYPE_IPADDR:
8311  return "::contrail::sandesh::protocol::T_IPADDR";
8312  case t_base_type::TYPE_STATIC_CONST_STRING:
8313  case t_base_type::TYPE_SANDESH_SYSTEM:
8314  case t_base_type::TYPE_SANDESH_REQUEST:
8315  case t_base_type::TYPE_SANDESH_RESPONSE:
8316  case t_base_type::TYPE_SANDESH_TRACE:
8317  case t_base_type::TYPE_SANDESH_TRACE_OBJECT:
8318  case t_base_type::TYPE_SANDESH_BUFFER:
8319  case t_base_type::TYPE_SANDESH_UVE:
8320  case t_base_type::TYPE_SANDESH_ALARM:
8321  case t_base_type::TYPE_SANDESH_OBJECT:
8322  case t_base_type::TYPE_SANDESH_FLOW:
8323  case t_base_type::TYPE_SANDESH_SESSION:
8324  return "::contrail::sandesh::protocol::T_STRING";
8325 #endif
8326  }
8327  } else if (type->is_enum()) {
8328  return "::contrail::sandesh::protocol::T_I32";
8329  } else if (type->is_struct()) {
8330  return "::contrail::sandesh::protocol::T_STRUCT";
8331  } else if (type->is_xception()) {
8332  return "::contrail::sandesh::protocol::T_STRUCT";
8333 #ifdef SANDESH
8334  } else if (type->is_sandesh()) {
8335  return "::contrail::sandesh::protocol::T_SANDESH";
8336 #endif
8337  } else if (type->is_map()) {
8338  return "::contrail::sandesh::protocol::T_MAP";
8339  } else if (type->is_set()) {
8340  return "::contrail::sandesh::protocol::T_SET";
8341  } else if (type->is_list()) {
8342  return "::contrail::sandesh::protocol::T_LIST";
8343  }
8344 
8345  throw "INVALID TYPE IN type_to_enum: " + type->get_name();
8346 }
8347 
8351 string t_cpp_generator::local_reflection_name(const char* prefix, t_type* ttype, bool external) {
8352  ttype = get_true_type(ttype);
8353 
8354  // We have to use the program name as part of the identifier because
8355  // if two thrift "programs" are compiled into one actual program
8356  // you would get a symbol collision if they both defined list<i32>.
8357  // trlo = Thrift Reflection LOcal.
8358  string prog;
8359  string name;
8360  string nspace;
8361 
8362  // TODO(dreiss): Would it be better to pregenerate the base types
8363  // and put them in Thrift.{h,cpp} ?
8364 
8365  if (ttype->is_base_type()) {
8366  prog = program_->get_name();
8367  name = ttype->get_ascii_fingerprint();
8368  } else if (ttype->is_enum()) {
8369  assert(ttype->get_program() != NULL);
8370  prog = ttype->get_program()->get_name();
8371  name = ttype->get_ascii_fingerprint();
8372  } else if (ttype->is_container()) {
8373  prog = program_->get_name();
8374  name = ttype->get_ascii_fingerprint();
8375  } else {
8376  assert(ttype->is_struct() || ttype->is_xception());
8377  assert(ttype->get_program() != NULL);
8378  prog = ttype->get_program()->get_name();
8379  name = ttype->get_ascii_fingerprint();
8380  }
8381 
8382  if (external &&
8383  ttype->get_program() != NULL &&
8384  ttype->get_program() != program_) {
8385  nspace = namespace_prefix(ttype->get_program()->get_namespace("cpp"));
8386  }
8387 
8388  return nspace + "trlo_" + prefix + "_" + prog + "_" + name;
8389 }
8390 
8391 string t_cpp_generator::get_include_prefix(const t_program& program) const {
8392  string include_prefix = program.get_include_prefix();
8393  if (!use_include_prefix_ ||
8394  (include_prefix.size() > 0 && include_prefix[0] == '/')) {
8395  // if flag is turned off or this is absolute path, return empty prefix
8396  return "";
8397  }
8398 
8399  string::size_type last_slash = string::npos;
8400  if ((last_slash = include_prefix.rfind("/")) != string::npos) {
8401  return include_prefix.substr(0, last_slash) + "/" + out_dir_base_ + "/";
8402  }
8403 
8404  return "";
8405 }
8406 
8407 void t_cpp_generator::generate_common_struct_reader_body(
8408  std::ofstream& out,
8409  t_struct_common* tstruct,
8410  bool pointers) {
8411  indent_up();
8412 
8413  std::string struct_type_name = tstruct->get_struct_type_name();
8414 
8415  const vector<t_field*>& fields = tstruct->get_members();
8416  vector<t_field*>::const_iterator f_iter;
8417 
8418  // Declare stack tmp variables
8419  out <<
8420  endl <<
8421  indent() << "int32_t xfer = 0, ret;" << endl <<
8422  indent() << "std::string fname;" << endl <<
8423  indent() << "::contrail::sandesh::protocol::TType ftype;" << endl <<
8424  indent() << "int16_t fid;" << endl <<
8425  endl <<
8426  indent() <<
8427  "if ((ret = iprot->read" << struct_type_name << "Begin(fname)) < 0) {" <<
8428  endl <<
8429  indent() << " return ret;" << endl <<
8430  indent() << "}" << endl <<
8431  indent() << "xfer += ret;" << endl <<
8432  endl;
8433 
8434  // Required variables aren't in __isset, so we need tmp vars to check them.
8435  for (f_iter = fields.begin(); f_iter != fields.end(); ++f_iter) {
8436  if ((*f_iter)->get_req() == t_field::T_REQUIRED)
8437  indent(out) << "bool isset_" << (*f_iter)->get_name() << " = false;" << endl;
8438  }
8439  out << endl;
8440 
8441  // Loop over reading in fields
8442  indent(out) <<
8443  "while (true)" << endl;
8444  scope_up(out);
8445 
8446  // Read beginning field marker
8447  indent(out) <<
8448  "if ((ret = iprot->readFieldBegin(fname, ftype, fid)) < 0) {" << endl;
8449  indent_up();
8450  indent(out) << "return ret;" << endl;
8451  scope_down(out);
8452  indent(out) << "xfer += ret;" << endl;
8453 
8454  // Check for field STOP marker
8455  out <<
8456  indent() << "if (ftype == ::contrail::sandesh::protocol::T_STOP) {" << endl <<
8457  indent() << " break;" << endl <<
8458  indent() << "}" << endl;
8459 
8460  // Generate 'switch' statement only when there is at least
8461  // one 'case' in it. Otherwise, generate just the contents
8462  // of 'default' label, without a 'switch'.
8463  if (fields.empty()) {
8464  out <<
8465  indent() << "if ((ret = iprot->skip(ftype)) < 0) {" << endl <<
8466  indent() << " return ret;" << endl <<
8467  indent() << "}" << endl <<
8468  indent() << "xfer += ret;" << endl;
8469  } else {
8470  // Switch statement on the field we are reading
8471  indent(out) <<
8472  "switch (fid)" << endl;
8473 
8474  scope_up(out);
8475 
8476  // Generate deserialization code for known cases
8477  for (f_iter = fields.begin(); f_iter != fields.end(); ++f_iter) {
8478  indent(out) <<
8479  "case " << (*f_iter)->get_key() << ":" << endl;
8480  indent_up();
8481  indent(out) <<
8482  "if (ftype == " << type_to_enum((*f_iter)->get_type()) << ") {" << endl;
8483  indent_up();
8484 
8485  const char *isset_prefix =
8486  ((*f_iter)->get_req() != t_field::T_REQUIRED) ? "this->__isset." : "isset_";
8487 
8488 #if 0
8489  // This code throws an exception if the same field is encountered twice.
8490  // We've decided to leave it out for performance reasons.
8491  // TODO(dreiss): Generate this code and "if" it out to make it easier
8492  // for people recompiling thrift to include it.
8493  out <<
8494  indent() << "if (" << isset_prefix << (*f_iter)->get_name() << ")" << endl <<
8495  indent() << " throw TProtocolException(TProtocolException::INVALID_DATA);" << endl;
8496 #endif
8497 
8498  if (pointers && !(*f_iter)->get_type()->is_xception()) {
8499  generate_deserialize_field(out, *f_iter, "(*(this->", "))");
8500  } else {
8501  generate_deserialize_field(out, *f_iter, "this->");
8502  }
8503  out <<
8504  indent() << isset_prefix << (*f_iter)->get_name() << " = true;" << endl;
8505  indent_down();
8506  out <<
8507  indent() << "} else {" << endl <<
8508  indent() << " if ((ret = iprot->skip(ftype)) < 0) {" << endl <<
8509  indent() << " return ret;" << endl <<
8510  indent() << " }" << endl <<
8511  indent() << " xfer += ret;" << endl <<
8512  // TODO(dreiss): Make this an option when thrift structs
8513  // have a common base class.
8514  // indent() << " throw TProtocolException(TProtocolException::INVALID_DATA);" << endl <<
8515  indent() << "}" << endl <<
8516  indent() << "break;" << endl;
8517  indent_down();
8518  }
8519 
8520  // In the default case we skip the field
8521  out <<
8522  indent() << "default:" << endl <<
8523  indent() << " if ((ret = iprot->skip(ftype)) < 0) {" << endl <<
8524  indent() << " return ret;" << endl <<
8525  indent() << " }" << endl <<
8526  indent() << " xfer += ret;" << endl <<
8527  indent() << " break;" << endl;
8528 
8529  scope_down(out);
8530  }
8531 
8532  // Read field end marker
8533  indent(out) <<
8534  "if ((ret = iprot->readFieldEnd()) < 0) {" << endl;
8535  indent_up();
8536  indent(out) << "return ret;" << endl;
8537  scope_down(out);
8538  indent(out) << "xfer += ret;" << endl;
8539 
8540  scope_down(out);
8541 
8542  out <<
8543  endl <<
8544  indent() <<
8545  "if ((ret = iprot->read" << struct_type_name << "End()) < 0) {" << endl;
8546 
8547  indent_up();
8548  indent(out) << "return ret;" << endl;
8549  scope_down(out);
8550  indent(out) << "xfer += ret;" << endl;
8551 
8552  // Throw if any required fields are missing.
8553  // We do this after reading the struct end so that
8554  // there might possibly be a chance of continuing.
8555  out << endl;
8556  for (f_iter = fields.begin(); f_iter != fields.end(); ++f_iter) {
8557  if ((*f_iter)->get_req() == t_field::T_REQUIRED)
8558  out <<
8559  indent() << "if (!isset_" << (*f_iter)->get_name() << ") {" << endl <<
8560  indent() << " LOG(ERROR, __func__ << \": Required field " << (*f_iter)->get_name()
8561  << " not set\");" << endl <<
8562  indent() << " return -1;" << endl <<
8563  indent() << "}" << endl;
8564  }
8565 
8566  indent(out) << "return xfer;" << endl;
8567 
8568  indent_down();
8569 }
8570 
8571 void t_cpp_generator::generate_common_struct_writer_body(
8572  std::ofstream& out,
8573  t_struct_common* tstruct,
8574  bool pointers) {
8575  indent_up();
8576 
8577  std::string struct_type_name = tstruct->get_struct_type_name();
8578 
8579  string name = tstruct->get_name();
8580  const vector<t_field*>& fields = tstruct->get_members();
8581  vector<t_field*>::const_iterator f_iter;
8582 
8583  out <<
8584  indent() << "int32_t xfer = 0, ret;" << endl;
8585 
8586  indent(out) <<
8587  "if ((ret = oprot->write" << struct_type_name << "Begin(\"" << name << "\")) < 0) {" <<
8588  endl;
8589 
8590  indent_up();
8591  indent(out) << "return ret;" << endl;
8592  scope_down(out);
8593  indent(out) << "xfer += ret;" << endl;
8594  for (f_iter = fields.begin(); f_iter != fields.end(); ++f_iter) {
8595  if ((*f_iter)->get_req() == t_field::T_OPTIONAL) {
8596  indent(out) << "if (this->__isset." << (*f_iter)->get_name() << ") {" << endl;
8597  indent_up();
8598  }
8599  // Write field header
8600  if (!((*f_iter)->annotations_.empty())) {
8601  out << indent() << "{" << endl;
8602  indent_up();
8603  out << indent() << "std::map<std::string, std::string> annotations_;" << endl;
8604  std::map<std::string, std::string>::iterator it;
8605  for (it = (*f_iter)->annotations_.begin(); it != (*f_iter)->annotations_.end(); it++) {
8606  if (!tstruct->is_sandesh() &&
8607  ((*f_iter)->get_name() == "name") && (it->first == "key")) {
8608  out << indent() << "assert(!table_.empty());" << endl;
8609  out << indent() << "annotations_.insert("
8610  "std::make_pair(\"key\", table_));" << endl;
8611  } else {
8612  out << indent() << "annotations_.insert(std::make_pair(\"" << (*it).first
8613  << "\"" << ", \"" << (*it).second << "\"));" << endl;
8614  }
8615  }
8616  out <<
8617  indent() << "if ((ret = oprot->writeFieldBegin(" <<
8618  "\"" << (*f_iter)->get_name() << "\", " <<
8619  type_to_enum((*f_iter)->get_type()) << ", " <<
8620  (*f_iter)->get_key() << ", &annotations_)) < 0) {" << endl;
8621  indent_up();
8622  indent(out) << "return ret;" << endl;
8623  scope_down(out);
8624  indent(out) << "xfer += ret;" << endl;
8625  indent_down();
8626  out << indent() << "}" << endl;
8627  } else {
8628  out <<
8629  indent() << "if ((ret = oprot->writeFieldBegin(" <<
8630  "\"" << (*f_iter)->get_name() << "\", " <<
8631  type_to_enum((*f_iter)->get_type()) << ", " <<
8632  (*f_iter)->get_key() << ")) < 0) {" << endl;
8633  indent_up();
8634  indent(out) << "return ret;" << endl;
8635  scope_down(out);
8636  indent(out) << "xfer += ret;" << endl;
8637  }
8638 
8639  // Write field contents
8640  if (pointers && !(*f_iter)->get_type()->is_xception()) {
8641  generate_serialize_field(out, *f_iter, "(*(this->", "))");
8642  } else {
8643  generate_serialize_field(out, *f_iter, "this->");
8644  }
8645  // Write field closer
8646  indent(out) <<
8647  "if ((ret = oprot->writeFieldEnd()) < 0) {" << endl;
8648  indent_up();
8649  indent(out) << "return ret;" << endl;
8650  scope_down(out);
8651  indent(out) << "xfer += ret;" << endl;
8652  if ((*f_iter)->get_req() == t_field::T_OPTIONAL) {
8653  indent_down();
8654  indent(out) << '}' << endl;
8655  }
8656  }
8657 
8658  // Write the struct map
8659  out <<
8660  indent() << "if ((ret = oprot->writeFieldStop()) < 0) {" << endl <<
8661  indent() << " return ret;" << endl <<
8662  indent() << "}" << endl <<
8663  indent() << "xfer += ret;" << endl <<
8664  indent() <<
8665  "if ((ret = oprot->write" << struct_type_name << "End()) < 0) {" <<
8666  endl <<
8667  indent() << " return ret;" << endl <<
8668  indent() << "}" << endl <<
8669  indent() << "xfer += ret;" << endl <<
8670  indent() << "return xfer;" << endl;
8671 
8672  indent_down();
8673 }
8674 
8675 
8676 THRIFT_REGISTER_GENERATOR(cpp, "C++",
8677 " cob_style: Generate \"Continuation OBject\"-style classes.\n"
8678 " no_client_completion:\n"
8679 " Omit calls to completion__() in CobClient class.\n"
8680 " templates: Generate templatized reader/writer methods.\n"
8681 " pure_enums: Generate pure enums instead of wrapper classes.\n"
8682 " dense: Generate type specifications for the dense protocol.\n"
8683 " include_prefix: Use full include paths in generated files.\n"
8684 )
ProcessorGenerator::indent
std::ostream & indent(std::ostream &os)
Definition: t_cpp_generator.cc:6089
ProcessorGenerator::f_out_
std::ofstream & f_out_
Definition: t_cpp_generator.cc:6103
ProcessorGenerator::type_name
std::string type_name(t_type *ttype, bool in_typedef=false, bool arg=false)
Definition: t_cpp_generator.cc:6082
ProcessorGenerator::ProcessorGenerator
ProcessorGenerator(t_cpp_generator *generator, t_service *service, const string &style)
Definition: t_cpp_generator.cc:6125
ProcessorGenerator::f_header_
std::ofstream & f_header_
Definition: t_cpp_generator.cc:6102
ProcessorGenerator::generator_
t_cpp_generator * generator_
Definition: t_cpp_generator.cc:6100
t_base_type
Definition: t_base_type.h:34
t_base_type::get_base
t_base get_base() const
Definition: t_base_type.h:81
t_base_type::is_string
bool is_string() const
Definition: t_base_type.h:89
t_base_type::t_base_name
static std::string t_base_name(t_base tbase)
Definition: t_base_type.h:218
t_base_type::t_base
t_base
Definition: t_base_type.h:39
t_base_type::TYPE_VOID
@ TYPE_VOID
Definition: t_base_type.h:40
t_base_type::TYPE_I64
@ TYPE_I64
Definition: t_base_type.h:61
t_base_type::TYPE_BYTE
@ TYPE_BYTE
Definition: t_base_type.h:58
t_base_type::TYPE_I16
@ TYPE_I16
Definition: t_base_type.h:59
t_base_type::TYPE_BOOL
@ TYPE_BOOL
Definition: t_base_type.h:57
t_base_type::TYPE_STRING
@ TYPE_STRING
Definition: t_base_type.h:41
t_base_type::TYPE_I32
@ TYPE_I32
Definition: t_base_type.h:60
t_base_type::TYPE_DOUBLE
@ TYPE_DOUBLE
Definition: t_base_type.h:71
t_const_value
Definition: t_const_value.h:36
t_const_value::get_uuid
std::string get_uuid() const
Definition: t_const_value.h:103
t_const_value::get_list
const std::vector< t_const_value * > & get_list() const
Definition: t_const_value.h:133
t_const_value::CV_INTEGER
@ CV_INTEGER
Definition: t_const_value.h:40
t_const_value::get_double
double get_double() const
Definition: t_const_value.h:99
t_const_value::get_string
std::string get_string() const
Definition: t_const_value.h:63
t_const_value::get_type
t_const_value_type get_type() const
Definition: t_const_value.h:177
t_const_value::get_map
const std::map< t_const_value *, t_const_value * > & get_map() const
Definition: t_const_value.h:121
t_const_value::get_integer
int64_t get_integer() const
Definition: t_const_value.h:72
t_container
Definition: t_container.h:25
t_container::has_cpp_name
bool has_cpp_name()
Definition: t_container.h:38
t_container::get_cpp_name
std::string get_cpp_name()
Definition: t_container.h:42
t_cpp_generator
Definition: t_cpp_generator.cc:47
t_cpp_generator::generate_xception
void generate_xception(t_struct *txception)
Definition: t_cpp_generator.cc:97
t_cpp_generator::generate_serialize_list_element
void generate_serialize_list_element(std::ofstream &out, t_list *tlist, std::string iter)
Definition: t_cpp_generator.cc:7862
t_cpp_generator::ns_open_
std::string ns_open_
Definition: t_cpp_generator.cc:429
t_cpp_generator::generate_service_async_skeleton
void generate_service_async_skeleton(t_service *tservice)
Definition: t_cpp_generator.cc:5377
t_cpp_generator::function_signature
std::string function_signature(t_function *tfunction, std::string style, std::string prefix="", bool name_params=true)
Definition: t_cpp_generator.cc:8169
t_cpp_generator::generate_common_struct_reader_body
void generate_common_struct_reader_body(std::ofstream &out, t_struct_common *tstruct, bool pointers=false)
Definition: t_cpp_generator.cc:8407
t_cpp_generator::generate_serialize_set_element
void generate_serialize_set_element(std::ofstream &out, t_set *tmap, std::string iter)
Definition: t_cpp_generator.cc:7852
t_cpp_generator::generate_deserialize_field
void generate_deserialize_field(std::ofstream &out, t_field *tfield, std::string prefix="", std::string suffix="")
Definition: t_cpp_generator.cc:7176
t_cpp_generator::generate_deserialize_struct
void generate_deserialize_struct(std::ofstream &out, t_struct *tstruct, std::string prefix="")
Definition: t_cpp_generator.cc:7285
t_cpp_generator::f_service_tcc_
std::ofstream f_service_tcc_
Definition: t_cpp_generator.cc:442
t_cpp_generator::t_cpp_generator
t_cpp_generator(t_program *program, const std::map< std::string, std::string > &parsed_options, const std::string &option_string)
Definition: t_cpp_generator.cc:49
t_cpp_generator::generate_typedef
void generate_typedef(t_typedef *ttypedef)
Definition: t_cpp_generator.cc:704
t_cpp_generator::generate_serialize_struct
void generate_serialize_struct(std::ofstream &out, t_struct *tstruct, std::string prefix="")
Definition: t_cpp_generator.cc:7673
t_cpp_generator::generate_struct_result_writer
void generate_struct_result_writer(std::ofstream &out, t_struct *tstruct, bool pointers=false)
Definition: t_cpp_generator.cc:4892
t_cpp_generator::generate_deserialize_list_element
void generate_deserialize_list_element(std::ofstream &out, t_list *tlist, std::string prefix, bool push_back, std::string index)
Definition: t_cpp_generator.cc:7514
t_cpp_generator::print_const_value
void print_const_value(std::ofstream &out, std::string name, t_type *type, t_const_value *value)
Definition: t_cpp_generator.cc:894
t_cpp_generator::f_types_tcc_
std::ofstream f_types_tcc_
Definition: t_cpp_generator.cc:439
t_cpp_generator::generate_service_skeleton
void generate_service_skeleton(t_service *tservice)
Definition: t_cpp_generator.cc:7090
t_cpp_generator::generate_serialize_container
void generate_serialize_container(std::ofstream &out, t_type *ttype, std::string prefix="")
Definition: t_cpp_generator.cc:7685
t_cpp_generator::generate_service_client
void generate_service_client(t_service *tservice, string style)
Definition: t_cpp_generator.cc:5584
t_cpp_generator::gen_templates_
bool gen_templates_
Definition: t_cpp_generator.cc:407
t_cpp_generator::gen_no_client_completion_
bool gen_no_client_completion_
Definition: t_cpp_generator.cc:423
t_cpp_generator::generate_function_call
void generate_function_call(ostream &out, t_function *tfunction, string target, string iface, string arg_prefix)
Definition: t_cpp_generator.cc:5350
t_cpp_generator::generate_function_helpers
void generate_function_helpers(t_service *tservice, t_function *tfunction)
Definition: t_cpp_generator.cc:6502
t_cpp_generator::base_type_name
std::string base_type_name(t_base_type::t_base tbase)
Definition: t_cpp_generator.cc:8037
t_cpp_generator::type_name
std::string type_name(t_type *ttype, bool in_typedef=false, bool arg=false)
Definition: t_cpp_generator.cc:7952
t_cpp_generator::generate_service_helpers
void generate_service_helpers(t_service *tservice)
Definition: t_cpp_generator.cc:5118
t_cpp_generator::generate_deserialize_container
void generate_deserialize_container(std::ofstream &out, t_type *ttype, std::string prefix="")
Definition: t_cpp_generator.cc:7297
t_cpp_generator::generate_service_interface_factory
void generate_service_interface_factory(t_service *tservice, string style)
Definition: t_cpp_generator.cc:5209
t_cpp_generator::gen_dense_
bool gen_dense_
Definition: t_cpp_generator.cc:402
t_cpp_generator::generate_local_reflection
void generate_local_reflection(std::ofstream &out, t_type *ttype, bool is_definition)
Definition: t_cpp_generator.cc:3262
t_cpp_generator::use_include_prefix_
bool use_include_prefix_
Definition: t_cpp_generator.cc:413
t_cpp_generator::render_const_value
std::string render_const_value(std::ofstream &out, std::string name, t_type *type, t_const_value *value)
Definition: t_cpp_generator.cc:960
t_cpp_generator::type_to_enum
std::string type_to_enum(t_type *ttype)
Definition: t_cpp_generator.cc:8275
t_cpp_generator::generate_struct_reader
void generate_struct_reader(std::ofstream &out, t_struct *tstruct, bool pointers=false)
Definition: t_cpp_generator.cc:3397
t_cpp_generator::generate_struct_fingerprint
void generate_struct_fingerprint(std::ofstream &out, t_struct *tstruct, bool is_definition)
Definition: t_cpp_generator.cc:3229
t_cpp_generator::generate_local_reflection_pointer
void generate_local_reflection_pointer(std::ofstream &out, t_type *ttype)
Definition: t_cpp_generator.cc:3379
t_cpp_generator::namespace_open
std::string namespace_open(std::string ns)
Definition: t_cpp_generator.cc:7905
t_cpp_generator::generate_cpp_struct
void generate_cpp_struct(t_struct *tstruct, bool is_exception)
Definition: t_cpp_generator.cc:1035
t_cpp_generator::set_use_include_prefix
void set_use_include_prefix(bool use_include_prefix)
Definition: t_cpp_generator.cc:367
t_cpp_generator::generate_process_function
void generate_process_function(t_service *tservice, t_function *tfunction, string style, bool specialized=false)
Definition: t_cpp_generator.cc:6541
t_cpp_generator::generate_service_multiface
void generate_service_multiface(t_service *tservice)
Definition: t_cpp_generator.cc:5461
t_cpp_generator::f_service_
std::ofstream f_service_
Definition: t_cpp_generator.cc:441
t_cpp_generator::declare_field
std::string declare_field(t_field *tfield, bool init=false, bool pointer=false, bool constant=false, bool reference=false)
Definition: t_cpp_generator.cc:8090
t_cpp_generator::f_header_
std::ofstream f_header_
Definition: t_cpp_generator.cc:440
t_cpp_generator::ns_close_
std::string ns_close_
Definition: t_cpp_generator.cc:430
t_cpp_generator::namespace_prefix
std::string namespace_prefix(std::string ns)
Definition: t_cpp_generator.cc:7875
t_cpp_generator::generate_struct
void generate_struct(t_struct *tstruct)
Definition: t_cpp_generator.cc:94
t_cpp_generator::generate_enum_constant_list
void generate_enum_constant_list(std::ofstream &f, const vector< t_enum_value * > &constants, const char *prefix, const char *suffix, bool include_values)
Definition: t_cpp_generator.cc:711
t_cpp_generator::generate_service_null
void generate_service_null(t_service *tservice, string style)
Definition: t_cpp_generator.cc:5298
t_cpp_generator::generate_deserialize_map_element
void generate_deserialize_map_element(std::ofstream &out, t_map *tmap, std::string prefix="")
Definition: t_cpp_generator.cc:7442
t_cpp_generator::local_reflection_name
std::string local_reflection_name(const char *, t_type *ttype, bool external=false)
Definition: t_cpp_generator.cc:8351
t_cpp_generator::get_include_prefix
std::string get_include_prefix(const t_program &program) const
Definition: t_cpp_generator.cc:8391
t_cpp_generator::generate_consts
void generate_consts(std::vector< t_const * > consts)
Definition: t_cpp_generator.cc:801
t_cpp_generator::cob_function_signature
std::string cob_function_signature(t_function *tfunction, std::string prefix="", bool name_params=true)
t_cpp_generator::generate_struct_definition
void generate_struct_definition(std::ofstream &out, t_struct *tstruct, bool is_exception=false, bool pointers=false, bool read=true, bool write=true)
Definition: t_cpp_generator.cc:2647
t_cpp_generator::reflected_fingerprints_
std::set< std::string > reflected_fingerprints_
Definition: t_cpp_generator.cc:464
t_cpp_generator::init_generator
void init_generator()
Definition: t_cpp_generator.cc:479
t_cpp_generator::generate_enum
void generate_enum(t_enum *tenum)
Definition: t_cpp_generator.cc:745
t_cpp_generator::argument_list
std::string argument_list(t_struct *tstruct, bool name_params=true, bool start_comma=false)
Definition: t_cpp_generator.cc:8251
t_cpp_generator::generate_serialize_field
void generate_serialize_field(std::ofstream &out, t_field *tfield, std::string prefix="", std::string suffix="")
Definition: t_cpp_generator.cc:7538
t_cpp_generator::namespace_close
std::string namespace_close(std::string ns)
Definition: t_cpp_generator.cc:7932
t_cpp_generator::gen_cob_style_
bool gen_cob_style_
Definition: t_cpp_generator.cc:418
t_cpp_generator::generate_serialize_map_element
void generate_serialize_map_element(std::ofstream &out, t_map *tmap, std::string iter)
Definition: t_cpp_generator.cc:7802
t_cpp_generator::generate_common_struct_writer_body
void generate_common_struct_writer_body(std::ofstream &out, t_struct_common *tstruct, bool pointers=false)
Definition: t_cpp_generator.cc:8571
t_cpp_generator::generate_struct_writer
void generate_struct_writer(std::ofstream &out, t_struct *tstruct, bool pointers=false)
Definition: t_cpp_generator.cc:3423
t_cpp_generator::generate_service
void generate_service(t_service *tservice)
Definition: t_cpp_generator.cc:4975
t_cpp_generator::is_complex_type
bool is_complex_type(t_type *ttype)
Definition: t_cpp_generator.cc:353
t_cpp_generator::gen_pure_enums_
bool gen_pure_enums_
Definition: t_cpp_generator.cc:397
t_cpp_generator::generate_service_interface
void generate_service_interface(t_service *tservice, string style)
Definition: t_cpp_generator.cc:5146
t_cpp_generator::f_types_impl_
std::ofstream f_types_impl_
Definition: t_cpp_generator.cc:438
t_cpp_generator::close_generator
void close_generator()
Definition: t_cpp_generator.cc:658
t_cpp_generator::generate_service_processor
void generate_service_processor(t_service *tservice, string style)
Definition: t_cpp_generator.cc:6491
t_cpp_generator::f_types_
std::ofstream f_types_
Definition: t_cpp_generator.cc:437
t_cpp_generator::generate_deserialize_set_element
void generate_deserialize_set_element(std::ofstream &out, t_set *tset, std::string prefix="")
Definition: t_cpp_generator.cc:7499
t_enum
Definition: t_enum.h:30
t_enum::get_constants
const std::vector< t_enum_value * > & get_constants()
Definition: t_enum.h:43
t_field
Definition: t_field.h:36
t_field::get_name
const std::string & get_name() const
Definition: t_field.h:95
t_field::get_type
t_type * get_type() const
Definition: t_field.h:91
t_field::T_OPTIONAL
@ T_OPTIONAL
Definition: t_field.h:105
t_field::T_REQUIRED
@ T_REQUIRED
Definition: t_field.h:104
t_field::get_req
e_req get_req() const
Definition: t_field.h:113
t_function
Definition: t_function.h:34
t_function::is_oneway
bool is_oneway() const
Definition: t_function.h:103
t_function::get_returntype
t_type * get_returntype() const
Definition: t_function.h:87
t_function::get_arglist
t_struct * get_arglist() const
Definition: t_function.h:95
t_function::get_xceptions
t_struct * get_xceptions() const
Definition: t_function.h:99
t_function::get_name
const std::string & get_name() const
Definition: t_function.h:91
t_generator::get_out_dir
virtual std::string get_out_dir() const
Definition: t_generator.h:148
t_generator::tmp
std::string tmp(std::string name)
Definition: t_generator.h:160
t_generator::service_name_
std::string service_name_
Definition: t_generator.h:288
t_generator::indent_up
void indent_up()
Definition: t_generator.h:170
t_generator::program_
t_program * program_
Definition: t_generator.h:276
t_generator::indent
std::string indent()
Definition: t_generator.h:181
t_generator::get_true_type
static t_type * get_true_type(t_type *type)
Definition: t_generator.h:268
t_generator::indent_down
void indent_down()
Definition: t_generator.h:174
t_generator::get_program
const t_program * get_program() const
Definition: t_generator.h:60
t_generator::out_dir_base_
std::string out_dir_base_
Definition: t_generator.h:301
t_list
Definition: t_list.h:29
t_list::get_elem_type
t_type * get_elem_type() const
Definition: t_list.h:34
t_map
Definition: t_map.h:30
t_map::get_key_type
t_type * get_key_type() const
Definition: t_map.h:36
t_map::get_val_type
t_type * get_val_type() const
Definition: t_map.h:40
t_oop_generator
Definition: t_oop_generator.h:37
t_oop_generator::scope_down
void scope_down(std::ostream &out)
Definition: t_oop_generator.h:51
t_oop_generator::scope_up
void scope_up(std::ostream &out)
Definition: t_oop_generator.h:46
t_program
Definition: t_program.h:64
t_program::get_include_prefix
const std::string & get_include_prefix() const
Definition: t_program.h:98
t_program::get_namespace
const std::string & get_namespace() const
Definition: t_program.h:95
t_program::get_name
const std::string & get_name() const
Definition: t_program.h:92
t_sandesh
Definition: t_sandesh.h:39
t_sandesh::has_key_annotation
virtual bool has_key_annotation() const
t_sandesh::is_level_category_supported
bool is_level_category_supported() const
Definition: t_sandesh.h:75
t_sandesh::get_type
const t_type * get_type()
Definition: t_sandesh.h:71
t_service
Definition: t_service.h:32
t_service::get_extends
t_service * get_extends()
Definition: t_service.h:60
t_service::get_functions
const std::vector< t_function * > & get_functions() const
Definition: t_service.h:56
t_set
Definition: t_set.h:29
t_set::get_elem_type
t_type * get_elem_type() const
Definition: t_set.h:34
t_struct_common
Definition: t_struct_common.h:23
t_struct_common::append
bool append(t_field *elem)
Definition: t_struct_common.h:39
t_struct_common::get_sorted_members
const members_type & get_sorted_members()
Definition: t_struct_common.h:35
t_struct_common::get_struct_type_name
virtual std::string get_struct_type_name()=0
t_struct_common::get_members
const members_type & get_members()
Definition: t_struct_common.h:31
t_struct
Definition: t_struct.h:37
t_type
Definition: t_type.h:48
t_type::get_binary_fingerprint
const uint8_t * get_binary_fingerprint() const
Definition: t_type.h:110
t_type::set_name
virtual void set_name(const std::string &name)
Definition: t_type.h:52
t_type::get_4byte_fingerprint
const uint32_t get_4byte_fingerprint() const
Definition: t_type.h:139
t_type::has_fingerprint
bool has_fingerprint() const
Definition: t_type.h:101
t_type::is_string
virtual bool is_string() const
Definition: t_type.h:62
t_type::is_set
virtual bool is_set() const
Definition: t_type.h:77
t_type::is_xception
virtual bool is_xception() const
Definition: t_type.h:74
t_type::fingerprint_len
static const int fingerprint_len
Definition: t_type.h:96
t_type::is_struct
virtual bool is_struct() const
Definition: t_type.h:73
t_type::is_enum
virtual bool is_enum() const
Definition: t_type.h:72
t_type::get_name
virtual const std::string & get_name() const
Definition: t_type.h:56
t_type::get_program
t_program * get_program()
Definition: t_type.h:81
t_type::byte_to_hex
static std::string byte_to_hex(uint8_t byte)
Definition: t_type.h:132
t_type::is_base_type
virtual bool is_base_type() const
Definition: t_type.h:61
t_type::get_fingerprint_material
virtual std::string get_fingerprint_material() const =0
t_type::get_ascii_fingerprint
std::string get_ascii_fingerprint() const
Definition: t_type.h:114
t_type::is_list
virtual bool is_list() const
Definition: t_type.h:76
t_type::annotations_
std::map< std::string, std::string > annotations_
Definition: t_type.h:171
t_type::is_container
virtual bool is_container() const
Definition: t_type.h:75
t_type::is_map
virtual bool is_map() const
Definition: t_type.h:78
t_type::is_void
virtual bool is_void() const
Definition: t_type.h:60
t_typedef
Definition: t_typedef.h:33
t_typedef::get_type
t_type * get_type() const
Definition: t_typedef.h:42
t_typedef::get_symbolic
const std::string & get_symbolic() const
Definition: t_typedef.h:46
namespace_prefix
static const char namespace_prefix[]
Definition: instance_manager.cc:34
type
uint8_t type
Definition: load_balance.h:2
LOG
#define LOG(_Level, _Msg)
Definition: logging.h:33
bgp_log_test::init
void init()
Definition: bgp_log.cc:37
contrail::sandesh::protocol::skip
int32_t skip(Protocol_ &prot, TType type)
Definition: TProtocol.h:187
MKDIR
#define MKDIR(x)
Definition: platform.h:28
THRIFT_REGISTER_GENERATOR
#define THRIFT_REGISTER_GENERATOR(language, long_name, doc)
Definition: t_generator_registry.h:97
t_oop_generator.h