TNonblockingServer 时必须使用
TFramedTransport ，不能使用 TBufferedTransport，因为前者会先写入这个消息的字节数。这样非阻塞时可以预知消息的大小。
1. 服务端获取请求方的IP
网上流传最多的是扩展
bool TDispatchProcessor::process(boost::shared_ptr<protocol::TProtocol> in,
boost::shared_ptr<protocol::TProtocol> out,
void* connectionContext)
从 in 获得 transport 然后获得 TSocket 对象。但这个方法只适合阻塞式的 TSimpleServer 和 TThreadPoolServer 等，并不适合 TNonblockingServer。使用 TNonblockingServer 时，可通过扩展 void TServerEventHandler::processContext(void* serverContext, boost::shared_ptr<TTransport> transport) ，这里 transport 参数是个 TSocket 对象，而这个函数总会在调用 processor 处理请求之前被调用。
static boost::thread_specific_ptr<std::string> thrift_client_ip; // thread specific
class MyServerEventHandler : public TServerEventHandler
{
virtual void processContext(void* serverContext, boost::shared_ptr<TTransport> transport)
{
TSocket *sock = static_cast<TSocket *>(transport.get());
if (sock)
{
//thrift_client_ip.reset(new string(sock->getPeerAddress())); // 0.9.2, 复用 TNonblockingServer::TConnection 导致 getPeerAddress() 返回脏数据, 见 https://issues.apache.org/jira/browse/THRIFT-3270
sock->getCachedAddress(); // use this api instead
}
}
};
// create nonblocking server
TNonblockingServer server(processor, protocolFactory, port, threadManager);
boost::shared_ptr<MyServerEventHandler> eventHandler(new MyServerEventHandler());
server.setServerEventHandler(eventHandler);
TNonblockingServer 复用 TNonblockingServer::TConnection 对象，但调用 TSocket::setCachedAddress() 并没有清掉 peerAddress_ 和 peerHost_ ，导致脏数据。
需要 TSocket::setCachedAddress() 函数中清除 这2个变量
lib/cpp/src/thrift/transport/TSocket.cpp
void TSocket::setCachedAddress(const sockaddr* addr, socklen_t len) {
if (!path_.empty()) {
return;
}
switch (addr->sa_family) {
case AF_INET:
if (len == sizeof(sockaddr_in)) {
memcpy((void*)&cachedPeerAddr_.ipv4, (void*)addr, len);
}
break;
case AF_INET6:
if (len == sizeof(sockaddr_in6)) {
memcpy((void*)&cachedPeerAddr_.ipv6, (void*)addr, len);
}
break;
}
peerAddress_.clear(); // ++
peerHost_.clear(); // ++
}
2. 禁用监听 IPv6，
阻塞式服务需要修改 lib/cpp/src/thrift/transport/TServerSocket.cpp 的
TServerSocket::listen() 函数，修改 hints.ai_family = PF_INET;
而 TNonblockingServer 需要修改 lib/cpp/src/thrift/transport/TNonblockingServer.cpp
TNonblockingServer::createAndListenOnSocket() 函数，修改 hints.ai_family = PF_INET;
3. 切换 libevent 为 libev
由于项目中用到了 libev，而 thrift 的 TNonblockingServer 用到 libevent，而它们头文件有名字冲突。libev 做了个简单的 libevent 适配接口，但 thrift 还用到了 创建 socketpair 设置 socket close on exec 等函数，我简单的从 libevent 拷贝过来，即编译成功。 (sockepair 本来理论上可以用 libev 的 ev_async 替代的，但由于这里 thrift 直接把 TConnection 的指针通过 socket 传递，有个队列的功能在里面了，稍花时间就没做)
因为前面 TNonblockingServer 复用 TConnection 和 TSocket 导致TSocket::getPeerAddress() 脏数据，需要修改 TSocket 源码，为了更好的发布，我把 TNonblockingServer 单独提到项目中，并直接禁用了复用 TConnection ，这样使用默认 thrift 库即可。
