Thrift开发实例

Thrift开发实例


准备工作
Thrift主页：http://thrift.apache.org/

Thrift下载：http://thrift.apache.org/download/

下载thrift-0.8.0.tar.gz和Thrift compiler for Windows (thrift-0.8.0.exe)

创建工作区../thrift；

将thrift-0.8.0.tar.gz和thrift-0.8.0.exe拷贝到../thrift下；

将thrift-0.8.0.tar.gz解压到当前目录；

创建start.bat，编辑其内容为：

cd

thrift-0.8.0 --gen java *.thrift

pause




简单实例
创建thrift脚本文件：

namespace java com.test.rpc

service TestService{

string getUserName(1:i64 id)

}


执行start.bat，thrift会根据脚本生成java代码../thrift/gen-java/com/test/rpc/ TestService.java;

创建java project；

创建package：com.test.rpc，将生成的TestService.java拷入；

创建接口实现类TestImpl

package com.test.rpc;



import org.apache.log4j.Logger;

import org.apache.thrift.TException;



import com.test.rpc.TestService.Iface;

import com.test.user.IUserService;

import com.test.user.UserServiceImpl;





public class TestImpl implements Iface {

private static final Logger LOG = Logger.getLogger(TestImpl.class);



@Override

public String getUserName(long id) throws TException {

LOG.info("TestImpl server get rpc msg :"+id);

IUserService userService = new UserServiceImpl();

String username = userService.getUser(id);

return username;

}



}


创建RPC server启动线程

package com.test.rpc;



import javax.xml.ws.Endpoint;



import org.apache.log4j.Logger;

import org.apache.thrift.TProcessor;

import org.apache.thrift.protocol.TBinaryProtocol;

import org.apache.thrift.protocol.TBinaryProtocol.Factory;

import org.apache.thrift.server.TServer;

import org.apache.thrift.server.TThreadPoolServer;

import org.apache.thrift.server.TThreadPoolServer.Args;

import org.apache.thrift.transport.TServerSocket;

import org.apache.thrift.transport.TTransportException;



import com.test.service.CommonService;











public class TestServer extends Thread {

private static final Logger LOG = Logger.getLogger(TestServer.class);



@Override

public void run() {

try {

TestImpl testimpl = new TestImpl();

TServerSocket serverTransport = new TServerSocket(7911);

Factory proFactory = new TBinaryProtocol.Factory();

TProcessor processor = new TestService.Processor<TestImpl>(testimpl);

Args rpcArgs = new Args(serverTransport);

rpcArgs.processor(processor);

rpcArgs.protocolFactory(proFactory);

TServer server = new TThreadPoolServer(rpcArgs);

LOG.info("Start TestService on port 7911..." + Thread.currentThread().getId() + "["

+ Thread.currentThread().getName() + "]");

server.serve();

} catch (TTransportException e) {

e.printStackTrace();

LOG.info("TestService.simpleRun", e);

}

}

/**

* @param args

*/

public static void main(String[] args) {

TestServer server = new TestServer();

server.start();

}



}








复杂对象实例
与简单实例基本相同，只是脚本不同，参数以对象方式传递，自动生成的java类多一些；

创建thrift脚本文件：

namespace java com.test.rpc

struct User {

1: i64 id

2: string name

}

service TestService{

User getUser(1:i64 id)

}


执行start.bat，thrift会根据脚本生成java代码../thrift/gen-java/com/test/rpc/目录下TestService.java、User.Java；

创建java project；

创建package：com.test.rpc，将生成的TestService.java、User.Java拷入；

其他步骤类同上例；





客户端代码



package com.test.rpc;



import org.apache.log4j.Logger;

import org.apache.thrift.TException;

import org.apache.thrift.protocol.TBinaryProtocol;

import org.apache.thrift.protocol.TProtocol;

import org.apache.thrift.transport.TSocket;

import org.apache.thrift.transport.TTransportException;

import com.test.rpc.TestService.Client;



public class TestClient {

private static final Logger LOG = Logger.getLogger(TestClient.class);



public static void main(String[] args){

try{

TSocket tsocket = new TSocket("localhost", 7911);

tsocket.open();

TProtocol protocol = new TBinaryProtocol(tsocket);

Client client = new Client(protocol);

User user = client.getUser(id);

LOG.info(user.getName());

}catch (Exception e) {

LOG.error(e.getMessage(), e);

}

}



}