发表于 | 分类于 程序设计 |
上一篇文章我们了解了thrift的概念以及类型系统，本文我们通过一个简单的实例来更深入地了解thrift的使用。我们的实例非常简单，就是实现一个登录注册功能，其用户名密码缓存在内存中。
编写thrift文件
我们编写一个account.thrift的文件。
namespace java me.wuchong.thrift.generated
enum Operation{
LOGIN = 1,
REGISTER = 2
}
struct Request{
1: string name,
2: string password,
3: Operation op
}
exception InvalidOperation{
1: i32 code,
2: string reason
}
service Account{
string doAction(1: Request request) throws (1: InvalidOperation e);
}
然后在命令行下运行如下命令：
thrift --gen java account.thrift
则会在当前目录生成gen-java目录，该目录下会按照namespace定义的路径名一次一层层生成文件夹，如下图所示，在指定的包路径下生成了4个类。

服务实现
到此为止，thrift已经完成了其工作。接下来我们需要做的就是实现Account接口里的具体逻辑。我们创建一个AccountService类，实现Account.Iface接口。逻辑非常简单，将用户账户信息缓存在内存中，实现登录注册的功能，并且对一些非法输入状况抛出异常。
package me.wuchong.thrift.impl;
import me.wuchong.thrift.generated.Account;
import me.wuchong.thrift.generated.InvalidOperation;
import me.wuchong.thrift.generated.Operation;
import me.wuchong.thrift.generated.Request;
import java.util.HashMap;
import java.util.Map;
/**
* Created by wuchong on 15/10/7.
*/
public class AccountService implements Account.Iface {
private static Map<String, String> accounts = new HashMap<>();
@Override
public String doAction(Request request) throws InvalidOperation {
String name = request.getName();
String pass = request.getPassword();
Operation op = request.getOp();
System.out.println(String.format("Get request[name:%s, pass:%s, op:%d]", name, pass, op.getValue()));
if (name == null || name.length() == 0){
throw new InvalidOperation(100, "param name should not be empty");
}
if (op == Operation.LOGIN) {
String password = accounts.get(name);
if (password != null && password.equals(pass)) {
return "Login success!! Hello " + name;
} else {
return "Login failed!! please check your username and password";
}
} else if (op == Operation.REGISTER) {
if (accounts.containsKey(name)) {
return String.format("The username '%s' has been registered, please change one.", name);
} else {
accounts.put(name, pass);
return "Register success!! Hello " + name;
}
} else {
throw new InvalidOperation(101, "unknown operation: " + op.getValue());
}
}
}
启动服务端和客户端
我们实现了服务的具体逻辑，接下来需要启动该服务。这里我们需要用到thrift的依赖包。在pom.xml中加入对thrift的依赖。
<dependency>
<groupId>org.apache.thrift</groupId>
<artifactId>libthrift</artifactId>
<version>0.9.2</version>
</dependency>
注：如果你的依赖中没有加入slf4j的实现，则需要加上slf4j-log4j12或者logback的依赖，因为thrift有用到slf4j
启动服务的实现如下：
package me.wuchong.thrift.impl;
import me.wuchong.thrift.generated.Account;
import org.apache.thrift.server.TServer;
import org.apache.thrift.server.TSimpleServer;
import org.apache.thrift.transport.TServerSocket;
/**
* Created by wuchong on 15/10/7.
*/
public class AccountServer {
public static void main(String[] args) throws Exception {
TServerSocket socket = new TServerSocket(9999);
Account.Processor processor = new Account.Processor<>(new AccountService());
TServer server = new TSimpleServer(new TServer.Args(socket).processor(processor));
System.out.println("Starting the Account server...");
server.serve();
}
}
运行之后，可以在控制台看到输出：
Starting the Account server...
目前服务已经启动，则在客户端就可以进行RPC调用了。启动客户端的代码如下：
package me.wuchong.thrift.impl;
import me.wuchong.thrift.generated.Account;
import me.wuchong.thrift.generated.InvalidOperation;
import me.wuchong.thrift.generated.Operation;
import me.wuchong.thrift.generated.Request;
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.TTransport;
/**
* Created by wuchong on 15/10/7.
*/
public class AccountClient {
public static void main(String[] args) throws TException {
TTransport transport = new TSocket("localhost", 9999);
transport.open(); //建立连接
TProtocol protocol = new TBinaryProtocol(transport);
Account.Client client = new Account.Client(protocol);
//第一个请求， 登录 wuchong 帐号
Request req = new Request("wuchong", "1234", Operation.LOGIN);
request(client, req);
//第二个请求， 注册 wuchong 帐号
req.setOp(Operation.REGISTER);
request(client, req);
//第三个请求， 登录 wuchong 帐号
req.setOp(Operation.LOGIN);
request(client, req);
//第四个请求， name 为空的请求
req.setName("");
request(client, req);
transport.close(); //关闭连接
}
public static void request(Account.Client client, Request req) throws TException{
try {
String result = client.doAction(req);
System.out.println(result);
} catch (InvalidOperation e) {
System.out.println(e.reason);
}
}
}
运行客户端，其结果如下所示。
Login failed!! please check your username and password
Register success!! Hello wuchong
Login success!! Hello wuchong
param name should not be empty
而此时，服务端会打印出收到的请求信息。
Starting the Account server...
Get request[name:wuchong, pass:1234, op:1]
Get request[name:wuchong, pass:1234, op:2]
Get request[name:wuchong, pass:1234, op:1]
Get request[name:, pass:1234, op:1]
你可以发现，只需要几行代码，我们就实现了高效的RPC通信。
##参考资料
