Apache Thrift学习小记

namespace java mytest.thrift.gen
namespace py mytest.thrift
struct User {
1: i32    userId,
2: string loginName,
3: string password,
4: string name
}
exception UserNotFound {
1: string message
}
service UserService {
User getUser(1:string loginName) throws (1:UserNotFound unf),
list<User> getUsers()
}

package myserver;
import java.util.ArrayList;
import java.util.List;
import mytest.thrift.gen.User;
import mytest.thrift.gen.UserNotFound;
import mytest.thrift.gen.UserService;
import org.apache.thrift.TException;
public class UserServiceHandler implements UserService.Iface {
@Override
public User getUser(String loginName) throws UserNotFound, TException {
if (!"login1".equals(loginName)) {
UserNotFound e = new UserNotFound("User not Found!");
throw e;
}
User user = new User();
user.setUserId(100);
user.setLoginName("login1");
user.setPassword("pwd1");
user.setName("user1");
return user;
}
@Override
public List<User> getUsers() throws TException {
List<User> list = new ArrayList<User>();
User user = new User();
user.setUserId(100);
user.setLoginName("login1");
user.setPassword("pwd1");
user.setName("user1");
list.add(user);
User user2 = new User();
user2.setUserId(200);
user2.setLoginName("login2");
user2.setPassword("pwd2");
user2.setName("user2");
list.add(user2);
return list;
}
}

package myserver;
import mytest.thrift.gen.UserService;
import org.apache.thrift.server.TServer;
import org.apache.thrift.server.TSimpleServer;
import org.apache.thrift.transport.TServerSocket;
import org.apache.thrift.transport.TServerTransport;
public class Server {
public static void main(String[] args) {
try {
UserServiceHandler handler = new UserServiceHandler();
UserService.Processor processor = new UserService.Processor(handler);
TServerTransport serverTransport = new TServerSocket(9090);
TServer server = new TSimpleServer(processor, serverTransport);
// Use this for a multithreaded server
// server = new TThreadPoolServer(processor, serverTransport);
System.out.println("Starting the server...");
server.serve();
} catch (Exception x) {
x.printStackTrace();
}
System.out.println("done.");
}
}

package myclient;
import java.util.Iterator;
import java.util.List;
import mytest.thrift.gen.User;
import mytest.thrift.gen.UserNotFound;
import mytest.thrift.gen.UserService;
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;
public class Client {
public static void main(String[] args) {
try {
TTransport transport = new TSocket("localhost", 9090);
TProtocol protocol = new TBinaryProtocol(transport);
UserService.Client client = new UserService.Client(protocol);
transport.open();
System.out.println("test1");
try {
User user1 = client.getUser("login1");
System.out.println("name=" + user1.getName());
} catch (UserNotFound e) {
System.out.println(e.getMessage());
}

System.out.println("test2");
try {
User user2 = client.getUser("login10");
System.out.println("name=" + user2.getName());
} catch (UserNotFound e) {
System.out.println(e.getMessage());
}

System.out.println("test3");
List<User> list = client.getUsers();
Iterator<User> it = list.iterator();
while(it.hasNext()){
User u = it.next();
System.out.println("name=" + u.getName());
}
transport.close();
} catch (TException x) {
x.printStackTrace();
}
}
}

from mytest.thrift import UserService
from mytest.thrift.ttypes import UserNotFound
from thrift import Thrift
from thrift.transport import TSocket
from thrift.transport import TTransport
from thrift.protocol import TBinaryProtocol
try:
# Make socket
transport = TSocket.TSocket('localhost', 9090)
# Buffering is critical. Raw sockets are very slow
transport = TTransport.TBufferedTransport(transport)
# Wrap in a protocol
protocol = TBinaryProtocol.TBinaryProtocol(transport)
# Create a client to use the protocol encoder
client = UserService.Client(protocol)
# Connect!
transport.open()
try:
user1 = client.getUser("login1")
print user1.name
except UserNotFound, io:
print '%r' % io
# Close!
transport.close()
except Thrift.TException, tx:
print '%s' % (tx.message)