boost asio簡易tcp服務器

 可用於一些需要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;
}

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章