可用於一些需要tcp連接的場景,代碼簡潔,不易出錯,經測試可用
#include <iostream>
#include <memory>
#include <array>
#include <boost/asio.hpp>
class session : public std::enable_shared_from_this<session>
{
public:
session(boost::asio::ip::tcp::socket s): socket_(std::move(s)){}
void async_read()
{
auto self(shared_from_this());
#if 0
// 讀操作完成時回調該函數, 讀取到固定長度纔會觸發回調
boost::asio::async_read(socket_, boost::asio::buffer(data_), [this, self](const boost::system::error_code &ec, size_t bytes_transferred) {
if (!ec)
{
std::cout << data_.data() << std::endl;
async_read();
}
});
#else
// 讀操作完成時回調該函數, 讀取到一些數據就會觸發回調
socket_.async_read_some(boost::asio::buffer(data_), [this, self](const boost::system::error_code &ec, size_t bytes) {
if (!ec)
{
std::cout << data_.data() << std::endl;
async_read();
}
});
#endif
}
void async_write(std::size_t length)
{
auto self(shared_from_this());
boost::asio::async_write(socket_, boost::asio::buffer(data_, length), [this, self](const boost::system::error_code &ec, size_t writed_bytes)
{
if (!ec)
{
// async_read();
}
});
}
private:
boost::asio::ip::tcp::socket socket_;
std::array<char, 10> data_;
};
class server
{
public:
server(boost::asio::io_service &io_service, short port):
acceptor_(io_service, boost::asio::ip::tcp::endpoint(boost::asio::ip::tcp::v4(), port)),
socket_(io_service){
async_accept();
}
private:
void async_accept()
{
acceptor_.async_accept(socket_, std::bind(&server::handle_accept, this, std::placeholders::_1));
}
void handle_accept(const boost::system::error_code &ec)
{
if (!ec)
{
session_ptr = std::shared_ptr<session>(new session(std::move(socket_)));
session_ptr->async_read();
}
async_accept();// 繼續監聽客戶端連接請求
}
boost::asio::ip::tcp::acceptor acceptor_;
boost::asio::ip::tcp::socket socket_;
std::shared_ptr<session> session_ptr;
};
int main(int argc, char* argv[])
{
boost::asio::io_service io_service;
boost::asio::io_service::work work_(io_service);
server s(io_service, 52014);
io_service.run();
return 0;
}