Thrift 获取客户端ip C#

看了网上很多资料,有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 input,

                                boost::shared_ptr output,

                                TTransport* transport); // 对于TNonblockingServer实际传递为TSocket

};

 

3) 不修改Thrift的实现。

在“收发数据:执行调用”的流程中,可以发现有对TServerEventHandler::processContext()的调用,而这里真好将TSocket作为第二个参数进行了传递,因此可以直接利用。

 

TServerEventHandler::createContext()和TServerEventHandler::processContext()的不同在于:前者只在建立连接时被调用一次,而后者每一个RPC调用时都会调用一次。

 

自己各种方法试了很几天,就是没有成功得到客户端的IP,processContext()中得到的Host始终为null,port始终为0.

最后只能开始一步一步的调试。

终于惊喜地发现,这个null来的原因(我用的是TServerSocket)。

一步一步调试进到这里,TServerSocket的源码中:

Thrift 获取客户端ip C#_第1张图片

 

进入TSocket的构造函数中,会发现并没有对host port进行赋值,这导致了后面得到的客户端的host始终是null.

Thrift 获取客户端ip C#_第2张图片

增加注释部分的代码。重新编译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");
            }
        }
    }

 

你可能感兴趣的:(c#,thrift)