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;
};
注意
当前的仅仅是一个例子,并没有考虑到客户端关闭连接,服务器线程及时退出