现在的位置: 首页 > 综合 > 正文

thrift之compiler源码详解-3

2018年03月30日 ⁄ 综合 ⁄ 共 6581字 ⁄ 字号 评论关闭

转至:http://www.cnblogs.com/brucewoo/archive/2012/04/25/2470687.html

生成C++语言代码的代码详解

      由于各种语言的代码生成结构和原理都差不多,所以我就只分析了生成C++代码的代码

这个功能是由t_cpp_generator类实现(在文件t_cpp_generator.cc定义和实现),直接继承至t_oop_generator类(这个类是所有面向对象语言生成器类的直接基类,封装了面向对象语言生成器共有的特征与行为),而t_oop_generator又从t_generator继承(上面已经介绍),下面详细分析这个类是怎样生成C++语言的代码文件的。这个还有从上面介绍的generate_program函数开始说起,因为这个函数才是控制整个代码生成的总枢纽。

首先执行的是构造函数,这个构造函数做了一些最基本的初始化,一个是传递拥有生成代码的符号资源的t_program对象到父类,第二个功能就是根据可选项参数初始化一些bool变量,以便后面根据这些bool变量做相应的处理,代码很简单就不列出来了,下面用一个表格说明各个bool变量的作用(或功能)。

 

gen_pure_enums_

是否生成纯净的枚举类型,而不是采用类包装的形式

gen_dense_

是否应该TDenseProtocol生成本地反射的元数据。

gen_templates_

是否要生成模板化的读/写方法

use_include_prefix_

是否应该为了thrift生成的其他头文件在#include中使用前缀路径

gen_cob_style_

是否应该生成继承扩展功能类(主要是异步)

gen_no_client_completion_

是否应该省略客户端类调用completion__()

 

构造函数只是做了最基本的初始化,更详细的初始化是上面介绍的代码生成器初始化函数init_generator,那我们看看C++代码生成器是怎么详细初始化的,都做了一些什么样的工作和实现了一些什么的功能。我们分步骤介绍这一个函数:

第一步:制作代码文件的输出目录:MKDIR(get_out_dir().c_str());MKDIR是一个宏函数,调用了mkdir来创建目录;

第二部:创建代码头文件和实现文件,如果需要生成模板化的读和写的方法还会创建一个文件单独实现,代码如下:

复制代码
 1 string f_types_name = get_out_dir()+program_name_+"_types.h";
 2 
 3   f_types_.open(f_types_name.c_str());
 4 
 5   string f_types_impl_name = get_out_dir()+program_name_+"_types.cpp";
 6 
 7   f_types_impl_.open(f_types_impl_name.c_str());
 8 
 9   if (gen_templates_) {
10 
11     string f_types_tcc_name = get_out_dir()+program_name_+"_types.tcc";
12 
13     f_types_tcc_.open(f_types_tcc_name.c_str());
14 
15   }
复制代码

 

这里需要说明几个用于输出流的成员变量,之所以定义成员变量是因为很多函数会用到,这样就不用用参数来传递它们了,它们定义和说明如下:

复制代码
 1 std::ofstream f_types_;//专门用于类型声明的输出流,也就是头文件(.h文件)
 2 
 3 std::ofstream f_types_impl_;//专门用于类型实现的输出流,也就是实现文件(.cpp文件)
 4 
 5   std::ofstream f_types_tcc_;//专门用于模板化的读和写方法实现的输出流
 6 
 7   std::ofstream f_header_;//专门用于服务声明生成的输出流
 8 
 9   std::ofstream f_service_;//专门用于服务实现生成的输出流
10 
11   std::ofstream f_service_tcc_;//专门用于模板的服务的输出流
复制代码

 

第三步:为每个文件打印头部注释,注释的作用就是说明这个文件是由Thrift自动生成的,代码如下:

1 f_types_ << autogen_comment();
2 
3 f_types_impl_ << autogen_comment();
4 
5 f_types_tcc_ << autogen_comment();

 

第四步:开始ifndef

第五步:包含各种头文件

第六步:打开命名空间,生成的代码都是在一个命令空间里面的。

以上步骤的功能都比较简单,主要就是注意输出格式和逻辑处理。通过这些功能基本内容都做好了,下面就是真正开始生成具体类型和服务的时候了,每一种数据类型都由一个单独的函数来负责生成为代码。

1)枚举类型生成函数generate_enum

首先在头文件中生成定义枚举类型的代码,具体的过程就是得到枚举的所有常量值和枚举类型的名称,然后根据C++定义枚举类型的语法输出代码到头文件,输出过程中根据是否需要用类来包装而所有不同,同时生成的代码也需要格式控制。具体实现如下:

复制代码
 1 vector<t_enum_value*> constants = tenum->get_constants();
 2 
 3   std::string enum_name = tenum->get_name();
 4 
 5   if (!gen_pure_enums_) {
 6 
 7     enum_name = "type";
 8 
 9     f_types_ << indent() << "struct " << tenum->get_name() << " {" << endl;
10 
11     indent_up();
12 
13   }
14 
15   f_types_ << indent() << "enum " << enum_name;
16 
17   generate_enum_constant_list(f_types_, constants, "", "", true);
18 
19  if (!gen_pure_enums_) {
20 
21     indent_down();
22 
23     f_types_ << "};" << endl;
24 
25   }
26 
27   f_types_ << endl;
复制代码

 

接着在后面在实现文件中定义一个整型数组和一个字符的数组并用定义的枚举类型的常量值来初始化这两个数组,后然在说这两个数组的值初始化一个map,其实这么做的目的就是为了测试这个枚举类型定义是否正确。

最后调用函数generate_local_reflection决定是否为TDenseProtocol协议生成对应类型的本地反射类型。这个函数功能比较复杂,后面单独详细讲解。

2)类型定义生成函数generate_typedef

此函数功能简单,就是在头文件中生成一个typedef的定义,就只有一句实现:

1 f_types_ <<  indent() << "typedef " << type_name(ttypedef->get_type(), true) << " " 
2 
3 << ttypedef->get_symbolic() << ";" << endl << endl;

 

3)常量类型生成函数generate_consts

常量类型的实现是采用一个类来包装所有的常量并且使用单独的文件来实现,所有首先创建常量类型定义头文件和实现文件,代码如下:

复制代码
 1 string f_consts_name = get_out_dir()+program_name_+"_constants.h";
 2 
 3 ofstream f_consts;
 4 
 5 f_consts.open(f_consts_name.c_str());
 6 
 7 string f_consts_impl_name = get_out_dir()+program_name_+"_constants.cpp";
 8 
 9 ofstream f_consts_impl;
10 
11 f_consts_impl.open(f_consts_impl_name.c_str());
复制代码

 

接着按照就开始按照类的定义格式在头文件中生成定义类的代码并在实现文件中定义这个类的常量类型;在这个类的构造函数中给定义的数据类型赋值:

复制代码
 1 f_consts_impl << "const " << program_name_ << "Constants g_" << program_name_ << "_constants;" << endl <<
 2 
 3     endl << program_name_ << "Constants::" << program_name_ << "Constants() {" << endl;
 4 
 5 indent_up();
 6 
 7 for (c_iter = consts.begin(); c_iter != consts.end(); ++c_iter) {
 8 
 9     print_const_value(f_consts_impl, (*c_iter)->get_name(), (*c_iter)->get_type(), (*c_iter)->get_value());
10 
11   }
12 
13 indent_down();
14 
15 indent(f_consts_impl) << "}" << endl;
复制代码

 

其中调用了print_const_value函数来根据数据类型来赋值,这些定义在类中的成员变量本身不是常量类型,只是在实现文件中定义了一个类的全局常量对象,在头文件中声明,以便其他地方可以被使用。

4)异常类型生成函数generate_xception

这个函数其实调用下面需要详细分析的一个函数实现的,就是generate_struct函数,因为异常也是通过结构体来定义和实现的。不过C++语言生成器中也自己实现了这个函数,不过它是调用generate_cpp_struct函数实现,C++generate_struct函数也是调用这个函数实现,只是传递一个bool变量来区分是否是异常类型,具体的实现在分析generate_struct函数时一起详细分析了,因为它们的基本实现功能都是相同的。

5)结构体类型生成函数generate_struct

上面已经说了这个函数也是调用generate_cpp_struct函数实现,也就是说异常类型和结构体类型都是用同样的流程实现的,它们都是定义为一个类,只是异常都从TException继承,而一般的结构体没有。首先调用函数generate_struct_definition在头文件中生成定义类的代码,这个过程如下:

第一步:得到所有的成员变量;

第二步:根据是否有可选成员决定是否定义一个结构体_XXX_isset,这个结果主要针对需要定义的类的可选成员而定义一些bool变量,来标识这些可选成员变量是否存在。设计这个功能的目的是为了灵活控制数据传输的结构;

第三步:开始生成定义类(IDL文件中定义的structC++都是用class来实现)的代码,生成的代码主要包括默认的构造函数、析构函数、各个字段、比较函数(等于、不等于和小于)等;

第四步:最后一步生成一个模板的读和写数据的函数的声明,模板参数是协议类型,实现代码如下:

复制代码
 1 if (read) {//读数据的模板函数
 2 
 3     if (gen_templates_) {
 4 
 5       out <<indent() << "template <class Protocol_>" << endl <<
 6 
 7         indent() << "uint32_t read(Protocol_* iprot);" << endl;
 8 
 9     } else {
10 
11       out << indent() << "uint32_t read(" << "::apache::thrift::protocol::TProtocol* iprot);" << endl;
12 
13     }
14 
15 }
16 
17 if (write) {//写数据的模板函数
18 
19     if (gen_templates_) {
20 
21       out << indent() << "template <class Protocol_>" << endl <<
22 
23         indent() << "uint32_t write(Protocol_* oprot) const;" << endl;
24 
25     } else {
26 
27       out << indent() << "uint32_t write(" << "::apache::thrift::protocol::TProtocol* oprot) const;" << endl;
28 
29     }
30 
31 }
复制代码

 

然后调用函数generate_struct_fingerprint在实现文件中初始化两个静态变量,一个是字符串,一个是8位的整型数组,这两个变量都是用来唯一的标识一个类。这个歌标识符的作用就是用于生成本地的反射类型,当使用TDenseProtocol协议传输数据时会用到。

接着两次调用generate_local_reflection函数分别来声明和定义用于类的本地反射的类型,调用generate_local_reflection_pointer函数来生成一个类的静态指针的本地反射类型。

最后分别调用函数generate_struct_reader和generate_struct_writer实现数据读和写函数。到此整个IDL定义的struct类型生成为C++的代码就完成了。

6)服务类型生成函数generate_service

这个函数的功能是最复杂的,它会做很多的工作(分别调用其它函数来实现),也会生成单独的头文件和实现文件。生成头文件的代码如下:

1 string f_header_name = get_out_dir()+svcname+".h";
2 
3   f_header_.open(f_header_name.c_str());

 

下面就开始在头文件中生成一些包含头文件的代码。

生成实现文件的代码:

1 string f_service_name = get_out_dir()+svcname+".cpp";
2 
3   f_service_.open(f_service_name.c_str());

 

后面也是生成一些包含头文件的代码。接着就开始生成正在的各种实现这个服务的代码了,如下:

复制代码
 1 generate_service_interface(tservice, "");//生成服务的接口类(在C++为抽象类)
 2 
 3   generate_service_null(tservice, "");//生成一个空实现服务接口类的类
 4 
 5   generate_service_helpers(tservice);//生成一些帮助类,如参数类、返回结果类等
 6 
 7   generate_service_client(tservice, "");//生成一个客户类
 8 
 9   generate_service_processor(tservice, "");//生成处理数据的类(就是生成用于远程调用)
10 
11   generate_service_multiface(tservice);//生成一个实现多接口的单一的服务器类
12 
13   generate_service_skeleton(tservice);//生成一个服务器的框架文件
复制代码

 

如果gen_cob_style_为true,还会生成一些扩展功能的类,代码如下:

复制代码
 1 if (gen_cob_style_) {
 2 
 3     generate_service_interface(tservice, "CobCl");
 4 
 5     generate_service_interface(tservice, "CobSv");
 6 
 7     generate_service_null(tservice, "CobSv");
 8 
 9     generate_service_client(tservice, "Cob");
10 
11     generate_service_processor(tservice, "Cob");
12 
13     generate_service_async_skeleton(tservice);
14 
15 }
复制代码

 

到此C++的代码的生成全部结束,最后调用close_generator函数来完成收尾工作和清理一些资源,如果关闭文件。

7)总结

对于生成C++代码这一块内容把基本的生成过程详细分析了一遍,主要集中在整个流程中。

抱歉!评论已关闭.