win32下boost::asio进一步封装

在网络通信中，我个人比较喜欢异步的方式。这样我程序就不会因为I/O的读写而导致线程阻塞。理想的工作方式是通知窗口的事件通知。windows中socket的底层其实是支持窗口事件通知的，但由于boost库比较强大，我就基于asio的库来实现这样的机制。

由于是异步方式，当事件处理完成后，我希望将结果传递给回调函数，因此类中有下面3个函数：

    virtual void handler_connect(const boost::system::error_code& error);    virtual void handler_send(const boost::system::error_code& error);    virtual void handler_receive(const boost::system::error_code& error);

参数传递过来的只是是否有错误。

在主线程中，我只需要调用

void connect();template<typename ConstBufferSequence>void send(const ConstBufferSequence& buffers);template<typename MutableBufferSequence>void receive(const MutableBufferSequence& buffers)

这三个函数即可，处理完成后会自动调用上面的回调函数，使用回调函数来处理结果。

 

asio中io_service的run会一直阻塞线程，所以需要将run在辅助线程中运行，但这样的话，回调函数就会在辅助线程中执行，为了保证线程安全性和消除MFC中不同线程执行后的代码异常，我需要将回调函数转入main线程中执行，这就应用了SendMessage函数了通知主线程窗口，下面详细的代码：

#define WM_ASIO_MESSAGE (WM_USER + 1)using boost::asio::ip::tcp;using boost::asio::deadline_timer;template <typename CWinWnd>class Win32TcpClient{    typedef Win32TcpClient<CWinWnd> MyClassName;    typedef void (MyClassName::*handler_ptr)(const boost::system::error_code&);public:    Win32TcpClient(boost::asio::io_service& ios,tcp::endpoint endpoint,CWinWnd* pWinWnd)        :io_service_(ios),win_wnd_(pWinWnd),socket_(ios),        endpoint_(endpoint),deadline_(ios),stop_(false),connect_timeout_(TIME_INF),handler_ptr_(NULL)    {        set_timeout(connect_timeout_);    }    enum {TIME_INF = -1};    void connect()    {        deadline_.async_wait(boost::bind(&MyClassName::check_deadline, this));        socket_.async_connect(endpoint_,            boost::bind(&MyClassName::run_main_thread_handler,this,&MyClassName::handler_connect,            boost::asio::placeholders::error));    }    void set_timeout(int timeout_seconds)    {        if (timeout_seconds == TIME_INF)            deadline_.expires_at(boost::posix_time::pos_infin);        else            deadline_.expires_from_now(boost::posix_time::seconds(timeout_seconds));    }    void close()    {        stop_ = true;        socket_.close();    }        template<typename ConstBufferSequence>    void send(const ConstBufferSequence& buffers)    {        boost::asio::async_write(socket_,buffers,            boost::bind(&MyClassName::run_main_thread_handler,this,&MyClassName::handler_send,            boost::asio::placeholders::error));    }    template<typename MutableBufferSequence>    void receive(const MutableBufferSequence& buffers)    {        boost::asio::async_read(socket_,buffers,            boost::bind(&MyClassName::run_main_thread_handler, this,&MyClassName::handler_receive,            boost::asio::placeholders::error));    }    void win_proc()    {        if(handler_ptr_ != NULL)            (this->*handler_ptr_)(error_code_);    }    virtual void handler_connect(const boost::system::error_code& error){}    virtual void handler_send(const boost::system::error_code& error){}    virtual void handler_receive(const boost::system::error_code& error){}private:        void run_main_thread_handler(handler_ptr handler, const boost::system::error_code& error)    {        handler_ptr_ = handler;        error_code_ = error;        ::SendMessage(win_wnd_->m_hWnd,WM_ASIO_MESSAGE,NULL,(LPARAM)this);    }    void check_deadline()    {        if(stop_)return;        if (deadline_.expires_at() <= deadline_timer::traits_type::now())        {            close();            deadline_.expires_at(boost::posix_time::pos_infin);        }        deadline_.async_wait(boost::bind(&MyClassName::check_deadline, this));    }protected:    CWinWnd* win_wnd_;    boost::asio::io_service& io_service_;    tcp::socket socket_;    tcp::endpoint endpoint_;    deadline_timer deadline_;    bool stop_;    std::size_t connect_timeout_;    handler_ptr handler_ptr_;    boost::system::error_code error_code_;    };

 

实际使用时，可以从上面的类中继承：

class CollectClient : public Win32TcpClient<CTestBoostTimerDlg>{public:    CollectClient(boost::asio::io_service& ios,tcp::endpoint endpoint,CTestBoostTimerDlg* pWinWnd):Win32TcpClient<CTestBoostTimerDlg>(ios,endpoint,pWinWnd){ memset(receive_buffer,0,100);}    virtual void handler_connect(const boost::system::error_code& error);    virtual void handler_send(const boost::system::error_code& error);    virtual void handler_receive(const boost::system::error_code& error);private:    char receive_buffer[200] ;};void CollectClient::handler_connect( const boost::system::error_code& error ){    char sendMessage[] = "0020abcdefghijklsdkjaslk";    if (!error)    {            win_wnd_->MessageBox("success");        receive(boost::asio::buffer(receive_buffer,30));        send(boost::asio::buffer(sendMessage,strlen(sendMessage)));    }    else    {            win_wnd_->MessageBox("fail");        if(!stop_)            close();    }}void CollectClient::handler_send( const boost::system::error_code& error ){    if(!error)    {        //win_wnd_->MessageBox("send success");    }    //close();}void CollectClient::handler_receive( const boost::system::error_code& error ){    if(!error)    {        win_wnd_->MessageBox(receive_buffer);        receive(boost::asio::buffer(receive_buffer,30));    }    else    {        win_wnd_->MessageBox(error.message().c_str());        }}

窗口类中需要添加处理代码是：

BEGIN_MESSAGE_MAP(CTestBoostTimerDlg, CDialogEx)    ON_WM_SYSCOMMAND()    ON_WM_PAINT()    ON_WM_QUERYDRAGICON()    ON_MESSAGE(WM_ASIO_MESSAGE,&CTestBoostTimerDlg::OnAsioProc)END_MESSAGE_MAP()

红色为自行添加的消息处理。

OnAsioProc的代码也是基本固定的。

声明：

afx_msg LRESULT OnAsioProc(WPARAM wParam,LPARAM lParam);

实现：

LRESULT CTestBoostTimerDlg::OnAsioProc( WPARAM wParam,LPARAM lParam ){    CollectClient* pc = (CollectClient*)lParam;    pc->win_proc();    return 0;}

在dialog的初始化OnInitDialog中

boost::asio::ip::tcp::endpoint endpoint(        boost::asio::ip::address::from_string("127.0.0.1"), 830);    pClient = new CollectClient(ios, endpoint,this);    pClient->connect();

 

这里的pClient和ios可以声明为全局变量，或者声明在窗口类中。

boost::asio::io_service ios;CollectClient* pClient;

 为了是异步工作起来，需要启动另一个线程！

boost::thread th(boost::bind(&io_service::run,&ios));th.detach();

ok，这样的client就可以应用了，关于网络的连接我们其实仅抽象了6个函数connect, handler_connect, send, handler_send, receive, handler_receive。

还有比较有用的是

　　set_timeout 这个函数设置超时的秒数。到这个时间后，socket会自动关闭，默认为inf，不会超时。

　　endpoint_ 该变量可查看连接的ip地址和端口号。

　　win_wnd_ 该变量为窗口类的指针，可以针对窗口做一系列的操作。