看了网上很多资料,有Java,C++,Python版本的。我以为是类似的,找着做,发现根本行不通。
为取得客户端的IP,有三个办法(转自网上查到资料):
1) 网上博文http://blog.csdn.net/hbuxiaoshe/article/details/38942869介绍的方法也是可行的,不过让人有些纠结;
2) 修改Thrift的实现,为TServerEventHandler::createContext()增加一个参数,将TSocket作为参数传递,这样就可以非常轻易的取得客户端的IP了。最简单的修改为:
class TServerEventHandler {
public:
virtual void* createContext(boost::shared_ptr<TProtocol> input,
boost::shared_ptr<TProtocol> output,
TTransport* transport); // 对于TNonblockingServer实际传递为TSocket
};
3) 不修改Thrift的实现。
在“收发数据:执行调用”的流程中,可以发现有对TServerEventHandler::processContext()的调用,而这里真好将TSocket作为第二个参数进行了传递,因此可以直接利用。
TServerEventHandler::createContext()和TServerEventHandler::processContext()的不同在于:前者只在建立连接时被调用一次,而后者每一个RPC调用时都会调用一次。
自己各种方法试了很几天,就是没有成功得到客户端的IP,processContext()中得到的Host始终为null,port始终为0.
最后只能开始一步一步的调试。
终于惊喜地发现,这个null来的原因(我用的是TServerSocket)。
一步一步调试进到这里,TServerSocket的源码中:
进入TSocket的构造函数中,会发现并没有对host port进行赋值,这导致了后面得到的客户端的host始终是null.
增加注释部分的代码。重新编译dll。
server.setEventHandler(new ServerEventHelper());
可以在下处获取Host,客户端的ip:
class ServerEventHelper: TServerEventHandler
{
//创建Context的时候,触发
//在server启动后,只会执行一次
public object createContext(TProtocol input, TProtocol output)
{
Console.WriteLine("createContext");
return null;
}
//删除Context的时候,触发
//在server启动后,只会执行一次
public void deleteContext(object serverContext, TProtocol input, TProtocol output)
{
Console.WriteLine("deleteContext");
Console.WriteLine(((TSocket)input.Transport).Host);
//断开的时候 input.Transport.Peek()为false
if (!input.Transport.IsOpen)
{
Console.WriteLine("有客户端断开连接");
}
}
//服务成功启动后执行
public void preServe()
{
Console.WriteLine("preServe");
}
//调用RPC服务的时候触发
//每调用一次方法,就会触发一次
public void processContext(object serverContext, TTransport transport)
{
Console.WriteLine(((TSocket)transport).Host);
if (transport.IsOpen)
{
Console.WriteLine("open");
}
}
}