thrift 服务器如何获取客户端地址

THRIFT服务器缺省无法获取客户端连接的地址。经过研究,发现通过重载transport factory+Thread local,能够简单解决。不需要修改thrift生成的代码。

>>>>>
arg.protocolFactory(protocolFactory);
arg.outputTransportFactory(new TFramedTransport.Factory());
arg.inputTransportFactory(new TFramedTransport.Factory() {
@Override
public TTransport getTransport(TTransport base) {
if(base instanceof TSocket) {
TSocket _base = (TSocket) base;
Socket socket = _base.getSocket();
if(socket != null) {
String remoteAddr = String.valueOf(socket.getRemoteSocketAddress());
ThreadLocalIpUtils.setIp(remoteAddr);
}
}

return new TFramedTransport(base) {
@Override
public void close() {
ThreadLocalIpUtils.remove();
super.close();
}
};
}
});

猜你喜欢

转载自tailorcai.iteye.com/blog/1756364