boost async_read_some 用法

Posted 2020-10-01

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;

};

注意

当前的仅仅是一个例子，并没有考虑到客户端关闭连接，服务器线程及时退出