首页 > 代码库 > thrift通过TServerEventHandler获取客户端ip
thrift通过TServerEventHandler获取客户端ip
用了这么久的thrift,一直不知道如何在接收端打印发送端的ip。一个笨的方法是在struct中添加字段并填充发送端的ip(或host),约定必须填充有效的ip,但这样不但增加了调用者的代码量,而且接收端还需要判断一个string字段是否为有效ip。
前段时间在网上查了一些资料,但无果。今天重新整理了这些资料,终于搞定了。
大致思路:http://web.archiveorange.com/archive/v/FzEy581cbgUiBLQh1GPv上讲的很清楚。我这里根据我的想法重复一下:
主要实现一个基类为TServerEventHandler的class,我叫它为ClientIPHandler。
1,定义一个map<uint64, string>,存放<pthread_self(), ip>。
2,在ClientIPHandler中createContext()里实现map的插入。
3,定义一个全局函数,获取map的值。这个函数供你的XServiceHandler(thrift自动生成的)使用。
4,在调用serve()前,定义shared_ptr<ClientIPHandler>变量a,调用setServerEventHandler设置eventHandler_(server/TServer.h:217).
map的读写需要加锁。
部分代码:
static map<uint64, std::string> thrift_client_ip; static base::lock::Mutex mutex; std::string GetThriftClientIp() { lock::MutexLock g(&mutex); return thrift_client_ip[pthread_self()]; } 。。。 class ClientIPHandler : virtual public TServerEventHandler { public: ClientIPHandler() { } virtual ~ClientIPHandler() { } virtual void preServe() { } virtual void* createContext(boost::shared_ptr<TProtocol> input, boost::shared_ptr<TProtocol> output) { TBufferedTransport *tbuf = dynamic_cast<TBufferedTransport *>(input->getTransport().get()); TSocket *sock = dynamic_cast<TSocket *>(tbuf->getUnderlyingTransport().get()); lock::MutexLock g(&mutex); thrift_client_ip[pthread_self()] = sock->getPeerAddress(); return NULL; } virtual void deleteContext(void* serverContext, boost::shared_ptr<TProtocol>input, boost::shared_ptr<TProtocol>output) { } virtual void processContext(void* serverContext, boost::shared_ptr<TTransport> transport) { } private: }; 。。。 TThreadedServer server(processor, serverTransport, transportFactory, protocolFactory); try { boost::shared_ptr<ClientIPHandler> ip_handler(new ClientIPHandler()); server.setServerEventHandler(ip_handler); LOG(INFO)<< "thrift server start, listen port:" << listen_port; server.serve(); } catch (apache::thrift::transport::TTransportException &ex) { LOG(FATAL) << ex.what(); }TServerEventHandler和TBufferedTransport需要相应的命名空间。
至此,我们可以在XServiceHandler中调用GetThriftClientIp()来获得客户端的ip了,终于不再烦恼数据是从哪里发来的了。
thrift通过TServerEventHandler获取客户端ip
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。