1. 程式人生 > >boost async_read_some 用法

boost async_read_some 用法

boost async_read_some

async_read_some讀到數據就會直接回調設置的函數,不管數據是否已經讀完。所以在這裏

會遇到一個非常棘手的問題,如何確定數據已經讀取完畢?常見的方式是在數據的後面添加

標誌位,例如添加/r/n/r/n作為結束符,然後停止讀取

async_read_some的基本原理是往IOCP的隊列裏面添加一個異步任務,沒有事情的時候,CSession::ContinueRead

不應該被調用


class CSession : public boost::enable_shared_from_this<CSession>

{

public:

CSession(boost::asio::io_service &io_service) : m_socket(io_service)

{

memset(m_szRecvBuffer, 0x00, 1024);

m_bStartRecv = false;

}


void Start()

{

static boost::asio::ip::tcp::no_delay option(true);

m_socket.set_option(option);

boost::function0<void> f = boost::bind(&CSession::StartThread, this);

boost::thread thrd(f);

}

/*

啟動線程函數的根本原因是需要向客戶端推送消息,而且在過程中需要等待接收消息

*/

void StartThread()

{

while (true)

{

/*

使用m_bStartRecv標誌位主要是為了避免多次設置回調,當正在接收的時候,不需要設置回調

*/

if (!m_bStartRecv)

{

m_bStartRecv = true;

m_socket.async_read_some(boost::asio::buffer(m_szRecvBuffer),

boost::bind(&CSession::ContinueRead, shared_from_this(),

boost::asio::placeholders::error,

boost::asio::placeholders::bytes_transferred));

}

char szAlarm[32] = "alarm";

boost::system::error_code ec;

m_socket.send(boost::asio::buffer(szAlarm), 0, ec);

boost::this_thread::sleep_for(boost::chrono::milliseconds(3000));

if (ec) break;

}

}


private:


/*

在當前接收回調函數中,我們還繼續設置了回調函數m_socket.async_read_some,避免僅僅是讀取數據包的一部分,

在這裏m_szRecvBuffer會一直作為接收的緩沖,而之前接收的數據也在裏面,並且剩下的數據,會根據偏移量,填充到

m_szRecvBuffer的後面,bytes_transferred參數代表當前已經接收的數據

*/

void ContinueRead(const boost::system::error_code &error, std::size_t bytes_transferred)

{

if (error) return;


m_strMatch = m_strMatch + m_szRecvBuffer;

int index = m_strMatch.find("\r\n\r\n", 0);

if (-1 != index)

{

int ret = m_socket.send(boost::asio::buffer(m_szRecvBuffer));

std::cout << m_szRecvBuffer << std::endl;

m_bStartRecv = false;

return;

}

m_socket.async_read_some(boost::asio::buffer((m_szRecvBuffer)),

boost::bind(&CSession::ContinueRead, shared_from_this(),

boost::asio::placeholders::error,

boost::asio::placeholders::bytes_transferred));

}


private:


boost::asio::ip::tcp::socket m_socket;

char m_szRecvBuffer[1024];

std::string m_strMatch;

bool m_bStartRecv;

};


註意

當前的僅僅是一個例子,並沒有考慮到客戶端關閉連接,服務器線程及時退出


boost async_read_some 用法