thrift小试牛刀

在Apache上http://thrift.apache.org/，thrift如是简单的介绍了自己：Thrift is a software framework for scalable cross-language services development. It combines a software stack with a code generation engine to build services that work efficiently and seamlessly between C++, Java, Python, PHP, Ruby, Erlang, Perl, Haskell, C#, Cocoa, Smalltalk, and OCaml.

 

在首页上也给出了一个不那么完整的例子，由于最近项目将使用thrift，所以就把这个例子拿来研究了一翻，这里就是自己初学thrift的一点笔记罢了，还望高人指点。

 

通过thrift搭建网络应用的第一步都是协议的设计；因此，首先我创建了一个user.thrift的文件，文件内容如下：

#!/usr/local/bin/thrift --gen cpp namespace cpp User struct UserProfile { 1: i32 uid, 2: string name, 3: string blurb } service UserStorage { void store(1: UserProfile user), UserProfile retrieve(1: i32 uid) } 

这次学习thrift主要是采用的C++，也会涉及一点python编写client，也许以后自己还会研究一下java吧。thrift的协议设计可以看成有两个部分组成，一个数据协议（此处的struct UserPorfile），另外一个就是接口方法（service UserStorage）了。通俗的说就是，client可以构造一个UserProfile的对象，这个对象包括了uid、name、blurb三个属性；然后，client就可以用构造的对象作为参数调用store方法，将这个对象存储到server端（也许是数据库中）；除此之外，client还可以通过uid调用retrieve方法来获取具体的user对象。store和retrieve的实现是在服务端，client只管使用即可，就像本地方法一样，这就是RPC。

 

user.thrift文件编写好后，就可以具体的语言框架程序了，在这里我首先要生成一个C++的RPC框架程序出来，那么我就只需要执行如下命令即可：

thrift --gen cpp user.thrift 

如果是要生成python的框架执行命令：

thrift --gen py user.thrift 

 

这里我就只管C++了，因此最后将产生一个gen-cpp的目录，这个目录中有一堆c++源文件，这就是整个网络服务的框架程序了。

 

 

这里有一个UserStorage_server.skeleton.cpp的源文件，显然是服务端的框架，那么就从这里开始实现我们自己的简单需求吧，其他源文件暂时不管了。我将修改skeleton程序，保存client发送过来的user。

// This autogenerated skeleton file illustrates how to build a server. // You should copy it to another filename to avoid overwriting it. #include "UserStorage.h" #include <protocol/TBinaryProtocol.h> #include <server/TSimpleServer.h> #include <transport/TServerSocket.h> #include <transport/TBufferTransports.h> #include <iostream> #include <vector> using namespace std; using namespace ::apache::thrift; using namespace ::apache::thrift::protocol; using namespace ::apache::thrift::transport; using namespace ::apache::thrift::server; using boost::shared_ptr; using namespace User; class UserStorageHandler : virtual public UserStorageIf { public: UserStorageHandler() { // Your initialization goes here } void store(const UserProfile& user) { // Your implementation goes here printf("store/n"); m_vUser.push_back(user); } void retrieve(UserProfile& _return, const int32_t uid) { // Your implementation goes here printf("retrieve/n"); vector<UserProfile>::iterator vit = m_vUser.begin(); for ( ; vit != m_vUser.end(); vit++) { if (vit->uid == uid) { _return = *vit; } } } private: vector<UserProfile> m_vUser; }; int main(int argc, char **argv) { int port = 9090; shared_ptr<UserStorageHandler> handler(new UserStorageHandler()); shared_ptr<TProcessor> processor(new UserStorageProcessor(handler)); shared_ptr<TServerTransport> serverTransport(new TServerSocket(port)); shared_ptr<TTransportFactory> transportFactory(new TBufferedTransportFactory()); shared_ptr<TProtocolFactory> protocolFactory(new TBinaryProtocolFactory()); TSimpleServer server(processor, serverTransport, transportFactory, protocolFactory); server.serve(); return 0; }

 

整个服务端的实现，很简单，就是编写了store和retrieve两个方法而已，很奇怪的是thrift生成的retrieve方法，返回值被移到了参数中去了，不知为何会这样，抽空研究下thrift源码再说吧。至此，server端的应用编写完了，编译下看看。

 

g++ -Wall -I/usr/local/include/thrift *.cpp -lthrift -o user_server

 

编译完后，目录中就产生了user_server这个可执行的服务器程序了哦，感觉很简单吧，完全没有网络编程的感觉吧，当把client实现完后，才真正的感觉网络服务的编写和本地程序没有什么区别。

 

 

我们将目录的UserStorage_server.skeleton.cpp文件删了，加入我们自己编写的客户端程序UserStorage_client.cpp，代码如下：

// This autogenerated skeleton file illustrates how to build a server. // You should copy it to another filename to avoid overwriting it. #include "UserStorage.h" #include <protocol/TBinaryProtocol.h> #include <transport/TSocket.h> #include <transport/TBufferTransports.h> #include <iostream> using namespace ::apache::thrift; using namespace ::apache::thrift::protocol; using namespace ::apache::thrift::transport; using boost::shared_ptr; using namespace User; using namespace std; int main(int argc, char **argv) { boost::shared_ptr<TSocket> socket(new TSocket("localhost", 9090)); boost::shared_ptr<TTransport> transport(new TBufferedTransport(socket)); boost::shared_ptr<TProtocol> protocol(new TBinaryProtocol(transport)); UserStorageClient client(protocol); transport->open(); UserProfile User1, User2, User3; User1.uid = 1; User1.name = "yixiao"; User1.blurb = "xxxx"; User2.uid = 2; User2.name = "buer"; User2.blurb = "bbb"; User3.uid = 3; User3.name = "wxw"; User3.blurb = "ccc"; client.store(User1); client.store(User2); client.store(User3); client.retrieve(User1, 2); client.retrieve(User2, 3); client.retrieve(User3, 1); cout << User1.uid << endl << User1.name << endl << User1.blurb << endl; cout << User2.uid << endl << User2.name << endl << User2.blurb << endl; cout << User3.uid << endl << User3.name << endl << User3.blurb << endl; transport->close(); return 0; }

 

client程序比server就更简单，就是构造了3个user对象，然后依次保存到server中，最后又其他顺序获取出来。client在存储和获取数据的时候，就和本地方法调用一样，你自己就很难察觉你是把数据保存到网络中的另外一个服务器上去了，呵呵。最后我们运行服务器，然后执行以下客户端程序看看结果如何。

 

 

 

client的结果符合我们预期的想象。

 

对thrift的小试牛刀就到此结束了，后面的路还很长，继续深入研究thrift的实现。

 

备注：thrift跨语言的强大特性，真的不错，这里我就不废话我的python客户端了，python is always simple and graceful.