http://git-wip-us.apache.org/repos/asf/incubator-hawq/blob/72ea8afd/depends/thirdparty/thrift/compiler/cpp/src/generate/t_rb_generator.cc ---------------------------------------------------------------------- diff --git a/depends/thirdparty/thrift/compiler/cpp/src/generate/t_rb_generator.cc b/depends/thirdparty/thrift/compiler/cpp/src/generate/t_rb_generator.cc deleted file mode 100644 index b25c248..0000000 --- a/depends/thirdparty/thrift/compiler/cpp/src/generate/t_rb_generator.cc +++ /dev/null @@ -1,1252 +0,0 @@ -/* - * Licensed to the Apache Software Foundation (ASF) under one - * or more contributor license agreements. See the NOTICE file - * distributed with this work for additional information - * regarding copyright ownership. The ASF licenses this file - * to you under the Apache License, Version 2.0 (the - * "License"); you may not use this file except in compliance - * with the License. You may obtain a copy of the License at - * - * http://www.apache.org/licenses/LICENSE-2.0 - * - * Unless required by applicable law or agreed to in writing, - * software distributed under the License is distributed on an - * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY - * KIND, either express or implied. See the License for the - * specific language governing permissions and limitations - * under the License. - * - * Contains some contributions under the Thrift Software License. - * Please see doc/old-thrift-license.txt in the Thrift distribution for - * details. - */ - -#include <string> -#include <fstream> -#include <iostream> -#include <vector> -#include <algorithm> - -#include <stdlib.h> -#include <sys/stat.h> -#include <sys/types.h> -#include <sstream> - -#include "t_oop_generator.h" -#include "platform.h" -#include "version.h" - -using std::map; -using std::ofstream; -using std::ostringstream; -using std::string; -using std::stringstream; -using std::vector; - -static const string endl = "\n"; // avoid ostream << std::endl flushes - -/** - * A subclass of std::ofstream that includes indenting functionality. - */ -class t_rb_ofstream : public std::ofstream { -private: - int indent_; - -public: - t_rb_ofstream() : std::ofstream(), indent_(0) {} - explicit t_rb_ofstream(const char* filename, - ios_base::openmode mode = ios_base::out, - int indent = 0) - : std::ofstream(filename, mode), indent_(indent) {} - - t_rb_ofstream& indent() { - for (int i = 0; i < indent_; ++i) { - *this << " "; - } - return *this; - } - - void indent_up() { indent_++; } - void indent_down() { indent_--; } -}; - -/** - * Ruby code generator. - * - */ -class t_rb_generator : public t_oop_generator { -public: - t_rb_generator(t_program* program, - const std::map<std::string, std::string>& parsed_options, - const std::string& option_string) - : t_oop_generator(program) { - (void)option_string; - out_dir_base_ = "gen-rb"; - - require_rubygems_ = (parsed_options.find("rubygems") != parsed_options.end()); - namespaced_ = (parsed_options.find("namespaced") != parsed_options.end()); - } - - /** - * Init and close methods - */ - - void init_generator(); - void close_generator(); - - /** - * Program-level generation functions - */ - - void generate_typedef(t_typedef* ttypedef); - void generate_enum(t_enum* tenum); - void generate_const(t_const* tconst); - void generate_struct(t_struct* tstruct); - void generate_union(t_struct* tunion); - void generate_xception(t_struct* txception); - void generate_service(t_service* tservice); - - t_rb_ofstream& render_const_value(t_rb_ofstream& out, t_type* type, t_const_value* value); - - /** - * Struct generation code - */ - - void generate_rb_struct(t_rb_ofstream& out, t_struct* tstruct, bool is_exception); - void generate_rb_struct_required_validator(t_rb_ofstream& out, t_struct* tstruct); - void generate_rb_union(t_rb_ofstream& out, t_struct* tstruct, bool is_exception); - void generate_rb_union_validator(t_rb_ofstream& out, t_struct* tstruct); - void generate_rb_function_helpers(t_function* tfunction); - void generate_rb_simple_constructor(t_rb_ofstream& out, t_struct* tstruct); - void generate_rb_simple_exception_constructor(t_rb_ofstream& out, t_struct* tstruct); - void generate_field_constants(t_rb_ofstream& out, t_struct* tstruct); - void generate_field_constructors(t_rb_ofstream& out, t_struct* tstruct); - void generate_field_defns(t_rb_ofstream& out, t_struct* tstruct); - void generate_field_data(t_rb_ofstream& out, - t_type* field_type, - const std::string& field_name, - t_const_value* field_value, - bool optional); - - /** - * Service-level generation functions - */ - - void generate_service_helpers(t_service* tservice); - void generate_service_interface(t_service* tservice); - void generate_service_client(t_service* tservice); - void generate_service_server(t_service* tservice); - void generate_process_function(t_service* tservice, t_function* tfunction); - - /** - * Serialization constructs - */ - - void generate_deserialize_field(t_rb_ofstream& out, - t_field* tfield, - std::string prefix = "", - bool inclass = false); - - void generate_deserialize_struct(t_rb_ofstream& out, t_struct* tstruct, std::string prefix = ""); - - void generate_deserialize_container(t_rb_ofstream& out, t_type* ttype, std::string prefix = ""); - - void generate_deserialize_set_element(t_rb_ofstream& out, t_set* tset, std::string prefix = ""); - - void generate_deserialize_map_element(t_rb_ofstream& out, t_map* tmap, std::string prefix = ""); - - void generate_deserialize_list_element(t_rb_ofstream& out, - t_list* tlist, - std::string prefix = ""); - - void generate_serialize_field(t_rb_ofstream& out, t_field* tfield, std::string prefix = ""); - - void generate_serialize_struct(t_rb_ofstream& out, t_struct* tstruct, std::string prefix = ""); - - void generate_serialize_container(t_rb_ofstream& out, t_type* ttype, std::string prefix = ""); - - void generate_serialize_map_element(t_rb_ofstream& out, - t_map* tmap, - std::string kiter, - std::string viter); - - void generate_serialize_set_element(t_rb_ofstream& out, t_set* tmap, std::string iter); - - void generate_serialize_list_element(t_rb_ofstream& out, t_list* tlist, std::string iter); - - void generate_rdoc(t_rb_ofstream& out, t_doc* tdoc); - - /** - * Helper rendering functions - */ - - std::string rb_autogen_comment(); - std::string render_require_thrift(); - std::string render_includes(); - std::string declare_field(t_field* tfield); - std::string type_name(t_type* ttype); - std::string full_type_name(t_type* ttype); - std::string function_signature(t_function* tfunction, std::string prefix = ""); - std::string argument_list(t_struct* tstruct); - std::string type_to_enum(t_type* ttype); - std::string rb_namespace_to_path_prefix(std::string rb_namespace); - - std::vector<std::string> ruby_modules(t_program* p) { - std::string ns = p->get_namespace("rb"); - std::vector<std::string> modules; - if (ns.empty()) { - return modules; - } - - std::string::iterator pos = ns.begin(); - while (true) { - std::string::iterator delim = std::find(pos, ns.end(), '.'); - modules.push_back(capitalize(std::string(pos, delim))); - pos = delim; - if (pos == ns.end()) { - break; - } - ++pos; - } - - return modules; - } - - void begin_namespace(t_rb_ofstream&, std::vector<std::string>); - void end_namespace(t_rb_ofstream&, std::vector<std::string>); - -private: - /** - * File streams - */ - - t_rb_ofstream f_types_; - t_rb_ofstream f_consts_; - t_rb_ofstream f_service_; - - std::string namespace_dir_; - std::string require_prefix_; - - /** If true, add a "require 'rubygems'" line to the top of each gen-rb file. */ - bool require_rubygems_; - - /** If true, generate files in idiomatic namespaced directories. */ - bool namespaced_; -}; - -/** - * Prepares for file generation by opening up the necessary file output - * streams. - * - * @param tprogram The program to generate - */ -void t_rb_generator::init_generator() { - string subdir = get_out_dir(); - - // Make output directory - MKDIR(subdir.c_str()); - - if (namespaced_) { - require_prefix_ = rb_namespace_to_path_prefix(program_->get_namespace("rb")); - - string dir = require_prefix_; - string::size_type loc; - - while ((loc = dir.find("/")) != string::npos) { - subdir = subdir + dir.substr(0, loc) + "/"; - MKDIR(subdir.c_str()); - dir = dir.substr(loc + 1); - } - } - - namespace_dir_ = subdir; - - // Make output file - string f_types_name = namespace_dir_ + underscore(program_name_) + "_types.rb"; - f_types_.open(f_types_name.c_str()); - - string f_consts_name = namespace_dir_ + underscore(program_name_) + "_constants.rb"; - f_consts_.open(f_consts_name.c_str()); - - // Print header - f_types_ << rb_autogen_comment() << endl << render_require_thrift() << render_includes() << endl; - begin_namespace(f_types_, ruby_modules(program_)); - - f_consts_ << rb_autogen_comment() << endl << render_require_thrift() << "require '" - << require_prefix_ << underscore(program_name_) << "_types'" << endl << endl; - begin_namespace(f_consts_, ruby_modules(program_)); -} - -/** - * Renders the require of thrift itself, and possibly of the rubygems dependency. - */ -string t_rb_generator::render_require_thrift() { - if (require_rubygems_) { - return "require 'rubygems'\nrequire 'thrift'\n"; - } else { - return "require 'thrift'\n"; - } -} - -/** - * Renders all the imports necessary for including another Thrift program - */ -string t_rb_generator::render_includes() { - const vector<t_program*>& includes = program_->get_includes(); - string result = ""; - for (size_t i = 0; i < includes.size(); ++i) { - if (namespaced_) { - t_program* included = includes[i]; - std::string included_require_prefix - = rb_namespace_to_path_prefix(included->get_namespace("rb")); - std::string included_name = included->get_name(); - result += "require '" + included_require_prefix + underscore(included_name) + "_types'\n"; - } else { - result += "require '" + underscore(includes[i]->get_name()) + "_types'\n"; - } - } - if (includes.size() > 0) { - result += "\n"; - } - return result; -} - -/** - * Autogen'd comment - */ -string t_rb_generator::rb_autogen_comment() { - return std::string("#\n") + "# Autogenerated by Thrift Compiler (" + THRIFT_VERSION + ")\n" - + "#\n" + "# DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING\n" + "#\n"; -} - -/** - * Closes the type files - */ -void t_rb_generator::close_generator() { - // Close types file - end_namespace(f_types_, ruby_modules(program_)); - end_namespace(f_consts_, ruby_modules(program_)); - f_types_.close(); - f_consts_.close(); -} - -/** - * Generates a typedef. This is not done in Ruby, types are all implicit. - * - * @param ttypedef The type definition - */ -void t_rb_generator::generate_typedef(t_typedef* ttypedef) { - (void)ttypedef; -} - -/** - * Generates code for an enumerated type. Done using a class to scope - * the values. - * - * @param tenum The enumeration - */ -void t_rb_generator::generate_enum(t_enum* tenum) { - f_types_.indent() << "module " << capitalize(tenum->get_name()) << endl; - f_types_.indent_up(); - - vector<t_enum_value*> constants = tenum->get_constants(); - vector<t_enum_value*>::iterator c_iter; - for (c_iter = constants.begin(); c_iter != constants.end(); ++c_iter) { - int value = (*c_iter)->get_value(); - - // Ruby class constants have to be capitalized... omg i am so on the fence - // about languages strictly enforcing capitalization why can't we just all - // agree and play nice. - string name = capitalize((*c_iter)->get_name()); - - generate_rdoc(f_types_, *c_iter); - f_types_.indent() << name << " = " << value << endl; - } - - // Create a hash mapping values back to their names (as strings) since ruby has no native enum - // type - f_types_.indent() << "VALUE_MAP = {"; - for (c_iter = constants.begin(); c_iter != constants.end(); ++c_iter) { - // Populate the hash - int value = (*c_iter)->get_value(); - if (c_iter != constants.begin()) - f_types_ << ", "; - f_types_ << value << " => \"" << capitalize((*c_iter)->get_name()) << "\""; - } - f_types_ << "}" << endl; - - // Create a set with valid values for this enum - f_types_.indent() << "VALID_VALUES = Set.new(["; - for (c_iter = constants.begin(); c_iter != constants.end(); ++c_iter) { - // Populate the set - if (c_iter != constants.begin()) - f_types_ << ", "; - f_types_ << capitalize((*c_iter)->get_name()); - } - f_types_ << "]).freeze" << endl; - - f_types_.indent_down(); - f_types_.indent() << "end" << endl << endl; -} - -/** - * Generate a constant value - */ -void t_rb_generator::generate_const(t_const* tconst) { - t_type* type = tconst->get_type(); - string name = tconst->get_name(); - t_const_value* value = tconst->get_value(); - - name[0] = toupper(name[0]); - - f_consts_.indent() << name << " = "; - render_const_value(f_consts_, type, value) << endl << endl; -} - -/** - * Prints the value of a constant with the given type. Note that type checking - * is NOT performed in this function as it is always run beforehand using the - * validate_types method in main.cc - */ -t_rb_ofstream& t_rb_generator::render_const_value(t_rb_ofstream& out, - t_type* type, - t_const_value* value) { - type = get_true_type(type); - if (type->is_base_type()) { - t_base_type::t_base tbase = ((t_base_type*)type)->get_base(); - switch (tbase) { - case t_base_type::TYPE_STRING: - out << "%q\"" << get_escaped_string(value) << '"'; - break; - case t_base_type::TYPE_BOOL: - out << (value->get_integer() > 0 ? "true" : "false"); - break; - case t_base_type::TYPE_BYTE: - case t_base_type::TYPE_I16: - case t_base_type::TYPE_I32: - case t_base_type::TYPE_I64: - out << value->get_integer(); - break; - case t_base_type::TYPE_DOUBLE: - if (value->get_type() == t_const_value::CV_INTEGER) { - out << value->get_integer(); - } else { - out << value->get_double(); - } - break; - default: - throw "compiler error: no const of base type " + t_base_type::t_base_name(tbase); - } - } else if (type->is_enum()) { - out.indent() << value->get_integer(); - } else if (type->is_struct() || type->is_xception()) { - out << full_type_name(type) << ".new({" << endl; - out.indent_up(); - const vector<t_field*>& fields = ((t_struct*)type)->get_members(); - vector<t_field*>::const_iterator f_iter; - const map<t_const_value*, t_const_value*>& val = value->get_map(); - map<t_const_value*, t_const_value*>::const_iterator v_iter; - for (v_iter = val.begin(); v_iter != val.end(); ++v_iter) { - t_type* field_type = NULL; - for (f_iter = fields.begin(); f_iter != fields.end(); ++f_iter) { - if ((*f_iter)->get_name() == v_iter->first->get_string()) { - field_type = (*f_iter)->get_type(); - } - } - if (field_type == NULL) { - throw "type error: " + type->get_name() + " has no field " + v_iter->first->get_string(); - } - out.indent(); - render_const_value(out, g_type_string, v_iter->first) << " => "; - render_const_value(out, field_type, v_iter->second) << "," << endl; - } - out.indent_down(); - out.indent() << "})"; - } else if (type->is_map()) { - t_type* ktype = ((t_map*)type)->get_key_type(); - t_type* vtype = ((t_map*)type)->get_val_type(); - out << "{" << endl; - out.indent_up(); - const map<t_const_value*, t_const_value*>& val = value->get_map(); - map<t_const_value*, t_const_value*>::const_iterator v_iter; - for (v_iter = val.begin(); v_iter != val.end(); ++v_iter) { - out.indent(); - render_const_value(out, ktype, v_iter->first) << " => "; - render_const_value(out, vtype, v_iter->second) << "," << endl; - } - out.indent_down(); - out.indent() << "}"; - } else if (type->is_list() || type->is_set()) { - t_type* etype; - if (type->is_list()) { - etype = ((t_list*)type)->get_elem_type(); - } else { - etype = ((t_set*)type)->get_elem_type(); - } - if (type->is_set()) { - out << "Set.new([" << endl; - } else { - out << "[" << endl; - } - out.indent_up(); - const vector<t_const_value*>& val = value->get_list(); - vector<t_const_value*>::const_iterator v_iter; - for (v_iter = val.begin(); v_iter != val.end(); ++v_iter) { - out.indent(); - render_const_value(out, etype, *v_iter) << "," << endl; - } - out.indent_down(); - if (type->is_set()) { - out.indent() << "])"; - } else { - out.indent() << "]"; - } - } else { - throw "CANNOT GENERATE CONSTANT FOR TYPE: " + type->get_name(); - } - return out; -} - -/** - * Generates a ruby struct - */ -void t_rb_generator::generate_struct(t_struct* tstruct) { - if (tstruct->is_union()) { - generate_rb_union(f_types_, tstruct, false); - } else { - generate_rb_struct(f_types_, tstruct, false); - } -} - -/** - * Generates a struct definition for a thrift exception. Basically the same - * as a struct but extends the Exception class. - * - * @param txception The struct definition - */ -void t_rb_generator::generate_xception(t_struct* txception) { - generate_rb_struct(f_types_, txception, true); -} - -/** - * Generates a ruby struct - */ -void t_rb_generator::generate_rb_struct(t_rb_ofstream& out, - t_struct* tstruct, - bool is_exception = false) { - generate_rdoc(out, tstruct); - out.indent() << "class " << type_name(tstruct); - if (is_exception) { - out << " < ::Thrift::Exception"; - } - out << endl; - - out.indent_up(); - out.indent() << "include ::Thrift::Struct, ::Thrift::Struct_Union" << endl; - - if (is_exception) { - generate_rb_simple_exception_constructor(out, tstruct); - } - - generate_field_constants(out, tstruct); - generate_field_defns(out, tstruct); - generate_rb_struct_required_validator(out, tstruct); - - out.indent() << "::Thrift::Struct.generate_accessors self" << endl; - - out.indent_down(); - out.indent() << "end" << endl << endl; -} - -/** - * Generates a ruby union - */ -void t_rb_generator::generate_rb_union(t_rb_ofstream& out, - t_struct* tstruct, - bool is_exception = false) { - (void)is_exception; - generate_rdoc(out, tstruct); - out.indent() << "class " << type_name(tstruct) << " < ::Thrift::Union" << endl; - - out.indent_up(); - out.indent() << "include ::Thrift::Struct_Union" << endl; - - generate_field_constructors(out, tstruct); - - generate_field_constants(out, tstruct); - generate_field_defns(out, tstruct); - generate_rb_union_validator(out, tstruct); - - out.indent() << "::Thrift::Union.generate_accessors self" << endl; - - out.indent_down(); - out.indent() << "end" << endl << endl; -} - -void t_rb_generator::generate_field_constructors(t_rb_ofstream& out, t_struct* tstruct) { - - out.indent() << "class << self" << endl; - out.indent_up(); - - const vector<t_field*>& fields = tstruct->get_members(); - vector<t_field*>::const_iterator f_iter; - - for (f_iter = fields.begin(); f_iter != fields.end(); ++f_iter) { - if (f_iter != fields.begin()) { - out << endl; - } - std::string field_name = (*f_iter)->get_name(); - - out.indent() << "def " << field_name << "(val)" << endl; - out.indent() << " " << tstruct->get_name() << ".new(:" << field_name << ", val)" << endl; - out.indent() << "end" << endl; - } - - out.indent_down(); - out.indent() << "end" << endl; - - out << endl; -} - -void t_rb_generator::generate_rb_simple_exception_constructor(t_rb_ofstream& out, - t_struct* tstruct) { - const vector<t_field*>& members = tstruct->get_members(); - - if (members.size() == 1) { - vector<t_field*>::const_iterator m_iter = members.begin(); - - if ((*m_iter)->get_type()->is_string()) { - string name = (*m_iter)->get_name(); - - out.indent() << "def initialize(message=nil)" << endl; - out.indent_up(); - out.indent() << "super()" << endl; - out.indent() << "self." << name << " = message" << endl; - out.indent_down(); - out.indent() << "end" << endl << endl; - - if (name != "message") { - out.indent() << "def message; " << name << " end" << endl << endl; - } - } - } -} - -void t_rb_generator::generate_field_constants(t_rb_ofstream& out, t_struct* tstruct) { - const vector<t_field*>& fields = tstruct->get_members(); - vector<t_field*>::const_iterator f_iter; - - for (f_iter = fields.begin(); f_iter != fields.end(); ++f_iter) { - std::string field_name = (*f_iter)->get_name(); - std::string cap_field_name = upcase_string(field_name); - - out.indent() << cap_field_name << " = " << (*f_iter)->get_key() << endl; - } - out << endl; -} - -void t_rb_generator::generate_field_defns(t_rb_ofstream& out, t_struct* tstruct) { - const vector<t_field*>& fields = tstruct->get_members(); - vector<t_field*>::const_iterator f_iter; - - out.indent() << "FIELDS = {" << endl; - out.indent_up(); - for (f_iter = fields.begin(); f_iter != fields.end(); ++f_iter) { - if (f_iter != fields.begin()) { - out << "," << endl; - } - - // generate the field docstrings within the FIELDS constant. no real better place... - generate_rdoc(out, *f_iter); - - out.indent() << upcase_string((*f_iter)->get_name()) << " => "; - - generate_field_data(out, - (*f_iter)->get_type(), - (*f_iter)->get_name(), - (*f_iter)->get_value(), - (*f_iter)->get_req() == t_field::T_OPTIONAL); - } - out.indent_down(); - out << endl; - out.indent() << "}" << endl << endl; - - out.indent() << "def struct_fields; FIELDS; end" << endl << endl; -} - -void t_rb_generator::generate_field_data(t_rb_ofstream& out, - t_type* field_type, - const std::string& field_name = "", - t_const_value* field_value = NULL, - bool optional = false) { - field_type = get_true_type(field_type); - - // Begin this field's defn - out << "{:type => " << type_to_enum(field_type); - - if (!field_name.empty()) { - out << ", :name => '" << field_name << "'"; - } - - if (field_value != NULL) { - out << ", :default => "; - render_const_value(out, field_type, field_value); - } - - if (!field_type->is_base_type()) { - if (field_type->is_struct() || field_type->is_xception()) { - out << ", :class => " << full_type_name((t_struct*)field_type); - } else if (field_type->is_list()) { - out << ", :element => "; - generate_field_data(out, ((t_list*)field_type)->get_elem_type()); - } else if (field_type->is_map()) { - out << ", :key => "; - generate_field_data(out, ((t_map*)field_type)->get_key_type()); - out << ", :value => "; - generate_field_data(out, ((t_map*)field_type)->get_val_type()); - } else if (field_type->is_set()) { - out << ", :element => "; - generate_field_data(out, ((t_set*)field_type)->get_elem_type()); - } - } else { - if (((t_base_type*)field_type)->is_binary()) { - out << ", :binary => true"; - } - } - - if (optional) { - out << ", :optional => true"; - } - - if (field_type->is_enum()) { - out << ", :enum_class => " << full_type_name(field_type); - } - - // End of this field's defn - out << "}"; -} - -void t_rb_generator::begin_namespace(t_rb_ofstream& out, vector<std::string> modules) { - for (vector<std::string>::iterator m_iter = modules.begin(); m_iter != modules.end(); ++m_iter) { - out.indent() << "module " << *m_iter << endl; - out.indent_up(); - } -} - -void t_rb_generator::end_namespace(t_rb_ofstream& out, vector<std::string> modules) { - for (vector<std::string>::reverse_iterator m_iter = modules.rbegin(); m_iter != modules.rend(); - ++m_iter) { - out.indent_down(); - out.indent() << "end" << endl; - } -} - -/** - * Generates a thrift service. - * - * @param tservice The service definition - */ -void t_rb_generator::generate_service(t_service* tservice) { - string f_service_name = namespace_dir_ + underscore(service_name_) + ".rb"; - f_service_.open(f_service_name.c_str()); - - f_service_ << rb_autogen_comment() << endl << render_require_thrift(); - - if (tservice->get_extends() != NULL) { - if (namespaced_) { - f_service_ << "require '" << rb_namespace_to_path_prefix( - tservice->get_extends()->get_program()->get_namespace("rb")) - << underscore(tservice->get_extends()->get_name()) << "'" << endl; - } else { - f_service_ << "require '" << require_prefix_ - << underscore(tservice->get_extends()->get_name()) << "'" << endl; - } - } - - f_service_ << "require '" << require_prefix_ << underscore(program_name_) << "_types'" << endl - << endl; - - begin_namespace(f_service_, ruby_modules(tservice->get_program())); - - f_service_.indent() << "module " << capitalize(tservice->get_name()) << endl; - f_service_.indent_up(); - - // Generate the three main parts of the service (well, two for now in PHP) - generate_service_client(tservice); - generate_service_server(tservice); - generate_service_helpers(tservice); - - f_service_.indent_down(); - f_service_.indent() << "end" << endl << endl; - - end_namespace(f_service_, ruby_modules(tservice->get_program())); - - // Close service file - f_service_.close(); -} - -/** - * Generates helper functions for a service. - * - * @param tservice The service to generate a header definition for - */ -void t_rb_generator::generate_service_helpers(t_service* tservice) { - vector<t_function*> functions = tservice->get_functions(); - vector<t_function*>::iterator f_iter; - - f_service_.indent() << "# HELPER FUNCTIONS AND STRUCTURES" << endl << endl; - - for (f_iter = functions.begin(); f_iter != functions.end(); ++f_iter) { - t_struct* ts = (*f_iter)->get_arglist(); - generate_rb_struct(f_service_, ts); - generate_rb_function_helpers(*f_iter); - } -} - -/** - * Generates a struct and helpers for a function. - * - * @param tfunction The function - */ -void t_rb_generator::generate_rb_function_helpers(t_function* tfunction) { - t_struct result(program_, tfunction->get_name() + "_result"); - t_field success(tfunction->get_returntype(), "success", 0); - if (!tfunction->get_returntype()->is_void()) { - result.append(&success); - } - - t_struct* xs = tfunction->get_xceptions(); - const vector<t_field*>& fields = xs->get_members(); - vector<t_field*>::const_iterator f_iter; - for (f_iter = fields.begin(); f_iter != fields.end(); ++f_iter) { - result.append(*f_iter); - } - generate_rb_struct(f_service_, &result); -} - -/** - * Generates a service client definition. - * - * @param tservice The service to generate a server for. - */ -void t_rb_generator::generate_service_client(t_service* tservice) { - string extends = ""; - string extends_client = ""; - if (tservice->get_extends() != NULL) { - extends = full_type_name(tservice->get_extends()); - extends_client = " < " + extends + "::Client "; - } - - f_service_.indent() << "class Client" << extends_client << endl; - f_service_.indent_up(); - - f_service_.indent() << "include ::Thrift::Client" << endl << endl; - - // Generate client method implementations - vector<t_function*> functions = tservice->get_functions(); - vector<t_function*>::const_iterator f_iter; - for (f_iter = functions.begin(); f_iter != functions.end(); ++f_iter) { - t_struct* arg_struct = (*f_iter)->get_arglist(); - const vector<t_field*>& fields = arg_struct->get_members(); - vector<t_field*>::const_iterator fld_iter; - string funname = (*f_iter)->get_name(); - - // Open function - f_service_.indent() << "def " << function_signature(*f_iter) << endl; - f_service_.indent_up(); - f_service_.indent() << "send_" << funname << "("; - - bool first = true; - for (fld_iter = fields.begin(); fld_iter != fields.end(); ++fld_iter) { - if (first) { - first = false; - } else { - f_service_ << ", "; - } - f_service_ << (*fld_iter)->get_name(); - } - f_service_ << ")" << endl; - - if (!(*f_iter)->is_oneway()) { - f_service_.indent(); - if (!(*f_iter)->get_returntype()->is_void()) { - f_service_ << "return "; - } - f_service_ << "recv_" << funname << "()" << endl; - } - f_service_.indent_down(); - f_service_.indent() << "end" << endl; - f_service_ << endl; - - f_service_.indent() << "def send_" << function_signature(*f_iter) << endl; - f_service_.indent_up(); - - std::string argsname = capitalize((*f_iter)->get_name() + "_args"); - std::string messageSendProc = (*f_iter)->is_oneway() ? "send_oneway_message" : "send_message"; - - f_service_.indent() << messageSendProc << "('" << funname << "', " << argsname; - - for (fld_iter = fields.begin(); fld_iter != fields.end(); ++fld_iter) { - f_service_ << ", :" << (*fld_iter)->get_name() << " => " << (*fld_iter)->get_name(); - } - - f_service_ << ")" << endl; - - f_service_.indent_down(); - f_service_.indent() << "end" << endl; - - if (!(*f_iter)->is_oneway()) { - std::string resultname = capitalize((*f_iter)->get_name() + "_result"); - t_struct noargs(program_); - - t_function recv_function((*f_iter)->get_returntype(), - string("recv_") + (*f_iter)->get_name(), - &noargs); - // Open function - f_service_ << endl; - f_service_.indent() << "def " << function_signature(&recv_function) << endl; - f_service_.indent_up(); - - // TODO(mcslee): Validate message reply here, seq ids etc. - - f_service_.indent() << "result = receive_message(" << resultname << ")" << endl; - - // Careful, only return _result if not a void function - if (!(*f_iter)->get_returntype()->is_void()) { - f_service_.indent() << "return result.success unless result.success.nil?" << endl; - } - - t_struct* xs = (*f_iter)->get_xceptions(); - const std::vector<t_field*>& xceptions = xs->get_members(); - vector<t_field*>::const_iterator x_iter; - for (x_iter = xceptions.begin(); x_iter != xceptions.end(); ++x_iter) { - f_service_.indent() << "raise result." << (*x_iter)->get_name() << " unless result." - << (*x_iter)->get_name() << ".nil?" << endl; - } - - // Careful, only return _result if not a void function - if ((*f_iter)->get_returntype()->is_void()) { - f_service_.indent() << "return" << endl; - } else { - f_service_.indent() << "raise " - "::Thrift::ApplicationException.new(::Thrift::ApplicationException::" - "MISSING_RESULT, '" << (*f_iter)->get_name() - << " failed: unknown result')" << endl; - } - - // Close function - f_service_.indent_down(); - f_service_.indent() << "end" << endl << endl; - } - } - - f_service_.indent_down(); - f_service_.indent() << "end" << endl << endl; -} - -/** - * Generates a service server definition. - * - * @param tservice The service to generate a server for. - */ -void t_rb_generator::generate_service_server(t_service* tservice) { - // Generate the dispatch methods - vector<t_function*> functions = tservice->get_functions(); - vector<t_function*>::iterator f_iter; - - string extends = ""; - string extends_processor = ""; - if (tservice->get_extends() != NULL) { - extends = full_type_name(tservice->get_extends()); - extends_processor = " < " + extends + "::Processor "; - } - - // Generate the header portion - f_service_.indent() << "class Processor" << extends_processor << endl; - f_service_.indent_up(); - - f_service_.indent() << "include ::Thrift::Processor" << endl << endl; - - // Generate the process subfunctions - for (f_iter = functions.begin(); f_iter != functions.end(); ++f_iter) { - generate_process_function(tservice, *f_iter); - } - - f_service_.indent_down(); - f_service_.indent() << "end" << endl << endl; -} - -/** - * Generates a process function definition. - * - * @param tfunction The function to write a dispatcher for - */ -void t_rb_generator::generate_process_function(t_service* tservice, t_function* tfunction) { - (void)tservice; - // Open function - f_service_.indent() << "def process_" << tfunction->get_name() << "(seqid, iprot, oprot)" << endl; - f_service_.indent_up(); - - string argsname = capitalize(tfunction->get_name()) + "_args"; - string resultname = capitalize(tfunction->get_name()) + "_result"; - - f_service_.indent() << "args = read_args(iprot, " << argsname << ")" << endl; - - t_struct* xs = tfunction->get_xceptions(); - const std::vector<t_field*>& xceptions = xs->get_members(); - vector<t_field*>::const_iterator x_iter; - - // Declare result for non oneway function - if (!tfunction->is_oneway()) { - f_service_.indent() << "result = " << resultname << ".new()" << endl; - } - - // Try block for a function with exceptions - if (xceptions.size() > 0) { - f_service_.indent() << "begin" << endl; - f_service_.indent_up(); - } - - // Generate the function call - t_struct* arg_struct = tfunction->get_arglist(); - const std::vector<t_field*>& fields = arg_struct->get_members(); - vector<t_field*>::const_iterator f_iter; - - f_service_.indent(); - if (!tfunction->is_oneway() && !tfunction->get_returntype()->is_void()) { - f_service_ << "result.success = "; - } - f_service_ << "@handler." << tfunction->get_name() << "("; - bool first = true; - for (f_iter = fields.begin(); f_iter != fields.end(); ++f_iter) { - if (first) { - first = false; - } else { - f_service_ << ", "; - } - f_service_ << "args." << (*f_iter)->get_name(); - } - f_service_ << ")" << endl; - - if (!tfunction->is_oneway() && xceptions.size() > 0) { - f_service_.indent_down(); - for (x_iter = xceptions.begin(); x_iter != xceptions.end(); ++x_iter) { - f_service_.indent() << "rescue " << full_type_name((*x_iter)->get_type()) << " => " - << (*x_iter)->get_name() << endl; - if (!tfunction->is_oneway()) { - f_service_.indent_up(); - f_service_.indent() << "result." << (*x_iter)->get_name() << " = " << (*x_iter)->get_name() - << endl; - f_service_.indent_down(); - } - } - f_service_.indent() << "end" << endl; - } - - // Shortcut out here for oneway functions - if (tfunction->is_oneway()) { - f_service_.indent() << "return" << endl; - f_service_.indent_down(); - f_service_.indent() << "end" << endl << endl; - return; - } - - f_service_.indent() << "write_result(result, oprot, '" << tfunction->get_name() << "', seqid)" - << endl; - - // Close function - f_service_.indent_down(); - f_service_.indent() << "end" << endl << endl; -} - -/** - * Renders a function signature of the form 'type name(args)' - * - * @param tfunction Function definition - * @return String of rendered function definition - */ -string t_rb_generator::function_signature(t_function* tfunction, string prefix) { - // TODO(mcslee): Nitpicky, no ',' if argument_list is empty - return prefix + tfunction->get_name() + "(" + argument_list(tfunction->get_arglist()) + ")"; -} - -/** - * Renders a field list - */ -string t_rb_generator::argument_list(t_struct* tstruct) { - string result = ""; - - const vector<t_field*>& fields = tstruct->get_members(); - vector<t_field*>::const_iterator f_iter; - bool first = true; - for (f_iter = fields.begin(); f_iter != fields.end(); ++f_iter) { - if (first) { - first = false; - } else { - result += ", "; - } - result += (*f_iter)->get_name(); - } - return result; -} - -string t_rb_generator::type_name(t_type* ttype) { - string prefix = ""; - - string name = ttype->get_name(); - if (ttype->is_struct() || ttype->is_xception() || ttype->is_enum()) { - name = capitalize(ttype->get_name()); - } - - return prefix + name; -} - -string t_rb_generator::full_type_name(t_type* ttype) { - string prefix = "::"; - vector<std::string> modules = ruby_modules(ttype->get_program()); - for (vector<std::string>::iterator m_iter = modules.begin(); m_iter != modules.end(); ++m_iter) { - prefix += *m_iter + "::"; - } - return prefix + type_name(ttype); -} - -/** - * Converts the parse type to a Ruby tyoe - */ -string t_rb_generator::type_to_enum(t_type* type) { - type = get_true_type(type); - - if (type->is_base_type()) { - t_base_type::t_base tbase = ((t_base_type*)type)->get_base(); - switch (tbase) { - case t_base_type::TYPE_VOID: - throw "NO T_VOID CONSTRUCT"; - case t_base_type::TYPE_STRING: - return "::Thrift::Types::STRING"; - case t_base_type::TYPE_BOOL: - return "::Thrift::Types::BOOL"; - case t_base_type::TYPE_BYTE: - return "::Thrift::Types::BYTE"; - case t_base_type::TYPE_I16: - return "::Thrift::Types::I16"; - case t_base_type::TYPE_I32: - return "::Thrift::Types::I32"; - case t_base_type::TYPE_I64: - return "::Thrift::Types::I64"; - case t_base_type::TYPE_DOUBLE: - return "::Thrift::Types::DOUBLE"; - } - } else if (type->is_enum()) { - return "::Thrift::Types::I32"; - } else if (type->is_struct() || type->is_xception()) { - return "::Thrift::Types::STRUCT"; - } else if (type->is_map()) { - return "::Thrift::Types::MAP"; - } else if (type->is_set()) { - return "::Thrift::Types::SET"; - } else if (type->is_list()) { - return "::Thrift::Types::LIST"; - } - - throw "INVALID TYPE IN type_to_enum: " + type->get_name(); -} - -string t_rb_generator::rb_namespace_to_path_prefix(string rb_namespace) { - string namespaces_left = rb_namespace; - string::size_type loc; - - string path_prefix = ""; - - while ((loc = namespaces_left.find(".")) != string::npos) { - path_prefix = path_prefix + underscore(namespaces_left.substr(0, loc)) + "/"; - namespaces_left = namespaces_left.substr(loc + 1); - } - if (namespaces_left.size() > 0) { - path_prefix = path_prefix + underscore(namespaces_left) + "/"; - } - return path_prefix; -} - -void t_rb_generator::generate_rdoc(t_rb_ofstream& out, t_doc* tdoc) { - if (tdoc->has_doc()) { - out.indent(); - generate_docstring_comment(out, "", "# ", tdoc->get_doc(), ""); - } -} - -void t_rb_generator::generate_rb_struct_required_validator(t_rb_ofstream& out, t_struct* tstruct) { - out.indent() << "def validate" << endl; - out.indent_up(); - - const vector<t_field*>& fields = tstruct->get_members(); - vector<t_field*>::const_iterator f_iter; - - for (f_iter = fields.begin(); f_iter != fields.end(); ++f_iter) { - t_field* field = (*f_iter); - if (field->get_req() == t_field::T_REQUIRED) { - out.indent() << "raise ::Thrift::ProtocolException.new(::Thrift::ProtocolException::UNKNOWN, " - "'Required field " << field->get_name() << " is unset!')"; - if (field->get_type()->is_bool()) { - out << " if @" << field->get_name() << ".nil?"; - } else { - out << " unless @" << field->get_name(); - } - out << endl; - } - } - - // if field is an enum, check that its value is valid - for (f_iter = fields.begin(); f_iter != fields.end(); ++f_iter) { - t_field* field = (*f_iter); - - if (field->get_type()->is_enum()) { - out.indent() << "unless @" << field->get_name() << ".nil? || " - << full_type_name(field->get_type()) << "::VALID_VALUES.include?(@" - << field->get_name() << ")" << endl; - out.indent_up(); - out.indent() << "raise ::Thrift::ProtocolException.new(::Thrift::ProtocolException::UNKNOWN, " - "'Invalid value of field " << field->get_name() << "!')" << endl; - out.indent_down(); - out.indent() << "end" << endl; - } - } - - out.indent_down(); - out.indent() << "end" << endl << endl; -} - -void t_rb_generator::generate_rb_union_validator(t_rb_ofstream& out, t_struct* tstruct) { - out.indent() << "def validate" << endl; - out.indent_up(); - - const vector<t_field*>& fields = tstruct->get_members(); - vector<t_field*>::const_iterator f_iter; - - out.indent() - << "raise(StandardError, 'Union fields are not set.') if get_set_field.nil? || get_value.nil?" - << endl; - - // if field is an enum, check that its value is valid - for (f_iter = fields.begin(); f_iter != fields.end(); ++f_iter) { - const t_field* field = (*f_iter); - - if (field->get_type()->is_enum()) { - out.indent() << "if get_set_field == :" << field->get_name() << endl; - out.indent() << " raise " - "::Thrift::ProtocolException.new(::Thrift::ProtocolException::UNKNOWN, " - "'Invalid value of field " << field->get_name() << "!') unless " - << full_type_name(field->get_type()) << "::VALID_VALUES.include?(get_value)" - << endl; - out.indent() << "end" << endl; - } - } - - out.indent_down(); - out.indent() << "end" << endl << endl; -} - -THRIFT_REGISTER_GENERATOR( - rb, - "Ruby", - " rubygems: Add a \"require 'rubygems'\" line to the top of each generated file.\n" - " namespaced: Generate files in idiomatic namespaced directories.\n")
http://git-wip-us.apache.org/repos/asf/incubator-hawq/blob/72ea8afd/depends/thirdparty/thrift/compiler/cpp/src/generate/t_st_generator.cc ---------------------------------------------------------------------- diff --git a/depends/thirdparty/thrift/compiler/cpp/src/generate/t_st_generator.cc b/depends/thirdparty/thrift/compiler/cpp/src/generate/t_st_generator.cc deleted file mode 100644 index 9235f68..0000000 --- a/depends/thirdparty/thrift/compiler/cpp/src/generate/t_st_generator.cc +++ /dev/null @@ -1,1049 +0,0 @@ -/* - * Licensed to the Apache Software Foundation (ASF) under one - * or more contributor license agreements. See the NOTICE file - * distributed with this work for additional information - * regarding copyright ownership. The ASF licenses this file - * to you under the Apache License, Version 2.0 (the - * "License"); you may not use this file except in compliance - * with the License. You may obtain a copy of the License at - * - * http://www.apache.org/licenses/LICENSE-2.0 - * - * Unless required by applicable law or agreed to in writing, - * software distributed under the License is distributed on an - * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY - * KIND, either express or implied. See the License for the - * specific language governing permissions and limitations - * under the License. - * - * Contains some contributions under the Thrift Software License. - * Please see doc/old-thrift-license.txt in the Thrift distribution for - * details. - */ - -#include <string> -#include <fstream> -#include <iostream> -#include <vector> - -#include <stdlib.h> -#include <time.h> -#include <sys/stat.h> -#include <sys/types.h> -#include <sstream> - -#include "platform.h" -#include "t_oop_generator.h" -#include "version.h" - -using std::map; -using std::ofstream; -using std::ostringstream; -using std::string; -using std::stringstream; -using std::vector; - -static const string endl = "\n"; // avoid ostream << std::endl flushes - -/** - * Smalltalk code generator. - * - */ -class t_st_generator : public t_oop_generator { -public: - t_st_generator(t_program* program, - const std::map<std::string, std::string>& parsed_options, - const std::string& option_string) - : t_oop_generator(program) { - (void)parsed_options; - (void)option_string; - out_dir_base_ = "gen-st"; - } - - /** - * Init and close methods - */ - - void init_generator(); - void close_generator(); - - /** - * Program-level generation functions - */ - - void generate_typedef(t_typedef* ttypedef); - void generate_enum(t_enum* tenum); - void generate_const(t_const* tconst); - void generate_struct(t_struct* tstruct); - void generate_xception(t_struct* txception); - void generate_service(t_service* tservice); - void generate_class_side_definition(); - void generate_force_consts(); - - std::string render_const_value(t_type* type, t_const_value* value); - - /** - * Struct generation code - */ - - void generate_st_struct(std::ofstream& out, t_struct* tstruct, bool is_exception); - void generate_accessors(std::ofstream& out, t_struct* tstruct); - - /** - * Service-level generation functions - */ - - void generate_service_client(t_service* tservice); - - void generate_send_method(t_function* tfunction); - void generate_recv_method(t_function* tfunction); - - std::string map_reader(t_map* tmap); - std::string list_reader(t_list* tlist); - std::string set_reader(t_set* tset); - std::string struct_reader(t_struct* tstruct, std::string clsName); - - std::string map_writer(t_map* tmap, std::string name); - std::string list_writer(t_list* tlist, std::string name); - std::string set_writer(t_set* tset, std::string name); - std::string struct_writer(t_struct* tstruct, std::string fname); - - std::string write_val(t_type* t, std::string fname); - std::string read_val(t_type* t); - - /** - * Helper rendering functions - */ - - std::string st_autogen_comment(); - - void st_class_def(std::ofstream& out, std::string name); - void st_method(std::ofstream& out, std::string cls, std::string name); - void st_method(std::ofstream& out, std::string cls, std::string name, std::string category); - void st_close_method(std::ofstream& out); - void st_class_method(std::ofstream& out, std::string cls, std::string name); - void st_class_method(std::ofstream& out, std::string cls, std::string name, std::string category); - void st_setter(std::ofstream& out, std::string cls, std::string name, std::string type); - void st_getter(std::ofstream& out, std::string cls, std::string name); - void st_accessors(std::ofstream& out, std::string cls, std::string name, std::string type); - - std::string class_name(); - static bool is_valid_namespace(const std::string& sub_namespace); - std::string client_class_name(); - std::string prefix(std::string name); - std::string declare_field(t_field* tfield); - std::string type_name(t_type* ttype); - - std::string function_signature(t_function* tfunction); - std::string argument_list(t_struct* tstruct); - std::string function_types_comment(t_function* fn); - - std::string type_to_enum(t_type* ttype); - std::string a_type(t_type* type); - bool is_vowel(char c); - std::string temp_name(); - std::string generated_category(); - -private: - /** - * File streams - */ - int temporary_var; - std::ofstream f_; -}; - -/** - * Prepares for file generation by opening up the necessary file output - * streams. - * - * @param tprogram The program to generate - */ -void t_st_generator::init_generator() { - // Make output directory - MKDIR(get_out_dir().c_str()); - - temporary_var = 0; - - // Make output file - string f_name = get_out_dir() + "/" + program_name_ + ".st"; - f_.open(f_name.c_str()); - - // Print header - f_ << st_autogen_comment() << endl; - - st_class_def(f_, program_name_); - generate_class_side_definition(); - - // Generate enums - vector<t_enum*> enums = program_->get_enums(); - vector<t_enum*>::iterator en_iter; - for (en_iter = enums.begin(); en_iter != enums.end(); ++en_iter) { - generate_enum(*en_iter); - } -} - -string t_st_generator::class_name() { - return capitalize(program_name_); -} - -bool t_st_generator::is_valid_namespace(const std::string& sub_namespace) { - return sub_namespace == "prefix" || sub_namespace == "category"; -} - -string t_st_generator::prefix(string class_name) { - string prefix = program_->get_namespace("smalltalk.prefix"); - string name = capitalize(class_name); - name = prefix.empty() ? name : (prefix + name); - return name; -} - -string t_st_generator::client_class_name() { - return capitalize(service_name_) + "Client"; -} - -/** - * Autogen'd comment - */ -string t_st_generator::st_autogen_comment() { - return std::string("'") + "Autogenerated by Thrift Compiler (" + THRIFT_VERSION + ")\n" + "\n" - + "DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING\n" + "'!\n"; -} - -void t_st_generator::generate_force_consts() { - f_ << prefix(class_name()) << " enums keysAndValuesDo: [:k :v | " << prefix(class_name()) - << " enums at: k put: v value].!" << endl; - - f_ << prefix(class_name()) << " constants keysAndValuesDo: [:k :v | " << prefix(class_name()) - << " constants at: k put: v value].!" << endl; -} - -void t_st_generator::close_generator() { - generate_force_consts(); - f_.close(); -} - -string t_st_generator::generated_category() { - string cat = program_->get_namespace("smalltalk.category"); - // For compatibility with the Thrift grammar, the category must - // be punctuated by dots. Replaces them with dashes here. - for (string::iterator iter = cat.begin(); iter != cat.end(); ++iter) { - if (*iter == '.') { - *iter = '-'; - } - } - return cat.size() ? cat : "Generated-" + class_name(); -} - -/** - * Generates a typedef. This is not done in Smalltalk, types are all implicit. - * - * @param ttypedef The type definition - */ -void t_st_generator::generate_typedef(t_typedef* ttypedef) { - (void)ttypedef; -} - -void t_st_generator::st_class_def(std::ofstream& out, string name) { - out << "Object subclass: #" << prefix(name) << endl; - indent_up(); - out << indent() << "instanceVariableNames: ''" << endl << indent() << "classVariableNames: ''" - << endl << indent() << "poolDictionaries: ''" << endl << indent() << "category: '" - << generated_category() << "'!" << endl << endl; -} - -void t_st_generator::st_method(std::ofstream& out, string cls, string name) { - st_method(out, cls, name, "as yet uncategorized"); -} - -void t_st_generator::st_class_method(std::ofstream& out, string cls, string name) { - st_method(out, cls + " class", name); -} - -void t_st_generator::st_class_method(std::ofstream& out, string cls, string name, string category) { - st_method(out, cls, name, category); -} - -void t_st_generator::st_method(std::ofstream& out, string cls, string name, string category) { - char timestr[50]; - time_t rawtime; - struct tm* tinfo; - - time(&rawtime); - tinfo = localtime(&rawtime); - strftime(timestr, 50, "%m/%d/%Y %H:%M", tinfo); - - out << "!" << prefix(cls) << " methodsFor: '" + category + "' stamp: 'thrift " << timestr - << "'!\n" << name << endl; - - indent_up(); - out << indent(); -} - -void t_st_generator::st_close_method(std::ofstream& out) { - out << "! !" << endl << endl; - indent_down(); -} - -void t_st_generator::st_setter(std::ofstream& out, - string cls, - string name, - string type = "anObject") { - st_method(out, cls, name + ": " + type); - out << name << " := " + type; - st_close_method(out); -} - -void t_st_generator::st_getter(std::ofstream& out, string cls, string name) { - st_method(out, cls, name + ""); - out << "^ " << name; - st_close_method(out); -} - -void t_st_generator::st_accessors(std::ofstream& out, - string cls, - string name, - string type = "anObject") { - st_setter(out, cls, name, type); - st_getter(out, cls, name); -} - -void t_st_generator::generate_class_side_definition() { - f_ << prefix(class_name()) << " class" << endl << "\tinstanceVariableNames: 'constants enums'!" - << endl << endl; - - st_accessors(f_, class_name() + " class", "enums"); - st_accessors(f_, class_name() + " class", "constants"); - - f_ << prefix(class_name()) << " enums: Dictionary new!" << endl; - f_ << prefix(class_name()) << " constants: Dictionary new!" << endl; - - f_ << endl; -} - -/** - * Generates code for an enumerated type. Done using a class to scope - * the values. - * - * @param tenum The enumeration - */ -void t_st_generator::generate_enum(t_enum* tenum) { - string cls_name = program_name_ + capitalize(tenum->get_name()); - - f_ << prefix(class_name()) << " enums at: '" << tenum->get_name() << "' put: [" - << "(Dictionary new " << endl; - - vector<t_enum_value*> constants = tenum->get_constants(); - vector<t_enum_value*>::iterator c_iter; - for (c_iter = constants.begin(); c_iter != constants.end(); ++c_iter) { - int value = (*c_iter)->get_value(); - f_ << "\tat: '" << (*c_iter)->get_name() << "' put: " << value << ";" << endl; - } - - f_ << "\tyourself)]!" << endl << endl; -} - -/** - * Generate a constant value - */ -void t_st_generator::generate_const(t_const* tconst) { - t_type* type = tconst->get_type(); - string name = tconst->get_name(); - t_const_value* value = tconst->get_value(); - - f_ << prefix(class_name()) << " constants at: '" << name << "' put: [" - << render_const_value(type, value) << "]!" << endl << endl; -} - -/** - * Prints the value of a constant with the given type. Note that type checking - * is NOT performed in this function as it is always run beforehand using the - * validate_types method in main.cc - */ -string t_st_generator::render_const_value(t_type* type, t_const_value* value) { - type = get_true_type(type); - std::ostringstream out; - if (type->is_base_type()) { - t_base_type::t_base tbase = ((t_base_type*)type)->get_base(); - switch (tbase) { - case t_base_type::TYPE_STRING: - out << '"' << get_escaped_string(value) << '"'; - break; - case t_base_type::TYPE_BOOL: - out << (value->get_integer() > 0 ? "true" : "false"); - break; - case t_base_type::TYPE_BYTE: - case t_base_type::TYPE_I16: - case t_base_type::TYPE_I32: - case t_base_type::TYPE_I64: - out << value->get_integer(); - break; - case t_base_type::TYPE_DOUBLE: - if (value->get_type() == t_const_value::CV_INTEGER) { - out << value->get_integer(); - } else { - out << value->get_double(); - } - break; - default: - throw "compiler error: no const of base type " + t_base_type::t_base_name(tbase); - } - } else if (type->is_enum()) { - indent(out) << value->get_integer(); - } else if (type->is_struct() || type->is_xception()) { - out << "(" << capitalize(type->get_name()) << " new " << endl; - indent_up(); - - const vector<t_field*>& fields = ((t_struct*)type)->get_members(); - vector<t_field*>::const_iterator f_iter; - const map<t_const_value*, t_const_value*>& val = value->get_map(); - map<t_const_value*, t_const_value*>::const_iterator v_iter; - - for (v_iter = val.begin(); v_iter != val.end(); ++v_iter) { - t_type* field_type = NULL; - for (f_iter = fields.begin(); f_iter != fields.end(); ++f_iter) { - if ((*f_iter)->get_name() == v_iter->first->get_string()) { - field_type = (*f_iter)->get_type(); - } - } - if (field_type == NULL) { - throw "type error: " + type->get_name() + " has no field " + v_iter->first->get_string(); - } - - out << indent() << v_iter->first->get_string() << ": " - << render_const_value(field_type, v_iter->second) << ";" << endl; - } - out << indent() << "yourself)"; - - indent_down(); - } else if (type->is_map()) { - t_type* ktype = ((t_map*)type)->get_key_type(); - t_type* vtype = ((t_map*)type)->get_val_type(); - out << "(Dictionary new" << endl; - indent_up(); - indent_up(); - const map<t_const_value*, t_const_value*>& val = value->get_map(); - map<t_const_value*, t_const_value*>::const_iterator v_iter; - for (v_iter = val.begin(); v_iter != val.end(); ++v_iter) { - out << indent() << indent(); - out << "at: " << render_const_value(ktype, v_iter->first); - out << " put: "; - out << render_const_value(vtype, v_iter->second); - out << ";" << endl; - } - out << indent() << indent() << "yourself)"; - indent_down(); - indent_down(); - } else if (type->is_list() || type->is_set()) { - t_type* etype; - if (type->is_list()) { - etype = ((t_list*)type)->get_elem_type(); - } else { - etype = ((t_set*)type)->get_elem_type(); - } - if (type->is_set()) { - out << "(Set new" << endl; - } else { - out << "(OrderedCollection new" << endl; - } - indent_up(); - indent_up(); - const vector<t_const_value*>& val = value->get_list(); - vector<t_const_value*>::const_iterator v_iter; - for (v_iter = val.begin(); v_iter != val.end(); ++v_iter) { - out << indent() << indent(); - out << "add: " << render_const_value(etype, *v_iter); - out << ";" << endl; - } - out << indent() << indent() << "yourself)"; - indent_down(); - indent_down(); - } else { - throw "CANNOT GENERATE CONSTANT FOR TYPE: " + type->get_name(); - } - return out.str(); -} - -/** - * Generates a Smalltalk struct - */ -void t_st_generator::generate_struct(t_struct* tstruct) { - generate_st_struct(f_, tstruct, false); -} - -/** - * Generates a struct definition for a thrift exception. Basically the same - * as a struct but extends the Exception class. - * - * @param txception The struct definition - */ -void t_st_generator::generate_xception(t_struct* txception) { - generate_st_struct(f_, txception, true); -} - -/** - * Generates a smalltalk class to represent a struct - */ -void t_st_generator::generate_st_struct(std::ofstream& out, - t_struct* tstruct, - bool is_exception = false) { - const vector<t_field*>& members = tstruct->get_members(); - vector<t_field*>::const_iterator m_iter; - - if (is_exception) - out << "Error"; - else - out << "Object"; - - out << " subclass: #" << prefix(type_name(tstruct)) << endl << "\tinstanceVariableNames: '"; - - if (members.size() > 0) { - for (m_iter = members.begin(); m_iter != members.end(); ++m_iter) { - if (m_iter != members.begin()) - out << " "; - out << camelcase((*m_iter)->get_name()); - } - } - - out << "'\n" - << "\tclassVariableNames: ''\n" - << "\tpoolDictionaries: ''\n" - << "\tcategory: '" << generated_category() << "'!\n\n"; - - generate_accessors(out, tstruct); -} - -bool t_st_generator::is_vowel(char c) { - switch (tolower(c)) { - case 'a': - case 'e': - case 'i': - case 'o': - case 'u': - return true; - } - return false; -} - -string t_st_generator::a_type(t_type* type) { - string prefix; - - if (is_vowel(type_name(type)[0])) - prefix = "an"; - else - prefix = "a"; - - return prefix + capitalize(type_name(type)); -} - -void t_st_generator::generate_accessors(std::ofstream& out, t_struct* tstruct) { - const vector<t_field*>& members = tstruct->get_members(); - vector<t_field*>::const_iterator m_iter; - string type; - string prefix; - - if (members.size() > 0) { - for (m_iter = members.begin(); m_iter != members.end(); ++m_iter) { - st_accessors(out, - capitalize(type_name(tstruct)), - camelcase((*m_iter)->get_name()), - a_type((*m_iter)->get_type())); - } - out << endl; - } -} - -/** - * Generates a thrift service. - * - * @param tservice The service definition - */ -void t_st_generator::generate_service(t_service* tservice) { - generate_service_client(tservice); - // generate_service_server(tservice); -} - -string t_st_generator::temp_name() { - std::ostringstream out; - out << "temp" << temporary_var++; - return out.str(); -} - -string t_st_generator::map_writer(t_map* tmap, string fname) { - std::ostringstream out; - string key = temp_name(); - string val = temp_name(); - - out << "[oprot writeMapBegin: (TMap new keyType: " << type_to_enum(tmap->get_key_type()) - << "; valueType: " << type_to_enum(tmap->get_val_type()) << "; size: " << fname << " size)." - << endl; - indent_up(); - - out << indent() << fname << " keysAndValuesDo: [:" << key << " :" << val << " |" << endl; - indent_up(); - - out << indent() << write_val(tmap->get_key_type(), key) << "." << endl << indent() - << write_val(tmap->get_val_type(), val); - indent_down(); - - out << "]." << endl << indent() << "oprot writeMapEnd] value"; - indent_down(); - - return out.str(); -} - -string t_st_generator::map_reader(t_map* tmap) { - std::ostringstream out; - string desc = temp_name(); - string val = temp_name(); - - out << "[|" << desc << " " << val << "| " << endl; - indent_up(); - - out << indent() << desc << " := iprot readMapBegin." << endl << indent() << val - << " := Dictionary new." << endl << indent() << desc << " size timesRepeat: [" << endl; - - indent_up(); - out << indent() << val << " at: " << read_val(tmap->get_key_type()) - << " put: " << read_val(tmap->get_val_type()); - indent_down(); - - out << "]." << endl << indent() << "iprot readMapEnd." << endl << indent() << val << "] value"; - indent_down(); - - return out.str(); -} - -string t_st_generator::list_writer(t_list* tlist, string fname) { - std::ostringstream out; - string val = temp_name(); - - out << "[oprot writeListBegin: (TList new elemType: " << type_to_enum(tlist->get_elem_type()) - << "; size: " << fname << " size)." << endl; - indent_up(); - - out << indent() << fname << " do: [:" << val << "|" << endl; - indent_up(); - - out << indent() << write_val(tlist->get_elem_type(), val) << endl; - indent_down(); - - out << "]." << endl << indent() << "oprot writeListEnd] value"; - indent_down(); - - return out.str(); -} - -string t_st_generator::list_reader(t_list* tlist) { - std::ostringstream out; - string desc = temp_name(); - string val = temp_name(); - - out << "[|" << desc << " " << val << "| " << desc << " := iprot readListBegin." << endl; - indent_up(); - - out << indent() << val << " := OrderedCollection new." << endl << indent() << desc - << " size timesRepeat: [" << endl; - - indent_up(); - out << indent() << val << " add: " << read_val(tlist->get_elem_type()); - indent_down(); - - out << "]." << endl << indent() << "iprot readListEnd." << endl << indent() << val << "] value"; - indent_down(); - - return out.str(); -} - -string t_st_generator::set_writer(t_set* tset, string fname) { - std::ostringstream out; - string val = temp_name(); - - out << "[oprot writeSetBegin: (TSet new elemType: " << type_to_enum(tset->get_elem_type()) - << "; size: " << fname << " size)." << endl; - indent_up(); - - out << indent() << fname << " do: [:" << val << "|" << endl; - indent_up(); - - out << indent() << write_val(tset->get_elem_type(), val) << endl; - indent_down(); - - out << "]." << endl << indent() << "oprot writeSetEnd] value"; - indent_down(); - - return out.str(); -} - -string t_st_generator::set_reader(t_set* tset) { - std::ostringstream out; - string desc = temp_name(); - string val = temp_name(); - - out << "[|" << desc << " " << val << "| " << desc << " := iprot readSetBegin." << endl; - indent_up(); - - out << indent() << val << " := Set new." << endl << indent() << desc << " size timesRepeat: [" - << endl; - - indent_up(); - out << indent() << val << " add: " << read_val(tset->get_elem_type()); - indent_down(); - - out << "]." << endl << indent() << "iprot readSetEnd." << endl << indent() << val << "] value"; - indent_down(); - - return out.str(); -} - -string t_st_generator::struct_writer(t_struct* tstruct, string sname) { - std::ostringstream out; - const vector<t_field*>& fields = tstruct->get_sorted_members(); - vector<t_field*>::const_iterator fld_iter; - - out << "[oprot writeStructBegin: " - << "(TStruct new name: '" + tstruct->get_name() + "')." << endl; - indent_up(); - - for (fld_iter = fields.begin(); fld_iter != fields.end(); ++fld_iter) { - bool optional = (*fld_iter)->get_req() == t_field::T_OPTIONAL; - string fname = camelcase((*fld_iter)->get_name()); - string accessor = sname + " " + camelcase(fname); - - if (optional) { - out << indent() << accessor << " ifNotNil: [" << endl; - indent_up(); - } - - out << indent() << "oprot writeFieldBegin: (TField new name: '" << fname - << "'; type: " << type_to_enum((*fld_iter)->get_type()) - << "; id: " << (*fld_iter)->get_key() << ")." << endl; - - out << indent() << write_val((*fld_iter)->get_type(), accessor) << "." << endl << indent() - << "oprot writeFieldEnd"; - - if (optional) { - out << "]"; - indent_down(); - } - - out << "." << endl; - } - - out << indent() << "oprot writeFieldStop; writeStructEnd] value"; - indent_down(); - - return out.str(); -} - -string t_st_generator::struct_reader(t_struct* tstruct, string clsName = "") { - std::ostringstream out; - const vector<t_field*>& fields = tstruct->get_members(); - vector<t_field*>::const_iterator fld_iter; - string val = temp_name(); - string desc = temp_name(); - string found = temp_name(); - - if (clsName.size() == 0) { - clsName = tstruct->get_name(); - } - - out << "[|" << desc << " " << val << "|" << endl; - indent_up(); - - // This is nasty, but without it we'll break things by prefixing TResult. - string name = ((capitalize(clsName) == "TResult") ? capitalize(clsName) : prefix(clsName)); - out << indent() << val << " := " << name << " new." << endl; - - out << indent() << "iprot readStructBegin." << endl << indent() << "[" << desc - << " := iprot readFieldBegin." << endl << indent() << desc - << " type = TType stop] whileFalse: [|" << found << "|" << endl; - indent_up(); - - for (fld_iter = fields.begin(); fld_iter != fields.end(); ++fld_iter) { - out << indent() << desc << " id = " << (*fld_iter)->get_key() << " ifTrue: [" << endl; - indent_up(); - - out << indent() << found << " := true." << endl << indent() << val << " " - << camelcase((*fld_iter)->get_name()) << ": " << read_val((*fld_iter)->get_type()); - indent_down(); - - out << "]." << endl; - } - - out << indent() << found << " ifNil: [iprot skip: " << desc << " type]]." << endl; - indent_down(); - - out << indent() << "oprot readStructEnd." << endl << indent() << val << "] value"; - indent_down(); - - return out.str(); -} - -string t_st_generator::write_val(t_type* t, string fname) { - t = get_true_type(t); - - if (t->is_base_type()) { - t_base_type::t_base tbase = ((t_base_type*)t)->get_base(); - switch (tbase) { - case t_base_type::TYPE_DOUBLE: - return "iprot writeDouble: " + fname + " asFloat"; - break; - case t_base_type::TYPE_BYTE: - case t_base_type::TYPE_I16: - case t_base_type::TYPE_I32: - case t_base_type::TYPE_I64: - return "iprot write" + capitalize(type_name(t)) + ": " + fname + " asInteger"; - default: - return "iprot write" + capitalize(type_name(t)) + ": " + fname; - } - } else if (t->is_map()) { - return map_writer((t_map*)t, fname); - } else if (t->is_struct() || t->is_xception()) { - return struct_writer((t_struct*)t, fname); - } else if (t->is_list()) { - return list_writer((t_list*)t, fname); - } else if (t->is_set()) { - return set_writer((t_set*)t, fname); - } else if (t->is_enum()) { - return "iprot writeI32: " + fname; - } else { - throw "Sorry, I don't know how to write this: " + type_name(t); - } -} - -string t_st_generator::read_val(t_type* t) { - t = get_true_type(t); - - if (t->is_base_type()) { - return "iprot read" + capitalize(type_name(t)); - } else if (t->is_map()) { - return map_reader((t_map*)t); - } else if (t->is_struct() || t->is_xception()) { - return struct_reader((t_struct*)t); - } else if (t->is_list()) { - return list_reader((t_list*)t); - } else if (t->is_set()) { - return set_reader((t_set*)t); - } else if (t->is_enum()) { - return "iprot readI32"; - } else { - throw "Sorry, I don't know how to read this: " + type_name(t); - } -} - -void t_st_generator::generate_send_method(t_function* function) { - string funname = function->get_name(); - string signature = function_signature(function); - t_struct* arg_struct = function->get_arglist(); - const vector<t_field*>& fields = arg_struct->get_members(); - vector<t_field*>::const_iterator fld_iter; - - st_method(f_, client_class_name(), "send" + capitalize(signature)); - f_ << "oprot writeMessageBegin:" << endl; - indent_up(); - - f_ << indent() << "(TCallMessage new" << endl; - indent_up(); - - f_ << indent() << "name: '" << funname << "'; " << endl << indent() << "seqid: self nextSeqid)." - << endl; - indent_down(); - indent_down(); - - f_ << indent() << "oprot writeStructBegin: " - << "(TStruct new name: '" + capitalize(camelcase(funname)) + "_args')." << endl; - - for (fld_iter = fields.begin(); fld_iter != fields.end(); ++fld_iter) { - string fname = camelcase((*fld_iter)->get_name()); - - f_ << indent() << "oprot writeFieldBegin: (TField new name: '" << fname - << "'; type: " << type_to_enum((*fld_iter)->get_type()) << "; id: " << (*fld_iter)->get_key() - << ")." << endl; - - f_ << indent() << write_val((*fld_iter)->get_type(), fname) << "." << endl << indent() - << "oprot writeFieldEnd." << endl; - } - - f_ << indent() << "oprot writeFieldStop; writeStructEnd; writeMessageEnd." << endl; - f_ << indent() << "oprot transport flush"; - - st_close_method(f_); -} - -// We only support receiving TResult structures (so this won't work on the server side) -void t_st_generator::generate_recv_method(t_function* function) { - string funname = camelcase(function->get_name()); - string signature = function_signature(function); - - t_struct result(program_, "TResult"); - t_field success(function->get_returntype(), "success", 0); - result.append(&success); - - t_struct* xs = function->get_xceptions(); - const vector<t_field*>& fields = xs->get_members(); - vector<t_field*>::const_iterator f_iter; - for (f_iter = fields.begin(); f_iter != fields.end(); ++f_iter) { - // duplicate the field, but call it "exception"... we don't need a dynamic name - t_field* exception = new t_field((*f_iter)->get_type(), "exception", (*f_iter)->get_key()); - result.append(exception); - } - - st_method(f_, client_class_name(), "recv" + capitalize(funname)); - f_ << "| f msg res | " << endl << indent() << "msg := oprot readMessageBegin." << endl << indent() - << "self validateRemoteMessage: msg." << endl << indent() - << "res := " << struct_reader(&result) << "." << endl << indent() << "oprot readMessageEnd." - << endl << indent() << "oprot transport flush." << endl << indent() - << "res exception ifNotNil: [res exception signal]." << endl << indent() << "^ res"; - st_close_method(f_); -} - -string t_st_generator::function_types_comment(t_function* fn) { - std::ostringstream out; - const vector<t_field*>& fields = fn->get_arglist()->get_members(); - vector<t_field*>::const_iterator f_iter; - - out << "\""; - - for (f_iter = fields.begin(); f_iter != fields.end(); ++f_iter) { - out << camelcase((*f_iter)->get_name()) << ": " << type_name((*f_iter)->get_type()); - if ((f_iter + 1) != fields.end()) { - out << ", "; - } - } - - out << "\""; - - return out.str(); -} - -/** - * Generates a service client definition. - * - * @param tservice The service to generate a server for. - */ -void t_st_generator::generate_service_client(t_service* tservice) { - string extends = ""; - string extends_client = "TClient"; - vector<t_function*> functions = tservice->get_functions(); - vector<t_function*>::iterator f_iter; - - if (tservice->get_extends() != NULL) { - extends = type_name(tservice->get_extends()); - extends_client = extends + "Client"; - } - - f_ << extends_client << " subclass: #" << prefix(client_class_name()) << endl - << "\tinstanceVariableNames: ''\n" - << "\tclassVariableNames: ''\n" - << "\tpoolDictionaries: ''\n" - << "\tcategory: '" << generated_category() << "'!\n\n"; - - for (f_iter = functions.begin(); f_iter != functions.end(); ++f_iter) { - string funname = camelcase((*f_iter)->get_name()); - string signature = function_signature(*f_iter); - - st_method(f_, client_class_name(), signature); - f_ << function_types_comment(*f_iter) << endl << indent() << "self send" - << capitalize(signature) << "." << endl; - - if (!(*f_iter)->is_oneway()) { - f_ << indent() << "^ self recv" << capitalize(funname) << " success " << endl; - } - - st_close_method(f_); - - generate_send_method(*f_iter); - if (!(*f_iter)->is_oneway()) { - generate_recv_method(*f_iter); - } - } -} - -/** - * Renders a function signature of the form 'type name(args)' - * - * @param tfunction Function definition - * @return String of rendered function definition - */ -string t_st_generator::function_signature(t_function* tfunction) { - return camelcase(tfunction->get_name()) + capitalize(argument_list(tfunction->get_arglist())); -} - -/** - * Renders a field list - */ -string t_st_generator::argument_list(t_struct* tstruct) { - string result = ""; - - const vector<t_field*>& fields = tstruct->get_members(); - vector<t_field*>::const_iterator f_iter; - bool first = true; - for (f_iter = fields.begin(); f_iter != fields.end(); ++f_iter) { - if (first) { - first = false; - } else { - result += " "; - } - string name = camelcase((*f_iter)->get_name()); - result += name + ": " + name; - } - return result; -} - -string t_st_generator::type_name(t_type* ttype) { - string prefix = ""; - t_program* program = ttype->get_program(); - if (program != NULL && program != program_) { - if (!ttype->is_service()) { - prefix = program->get_name() + "_types."; - } - } - - string name = ttype->get_name(); - if (ttype->is_struct() || ttype->is_xception()) { - name = capitalize(ttype->get_name()); - } - - return prefix + name; -} - -/* Convert t_type to Smalltalk type code */ -string t_st_generator::type_to_enum(t_type* type) { - type = get_true_type(type); - - if (type->is_base_type()) { - t_base_type::t_base tbase = ((t_base_type*)type)->get_base(); - switch (tbase) { - case t_base_type::TYPE_VOID: - throw "NO T_VOID CONSTRUCT"; - case t_base_type::TYPE_STRING: - return "TType string"; - case t_base_type::TYPE_BOOL: - return "TType bool"; - case t_base_type::TYPE_BYTE: - return "TType byte"; - case t_base_type::TYPE_I16: - return "TType i16"; - case t_base_type::TYPE_I32: - return "TType i32"; - case t_base_type::TYPE_I64: - return "TType i64"; - case t_base_type::TYPE_DOUBLE: - return "TType double"; - } - } else if (type->is_enum()) { - return "TType i32"; - } else if (type->is_struct() || type->is_xception()) { - return "TType struct"; - } else if (type->is_map()) { - return "TType map"; - } else if (type->is_set()) { - return "TType set"; - } else if (type->is_list()) { - return "TType list"; - } - - throw "INVALID TYPE IN type_to_enum: " + type->get_name(); -} - -THRIFT_REGISTER_GENERATOR(st, "Smalltalk", "") http://git-wip-us.apache.org/repos/asf/incubator-hawq/blob/72ea8afd/depends/thirdparty/thrift/compiler/cpp/src/generate/t_xsd_generator.cc ---------------------------------------------------------------------- diff --git a/depends/thirdparty/thrift/compiler/cpp/src/generate/t_xsd_generator.cc b/depends/thirdparty/thrift/compiler/cpp/src/generate/t_xsd_generator.cc deleted file mode 100644 index ed76bec..0000000 --- a/depends/thirdparty/thrift/compiler/cpp/src/generate/t_xsd_generator.cc +++ /dev/null @@ -1,355 +0,0 @@ -/* - * Licensed to the Apache Software Foundation (ASF) under one - * or more contributor license agreements. See the NOTICE file - * distributed with this work for additional information - * regarding copyright ownership. The ASF licenses this file - * to you under the Apache License, Version 2.0 (the - * "License"); you may not use this file except in compliance - * with the License. You may obtain a copy of the License at - * - * http://www.apache.org/licenses/LICENSE-2.0 - * - * Unless required by applicable law or agreed to in writing, - * software distributed under the License is distributed on an - * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY - * KIND, either express or implied. See the License for the - * specific language governing permissions and limitations - * under the License. - */ - -#include <fstream> -#include <iostream> -#include <sstream> - -#include <stdlib.h> -#include <sys/stat.h> -#include <sstream> -#include "t_generator.h" -#include "platform.h" - -using std::map; -using std::ofstream; -using std::ostream; -using std::ostringstream; -using std::string; -using std::stringstream; -using std::vector; - -static const string endl = "\n"; // avoid ostream << std::endl flushes - -/** - * XSD generator, creates an XSD for the base types etc. - * - */ -class t_xsd_generator : public t_generator { -public: - t_xsd_generator(t_program* program, - const std::map<std::string, std::string>& parsed_options, - const std::string& option_string) - : t_generator(program) { - (void)parsed_options; - (void)option_string; - out_dir_base_ = "gen-xsd"; - } - - virtual ~t_xsd_generator() {} - - /** - * Init and close methods - */ - - void init_generator(); - void close_generator(); - - /** - * Program-level generation functions - */ - - void generate_typedef(t_typedef* ttypedef); - void generate_enum(t_enum* tenum) { (void)tenum; } - - void generate_service(t_service* tservice); - void generate_struct(t_struct* tstruct); - -private: - void generate_element(std::ostream& out, - std::string name, - t_type* ttype, - t_struct* attrs = NULL, - bool optional = false, - bool nillable = false, - bool list_element = false); - - std::string ns(std::string in, std::string ns) { return ns + ":" + in; } - - std::string xsd(std::string in) { return ns(in, "xsd"); } - - std::string type_name(t_type* ttype); - std::string base_type_name(t_base_type::t_base tbase); - - /** - * Output xsd/php file - */ - std::ofstream f_xsd_; - std::ofstream f_php_; - - /** - * Output string stream - */ - std::ostringstream s_xsd_types_; -}; - -void t_xsd_generator::init_generator() { - // Make output directory - MKDIR(get_out_dir().c_str()); - - // Make output file - string f_php_name = get_out_dir() + program_->get_name() + "_xsd.php"; - f_php_.open(f_php_name.c_str()); - - f_php_ << "<?php" << endl; -} - -void t_xsd_generator::close_generator() { - f_php_ << "?>" << endl; - f_php_.close(); -} - -void t_xsd_generator::generate_typedef(t_typedef* ttypedef) { - indent(s_xsd_types_) << "<xsd:simpleType name=\"" << ttypedef->get_name() << "\">" << endl; - indent_up(); - if (ttypedef->get_type()->is_string() && ((t_base_type*)ttypedef->get_type())->is_string_enum()) { - indent(s_xsd_types_) << "<xsd:restriction base=\"" << type_name(ttypedef->get_type()) << "\">" - << endl; - indent_up(); - const vector<string>& values = ((t_base_type*)ttypedef->get_type())->get_string_enum_vals(); - vector<string>::const_iterator v_iter; - for (v_iter = values.begin(); v_iter != values.end(); ++v_iter) { - indent(s_xsd_types_) << "<xsd:enumeration value=\"" << (*v_iter) << "\" />" << endl; - } - indent_down(); - indent(s_xsd_types_) << "</xsd:restriction>" << endl; - } else { - indent(s_xsd_types_) << "<xsd:restriction base=\"" << type_name(ttypedef->get_type()) << "\" />" - << endl; - } - indent_down(); - indent(s_xsd_types_) << "</xsd:simpleType>" << endl << endl; -} - -void t_xsd_generator::generate_struct(t_struct* tstruct) { - vector<t_field*>::const_iterator m_iter; - const vector<t_field*>& members = tstruct->get_members(); - bool xsd_all = tstruct->get_xsd_all(); - - indent(s_xsd_types_) << "<xsd:complexType name=\"" << tstruct->get_name() << "\">" << endl; - indent_up(); - if (xsd_all) { - indent(s_xsd_types_) << "<xsd:all>" << endl; - } else { - indent(s_xsd_types_) << "<xsd:sequence>" << endl; - } - indent_up(); - - for (m_iter = members.begin(); m_iter != members.end(); ++m_iter) { - generate_element(s_xsd_types_, - (*m_iter)->get_name(), - (*m_iter)->get_type(), - (*m_iter)->get_xsd_attrs(), - (*m_iter)->get_xsd_optional() || xsd_all, - (*m_iter)->get_xsd_nillable()); - } - - indent_down(); - if (xsd_all) { - indent(s_xsd_types_) << "</xsd:all>" << endl; - } else { - indent(s_xsd_types_) << "</xsd:sequence>" << endl; - } - indent_down(); - indent(s_xsd_types_) << "</xsd:complexType>" << endl << endl; -} - -void t_xsd_generator::generate_element(ostream& out, - string name, - t_type* ttype, - t_struct* attrs, - bool optional, - bool nillable, - bool list_element) { - string sminOccurs = (optional || list_element) ? " minOccurs=\"0\"" : ""; - string smaxOccurs = list_element ? " maxOccurs=\"unbounded\"" : ""; - string soptional = sminOccurs + smaxOccurs; - string snillable = nillable ? " nillable=\"true\"" : ""; - - if (ttype->is_void() || ttype->is_list()) { - indent(out) << "<xsd:element name=\"" << name << "\"" << soptional << snillable << ">" << endl; - indent_up(); - if (attrs == NULL && ttype->is_void()) { - indent(out) << "<xsd:complexType />" << endl; - } else { - indent(out) << "<xsd:complexType>" << endl; - indent_up(); - if (ttype->is_list()) { - indent(out) << "<xsd:sequence minOccurs=\"0\" maxOccurs=\"unbounded\">" << endl; - indent_up(); - string subname; - t_type* subtype = ((t_list*)ttype)->get_elem_type(); - if (subtype->is_base_type() || subtype->is_container()) { - subname = name + "_elt"; - } else { - subname = type_name(subtype); - } - f_php_ << "$GLOBALS['" << program_->get_name() << "_xsd_elt_" << name << "'] = '" << subname - << "';" << endl; - generate_element(out, subname, subtype, NULL, false, false, true); - indent_down(); - indent(out) << "</xsd:sequence>" << endl; - indent(out) << "<xsd:attribute name=\"list\" type=\"xsd:boolean\" />" << endl; - } - if (attrs != NULL) { - const vector<t_field*>& members = attrs->get_members(); - vector<t_field*>::const_iterator a_iter; - for (a_iter = members.begin(); a_iter != members.end(); ++a_iter) { - indent(out) << "<xsd:attribute name=\"" << (*a_iter)->get_name() << "\" type=\"" - << type_name((*a_iter)->get_type()) << "\" />" << endl; - } - } - indent_down(); - indent(out) << "</xsd:complexType>" << endl; - } - indent_down(); - indent(out) << "</xsd:element>" << endl; - } else { - if (attrs == NULL) { - indent(out) << "<xsd:element name=\"" << name << "\"" - << " type=\"" << type_name(ttype) << "\"" << soptional << snillable << " />" - << endl; - } else { - // Wow, all this work for a SIMPLE TYPE with attributes?!?!?! - indent(out) << "<xsd:element name=\"" << name << "\"" << soptional << snillable << ">" - << endl; - indent_up(); - indent(out) << "<xsd:complexType>" << endl; - indent_up(); - indent(out) << "<xsd:complexContent>" << endl; - indent_up(); - indent(out) << "<xsd:extension base=\"" << type_name(ttype) << "\">" << endl; - indent_up(); - const vector<t_field*>& members = attrs->get_members(); - vector<t_field*>::const_iterator a_iter; - for (a_iter = members.begin(); a_iter != members.end(); ++a_iter) { - indent(out) << "<xsd:attribute name=\"" << (*a_iter)->get_name() << "\" type=\"" - << type_name((*a_iter)->get_type()) << "\" />" << endl; - } - indent_down(); - indent(out) << "</xsd:extension>" << endl; - indent_down(); - indent(out) << "</xsd:complexContent>" << endl; - indent_down(); - indent(out) << "</xsd:complexType>" << endl; - indent_down(); - indent(out) << "</xsd:element>" << endl; - } - } -} - -void t_xsd_generator::generate_service(t_service* tservice) { - // Make output file - string f_xsd_name = get_out_dir() + tservice->get_name() + ".xsd"; - f_xsd_.open(f_xsd_name.c_str()); - - string ns = program_->get_namespace("xsd"); - if (ns.size() > 0) { - ns = " targetNamespace=\"" + ns + "\" xmlns=\"" + ns + "\" " - + "elementFormDefault=\"qualified\""; - } - - // Print the XSD header - f_xsd_ << "<?xml version=\"1.0\" encoding=\"UTF-8\" ?>" << endl - << "<xsd:schema xmlns:xsd=\"http://www.w3.org/2001/XMLSchema\""; << ns << ">" << endl - << endl << "<!-- Yo yo yo, this XSD woz be generated by Thrift. -->" << endl << endl; - - // Print out the type definitions - indent(f_xsd_) << s_xsd_types_.str(); - - // Keep a list of all the possible exceptions that might get thrown - map<string, t_struct*> all_xceptions; - - // List the elements that you might actually get - vector<t_function*> functions = tservice->get_functions(); - vector<t_function*>::iterator f_iter; - for (f_iter = functions.begin(); f_iter != functions.end(); ++f_iter) { - string elemname = (*f_iter)->get_name() + "_response"; - t_type* returntype = (*f_iter)->get_returntype(); - generate_element(f_xsd_, elemname, returntype); - f_xsd_ << endl; - - t_struct* xs = (*f_iter)->get_xceptions(); - const std::vector<t_field*>& xceptions = xs->get_members(); - vector<t_field*>::const_iterator x_iter; - for (x_iter = xceptions.begin(); x_iter != xceptions.end(); ++x_iter) { - all_xceptions[(*x_iter)->get_name()] = (t_struct*)((*x_iter)->get_type()); - } - } - - map<string, t_struct*>::iterator ax_iter; - for (ax_iter = all_xceptions.begin(); ax_iter != all_xceptions.end(); ++ax_iter) { - generate_element(f_xsd_, ax_iter->first, ax_iter->second); - } - - // Close the XSD document - f_xsd_ << endl << "</xsd:schema>" << endl; - f_xsd_.close(); -} - -string t_xsd_generator::type_name(t_type* ttype) { - if (ttype->is_typedef()) { - return ttype->get_name(); - } - - if (ttype->is_base_type()) { - return xsd(base_type_name(((t_base_type*)ttype)->get_base())); - } - - if (ttype->is_enum()) { - return xsd("int"); - } - - if (ttype->is_struct() || ttype->is_xception()) { - return ttype->get_name(); - } - - return "container"; -} - -/** - * Returns the XSD type that corresponds to the thrift type. - * - * @param tbase The base type - * @return Explicit XSD type, i.e. xsd:string - */ -string t_xsd_generator::base_type_name(t_base_type::t_base tbase) { - switch (tbase) { - case t_base_type::TYPE_VOID: - return "void"; - case t_base_type::TYPE_STRING: - return "string"; - case t_base_type::TYPE_BOOL: - return "boolean"; - case t_base_type::TYPE_BYTE: - return "byte"; - case t_base_type::TYPE_I16: - return "short"; - case t_base_type::TYPE_I32: - return "int"; - case t_base_type::TYPE_I64: - return "long"; - case t_base_type::TYPE_DOUBLE: - return "decimal"; - default: - throw "compiler error: no XSD base type name for base type " + t_base_type::t_base_name(tbase); - } -} - -THRIFT_REGISTER_GENERATOR(xsd, "XSD", "")
