thrift通过TServerEventHandler获取客户端ip

标签：style   blog   http   color   os   使用   io   strong   ar   

用了这么久的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();
  }

至此，我们可以在XServiceHandler中调用GetThriftClientIp()来获得客户端的ip了，终于不再烦恼数据是从哪里发来的了。

thrift通过TServerEventHandler获取客户端ip

标签：style   blog   http   color   os   使用   io   strong   ar   

原文地址：http://blog.csdn.net/hbuxiaoshe/article/details/38942869