boost asio 异步实现tcp通讯

最新推荐文章于 2024-05-23 00:22:01 发布
最新推荐文章于 2024-05-23 00:22:01 发布
阅读量7.5k 收藏 8
点赞数 1
分类专栏： boost

转自 http://blog.csdn.net/fhxpp_27/article/details/8569013

一、前言

boost asio可算是一个简单易用，功能又强大可跨平台的C++通讯库，效率也表现的不错，linux环境是epoll实现的，而windows环境是iocp实现的。而tcp通讯是项目当中经常用到通讯方式之一，实现的方法有各式各样，因此总结一套适用于自己项目的方法是很有必要，很可能下一个项目直接套上去就可以用了。


二、实现思路

1.通讯包数据结构


Tag：检查数据包是否合法，具体会在下面讲解；

Length：描述Body的长度；

Command：表示数据包的类型，0表示心跳包（长连接需要心跳来检测连接是否正常），1表示注册包（客户端连接上服务器之后要将相关信息注册给服务器），2表示业务消息包；

business_type：业务消息包类型，服务器会根据业务消息包类型将数据路由到对应的客户端（客户端是有业务类型分类的）；

app_id：客户端唯一标识符；

Data：消息数据；

2.连接对象

客户端连接上服务器之后，双方都会产生一个socket连接对象，通过这个对象可以收发数据，因此我定义为socket_session。

//socket_session.h

[cpp] view plain copy print ?
  1. #pragma once   
  2. #include <iostream>   
  3. #include <list>   
  4. #include <hash_map>   
  5. #include <boost/bind.hpp>   
  6. #include <boost/asio.hpp>   
  7. #include <boost/shared_ptr.hpp>   
  8. #include <boost/make_shared.hpp>   
  9. #include <boost/thread.hpp>   
  10. #include <boost/thread/mutex.hpp>   
  11. #include <boost/enable_shared_from_this.hpp>   
  12. #include <firebird/log/logger_log4.hpp>   
  13. #include <firebird/detail/config.hpp>   
  14. #include <firebird/socket_utils/message_archive.hpp>   
  15.   
  16. using boost::asio::ip::tcp;  
  17.   
  18. namespace firebird{  
  19.     enum command{ heartbeat = 0, regist, normal};  
  20.   
  21.     const std::string tag = "KDS";  
  22.   
  23.     class FIREBIRD_DECL socket_session;  
  24.     typedef boost::shared_ptr<socket_session> socket_session_ptr;  
  25.   
  26.     class FIREBIRD_DECL socket_session:  
  27.         public boost::enable_shared_from_this<socket_session>,  
  28.         private boost::noncopyable  
  29.     {  
  30.     public:  
  31.         typedef boost::function<void(socket_session_ptr)> close_callback;  
  32.         typedef boost::function<void(  
  33.             const boost::system::error_code&,   
  34.             socket_session_ptr, message&)> read_data_callback;  
  35.   
  36.         socket_session(boost::asio::io_service& io_service);  
  37.         ~socket_session(void);  
  38.   
  39.         DWORD id() { return m_id; }  
  40.         WORD get_business_type(){ return m_business_type; }  
  41.         void set_business_type(WORD type) { m_business_type = type; }  
  42.         DWORD get_app_id(){ return m_app_id; }  
  43.         void set_app_id(DWORD app_id) { m_app_id = app_id; }  
  44.         std::string& get_remote_addr() { return m_name; }  
  45.         void set_remote_addr(std::string& name) { m_name = name; }  
  46.         tcp::socket& socket() { return m_socket; }  
  47.   
  48.         void installCloseCallBack(close_callback cb){ close_cb = cb; }  
  49.         void installReadDataCallBack(read_data_callback cb) { read_data_cb = cb; }  
  50.   
  51.         void start();  
  52.         void close();  
  53.         void async_write(const std::string& sMsg);  
  54.         void async_write(message& msg);  
  55.   
  56.         bool is_timeout();  
  57.         void set_op_time(){std::time(&m_last_op_time);}  
  58.   
  59.     private:  
  60.         static boost::detail::atomic_count m_last_id;  
  61.   
  62.         DWORD m_id;  
  63.         WORD  m_business_type;  
  64.         DWORD m_app_id;  
  65.         std::string m_name;  
  66.         boost::array<char, 7> sHeader;  
  67.         std::string sBody;  
  68.   
  69.         tcp::socket m_socket;  
  70.         boost::asio::io_service& m_io_service;  
  71.   
  72.         std::time_t m_last_op_time;  
  73.   
  74.         close_callback close_cb;  
  75.         read_data_callback read_data_cb;  
  76.   
  77.         //发送消息   
  78.         void handle_write(const boost::system::error_code& e,   
  79.             std::size_t bytes_transferred, std::string* pmsg);  
  80.   
  81.         //读消息头   
  82.         void handle_read_header(const boost::system::error_code& error);  
  83.         //读消息体   
  84.         void handle_read_body(const boost::system::error_code& error);  
  85.   
  86.         void handle_close();  
  87.     };  
  88. }  
#pragma once
#include <iostream>
#include <list>
#include <hash_map>
#include <boost/bind.hpp>
#include <boost/asio.hpp>
#include <boost/shared_ptr.hpp>
#include <boost/make_shared.hpp>
#include <boost/thread.hpp>
#include <boost/thread/mutex.hpp>
#include <boost/enable_shared_from_this.hpp>
#include <firebird/log/logger_log4.hpp>
#include <firebird/detail/config.hpp>
#include <firebird/socket_utils/message_archive.hpp>

using boost::asio::ip::tcp;

namespace firebird{
	enum command{ heartbeat = 0, regist, normal};

	const std::string tag = "KDS";

	class FIREBIRD_DECL socket_session;
	typedef boost::shared_ptr<socket_session> socket_session_ptr;

	class FIREBIRD_DECL socket_session:
		public boost::enable_shared_from_this<socket_session>,
		private boost::noncopyable
	{
	public:
		typedef boost::function<void(socket_session_ptr)> close_callback;
		typedef boost::function<void(
			const boost::system::error_code&, 
			socket_session_ptr, message&)> read_data_callback;

		socket_session(boost::asio::io_service& io_service);
		~socket_session(void);

		DWORD id() { return m_id; }
		WORD get_business_type(){ return m_business_type; }
		void set_business_type(WORD type) { m_business_type = type; }
		DWORD get_app_id(){ return m_app_id; }
		void set_app_id(DWORD app_id) { m_app_id = app_id; }
		std::string& get_remote_addr() { return m_name; }
		void set_remote_addr(std::string& name) { m_name = name; }
		tcp::socket& socket() { return m_socket; }

		void installCloseCallBack(close_callback cb){ close_cb = cb; }
		void installReadDataCallBack(read_data_callback cb) { read_data_cb = cb; }

		void start();
		void close();
		void async_write(const std::string& sMsg);
		void async_write(message& msg);

		bool is_timeout();
		void set_op_time(){std::time(&m_last_op_time);}

	private:
		static boost::detail::atomic_count m_last_id;

		DWORD m_id;
		WORD  m_business_type;
		DWORD m_app_id;
		std::string m_name;
		boost::array<char, 7> sHeader;
		std::string sBody;

		tcp::socket m_socket;
		boost::asio::io_service& m_io_service;

		std::time_t m_last_op_time;

		close_callback close_cb;
		read_data_callback read_data_cb;

		//发送消息
		void handle_write(const boost::system::error_code& e, 
			std::size_t bytes_transferred, std::string* pmsg);

		//读消息头
		void handle_read_header(const boost::system::error_code& error);
		//读消息体
		void handle_read_body(const boost::system::error_code& error);

		void handle_close();
	};
}


这里注意的是，定义了一个tag="KDS"，目的是为了检查收到的数据包是否有效，每一个数据包前3个字节不为“KDS”，那么就认为是非法的请求包，你也可以定义tag等于其它字符串，只要按协议发包就正常，当然这是比较简单的数据包检查方法了。比较严谨的方法是双方使用哈希算法来检查的，怎么做，这里先不做详解。

//socket_session.cpp

[cpp] view plain copy print ?
  1. #include "socket_session.h"   
  2.   
  3. namespace firebird{  
  4.     boost::detail::atomic_count socket_session::m_last_id(0);  
  5.   
  6.     socket_session::socket_session(boost::asio::io_service& io_srv)  
  7.         :m_io_service(io_srv), m_socket(io_srv),   
  8.         m_business_type(0), m_app_id(0)  
  9.     {  
  10.         m_id = ++socket_session::m_last_id;  
  11.     }  
  12.   
  13.     socket_session::~socket_session(void)  
  14.     {  
  15.         m_socket.close();  
  16.     }  
  17.   
  18.     void socket_session::start()  
  19.     {  
  20.         m_socket.set_option(boost::asio::ip::tcp::acceptor::linger(true, 0));  
  21.         m_socket.set_option(boost::asio::socket_base::keep_alive(true));  
  22.         std::time(&m_last_op_time);  
  23.         const boost::system::error_code error;  
  24.         handle_read_header(error);  
  25.     }  
  26.   
  27.     void socket_session::handle_close()  
  28.     {  
  29.         try{  
  30.             m_socket.close();  
  31.             close_cb(shared_from_this());  
  32.         }  
  33.         catch(std::exception& e)  
  34.         {  
  35.             LOG4CXX_ERROR(firebird_log, KDS_CODE_INFO << "连接远程地址:[" << get_remote_addr() << "],socket异常:[" << e.what() << "]");  
  36.         }  
  37.         catch(...)  
  38.         {  
  39.             LOG4CXX_ERROR(firebird_log, KDS_CODE_INFO << "连接远程地址:[" << get_remote_addr() << "],socket异常:[未知异常]");  
  40.         }  
  41.     }  
  42.   
  43.     void socket_session::close()  
  44.     {             
  45.         //由于回调中有加锁的情况，必须提交到另外一个线程去做，不然会出现死锁   
  46.         m_io_service.post(boost::bind(&socket_session::handle_close, shared_from_this()));  
  47.     }  
  48.   
  49.     static int connection_timeout = 60;  
  50.   
  51.     bool socket_session::is_timeout()  
  52.     {  
  53.         std::time_t now;  
  54.         std::time(&now);      
  55.         return now - m_last_op_time > connection_timeout;  
  56.     }  
  57.   
  58.     //读消息头   
  59.     void socket_session::handle_read_header(const boost::system::error_code& error)  
  60.     {  
  61.         LOG4CXX_DEBUG(firebird_log, KDS_CODE_INFO  << "enter.");  
  62.   
  63.         try{  
  64.             if(error)  
  65.             {  
  66.                 LOG4CXX_ERROR(firebird_log, KDS_CODE_INFO  << "连接远程地址:[" << get_remote_addr() << "],socket异常:[" << error.message().c_str() << "]");  
  67.                 close();  
  68.                 return;  
  69.             }  
  70.   
  71.             std::string data;  
  72.             data.swap(sBody);  
  73.             boost::asio::async_read(m_socket,   
  74.                 boost::asio::buffer(sHeader),  
  75.                 boost::bind(&socket_session::handle_read_body, shared_from_this(),  
  76.                 boost::asio::placeholders::error));  
  77.   
  78.             if (data.length() > 0 && data != "")  
  79.             {//读到数据回调注册的READ_DATA函数   
  80.                 message msg;  
  81.                 message_iarchive(msg, data);  
  82.   
  83.                 read_data_cb(error, shared_from_this(), msg);  
  84.             }  
  85.         }  
  86.         catch(std::exception& e)  
  87.         {  
  88.             LOG4CXX_ERROR(firebird_log, KDS_CODE_INFO << "连接远程地址:[" << get_remote_addr() << "],socket异常:[" << e.what() << "]");  
  89.             close();  
  90.         }  
  91.         catch(...)  
  92.         {  
  93.             LOG4CXX_ERROR(firebird_log, KDS_CODE_INFO << "连接远程地址:[" << get_remote_addr() << "],socket异常:[未知异常]");  
  94.             close();  
  95.         }  
  96.     }  
  97.   
  98.     //读消息体   
  99.     void socket_session::handle_read_body(const boost::system::error_code& error)  
  100.     {  
  101.         LOG4CXX_DEBUG(firebird_log, KDS_CODE_INFO << "enter.");  
  102.   
  103.         try{  
  104.             if(error)  
  105.             {  
  106.                 LOG4CXX_ERROR(firebird_log, KDS_CODE_INFO << "连接远程地址:[" << get_remote_addr() << "],socket异常:[" << error.message().c_str() << "]");  
  107.                 close();  
  108.                 return;  
  109.             }  
  110.   
  111.             if (tag.compare(0, tag.length(), sHeader.data(), 0, tag.length()))  
  112.             {  
  113.                 LOG4CXX_ERROR(firebird_log, KDS_CODE_INFO <<  "连接远程地址:[" << get_remote_addr() << "],socket异常:[这是个非法连接!]");  
  114.                 close();  
  115.                 return;  
  116.             }  
  117.   
  118.             DWORD dwLength = 0;  
  119.   
  120.             char* len = (char*)&dwLength;  
  121.             memcpy(len, &sHeader[tag.length()], sizeof(dwLength));  
  122.   
  123.             sBody.resize(dwLength);  
  124.             char* pBody = &sBody[0];  
  125.   
  126.             boost::asio::async_read(m_socket,   
  127.                 boost::asio::buffer(pBody, dwLength),  
  128.                 boost::bind(&socket_session::handle_read_header, shared_from_this(),  
  129.                 boost::asio::placeholders::error));  
  130.         }  
  131.         catch(std::exception& e)  
  132.         {  
  133.             LOG4CXX_ERROR(firebird_log, KDS_CODE_INFO << "连接远程地址:[" << get_remote_addr() << "],socket异常:[" << e.what() << "]");  
  134.             close();  
  135.         }  
  136.         catch(...)  
  137.         {  
  138.             LOG4CXX_ERROR(firebird_log, KDS_CODE_INFO << "连接远程地址:[" << get_remote_addr() << "],socket异常:[未知异常]");  
  139.             close();  
  140.         }  
  141.     }  
  142.   
  143.     void socket_session::handle_write(const boost::system::error_code& error,   
  144.         std::size_t bytes_transferred, std::string* pmsg)  
  145.     {  
  146.         //数据发送成功就销毁   
  147.         if (pmsg != NULL)  
  148.         {  
  149.             delete pmsg;  
  150.         }  
  151.   
  152.         if(error)  
  153.         {  
  154.             LOG4CXX_ERROR(firebird_log, KDS_CODE_INFO << "连接远程地址:[" << get_remote_addr() << "],socket异常:[" << error.message().c_str() << "]");  
  155.             close();  
  156.             return;  
  157.         }  
  158.     }  
  159.   
  160.     void socket_session::async_write(const std::string& sMsg)  
  161.     {  
  162.         LOG4CXX_DEBUG(firebird_log, KDS_CODE_INFO  << "enter.")  
  163.   
  164.         try  
  165.         {  
  166.             DWORD dwLength = sMsg.size();  
  167.             char* pLen = (char*)&dwLength;  
  168.   
  169.             //由于是异步发送，要保证数据发送完整时，才把数据销毁   
  170.             std::string* msg = new std::string();  
  171.             msg->append(tag);  
  172.             msg->append(pLen, sizeof(dwLength));  
  173.             msg->append(sMsg);  
  174.   
  175.             boost::asio::async_write(m_socket,boost::asio::buffer(*msg, msg->size()),   
  176.                 boost::bind(&socket_session::handle_write, shared_from_this(),  
  177.                 boost::asio::placeholders::error, boost::asio::placeholders::bytes_transferred,  
  178.                 msg));  
  179.   
  180.         }  
  181.         catch(std::exception& e)  
  182.         {  
  183.             LOG4CXX_ERROR(firebird_log, KDS_CODE_INFO << "连接远程地址:[" << get_remote_addr() << "],socket异常:[" << e.what() << "]");  
  184.             close();  
  185.         }  
  186.         catch(...)  
  187.         {  
  188.             LOG4CXX_ERROR(firebird_log, KDS_CODE_INFO << "连接远程地址:[" << get_remote_addr() << "],socket异常:[未知异常]");  
  189.             close();  
  190.         }  
  191.     }  
  192.   
  193.     void socket_session::async_write(message& msg)  
  194.     {  
  195.         std::string data;  
  196.         message_oarchive(data, msg);  
  197.           
  198.         async_write(data);  
  199.     }  
  200. }  
#include "socket_session.h"

namespace firebird{
	boost::detail::atomic_count socket_session::m_last_id(0);

	socket_session::socket_session(boost::asio::io_service& io_srv)
		:m_io_service(io_srv), m_socket(io_srv), 
		m_business_type(0), m_app_id(0)
	{
		m_id = ++socket_session::m_last_id;
	}

	socket_session::~socket_session(void)
	{
		m_socket.close();
	}

	void socket_session::start()
	{
		m_socket.set_option(boost::asio::ip::tcp::acceptor::linger(true, 0));
		m_socket.set_option(boost::asio::socket_base::keep_alive(true));
		std::time(&m_last_op_time);
		const boost::system::error_code error;
		handle_read_header(error);
	}

	void socket_session::handle_close()
	{
		try{
			m_socket.close();
			close_cb(shared_from_this());
		}
		catch(std::exception& e)
		{
			LOG4CXX_ERROR(firebird_log, KDS_CODE_INFO << "连接远程地址:[" << get_remote_addr() << "],socket异常:[" << e.what() << "]");
		}
		catch(...)
		{
			LOG4CXX_ERROR(firebird_log, KDS_CODE_INFO << "连接远程地址:[" << get_remote_addr() << "],socket异常:[未知异常]");
		}
	}

	void socket_session::close()
	{			
		//由于回调中有加锁的情况，必须提交到另外一个线程去做，不然会出现死锁
		m_io_service.post(boost::bind(&socket_session::handle_close, shared_from_this()));
	}

	static int connection_timeout = 60;

	bool socket_session::is_timeout()
	{
		std::time_t now;
		std::time(&now);	
		return now - m_last_op_time > connection_timeout;
	}

	//读消息头
	void socket_session::handle_read_header(const boost::system::error_code& error)
	{
		LOG4CXX_DEBUG(firebird_log, KDS_CODE_INFO  << "enter.");

		try{
			if(error)
			{
				LOG4CXX_ERROR(firebird_log, KDS_CODE_INFO  << "连接远程地址:[" << get_remote_addr() << "],socket异常:[" << error.message().c_str() << "]");
				close();
				return;
			}

			std::string data;
			data.swap(sBody);
			boost::asio::async_read(m_socket, 
				boost::asio::buffer(sHeader),
				boost::bind(&socket_session::handle_read_body, shared_from_this(),
				boost::asio::placeholders::error));

			if (data.length() > 0 && data != "")
			{//读到数据回调注册的READ_DATA函数
				message msg;
				message_iarchive(msg, data);

				read_data_cb(error, shared_from_this(), msg);
			}
		}
		catch(std::exception& e)
		{
			LOG4CXX_ERROR(firebird_log, KDS_CODE_INFO << "连接远程地址:[" << get_remote_addr() << "],socket异常:[" << e.what() << "]");
			close();
		}
		catch(...)
		{
			LOG4CXX_ERROR(firebird_log, KDS_CODE_INFO << "连接远程地址:[" << get_remote_addr() << "],socket异常:[未知异常]");
			close();
		}
	}

	//读消息体
	void socket_session::handle_read_body(const boost::system::error_code& error)
	{
		LOG4CXX_DEBUG(firebird_log, KDS_CODE_INFO << "enter.");

		try{
			if(error)
			{
				LOG4CXX_ERROR(firebird_log, KDS_CODE_INFO << "连接远程地址:[" << get_remote_addr() << "],socket异常:[" << error.message().c_str() << "]");
				close();
				return;
			}

			if (tag.compare(0, tag.length(), sHeader.data(), 0, tag.length()))
			{
				LOG4CXX_ERROR(firebird_log, KDS_CODE_INFO <<  "连接远程地址:[" << get_remote_addr() << "],socket异常:[这是个非法连接!]");
				close();
				return;
			}

			DWORD dwLength = 0;

			char* len = (char*)&dwLength;
			memcpy(len, &sHeader[tag.length()], sizeof(dwLength));

			sBody.resize(dwLength);
			char* pBody = &sBody[0];

			boost::asio::async_read(m_socket, 
				boost::asio::buffer(pBody, dwLength),
				boost::bind(&socket_session::handle_read_header, shared_from_this(),
				boost::asio::placeholders::error));
		}
		catch(std::exception& e)
		{
			LOG4CXX_ERROR(firebird_log, KDS_CODE_INFO << "连接远程地址:[" << get_remote_addr() << "],socket异常:[" << e.what() << "]");
			close();
		}
		catch(...)
		{
			LOG4CXX_ERROR(firebird_log, KDS_CODE_INFO << "连接远程地址:[" << get_remote_addr() << "],socket异常:[未知异常]");
			close();
		}
	}

	void socket_session::handle_write(const boost::system::error_code& error, 
		std::size_t bytes_transferred, std::string* pmsg)
	{
		//数据发送成功就销毁
		if (pmsg != NULL)
		{
			delete pmsg;
		}

		if(error)
		{
			LOG4CXX_ERROR(firebird_log, KDS_CODE_INFO << "连接远程地址:[" << get_remote_addr() << "],socket异常:[" << error.message().c_str() << "]");
			close();
			return;
		}
	}

	void socket_session::async_write(const std::string& sMsg)
	{
		LOG4CXX_DEBUG(firebird_log, KDS_CODE_INFO  << "enter.")

		try
		{
			DWORD dwLength = sMsg.size();
			char* pLen = (char*)&dwLength;

			//由于是异步发送，要保证数据发送完整时，才把数据销毁
			std::string* msg = new std::string();
			msg->append(tag);
			msg->append(pLen, sizeof(dwLength));
			msg->append(sMsg);

			boost::asio::async_write(m_socket,boost::asio::buffer(*msg, msg->size()), 
				boost::bind(&socket_session::handle_write, shared_from_this(),
				boost::asio::placeholders::error, boost::asio::placeholders::bytes_transferred,
				msg));

		}
		catch(std::exception& e)
		{
			LOG4CXX_ERROR(firebird_log, KDS_CODE_INFO << "连接远程地址:[" << get_remote_addr() << "],socket异常:[" << e.what() << "]");
			close();
		}
		catch(...)
		{
			LOG4CXX_ERROR(firebird_log, KDS_CODE_INFO << "连接远程地址:[" << get_remote_addr() << "],socket异常:[未知异常]");
			close();
		}
	}

	void socket_session::async_write(message& msg)
	{
		std::string data;
		message_oarchive(data, msg);
		
		async_write(data);
	}
}


接受数据时，socket_session会先读取7个字节的head，比较前3个字节“KDS”，然后取得4个字节的Length，再读出Length长度的数据，最后将该数据传给read_data_cb回调函数处理，read_data_cb回调函数是在外部注册的。

3.连接管理器

对于服务器来说，它同时服务多个客户端，为了有效的管理，因此需要一个连接管理器，我定义为session_manager。session_manager主要是对socket_session的增删改查，和有效性检查。

//session_manager.h

[cpp] view plain copy print ?
  1. #pragma once   
  2. #include "socket_session.h"   
  3. #include "filter_container.h"   
  4. #include <boost/date_time/posix_time/posix_time.hpp>   
  5. #include <boost/multi_index_container.hpp>   
  6. #include <boost/multi_index/member.hpp>   
  7. #include <boost/multi_index/ordered_index.hpp>   
  8. #include <boost/typeof/typeof.hpp>   
  9. #include <boost/random.hpp>   
  10. #include <boost/pool/detail/singleton.hpp>   
  11.   
  12. namespace firebird{  
  13.     template<typename T>  
  14.     class var_gen_wraper  
  15.     {  
  16.     public:  
  17.         var_gen_wraper(): gen(boost::mt19937((boost::int32_t)std::time(0)),   
  18.             boost::uniform_smallint<>(1, 100)) {}  
  19.         typename T::result_type operator() () { return gen(); }  
  20.     private:  
  21.         T gen;  
  22.     };  
  23.   
  24.     struct  session_stu  
  25.     {  
  26.         DWORD   id;  
  27.         WORD    business_type;  
  28.         std::string address;  
  29.         DWORD   app_id;  
  30.         socket_session_ptr session;  
  31.     };  
  32.   
  33.     struct sid{};  
  34.     struct sbusiness_type{};  
  35.     struct saddress{};  
  36.     struct sapp_id{};  
  37.   
  38.     enum session_idx_member{ session_id = 0, session_business_type, session_address, app_id};  
  39. #define CLIENT 0   
  40. #define SERVER 1   
  41.   
  42.     typedef boost::multi_index::multi_index_container<  
  43.         session_stu,   
  44.         boost::multi_index::indexed_by<  
  45.         boost::multi_index::ordered_unique<  
  46.         boost::multi_index::tag<sid>, BOOST_MULTI_INDEX_MEMBER(session_stu, DWORD, id)>,  
  47.         boost::multi_index::ordered_non_unique<  
  48.         boost::multi_index::tag<sbusiness_type>, BOOST_MULTI_INDEX_MEMBER(session_stu, WORD, business_type)>,  
  49.         boost::multi_index::ordered_non_unique<  
  50.         boost::multi_index::tag<saddress>, BOOST_MULTI_INDEX_MEMBER(session_stu, std::string, address)>,  
  51.         boost::multi_index::ordered_non_unique<  
  52.         boost::multi_index::tag<sapp_id>, BOOST_MULTI_INDEX_MEMBER(session_stu, DWORD, app_id)>  
  53.         >  
  54.     > session_set;  
  55.   
  56. #define MULTI_MEMBER_CON(Tag) boost::multi_index::index<session_set,Tag>::type&   
  57. #define MULTI_MEMBER_ITR(Tag) boost::multi_index::index<session_set,Tag>::type::iterator   
  58.   
  59.     struct is_business_type {  
  60.         is_business_type(WORD type)  
  61.             :m_type(type)  
  62.         {  
  63.   
  64.         }  
  65.         bool operator()(const session_stu& s)   
  66.         {  
  67.             return (s.business_type == m_type);  
  68.         }  
  69.   
  70.         WORD m_type;  
  71.     };  
  72.   
  73.     class session_manager  
  74.     {  
  75.     public:  
  76.         typedef boost::shared_lock<boost::shared_mutex> readLock;  
  77.         typedef boost:: unique_lock<boost::shared_mutex> writeLock;  
  78.   
  79.         session_manager(boost::asio::io_service& io_srv, int type, int expires_time);  
  80.         ~session_manager();  
  81.   
  82.         void add_session(socket_session_ptr p);  
  83.         void update_session(socket_session_ptr p);  
  84.   
  85.         template<typename Tag, typename Member>  
  86.         void del_session(Member m)  
  87.         {  
  88.             writeLock lock(m_mutex);  
  89.             if (m_sessions.empty())  
  90.             {  
  91.                 return ;  
  92.             }  
  93.   
  94.             MULTI_MEMBER_CON(Tag) idx = boost::multi_index::get<Tag>(m_sessions);  
  95.             //BOOST_AUTO(idx, boost::multi_index::get<Tag>(m_sessions));   
  96.             BOOST_AUTO(iter, idx.find(m));  
  97.   
  98.             if (iter != idx.end())  
  99.             {  
  100.                 idx.erase(iter);  
  101.             }  
  102.         }  
  103.   
  104.         //获取容器中的第一个session   
  105.         template<typename Tag, typename Member>  
  106.         socket_session_ptr get_session(Member m)  
  107.         {  
  108.             readLock lock(m_mutex);  
  109.   
  110.             if (m_sessions.empty())  
  111.             {  
  112.                 return socket_session_ptr();  
  113.             }  
  114.   
  115.             MULTI_MEMBER_CON(Tag) idx = boost::multi_index::get<Tag>(m_sessions);  
  116.             BOOST_AUTO(iter, idx.find(m));  
  117.             return iter != boost::end(idx) ? iter->session : socket_session_ptr();  
  118.         }  
  119.   
  120.         //随机获取容器中的session   
  121.         template<typename Tag>  
  122.         socket_session_ptr get_session_by_business_type(WORD m)  
  123.         {  
  124.             typedef filter_container<is_business_type, MULTI_MEMBER_ITR(Tag)> FilterContainer;  
  125.             readLock lock(m_mutex);  
  126.   
  127.             if (m_sessions.empty())  
  128.             {  
  129.                 return socket_session_ptr();  
  130.             }  
  131.   
  132.             MULTI_MEMBER_CON(Tag) idx = boost::multi_index::get<Tag>(m_sessions);  
  133.   
  134.             //对容器的元素条件过滤   
  135.             is_business_type predicate(m);  
  136.             FilterContainer fc(predicate, idx.begin(), idx.end());  
  137.             FilterContainer::FilterIter iter = fc.begin();  
  138.   
  139.             if (fc.begin() == fc.end())  
  140.             {  
  141.                 return socket_session_ptr();  
  142.             }  
  143.   
  144.             //typedef boost::variate_generator<boost::mt19937, boost::uniform_smallint<>> var_gen;   
  145.             //typedef boost::details::pool::singleton_default<var_gen_wraper<var_gen>> s_var_gen;   
  146.             根据随机数产生session   
  147.             //s_var_gen::object_type &gen = s_var_gen::instance();   
  148.             //int step = gen() % fc.szie();   
  149.   
  150.             int step = m_next_session % fc.szie();  
  151.             ++m_next_session;  
  152.   
  153.             for (int i = 0; i < step; ++i)  
  154.             {  
  155.                 iter++;  
  156.             }  
  157.   
  158.             return iter != fc.end() ? iter->session : socket_session_ptr();  
  159.         }  
  160.   
  161.         //根据类型和地址取session   
  162.         template<typename Tag>  
  163.         socket_session_ptr get_session_by_type_ip(WORD m, std::string& ip)  
  164.         {  
  165.             typedef filter_container<is_business_type, MULTI_MEMBER_ITR(Tag)> FilterContainer;  
  166.             readLock lock(m_mutex);  
  167.   
  168.             if (m_sessions.empty())  
  169.             {  
  170.                 return socket_session_ptr();  
  171.             }  
  172.   
  173.             MULTI_MEMBER_CON(Tag) idx = boost::multi_index::get<Tag>(m_sessions);  
  174.   
  175.             //对容器的元素条件过滤   
  176.             is_business_type predicate(m);  
  177.             FilterContainer fc(predicate, idx.begin(), idx.end());  
  178.             FilterContainer::FilterIter iter = fc.begin();  
  179.   
  180.             if (fc.begin() == fc.end())  
  181.             {  
  182.                 return socket_session_ptr();  
  183.             }  
  184.   
  185.             while (iter != fc.end())  
  186.             {  
  187.                 if (iter->session->get_remote_addr().find(ip) != std::string::npos)  
  188.                 {  
  189.                     break;  
  190.                 }  
  191.   
  192.                 iter++;  
  193.             }  
  194.   
  195.             return iter != fc.end() ? iter->session : socket_session_ptr();  
  196.         }  
  197.   
  198.         //根据类型和app_id取session   
  199.         template<typename Tag>  
  200.         socket_session_ptr get_session_by_type_appid(WORD m, DWORD app_id)  
  201.         {  
  202.             typedef filter_container<is_business_type, MULTI_MEMBER_ITR(Tag)> FilterContainer;  
  203.             readLock lock(m_mutex);  
  204.   
  205.             if (m_sessions.empty())  
  206.             {  
  207.                 return socket_session_ptr();  
  208.             }  
  209.   
  210.             MULTI_MEMBER_CON(Tag) idx = boost::multi_index::get<Tag>(m_sessions);  
  211.   
  212.             //对容器的元素条件过滤   
  213.             is_business_type predicate(m);  
  214.             FilterContainer fc(predicate, idx.begin(), idx.end());  
  215.             FilterContainer::FilterIter iter = fc.begin();  
  216.   
  217.             if (fc.begin() == fc.end())  
  218.             {  
  219.                 return socket_session_ptr();  
  220.             }  
  221.   
  222.             while (iter != fc.end())  
  223.             {  
  224.                 if (iter->session->get_app_id() == app_id)  
  225.                 {  
  226.                     break;  
  227.                 }  
  228.   
  229.                 iter++;  
  230.             }  
  231.   
  232.             return iter != fc.end() ? iter->session : socket_session_ptr();  
  233.         }  
  234.   
  235.     private:  
  236.         int m_type;  
  237.         int m_expires_time;  
  238.         boost::asio::io_service& m_io_srv;  
  239.         boost::asio::deadline_timer m_check_tick;  
  240.         boost::shared_mutex m_mutex;  
  241.         unsigned short m_next_session;  
  242.   
  243.         session_set m_sessions;  
  244.   
  245.         void check_connection();  
  246.     };  
  247. }  
#pragma once
#include "socket_session.h"
#include "filter_container.h"
#include <boost/date_time/posix_time/posix_time.hpp>
#include <boost/multi_index_container.hpp>
#include <boost/multi_index/member.hpp>
#include <boost/multi_index/ordered_index.hpp>
#include <boost/typeof/typeof.hpp>
#include <boost/random.hpp>
#include <boost/pool/detail/singleton.hpp>

namespace firebird{
	template<typename T>
	class var_gen_wraper
	{
	public:
		var_gen_wraper(): gen(boost::mt19937((boost::int32_t)std::time(0)), 
			boost::uniform_smallint<>(1, 100)) {}
		typename T::result_type operator() () { return gen(); }
	private:
		T gen;
	};

	struct  session_stu
	{
		DWORD	id;
		WORD	business_type;
		std::string address;
		DWORD	app_id;
		socket_session_ptr session;
	};

	struct sid{};
	struct sbusiness_type{};
	struct saddress{};
	struct sapp_id{};

	enum session_idx_member{ session_id = 0, session_business_type, session_address, app_id};
#define CLIENT 0
#define SERVER 1

	typedef boost::multi_index::multi_index_container<
		session_stu, 
		boost::multi_index::indexed_by<
		boost::multi_index::ordered_unique<
		boost::multi_index::tag<sid>, BOOST_MULTI_INDEX_MEMBER(session_stu, DWORD, id)>,
		boost::multi_index::ordered_non_unique<
		boost::multi_index::tag<sbusiness_type>, BOOST_MULTI_INDEX_MEMBER(session_stu, WORD, business_type)>,
		boost::multi_index::ordered_non_unique<
		boost::multi_index::tag<saddress>, BOOST_MULTI_INDEX_MEMBER(session_stu, std::string, address)>,
		boost::multi_index::ordered_non_unique<
		boost::multi_index::tag<sapp_id>, BOOST_MULTI_INDEX_MEMBER(session_stu, DWORD, app_id)>
		>
	> session_set;

#define MULTI_MEMBER_CON(Tag) boost::multi_index::index<session_set,Tag>::type&
#define MULTI_MEMBER_ITR(Tag) boost::multi_index::index<session_set,Tag>::type::iterator

	struct is_business_type {
		is_business_type(WORD type)
			:m_type(type)
		{

		}
		bool operator()(const session_stu& s) 
		{
			return (s.business_type == m_type);
		}

		WORD m_type;
	};

	class session_manager
	{
	public:
		typedef boost::shared_lock<boost::shared_mutex> readLock;
		typedef boost:: unique_lock<boost::shared_mutex> writeLock;

		session_manager(boost::asio::io_service& io_srv, int type, int expires_time);
		~session_manager();

		void add_session(socket_session_ptr p);
		void update_session(socket_session_ptr p);

		template<typename Tag, typename Member>
		void del_session(Member m)
		{
			writeLock lock(m_mutex);
			if (m_sessions.empty())
			{
				return ;
			}

			MULTI_MEMBER_CON(Tag) idx = boost::multi_index::get<Tag>(m_sessions);
			//BOOST_AUTO(idx, boost::multi_index::get<Tag>(m_sessions));
			BOOST_AUTO(iter, idx.find(m));

			if (iter != idx.end())
			{
				idx.erase(iter);
			}
		}

		//获取容器中的第一个session
		template<typename Tag, typename Member>
		socket_session_ptr get_session(Member m)
		{
			readLock lock(m_mutex);

			if (m_sessions.empty())
			{
				return socket_session_ptr();
			}

			MULTI_MEMBER_CON(Tag) idx = boost::multi_index::get<Tag>(m_sessions);
			BOOST_AUTO(iter, idx.find(m));
			return iter != boost::end(idx) ? iter->session : socket_session_ptr();
		}

		//随机获取容器中的session
		template<typename Tag>
		socket_session_ptr get_session_by_business_type(WORD m)
		{
			typedef filter_container<is_business_type, MULTI_MEMBER_ITR(Tag)> FilterContainer;
			readLock lock(m_mutex);

			if (m_sessions.empty())
			{
				return socket_session_ptr();
			}

			MULTI_MEMBER_CON(Tag) idx = boost::multi_index::get<Tag>(m_sessions);

			//对容器的元素条件过滤
			is_business_type predicate(m);
			FilterContainer fc(predicate, idx.begin(), idx.end());
			FilterContainer::FilterIter iter = fc.begin();

			if (fc.begin() == fc.end())
			{
				return socket_session_ptr();
			}

			//typedef boost::variate_generator<boost::mt19937, boost::uniform_smallint<>> var_gen;
			//typedef boost::details::pool::singleton_default<var_gen_wraper<var_gen>> s_var_gen;
			根据随机数产生session
			//s_var_gen::object_type &gen = s_var_gen::instance();
			//int step = gen() % fc.szie();

			int step = m_next_session % fc.szie();
			++m_next_session;

			for (int i = 0; i < step; ++i)
			{
				iter++;
			}

			return iter != fc.end() ? iter->session : socket_session_ptr();
		}

		//根据类型和地址取session
		template<typename Tag>
		socket_session_ptr get_session_by_type_ip(WORD m, std::string& ip)
		{
			typedef filter_container<is_business_type, MULTI_MEMBER_ITR(Tag)> FilterContainer;
			readLock lock(m_mutex);

			if (m_sessions.empty())
			{
				return socket_session_ptr();
			}

			MULTI_MEMBER_CON(Tag) idx = boost::multi_index::get<Tag>(m_sessions);

			//对容器的元素条件过滤
			is_business_type predicate(m);
			FilterContainer fc(predicate, idx.begin(), idx.end());
			FilterContainer::FilterIter iter = fc.begin();

			if (fc.begin() == fc.end())
			{
				return socket_session_ptr();
			}

			while (iter != fc.end())
			{
				if (iter->session->get_remote_addr().find(ip) != std::string::npos)
				{
					break;
				}

				iter++;
			}

			return iter != fc.end() ? iter->session : socket_session_ptr();
		}

		//根据类型和app_id取session
		template<typename Tag>
		socket_session_ptr get_session_by_type_appid(WORD m, DWORD app_id)
		{
			typedef filter_container<is_business_type, MULTI_MEMBER_ITR(Tag)> FilterContainer;
			readLock lock(m_mutex);

			if (m_sessions.empty())
			{
				return socket_session_ptr();
			}

			MULTI_MEMBER_CON(Tag) idx = boost::multi_index::get<Tag>(m_sessions);

			//对容器的元素条件过滤
			is_business_type predicate(m);
			FilterContainer fc(predicate, idx.begin(), idx.end());
			FilterContainer::FilterIter iter = fc.begin();

			if (fc.begin() == fc.end())
			{
				return socket_session_ptr();
			}

			while (iter != fc.end())
			{
				if (iter->session->get_app_id() == app_id)
				{
					break;
				}

				iter++;
			}

			return iter != fc.end() ? iter->session : socket_session_ptr();
		}

	private:
		int m_type;
		int m_expires_time;
		boost::asio::io_service& m_io_srv;
		boost::asio::deadline_timer m_check_tick;
		boost::shared_mutex m_mutex;
		unsigned short m_next_session;

		session_set m_sessions;

		void check_connection();
	};
}


这里主要用到了boost的multi_index容器，这是一个非常有用方便的容器，可实现容器的多列索引，具体的使用方法，在这里不多做详解。

//session_manager.cpp

[cpp] view plain copy print ?
  1. #include "session_manager.h"   
  2.   
  3. namespace firebird{  
  4.     session_manager::session_manager(boost::asio::io_service& io_srv, int type, int expires_time)  
  5.         :m_io_srv(io_srv), m_check_tick(io_srv), m_type(type), m_expires_time(expires_time),m_next_session(0)  
  6.     {  
  7.         check_connection();  
  8.     }  
  9.   
  10.     session_manager::~session_manager()  
  11.     {  
  12.   
  13.     }  
  14.   
  15.     //检查服务器所有session的连接状态   
  16.     void session_manager::check_connection()  
  17.     {  
  18.         try{  
  19.             writeLock lock(m_mutex);  
  20.   
  21.             session_set::iterator iter = m_sessions.begin();  
  22.             while (iter != m_sessions.end())  
  23.             {  
  24.                 LOG4CXX_DEBUG(firebird_log, "循环");  
  25.                 if (CLIENT == m_type)//客户端的方式   
  26.                 {  
  27.                     if (!iter->session->socket().is_open())//已断开，删除已断开的连接   
  28.                     {  
  29.                         LOG4CXX_INFO(firebird_log, "重新连接[" << iter->address << "]");  
  30.                         iter->session->close(); //通过关闭触发客户端重连   
  31.                     }  
  32.                     else{//连接中，发送心跳   
  33.                         message msg;  
  34.                         msg.command = heartbeat;  
  35.                         msg.business_type = iter->session->get_business_type();  
  36.                         msg.app_id = iter->session->get_app_id();  
  37.                         msg.data() = "H";  
  38.   
  39.                         iter->session->async_write(msg);  
  40.                         iter->session->set_op_time();  
  41.                     }  
  42.                 }  
  43.                 else if (SERVER == m_type)//服务器的方式   
  44.                 {  
  45.                     if (!iter->session->socket().is_open())//已断开，删除已断开的连接   
  46.                     {  
  47.                         LOG4CXX_INFO(firebird_log, KDS_CODE_INFO << "删除已关闭的session:[" << iter->session->get_remote_addr() << "]");  
  48.                         iter = m_sessions.erase(iter);  
  49.                         continue;  
  50.                     }  
  51.                     else{//连接中，设定每30秒检查一次   
  52.                         if (iter->session->is_timeout()) //如果session已长时间没操作，则关闭   
  53.                         {  
  54.                             LOG4CXX_INFO(firebird_log, KDS_CODE_INFO << "删除已超时的session:[" << iter->session->get_remote_addr() << "]");  
  55.                             iter->session->close();//通过关闭触发删除session   
  56.                         }  
  57.                     }  
  58.   
  59.                     iter->session->set_op_time();  
  60.                 }  
  61.                 else{  
  62.                     LOG4CXX_ERROR(firebird_log, KDS_CODE_INFO << "unknown manager_type");  
  63.                 }  
  64.                 ++iter;  
  65.             }  
  66.   
  67.             LOG4CXX_DEBUG(firebird_log, "定时检查");  
  68.             m_check_tick.expires_from_now(boost::posix_time::seconds(m_expires_time));  
  69.             m_check_tick.async_wait(boost::bind(&session_manager::check_connection, this));  
  70.         }  
  71.         catch(std::exception& e)  
  72.         {  
  73.             LOG4CXX_ERROR(firebird_log, KDS_CODE_INFO << "[" << e.what() << "]");  
  74.         }  
  75.         catch(...)  
  76.         {  
  77.             LOG4CXX_ERROR(firebird_log, KDS_CODE_INFO << "unknown exception.");  
  78.         }  
  79.     }  
  80.   
  81.     void session_manager::add_session(socket_session_ptr p)  
  82.     {  
  83.         writeLock lock(m_mutex);  
  84.         session_stu stuSession;  
  85.         stuSession.id = p->id();  
  86.         stuSession.business_type = 0;  
  87.         stuSession.address = p->get_remote_addr();  
  88.         stuSession.app_id = p->get_app_id();  
  89.         stuSession.session = p;  
  90.         m_sessions.insert(stuSession);  
  91.     }  
  92.   
  93.     void session_manager::update_session(socket_session_ptr p)  
  94.     {  
  95.         writeLock lock(m_mutex);  
  96.         if (m_sessions.empty())  
  97.         {  
  98.             return ;  
  99.         }  
  100.   
  101.         MULTI_MEMBER_CON(sid) idx = boost::multi_index::get<sid>(m_sessions);  
  102.         BOOST_AUTO(iter, idx.find(p->id()));  
  103.   
  104.         if (iter != idx.end())  
  105.         {  
  106.             const_cast<session_stu&>(*iter).business_type = p->get_business_type();  
  107.             const_cast<session_stu&>(*iter).app_id = p->get_app_id();  
  108.         }  
  109.     }  
  110. }  
#include "session_manager.h"

namespace firebird{
	session_manager::session_manager(boost::asio::io_service& io_srv, int type, int expires_time)
		:m_io_srv(io_srv), m_check_tick(io_srv), m_type(type), m_expires_time(expires_time),m_next_session(0)
	{
		check_connection();
	}

	session_manager::~session_manager()
	{

	}

	//检查服务器所有session的连接状态
	void session_manager::check_connection()
	{
		try{
			writeLock lock(m_mutex);

			session_set::iterator iter = m_sessions.begin();
			while (iter != m_sessions.end())
			{
				LOG4CXX_DEBUG(firebird_log, "循环");
				if (CLIENT == m_type)//客户端的方式
				{
					if (!iter->session->socket().is_open())//已断开，删除已断开的连接
					{
						LOG4CXX_INFO(firebird_log, "重新连接[" << iter->address << "]");
						iter->session->close(); //通过关闭触发客户端重连
					}
					else{//连接中，发送心跳
						message msg;
						msg.command = heartbeat;
						msg.business_type = iter->session->get_business_type();
						msg.app_id = iter->session->get_app_id();
						msg.data() = "H";

						iter->session->async_write(msg);
						iter->session->set_op_time();
					}
				}
				else if (SERVER == m_type)//服务器的方式
				{
					if (!iter->session->socket().is_open())//已断开，删除已断开的连接
					{
						LOG4CXX_INFO(firebird_log, KDS_CODE_INFO << "删除已关闭的session:[" << iter->session->get_remote_addr() << "]");
						iter = m_sessions.erase(iter);
						continue;
					}
					else{//连接中，设定每30秒检查一次
						if (iter->session->is_timeout()) //如果session已长时间没操作，则关闭
						{
							LOG4CXX_INFO(firebird_log, KDS_CODE_INFO << "删除已超时的session:[" << iter->session->get_remote_addr() << "]");
							iter->session->close();//通过关闭触发删除session
						}
					}

					iter->session->set_op_time();
				}
				else{
					LOG4CXX_ERROR(firebird_log, KDS_CODE_INFO << "unknown manager_type");
				}
				++iter;
			}

			LOG4CXX_DEBUG(firebird_log, "定时检查");
			m_check_tick.expires_from_now(boost::posix_time::seconds(m_expires_time));
			m_check_tick.async_wait(boost::bind(&session_manager::check_connection, this));
		}
		catch(std::exception& e)
		{
			LOG4CXX_ERROR(firebird_log, KDS_CODE_INFO << "[" << e.what() << "]");
		}
		catch(...)
		{
			LOG4CXX_ERROR(firebird_log, KDS_CODE_INFO << "unknown exception.");
		}
	}

	void session_manager::add_session(socket_session_ptr p)
	{
		writeLock lock(m_mutex);
		session_stu stuSession;
		stuSession.id = p->id();
		stuSession.business_type = 0;
		stuSession.address = p->get_remote_addr();
		stuSession.app_id = p->get_app_id();
		stuSession.session = p;
		m_sessions.insert(stuSession);
	}

	void session_manager::update_session(socket_session_ptr p)
	{
		writeLock lock(m_mutex);
		if (m_sessions.empty())
		{
			return ;
		}

		MULTI_MEMBER_CON(sid) idx = boost::multi_index::get<sid>(m_sessions);
		BOOST_AUTO(iter, idx.find(p->id()));

		if (iter != idx.end())
		{
			const_cast<session_stu&>(*iter).business_type = p->get_business_type();
			const_cast<session_stu&>(*iter).app_id = p->get_app_id();
		}
	}
}


这个时候，我就可以使用id、business_type、address、app_id当做key来索引socket_session了，单使用map容器是做不到的。

还有索引时，需要的一个条件过滤器

//filter_container.h

[cpp] view plain copy print ?
  1. #pragma once   
  2. #include <boost/iterator/filter_iterator.hpp>   
  3.   
  4. namespace firebird{  
  5.     template <class Predicate, class Iterator>  
  6.     class filter_container  
  7.     {  
  8.     public:  
  9.         typedef boost::filter_iterator<Predicate, Iterator> FilterIter;  
  10.   
  11.         filter_container(Predicate p, Iterator begin, Iterator end)  
  12.             :m_begin(p, begin, end),  
  13.             m_end(p, end, end)  
  14.         {  
  15.   
  16.         }  
  17.         ~filter_container() {}  
  18.   
  19.         FilterIter begin() { return m_begin; }  
  20.         FilterIter end()   { return m_end; }  
  21.         int szie() {  
  22.             int i = 0;  
  23.             FilterIter fi = m_begin;  
  24.             while(fi != m_end)  
  25.             {  
  26.                 ++i;  
  27.                 ++fi;  
  28.             }  
  29.   
  30.             return i;  
  31.         }  
  32.   
  33.     private:  
  34.         FilterIter m_begin;  
  35.         FilterIter m_end;  
  36.     };  
  37. }  
#pragma once
#include <boost/iterator/filter_iterator.hpp>

namespace firebird{
	template <class Predicate, class Iterator>
	class filter_container
	{
	public:
		typedef boost::filter_iterator<Predicate, Iterator> FilterIter;

		filter_container(Predicate p, Iterator begin, Iterator end)
			:m_begin(p, begin, end),
			m_end(p, end, end)
		{

		}
		~filter_container() {}

		FilterIter begin() { return m_begin; }
		FilterIter end()   { return m_end; }
		int szie() {
			int i = 0;
			FilterIter fi = m_begin;
			while(fi != m_end)
			{
				++i;
				++fi;
			}

			return i;
		}

	private:
		FilterIter m_begin;
		FilterIter m_end;
	};
}


 

4.服务器端的实现

服务器我定义为server_socket_utils，拥有一个session_manager，每当accept成功得到一个socket_session时，都会将其增加到session_manager去管理，注册相关回调函数。

read_data_callback   接收到数据的回调函数

收到数据之后，也就是数据包的body部分，反序列化出command、business_type、app_id和data（我使用到了thrift），如果command==normal正常的业务包，会调用handle_read_data传入data。

close_callback 关闭socket_session触发的回调函数

根据id将该连接从session_manager中删除掉

//server_socket_utils.h

[cpp] view plain copy print ?
  1. #pragma once   
  2. #include "socket_session.h"   
  3. #include "session_manager.h"   
  4. #include <boost/format.hpp>   
  5. #include <firebird/message/message.hpp>   
  6.   
  7. namespace firebird{  
  8.     using boost::asio::ip::tcp;  
  9.   
  10.     class FIREBIRD_DECL server_socket_utils  
  11.     {  
  12.     private:  
  13.         boost::asio::io_service m_io_srv;  
  14.         boost::asio::io_service::work m_work;  
  15.         tcp::acceptor m_acceptor;  
  16.   
  17.         void handle_accept(socket_session_ptr session, const boost::system::error_code& error);  
  18.   
  19.         void close_callback(socket_session_ptr session);  
  20.         void read_data_callback(const boost::system::error_code& e,   
  21.             socket_session_ptr session, message& msg);  
  22.   
  23.     protected:  
  24.         virtual void handle_read_data(message& msg, socket_session_ptr pSession) = 0;  
  25.   
  26.     public:  
  27.         server_socket_utils(int port);  
  28.         ~server_socket_utils(void);  
  29.   
  30.         void start();  
  31.         boost::asio::io_service& get_io_service() { return m_io_srv; }  
  32.   
  33.         session_manager m_manager;  
  34.     };  
  35. }  
#pragma once
#include "socket_session.h"
#include "session_manager.h"
#include <boost/format.hpp>
#include <firebird/message/message.hpp>

namespace firebird{
	using boost::asio::ip::tcp;

	class FIREBIRD_DECL server_socket_utils
	{
	private:
		boost::asio::io_service m_io_srv;
		boost::asio::io_service::work m_work;
		tcp::acceptor m_acceptor;

		void handle_accept(socket_session_ptr session, const boost::system::error_code& error);

		void close_callback(socket_session_ptr session);
		void read_data_callback(const boost::system::error_code& e, 
			socket_session_ptr session, message& msg);

	protected:
		virtual void handle_read_data(message& msg, socket_session_ptr pSession) = 0;

	public:
		server_socket_utils(int port);
		~server_socket_utils(void);

		void start();
		boost::asio::io_service& get_io_service() { return m_io_srv; }

		session_manager m_manager;
	};
}

//server_socket_utils.cpp

[cpp] view plain copy print ?
  1. #include "server_socket_utils.h"   
  2.   
  3. namespace firebird{  
  4.     server_socket_utils::server_socket_utils(int port)  
  5.         :m_work(m_io_srv),  
  6.         m_acceptor(m_io_srv, tcp::endpoint(tcp::v4(), port)),  
  7.         m_manager(m_io_srv, SERVER, 3)  
  8.     {  
  9.         //m_acceptor.set_option(boost::asio::ip::tcp::acceptor::reuse_address(true));   
  10.          关闭连接前留0秒给客户接收数据   
  11.         //m_acceptor.set_option(boost::asio::ip::tcp::acceptor::linger(true, 0));   
  12.         //m_acceptor.set_option(boost::asio::ip::tcp::no_delay(true));   
  13.         //m_acceptor.set_option(boost::asio::socket_base::keep_alive(true));   
  14.         //m_acceptor.set_option(boost::asio::socket_base::receive_buffer_size(16384));   
  15.     }  
  16.   
  17.     server_socket_utils::~server_socket_utils(void)  
  18.     {  
  19.     }  
  20.   
  21.     void server_socket_utils::start()  
  22.     {  
  23.         try{  
  24.             socket_session_ptr new_session(new socket_session(m_io_srv));  
  25.             m_acceptor.async_accept(new_session->socket(),  
  26.                 boost::bind(&server_socket_utils::handle_accept, this, new_session,  
  27.                 boost::asio::placeholders::error));  
  28.         }  
  29.         catch(std::exception& e)  
  30.         {  
  31.             LOG4CXX_ERROR(firebird_log, KDS_CODE_INFO << "socket异常:[" << e.what() << "]");  
  32.         }  
  33.         catch(...)  
  34.         {  
  35.             LOG4CXX_ERROR(firebird_log, KDS_CODE_INFO << "socket异常:[未知异常]");  
  36.         }  
  37.     }  
  38.   
  39.     void server_socket_utils::handle_accept(socket_session_ptr session, const boost::system::error_code& error)  
  40.     {  
  41.         if (!error)  
  42.         {  
  43.             try{  
  44.                 socket_session_ptr new_session(new socket_session(m_io_srv));  
  45.                 m_acceptor.async_accept(new_session->socket(),  
  46.                     boost::bind(&server_socket_utils::handle_accept, this, new_session,  
  47.                     boost::asio::placeholders::error));  
  48.   
  49.                 if (session != NULL)  
  50.                 {  
  51.                     //注册关闭回调函数   
  52.                     session->installCloseCallBack(boost::bind(&server_socket_utils::close_callback, this, _1));  
  53.                     //注册读到数据回调函数   
  54.                     session->installReadDataCallBack(boost::bind(&server_socket_utils::read_data_callback, this, _1, _2, _3));  
  55.   
  56.                     boost::format fmt("%1%:%2%");  
  57.                     fmt % session->socket().remote_endpoint().address().to_string();  
  58.                     fmt % session->socket().remote_endpoint().port();  
  59.                     session->set_remote_addr(fmt.str());  
  60.   
  61.                     session->start();  
  62.                     m_manager.add_session(session);  
  63.                 }  
  64.             }  
  65.             catch(std::exception& e)  
  66.             {  
  67.                 LOG4CXX_ERROR(firebird_log, KDS_CODE_INFO << "socket异常:[" << e.what() << "]");  
  68.             }  
  69.             catch(...)  
  70.             {  
  71.                 LOG4CXX_ERROR(firebird_log, KDS_CODE_INFO << "socket异常:[未知异常]");  
  72.             }  
  73.   
  74.         }  
  75.     }  
  76.   
  77.     void server_socket_utils::close_callback(socket_session_ptr session)  
  78.     {  
  79.         LOG4CXX_DEBUG(firebird_log, "close_callback");  
  80.         try{  
  81.             m_manager.del_session<sid>(session->id());  
  82.         }  
  83.         catch(std::exception& e)  
  84.         {  
  85.             LOG4CXX_ERROR(firebird_log, KDS_CODE_INFO << "socket异常:[" << e.what() << "]");  
  86.         }  
  87.         catch(...)  
  88.         {  
  89.             LOG4CXX_ERROR(firebird_log, KDS_CODE_INFO << "socket异常:[未知异常]");  
  90.         }  
  91.   
  92.     }  
  93.   
  94.     void server_socket_utils::read_data_callback(const boost::system::error_code& e,   
  95.         socket_session_ptr session, message& msg)  
  96.     {  
  97.         try{  
  98.             LOG4CXX_DEBUG(firebird_log, "command =[" << msg.command << "],["   
  99.                 << msg.business_type << "],[" << msg.data() << "]");  
  100.   
  101.             if (msg.command == heartbeat)  
  102.             {//心跳   
  103.                 session->async_write(msg);  
  104.             }  
  105.             else if (msg.command == regist)  
  106.             {//注册   
  107.                 session->set_business_type(msg.business_type);  
  108.                 session->set_app_id(msg.app_id);  
  109.                 m_manager.update_session(session);  
  110.   
  111.                 session->async_write(msg);  
  112.                 LOG4CXX_FATAL(firebird_log, "远程地址:[" << session->get_remote_addr() << "],服务器类型:[" <<  
  113.                     session->get_business_type() << "],服务器ID:[" << session->get_app_id() << "]注册成功!");  
  114.             }  
  115.             else if (msg.command == normal)  
  116.             {//业务数据   
  117.                 handle_read_data(msg, session);  
  118.             }  
  119.             else   
  120.             {  
  121.                 LOG4CXX_ERROR(firebird_log, KDS_CODE_INFO << "收到非法消息包!");  
  122.             }  
  123.         }  
  124.         catch(std::exception& e)  
  125.         {  
  126.             LOG4CXX_ERROR(firebird_log, KDS_CODE_INFO << "socket异常:[" << e.what() << "]");  
  127.         }  
  128.         catch(...)  
  129.         {  
  130.             LOG4CXX_ERROR(firebird_log, KDS_CODE_INFO << "socket异常:[未知异常]");  
  131.         }  
  132.     }  
  133. }  
#include "server_socket_utils.h"

namespace firebird{
	server_socket_utils::server_socket_utils(int port)
		:m_work(m_io_srv),
		m_acceptor(m_io_srv, tcp::endpoint(tcp::v4(), port)),
		m_manager(m_io_srv, SERVER, 3)
	{
		//m_acceptor.set_option(boost::asio::ip::tcp::acceptor::reuse_address(true));
		 关闭连接前留0秒给客户接收数据
		//m_acceptor.set_option(boost::asio::ip::tcp::acceptor::linger(true, 0));
		//m_acceptor.set_option(boost::asio::ip::tcp::no_delay(true));
		//m_acceptor.set_option(boost::asio::socket_base::keep_alive(true));
		//m_acceptor.set_option(boost::asio::socket_base::receive_buffer_size(16384));
	}

	server_socket_utils::~server_socket_utils(void)
	{
	}

	void server_socket_utils::start()
	{
		try{
			socket_session_ptr new_session(new socket_session(m_io_srv));
			m_acceptor.async_accept(new_session->socket(),
				boost::bind(&server_socket_utils::handle_accept, this, new_session,
				boost::asio::placeholders::error));
		}
		catch(std::exception& e)
		{
			LOG4CXX_ERROR(firebird_log, KDS_CODE_INFO << "socket异常:[" << e.what() << "]");
		}
		catch(...)
		{
			LOG4CXX_ERROR(firebird_log, KDS_CODE_INFO << "socket异常:[未知异常]");
		}
	}

	void server_socket_utils::handle_accept(socket_session_ptr session, const boost::system::error_code& error)
	{
		if (!error)
		{
			try{
				socket_session_ptr new_session(new socket_session(m_io_srv));
				m_acceptor.async_accept(new_session->socket(),
					boost::bind(&server_socket_utils::handle_accept, this, new_session,
					boost::asio::placeholders::error));

				if (session != NULL)
				{
					//注册关闭回调函数
					session->installCloseCallBack(boost::bind(&server_socket_utils::close_callback, this, _1));
					//注册读到数据回调函数
					session->installReadDataCallBack(boost::bind(&server_socket_utils::read_data_callback, this, _1, _2, _3));

					boost::format fmt("%1%:%2%");
					fmt % session->socket().remote_endpoint().address().to_string();
					fmt % session->socket().remote_endpoint().port();
					session->set_remote_addr(fmt.str());

					session->start();
					m_manager.add_session(session);
				}
			}
			catch(std::exception& e)
			{
				LOG4CXX_ERROR(firebird_log, KDS_CODE_INFO << "socket异常:[" << e.what() << "]");
			}
			catch(...)
			{
				LOG4CXX_ERROR(firebird_log, KDS_CODE_INFO << "socket异常:[未知异常]");
			}

		}
	}

	void server_socket_utils::close_callback(socket_session_ptr session)
	{
		LOG4CXX_DEBUG(firebird_log, "close_callback");
		try{
			m_manager.del_session<sid>(session->id());
		}
		catch(std::exception& e)
		{
			LOG4CXX_ERROR(firebird_log, KDS_CODE_INFO << "socket异常:[" << e.what() << "]");
		}
		catch(...)
		{
			LOG4CXX_ERROR(firebird_log, KDS_CODE_INFO << "socket异常:[未知异常]");
		}

	}

	void server_socket_utils::read_data_callback(const boost::system::error_code& e, 
		socket_session_ptr session, message& msg)
	{
		try{
			LOG4CXX_DEBUG(firebird_log, "command =[" << msg.command << "],[" 
				<< msg.business_type << "],[" << msg.data() << "]");

			if (msg.command == heartbeat)
			{//心跳
				session->async_write(msg);
			}
			else if (msg.command == regist)
			{//注册
				session->set_business_type(msg.business_type);
				session->set_app_id(msg.app_id);
				m_manager.update_session(session);

				session->async_write(msg);
				LOG4CXX_FATAL(firebird_log, "远程地址:[" << session->get_remote_addr() << "],服务器类型:[" <<
					session->get_business_type() << "],服务器ID:[" << session->get_app_id() << "]注册成功!");
			}
			else if (msg.command == normal)
			{//业务数据
				handle_read_data(msg, session);
			}
			else 
			{
				LOG4CXX_ERROR(firebird_log, KDS_CODE_INFO << "收到非法消息包!");
			}
		}
		catch(std::exception& e)
		{
			LOG4CXX_ERROR(firebird_log, KDS_CODE_INFO << "socket异常:[" << e.what() << "]");
		}
		catch(...)
		{
			LOG4CXX_ERROR(firebird_log, KDS_CODE_INFO << "socket异常:[未知异常]");
		}
	}
}


 

5.客户端

客户端与服务器的逻辑也差不多，区别就是在于客户端通过connect得到socket_session，而服务器是通过accept得到socket_session。

//client_socket_utils.h

[cpp] view plain copy print ?
  1. #pragma once   
  2. #include "socket_session.h"   
  3. #include "session_manager.h"   
  4. #include <boost/algorithm/string.hpp>   
  5. #include <firebird/message/message.hpp>   
  6.   
  7. namespace firebird{  
  8.     class FIREBIRD_DECL client_socket_utils  
  9.     {  
  10.     public:  
  11.         client_socket_utils();  
  12.         ~client_socket_utils();  
  13.   
  14.         void session_connect(std::vector<socket_session_ptr>& vSession);  
  15.         void session_connect(socket_session_ptr pSession);  
  16.         //socket_session_ptr get_session(std::string& addr);   
  17.         boost::asio::io_service& get_io_service() { return m_io_srv; }  
  18.   
  19.     protected:  
  20.         virtual void handle_read_data(message& msg, socket_session_ptr pSession) = 0;  
  21.   
  22.     private:  
  23.         boost::asio::io_service m_io_srv;  
  24.         boost::asio::io_service::work m_work;  
  25.         session_manager m_manager;  
  26.   
  27.         void handle_connect(const boost::system::error_code& error,  
  28.             tcp::resolver::iterator endpoint_iterator, socket_session_ptr pSession);  
  29.   
  30.         void close_callback(socket_session_ptr session);  
  31.         void read_data_callback(const boost::system::error_code& e,   
  32.             socket_session_ptr session, message& msg);  
  33.     };  
  34. }  
#pragma once
#include "socket_session.h"
#include "session_manager.h"
#include <boost/algorithm/string.hpp>
#include <firebird/message/message.hpp>

namespace firebird{
	class FIREBIRD_DECL client_socket_utils
	{
	public:
		client_socket_utils();
		~client_socket_utils();

		void session_connect(std::vector<socket_session_ptr>& vSession);
		void session_connect(socket_session_ptr pSession);
		//socket_session_ptr get_session(std::string& addr);
		boost::asio::io_service& get_io_service() { return m_io_srv; }

	protected:
		virtual void handle_read_data(message& msg, socket_session_ptr pSession) = 0;

	private:
		boost::asio::io_service m_io_srv;
		boost::asio::io_service::work m_work;
		session_manager m_manager;

		void handle_connect(const boost::system::error_code& error,
			tcp::resolver::iterator endpoint_iterator, socket_session_ptr pSession);

		void close_callback(socket_session_ptr session);
		void read_data_callback(const boost::system::error_code& e, 
			socket_session_ptr session, message& msg);
	};
}


//client_socket_utils.cpp

[cpp] view plain copy print ?
  1. #include "client_socket_utils.h"   
  2.   
  3. namespace firebird{  
  4.     client_socket_utils::client_socket_utils()  
  5.         :m_work(m_io_srv), m_manager(m_io_srv, CLIENT, 3)  
  6.     {  
  7.     }  
  8.   
  9.     client_socket_utils::~client_socket_utils()  
  10.     {  
  11.     }  
  12.   
  13.     void client_socket_utils::session_connect(std::vector<socket_session_ptr>& vSession)  
  14.     {  
  15.         for (int i = 0; i < vSession.size(); ++i)  
  16.         {  
  17.             session_connect(vSession[i]);  
  18.         }  
  19.     }  
  20.   
  21.     void client_socket_utils::session_connect(socket_session_ptr pSession)  
  22.     {  
  23.         std::string& addr = pSession->get_remote_addr();  
  24.         try{  
  25.             //注册关闭回调函数   
  26.             pSession->installCloseCallBack(boost::bind(&client_socket_utils::close_callback, this, _1));  
  27.             //注册读到数据回调函数   
  28.             pSession->installReadDataCallBack(boost::bind(&client_socket_utils::read_data_callback, this, _1, _2, _3));  
  29.   
  30.             std::vector<std::string> ip_port;  
  31.             boost::split(ip_port, addr, boost::is_any_of(":"));  
  32.   
  33.             if (ip_port.size() < 2)  
  34.             {  
  35.                 //throw std::runtime_error("ip 格式不正确！");   
  36.                 LOG4CXX_ERROR(firebird_log, "[" << addr << "] ip 格式不正确!");  
  37.                 return;  
  38.             }  
  39.   
  40.             tcp::resolver resolver(pSession->socket().get_io_service());  
  41.             tcp::resolver::query query(ip_port[0], ip_port[1]);  
  42.             tcp::resolver::iterator endpoint_iterator = resolver.resolve(query);  
  43.             //pSession->set_begin_endpoint(endpoint_iterator);//设置起始地址，以便重连   
  44.   
  45.             //由于客户端是不断重连的，即使还未连接也要保存该session   
  46.             m_manager.add_session(pSession);  
  47.   
  48.             tcp::endpoint endpoint = *endpoint_iterator;  
  49.             pSession->socket().async_connect(endpoint,  
  50.                 boost::bind(&client_socket_utils::handle_connect, this,  
  51.                 boost::asio::placeholders::error, ++endpoint_iterator, pSession));  
  52.         }  
  53.         catch(std::exception& e)  
  54.         {  
  55.             LOG4CXX_ERROR(firebird_log, KDS_CODE_INFO << "连接远程地址:[" << addr << "],socket异常:[" << e.what() << "]");  
  56.         }  
  57.         catch(...)  
  58.         {  
  59.             LOG4CXX_ERROR(firebird_log, KDS_CODE_INFO << "连接远程地址:[" << addr << "],socket异常:[未知异常]");  
  60.         }  
  61.     }  
  62.   
  63.     void client_socket_utils::handle_connect(const boost::system::error_code& error,  
  64.         tcp::resolver::iterator endpoint_iterator, socket_session_ptr pSession)  
  65.     {  
  66.         LOG4CXX_DEBUG(firebird_log, KDS_CODE_INFO << " enter.");  
  67.         std::string sLog;  
  68.         try{  
  69.             if (!error)  
  70.             {  
  71.                 LOG4CXX_FATAL(firebird_log, "服务器:[" << pSession->get_business_type() <<"],连接远程地址:[" << pSession->get_remote_addr().c_str() << "]成功!");  
  72.                 pSession->start();  
  73.   
  74.                 //向服务器注册服务类型   
  75.                 message msg;  
  76.                 msg.command = regist;  
  77.                 msg.business_type = pSession->get_business_type();  
  78.                 msg.app_id = pSession->get_app_id();  
  79.                 msg.data() = "R";  
  80.   
  81.                 pSession->async_write(msg);  
  82.             }  
  83.             else if (endpoint_iterator != tcp::resolver::iterator())  
  84.             {  
  85.                 LOG4CXX_ERROR(firebird_log, "连接远程地址:[" << pSession->get_remote_addr().c_str() << "]失败，试图重连下一个地址。");  
  86.                 pSession->socket().close();//此处用socket的close，不应用session的close触发连接，不然会导致一直重连   
  87.                 tcp::endpoint endpoint = *endpoint_iterator;  
  88.                 pSession->socket().async_connect(endpoint,  
  89.                     boost::bind(&client_socket_utils::handle_connect, this,  
  90.                     boost::asio::placeholders::error, ++endpoint_iterator, pSession));  
  91.             }  
  92.             else  
  93.             {  
  94.                 LOG4CXX_ERROR(firebird_log, KDS_CODE_INFO << "连接远程地址:[" << pSession->get_remote_addr().c_str() << "]失败！");  
  95.                 pSession->socket().close();//此处用socket的close，不应用session的close触发连接，不然会导致一直重连   
  96.             }  
  97.         }  
  98.         catch(std::exception& e)  
  99.         {  
  100.             LOG4CXX_ERROR(firebird_log, KDS_CODE_INFO << "连接远程地址:[" << pSession->get_remote_addr().c_str() <<"],socket异常:[" << e.what() << "]");  
  101.         }  
  102.         catch(...)  
  103.         {  
  104.             LOG4CXX_ERROR(firebird_log, KDS_CODE_INFO << "连接远程地址:[" << pSession->get_remote_addr().c_str() <<"],socket异常:[未知异常]");  
  105.         }  
  106.     }  
  107.   
  108.     void client_socket_utils::read_data_callback(const boost::system::error_code& e,   
  109.         socket_session_ptr session, message& msg)  
  110.     {  
  111.         LOG4CXX_DEBUG(firebird_log, "command =[" << msg.command << "],["   
  112.             << msg.business_type << "],[" << msg.data() << "]");  
  113.   
  114.         if (msg.command == heartbeat)  
  115.         {//心跳   
  116.         }  
  117.         else if (msg.command == regist)  
  118.         {//注册   
  119.             LOG4CXX_FATAL(firebird_log, "服务器:[" << session->get_business_type() <<"]注册成功。");  
  120.         }  
  121.         else if (msg.command == normal)  
  122.         {//业务数据   
  123.             handle_read_data(msg, session);  
  124.         }  
  125.         else   
  126.         {  
  127.             LOG4CXX_ERROR(firebird_log, KDS_CODE_INFO << "收到非法消息包!");  
  128.         }  
  129.     }  
  130.   
  131.     //关闭session就会重连   
  132.     void client_socket_utils::close_callback(socket_session_ptr session)  
  133.     {  
  134.         LOG4CXX_DEBUG(firebird_log, KDS_CODE_INFO << "enter.");  
  135.   
  136.         try{  
  137.             //tcp::resolver::iterator endpoint_iterator = context.session->get_begin_endpoint();   
  138.   
  139.             std::string& addr = session->get_remote_addr();  
  140.   
  141.             std::vector<std::string> ip_port;  
  142.             boost::split(ip_port, addr, boost::is_any_of(":"));  
  143.   
  144.             if (ip_port.size() < 2)  
  145.             {  
  146.                 LOG4CXX_ERROR(firebird_log, "[" << addr << "] ip 格式不正确!");  
  147.                 return;  
  148.             }  
  149.   
  150.             tcp::resolver resolver(session->socket().get_io_service());  
  151.             tcp::resolver::query query(ip_port[0], ip_port[1]);  
  152.             tcp::resolver::iterator endpoint_iterator = resolver.resolve(query);  
  153.   
  154.             tcp::endpoint endpoint = *endpoint_iterator;  
  155.             session->socket().async_connect(endpoint,  
  156.                 boost::bind(&client_socket_utils::handle_connect, this,  
  157.                 boost::asio::placeholders::error, ++endpoint_iterator, session));  
  158.         }  
  159.         catch(std::exception& e)  
  160.         {  
  161.             LOG4CXX_ERROR(firebird_log, KDS_CODE_INFO << "连接远程地址:[" << session->get_remote_addr().c_str() <<"],socket异常:[" << e.what() << "]");  
  162.         }  
  163.         catch(...)  
  164.         {  
  165.             LOG4CXX_ERROR(firebird_log, KDS_CODE_INFO << "连接远程地址:[" << session->get_remote_addr().c_str() <<"],socket异常:[未知异常]");  
  166.         }  
  167.     }  
  168. }  
#include "client_socket_utils.h"

namespace firebird{
	client_socket_utils::client_socket_utils()
		:m_work(m_io_srv), m_manager(m_io_srv, CLIENT, 3)
	{
	}

	client_socket_utils::~client_socket_utils()
	{
	}

	void client_socket_utils::session_connect(std::vector<socket_session_ptr>& vSession)
	{
		for (int i = 0; i < vSession.size(); ++i)
		{
			session_connect(vSession[i]);
		}
	}

	void client_socket_utils::session_connect(socket_session_ptr pSession)
	{
		std::string& addr = pSession->get_remote_addr();
		try{
			//注册关闭回调函数
			pSession->installCloseCallBack(boost::bind(&client_socket_utils::close_callback, this, _1));
			//注册读到数据回调函数
			pSession->installReadDataCallBack(boost::bind(&client_socket_utils::read_data_callback, this, _1, _2, _3));

			std::vector<std::string> ip_port;
			boost::split(ip_port, addr, boost::is_any_of(":"));

			if (ip_port.size() < 2)
			{
				//throw std::runtime_error("ip 格式不正确！");
				LOG4CXX_ERROR(firebird_log, "[" << addr << "] ip 格式不正确!");
				return;
			}

			tcp::resolver resolver(pSession->socket().get_io_service());
			tcp::resolver::query query(ip_port[0], ip_port[1]);
			tcp::resolver::iterator endpoint_iterator = resolver.resolve(query);
			//pSession->set_begin_endpoint(endpoint_iterator);//设置起始地址，以便重连

			//由于客户端是不断重连的，即使还未连接也要保存该session
			m_manager.add_session(pSession);

			tcp::endpoint endpoint = *endpoint_iterator;
			pSession->socket().async_connect(endpoint,
				boost::bind(&client_socket_utils::handle_connect, this,
				boost::asio::placeholders::error, ++endpoint_iterator, pSession));
		}
		catch(std::exception& e)
		{
			LOG4CXX_ERROR(firebird_log, KDS_CODE_INFO << "连接远程地址:[" << addr << "],socket异常:[" << e.what() << "]");
		}
		catch(...)
		{
			LOG4CXX_ERROR(firebird_log, KDS_CODE_INFO << "连接远程地址:[" << addr << "],socket异常:[未知异常]");
		}
	}

	void client_socket_utils::handle_connect(const boost::system::error_code& error,
		tcp::resolver::iterator endpoint_iterator, socket_session_ptr pSession)
	{
		LOG4CXX_DEBUG(firebird_log, KDS_CODE_INFO << " enter.");
		std::string sLog;
		try{
			if (!error)
			{
				LOG4CXX_FATAL(firebird_log, "服务器:[" << pSession->get_business_type() <<"],连接远程地址:[" << pSession->get_remote_addr().c_str() << "]成功!");
				pSession->start();

				//向服务器注册服务类型
				message msg;
				msg.command = regist;
				msg.business_type = pSession->get_business_type();
				msg.app_id = pSession->get_app_id();
				msg.data() = "R";

				pSession->async_write(msg);
			}
			else if (endpoint_iterator != tcp::resolver::iterator())
			{
				LOG4CXX_ERROR(firebird_log, "连接远程地址:[" << pSession->get_remote_addr().c_str() << "]失败，试图重连下一个地址。");
				pSession->socket().close();//此处用socket的close，不应用session的close触发连接，不然会导致一直重连
				tcp::endpoint endpoint = *endpoint_iterator;
				pSession->socket().async_connect(endpoint,
					boost::bind(&client_socket_utils::handle_connect, this,
					boost::asio::placeholders::error, ++endpoint_iterator, pSession));
			}
			else
			{
				LOG4CXX_ERROR(firebird_log, KDS_CODE_INFO << "连接远程地址:[" << pSession->get_remote_addr().c_str() << "]失败！");
				pSession->socket().close();//此处用socket的close，不应用session的close触发连接，不然会导致一直重连
			}
		}
		catch(std::exception& e)
		{
			LOG4CXX_ERROR(firebird_log, KDS_CODE_INFO << "连接远程地址:[" << pSession->get_remote_addr().c_str() <<"],socket异常:[" << e.what() << "]");
		}
		catch(...)
		{
			LOG4CXX_ERROR(firebird_log, KDS_CODE_INFO << "连接远程地址:[" << pSession->get_remote_addr().c_str() <<"],socket异常:[未知异常]");
		}
	}

	void client_socket_utils::read_data_callback(const boost::system::error_code& e, 
		socket_session_ptr session, message& msg)
	{
		LOG4CXX_DEBUG(firebird_log, "command =[" << msg.command << "],[" 
			<< msg.business_type << "],[" << msg.data() << "]");

		if (msg.command == heartbeat)
		{//心跳
		}
		else if (msg.command == regist)
		{//注册
			LOG4CXX_FATAL(firebird_log, "服务器:[" << session->get_business_type() <<"]注册成功。");
		}
		else if (msg.command == normal)
		{//业务数据
			handle_read_data(msg, session);
		}
		else 
		{
			LOG4CXX_ERROR(firebird_log, KDS_CODE_INFO << "收到非法消息包!");
		}
	}

	//关闭session就会重连
	void client_socket_utils::close_callback(socket_session_ptr session)
	{
		LOG4CXX_DEBUG(firebird_log, KDS_CODE_INFO << "enter.");

		try{
			//tcp::resolver::iterator endpoint_iterator = context.session->get_begin_endpoint();

			std::string& addr = session->get_remote_addr();

			std::vector<std::string> ip_port;
			boost::split(ip_port, addr, boost::is_any_of(":"));

			if (ip_port.size() < 2)
			{
				LOG4CXX_ERROR(firebird_log, "[" << addr << "] ip 格式不正确!");
				return;
			}

			tcp::resolver resolver(session->socket().get_io_service());
			tcp::resolver::query query(ip_port[0], ip_port[1]);
			tcp::resolver::iterator endpoint_iterator = resolver.resolve(query);

			tcp::endpoint endpoint = *endpoint_iterator;
			session->socket().async_connect(endpoint,
				boost::bind(&client_socket_utils::handle_connect, this,
				boost::asio::placeholders::error, ++endpoint_iterator, session));
		}
		catch(std::exception& e)
		{
			LOG4CXX_ERROR(firebird_log, KDS_CODE_INFO << "连接远程地址:[" << session->get_remote_addr().c_str() <<"],socket异常:[" << e.what() << "]");
		}
		catch(...)
		{
			LOG4CXX_ERROR(firebird_log, KDS_CODE_INFO << "连接远程地址:[" << session->get_remote_addr().c_str() <<"],socket异常:[未知异常]");
		}
	}
}


 

5.对象串行化

socket_session发送和接收数据包的时候使用到了对象串行化，我这里是通过thrift实现的，其实boost的serialization库也提供了这样的功能，使用起来更为方便，但我在测试过程中，thrift相比之下性能会高很多，因此就坚持使用thrift了，感兴趣的话可以看我之前写的《使用thrift串行化对象》和《轻量级序列化库boost serialization》。

5.1字符串与thrift对象的相互转换

[cpp] view plain copy print ?
  1. #pragma once   
  2. #include <boost/shared_ptr.hpp>   
  3. #include <transport/TBufferTransports.h>   
  4. #include <protocol/TProtocol.h>   
  5. #include <protocol/TBinaryProtocol.h>   
  6.   
  7. namespace firebird{  
  8.     using namespace apache::thrift;  
  9.     using namespace apache::thrift::transport;  
  10.     using namespace apache::thrift::protocol;  
  11.   
  12.     template<typename T>  
  13.     void thrift_iserialize(T& stu, std::string& s)  
  14.     {  
  15.         boost::shared_ptr<TMemoryBuffer> trans(new TMemoryBuffer((uint8_t*)&s[0], s.size()));  
  16.         boost::shared_ptr<TProtocol> proto(new TBinaryProtocol(trans));  
  17.         stu.read(proto.get());  
  18.     }  
  19.   
  20.     template<typename T>  
  21.     void thrift_oserialize(T& stu, std::string& s)  
  22.     {  
  23.         boost::shared_ptr<TMemoryBuffer> trans(new TMemoryBuffer());  
  24.         boost::shared_ptr<TProtocol> proto(new TBinaryProtocol(trans));  
  25.         stu.write(proto.get());  
  26.         s = trans->getBufferAsString();  
  27.     }  
  28. }  
#pragma once
#include <boost/shared_ptr.hpp>
#include <transport/TBufferTransports.h>
#include <protocol/TProtocol.h>
#include <protocol/TBinaryProtocol.h>

namespace firebird{
	using namespace apache::thrift;
	using namespace apache::thrift::transport;
	using namespace apache::thrift::protocol;

	template<typename T>
	void thrift_iserialize(T& stu, std::string& s)
	{
		boost::shared_ptr<TMemoryBuffer> trans(new TMemoryBuffer((uint8_t*)&s[0], s.size()));
		boost::shared_ptr<TProtocol> proto(new TBinaryProtocol(trans));
		stu.read(proto.get());
	}

	template<typename T>
	void thrift_oserialize(T& stu, std::string& s)
	{
		boost::shared_ptr<TMemoryBuffer> trans(new TMemoryBuffer());
		boost::shared_ptr<TProtocol> proto(new TBinaryProtocol(trans));
		stu.write(proto.get());
		s = trans->getBufferAsString();
	}
}


5.2通过thrift对象，普通的对象与字符串的相互转换

[cpp] view plain copy print ?
  1. #pragma once   
  2.   
  3. #include "message_archive.hpp"   
  4. #include <firebird/archive/thrift_archive.hpp>   
  5. #include <firebird/message/TMessage_types.h>   
  6.   
  7. namespace firebird  
  8. {  
  9.     /*** message to ThriftMessage ***/  
  10.     void msg_to_tmsg(TMessage& tmsg, message& msg)  
  11.     {  
  12.         //设置   
  13.         tmsg.command = msg.command;  
  14.         tmsg.business_type = msg.business_type;  
  15.         tmsg.app_id = msg.app_id;  
  16.         //设置context   
  17.         tmsg.context.cmdVersion = msg.context().cmdVersion;  
  18.         tmsg.context.cpid.swap(msg.context().cpid);  
  19.         tmsg.context.remote_ip.swap(msg.context().remote_ip);  
  20.         tmsg.context.wSerialNumber = msg.context().wSerialNumber;  
  21.         tmsg.context.session_id = msg.context().session_id;  
  22.   
  23.         //设置source   
  24.         for (int i = 0; i < msg.source().size(); ++i)  
  25.         {  
  26.             tmsg.source.push_back(msg.source()[i]);  
  27.         }  
  28.   
  29.         //设置destination   
  30.         for (int i = 0; i < msg.destination().size(); ++i)  
  31.         {  
  32.             tmsg.destination.push_back(msg.destination()[i]);  
  33.         }  
  34.   
  35.         //设置data   
  36.         tmsg.data = msg.data();  
  37.     }  
  38.   
  39.     /*** ThriftMessage to message ***/  
  40.     void tmsg_to_msg(message& msg, TMessage& tmsg)  
  41.     {  
  42.         //设置   
  43.         msg.command = tmsg.command;  
  44.         msg.business_type = tmsg.business_type;  
  45.         msg.app_id = tmsg.app_id;  
  46.   
  47.         //设置context   
  48.         msg.context().cmdVersion = tmsg.context.cmdVersion;  
  49.         msg.context().cpid = tmsg.context.cpid;  
  50.         msg.context().remote_ip = tmsg.context.remote_ip;  
  51.         msg.context().wSerialNumber = tmsg.context.wSerialNumber;  
  52.         msg.context().session_id = tmsg.context.session_id;  
  53.   
  54.         //设置source   
  55.         for (int i = 0; i < tmsg.source.size(); ++i)  
  56.         {  
  57.             msg.source() << tmsg.source[i];  
  58.         }  
  59.   
  60.         //设置destination   
  61.         for (int i = 0; i < tmsg.destination.size(); ++i)  
  62.         {  
  63.             msg.destination() << tmsg.destination[i];  
  64.         }  
  65.   
  66.         //设置data   
  67.         msg.data() = tmsg.data;  
  68.     }  
  69.   
  70.     void message_iarchive(message& msg, std::string& s)  
  71.     {  
  72.         TMessage tmsg;  
  73.         thrift_iserialize(tmsg, s);  
  74.         tmsg_to_msg(msg, tmsg);  
  75.     }  
  76.   
  77.     void message_oarchive(std::string& s, message& msg)  
  78.     {  
  79.         TMessage tmsg;  
  80.         msg_to_tmsg(tmsg, msg);  
  81.         thrift_oserialize(tmsg, s);  
  82.     }  
  83. }  

 

yinchuofan1985
关注
  • 1
    点赞
  • 踩
  • 8
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
专栏目录
Boost.asio实现同步的TCP/IP通信
KnightOnHorse的博客
05-05 1753
服务器【如果想保留套接字之后继续通信，可以动态申请socket_type，保存指针，因为socket_type貌似不能拷贝】#include "stdafx.h" #include &lt;iostream&gt; #include &lt;boost/asio.hpp&gt; using namespace boost::asio; using namespace std; int main...
基于boost::asio的异步tcp通信框架
05-07 2949
最近在做ROS与树莓派进行通信时，发现boost:asio库还蛮强大的，就试着用asio写了一个简单的使用tcp传输轨迹数据到树莓派，让树莓派控制步进电机旋转的代码。其实也就是将asio的功能封装成了两个类而已。asio_server类运行于树莓派上作为服务器，asio_client运行于PC上作为客户端。刚开始写的代码仅仅是客户端发送，服务器端接收这么简单。但后来一想，树莓派不是还要返回数据给R...
VS2022通过C++网络库Boost.asio搭建一个简单TCP异步服务器和客户端
最新发布
LorraineWen的博客
05-23 1381
上一篇博客我们介绍了通过Boost.asio搭建一个TCP同步服务器和客户端，这次我们再通过asio搭建一个异步通信的服务器和客户端系统，由于这是一个简单异步服务器，所以我们的异步特指异步服务器而不是异步客户端，同步服务器在处理一个请求时会阻塞其他请求，而异步服务器可以同时处理多个请求，不会阻塞其他请求的处理，客户端一般是不会处理其他客户端请求的，所以客户端仍旧使用同步模式。（本次博客使用的Boost库版本是1.84.0)
boost::asio 中async_read与async_receive的区别
热门推荐
无与伦比BLOG
04-13 1万+
boost::asio 中async_read与async_receive的区别 现象：在调用async_read后，即使收到数据包，也没有调用相应的ReadHandler 回调函数 void handle_connect(const boost::system::error_code & error,        tcp::resolver::iterator endpoin
boost::asio::async_read、boost::asio::async_write 模板函数的简化版实现逻辑。
liulilittle的博客
08-03 1250
write_yield 等价 boost::asio::async_write 模板函数（stackful 协同程式）read_yield 等价 boost::asio::async_read 模板函数（stackful 协同程式）write_async 等价 boost::asio::async_write 模板函数。read_async 等价 boost::asio::async_read 模板函数。...............
Ubuntu 上采用 Boost asio 异步 TCP 通讯boostasiotcp-master.zip
05-08
本教程将详细探讨如何在 Ubuntu 操作系统上利用 Boost.Asio 实现高效的异步 TCP 通信。Ubuntu 的稳定性和跨平台兼容性使其成为开发此类应用的理想选择。 首先，我们需要了解 Boost.Asio 是什么。Boost.Asio 是 ...
boost asio tcp服务器的实现
04-24
总的来说，实现一个`boost::asio`的TCP服务器需要掌握网络编程的基本概念，理解TCP协议的工作原理，熟悉Boost.Asio库的使用，特别是异步I/O操作。同时，利用C++11的智能指针增强代码的健壮性，结合IOCP提升服务器...
Ubuntu 上采用 Boost asio 异步 TCP 通讯，示例代码.rar
06-05
基于boost asio实现server端通信，采用one by one的同步处理方式，并且设置连接等待超时。下面给出了string和byte两种数据类型的通信方式，可覆盖基本通信场景需求。  规定server与client双方交互的数据格式是...
Boost asio异步TCP通讯及tcp粘包解包解决方案.doc
09-26
Boost asio异步TCP通讯及tcp粘包解包解决方案.doc
c++ boost asio网络库 异步tcp方式文件传输 速度在10M左右.zip
01-18
在这个场景中，我们讨论的是使用Boost.ASIO实现的异步TCP文件传输，其传输速度可以达到大约10MB/s。 1. **Boost.ASIO库基础** Boost.ASIO是Boost库的一部分，它提供了跨平台的网络编程接口。它支持多种协议，包括...
boost库之tcp客户端实例
深之JohnChen的专栏
02-03 4756
tcpclient.h   #pragma once #include &lt;boost/function.hpp&gt; #include &lt;boost/bind.hpp&gt; #include &lt;boost/asio.hpp&gt; #include &lt;boost/shared_ptr.hpp&gt; #include &lt;iostream&gt; #incl...
async_read_some async_read区别
飞翔的大象
04-17 5831
今天用asio写一个简单的server时，当连接成功时做如下工作1 asio::async_read(socket_, asio::buffer(&m_packet, sizeof(m_packet)),2 boost::bind(&tcp_connection::HandleReadCardKey, shared_from_this(),3 asio::placeholders::error))
asio入门资料
jacke121的专栏
04-20 2915
下载地址： https://think-async.com/Asio/ 选择asio独立的版本。 server：这个server 有时会有乱码的问题： // asio_server.cpp : 定义控制台应用程序的入口点。 // #include "stdafx.h" #include <asio.hpp> using namespace asio; usin...
boost.asio无锁异步并发
Erick Lv的笔记
02-21 2059
简介 给出一个不安全的情况，假设我们有一个socket用于和用户通信，用户会发送多个请求，我们处理请求需要一定的时间，同时使用异步并发的模型来处理对应的请求，即一个io_context::run会有多个线程执行。那么boost::asio的基本代码框架如下，给出伪代码示例： var scoket; // 一个socket async_read(socket, buf, [](){ // 处理一...
C++ 建立本地网络服务器 （Boost.Asio库）
学习是一种信仰。
03-12 8206
详细的注释讲解利用Boost库建立服务器，最终自己也写了一个具有并发功能的boost服务器，可供直接调用。
boost_asio学习笔记[2] - 客户端异步通讯
weixin_34087307的博客
06-22 149
2019独角兽企业重金招聘Python工程师标准>>> ...
【boost网络库从青铜到王者】第六篇:asio网络编程中的socket异步读(接收)写(发送)
森明帮大于黑虎帮的博客
08-24 1449
【boost网络库从青铜到王者】第六篇:asio网络编程中的socket异步读(接收)写(发送)
Boost Asio异步发送数据（async_write）崩溃问题记录
guotianqing的博客
03-03 3749
背景 服务端与客户端之间的网络通信（使用Boost Asio库异步编程模式实现），客户端会向服务端请求数据。 在刚开始的测试中，是没有出现问题的。后来有一次测试时，服务端查询完数据后，向客户端发送时总是崩溃。 通过gdb调试，可以发现是在调用到异步发送函数（boost::asio::async_write）后崩溃的。 打印的栈信息如下： Program terminated with signa...
boost asio tcp异步通信
09-13
要使用 Boost.Asio 进行 TCP 异步通信，首先需要创建一个 io_context 对象，它负责处理底层的 I/O 操作。然后，创建一个 tcp::socket 对象，用于建立和管理 TCP 连接。接下来，使用异步操作来进行连接、读取和写入...

“相关推荐”对你有帮助么？

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值