C++ Thrift服务端记录调用者IP和被调接口方法

9 篇文章 0 订阅

 

Apache开源的Thrift(http://thrift.apache.org)有着广泛的使用,有时候需要知道谁调用了指定的函数,比如在下线一起老的接口之前,需要确保对这些老接口的访问已全部迁移到新口。Thrift提供了支持,在《Thrift结构分析及增加取客户端IP功能实现》一文中已做过介绍,但不够具体。

本文对这个做一个详细的介绍,过程中使用到了开源的C++ Thrift服务端的辅助类CThriftServerHelper(对应的客户端辅助类为CThriftClientHelper),源代码网址为:

https://github.com/eyjian/libmooon/blob/master/include/mooon/net/thrift_helper.h

 

为达到目标,需要提供一个Context结构体和两个回调接口实现类。

1) Contex结构体ThriftServerContext

结构体的内容完成自定义,这里定义一个peer成员用来保存客户端的IP和端口号,根据实际需要也可分成两个字段。

struct ThriftServerContext

{

    std::string peer; // 客户端的IP和端口号,格式为标准的“IP:PORT”

};

 

2) ServerEvent回调接口实现类MyServerEventHandler

class MyServerEventHandler: public apache::thrift::server::TServerEventHandler

{

private:

    virtual void* createContext(

            boost::shared_ptr<apache::thrift::server::TProtocol> input,

            boost::shared_ptr<apache::thrift::server::TProtocol> output);

    virtual void deleteContext(

            void* serverContext,

            boost::shared_ptr<apache::thrift::server::TProtocol>input,

            boost::shared_ptr<apache::thrift::server::TProtocol>output);

    virtual void processContext(void* serverContext, boost::shared_ptr<apache::thrift::server::TTransport> transport);

};

 

3) ProcessorEvent回调接口实现类MyProcessorEventHandler

class MyProcessorEventHandler: public apache::thrift::TProcessorEventHandler

{

private:

    virtual void* getContext(const char* fn_name, void* serverContext);

};

 

4) 相关的实现

void* MyServerEventHandler::createContext(

        boost::shared_ptr<apache::thrift::server::TProtocol> input,

        boost::shared_ptr<apache::thrift::server::TProtocol> output)

{

    // 以下针对TNonblockingServer

    // in_transport和out_transport实际为apache::thrift::server::TMemoryBuffer

    //boost::shared_ptr<apache::thrift::server::TTransport> in_transport = input->getTransport();

    //boost::shared_ptr<apache::thrift::server::TTransport> output_transport = output->getTransport();

    return new ThriftServerContext; // 创建Context,Context每一个客户端连接是一对一的关系

}

 

void MyServerEventHandler::deleteContext(

        void* serverContext,

        boost::shared_ptr<apache::thrift::server::TProtocol>input,

        boost::shared_ptr<apache::thrift::server::TProtocol>output)

{

    delete (ThriftServerContext*)serverContext; // 释放Context,否则内存泄漏,连接被关闭时调用

}

 

void MyServerEventHandler::processContext(

        void* serverContext,

        boost::shared_ptr<apache::thrift::server::TTransport> transport)

{

#if 1

    // 如果是TNonblockingServer,则TTransport::getOrigin返回的是IP地址

    //MYLOG_DEBUG("Called from %s\n", transport->getOrigin().c_str());

 

    ThriftServerContext* ctx = (ThriftServerContext*)serverContext;

    // 保存客户端的IP和端口号,以便MyProcessorEventHandler::getContext中可用

    ctx->peer = transport->getOrigin();

#else

    // 以下针对TNonblockingServer有效

    apache::thrift::server::TSocket* socket = dynamic_cast<apache::thrift::server::TSocket*>(transport.get());

    if (socket != NULL)

    {

        // TSocket::getPeerAddress返回的是IP地址,

        // 如果调用TSocket::getPeerHost(),则返回的可能是IP对应的hostname

        MYLOG_DEBUG("Called from %s:%d\n", socket->getPeerAddress().c_str(), socket->getPeerPort());

    }

#endif

}

 

// 参数fn_name为被调用接口名

// serverContext承载了客户端的IP和端口号数据

//

// 在getContext中,还可为每个调用创建自己的Context,但注意区别Server的Context

void* MyProcessorEventHandler::getContext(const char* fn_name, void* serverContext)

{

    ThriftServerContext* ctx = (ThriftServerContext*)serverContext;

    MYLOG_INFO("%s called by %s\n", fn_name, ctx->peer.c_str());

    return NULL; // 如果为本次调用创建Context,则需要实现freeContext以释放Context

}

 

5) 应用示例

class CMyServer

{

public:

  CMyServer();

  void start();

 

private:

  boost::shared_ptr<MyServerEventHandler> _server_event_handler;

  boost::shared_ptr<MyProcessorEventHandler> _processor_event_handler;

  mooon::net::CThriftServerHelper<CMyHandler, MyServiceProcessor> _thrift_server;

};

 

CMyServer::CMyServer()

  : _server_event_handler(new MyServerEventHandler),

    _processor_event_handler(new MyProcessorEventHandler),

    _thrift_server(_server_event_handler, _processor_event_handler)

{

}

 

void CMyServer::start()

{

  // 启动Thrift服务,

  // 注意调用线程在这里会阻塞,

  // 直到调用_thrift_server.stop()停止Thrift服务。

  _thrift_server.serve(

    mooon::argument::port->value(), // 监听端口号

    mooon::argument::wkthreads->value(), // 工作线程数

    mooon::argument::iothreads->value()); // IO线程数

}

 

有关细节请参见《Thrift结构分析及增加取客户端IP功能实现》,以及编译thrift文件后生成的Service.cpp文件:

https://blog.csdn.net/Aquester/article/details/48261609

 

查看回调接口TProcessorEventHandler和TServerEventHandler可了解更多的使用。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
TCP,UDP服务端口,HMODULE hIpDLL = LoadLibrary( "iphlpapi.dll"); if ( !hIpDLL) return; PMIB_TCPTABLE_OWNER_PID pTcpTable(NULL); DWORD dwSize(0); PGet_Extended_TcpTable pGetExtendedTcpTable = NULL; pGetExtendedTcpTable = (PGet_Extended_TcpTable) GetProcAddress(hIpDLL, "GetExtendedTcpTable"); if(pGetExtendedTcpTable==NULL) { FreeLibrary(hIpDLL); return; } if(pGetExtendedTcpTable(pTcpTable, &dwSize, TRUE, AF_INET, TCP_TABLE_OWNER_PID_ALL, 0) == ERROR_INSUFFICIENT_BUFFER) pTcpTable = (MIB_TCPTABLE_OWNER_PID *)new char[dwSize];//重新分配缓冲区 if (pGetExtendedTcpTable(pTcpTable, &dwSize, TRUE, AF_INET, TCP_TABLE_OWNER_PID_ALL, 0) != NO_ERROR) { delete pTcpTable; return; } CString sTemp; m_PortList.DeleteAllItems(); int nNum = (int)pTcpTable->dwNumEntries; //TCP连接的数目 // nNum = 0; for (int i = 0; itable[i].dwLocalAddr))); sTemp.Format("%u",htons((WORD)pTcpTable->table[i].dwLocalPort)); m_PortList.SetItemText(i,4,sTemp); m_PortList.SetItemBgColor(i,4,RGB(210,0,0)); m_PortList.SetItemText(i,5,FormatNumToIpv4(htonl(pTcpTable->table[i].dwRemoteAddr))); sTemp.Format("%u",htons((WORD) pTcpTable->table[i].dwRemotePort)); m_PortList.SetItemText(i,6,sTemp); sTemp.Format("%u",pTcpTable->table[i].dwOwningPid); m_PortList.SetItemText(i,8,sTemp); //ProcessPidToName(pTcpTable->table[i].dwOwningPid,i); ProcessPidToNameX(pTcpTable->table[i].dwOwningPid,i); switch (pTcpTable->table[i].dwState) { case MIB_TCP_STATE_CLOSED: m_PortList.SetItemText(i,2,"CLOSED"); break; case MIB_TCP_STATE_LISTEN: m_PortList.SetItemText(i,2,"LISTEN"); break; case MIB_TCP_STATE_SYN_SENT: m_PortList.SetItemText(i,2,"SYN

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值