数据库MySql类库系列(九)-DBServicePool

第七个工具类：DBServicePool

是针对前文提到的DBService的一个管理器

用于多线程环境下，可能有需求需要创建多个DBService，使得每个线程有机会可以独享一个DBService

所以有了这样一个DBService的管理器

提供Start，Stop，用来运行/停止所有DBService对象

提供Ping，用来维持所有DBService对象的心跳

提供一个虚接口Service，子类可以实现管理自己的继承自DBService的子类对象

比如：

实现一个MyDBService : public DBService

实现一个MyDBServicePool : public DBServicePool

DBServicePool的Service接口，实现的是创建一个DBService对象，返回其shared_ptr

MyDBServicePool 的Service接口，可以自行实现创建一个MyDBService对象，返回其shared_ptr

2017.3.10修改：

改为模板实现

模板类型T，T要求为DBService或其子类

Start并入构造函数，构造时创建指定个数的T对象，可以指定为0个，即不创建任何DBService或其子类对象

虚接口Service删除，不再需要，创建时，直接new T(...)

GetDBService：获取一个DBService对象，当Pool中没有DBService对象时（构造时指定了个数为0），抛一个异常

修改后代码：

DBServicePool.h

#ifndef __DBServicePool_H__
#define	__DBServicePool_H__

#include <boost/noncopyable.hpp>
#include <boost/thread.hpp>
#include <boost/shared_ptr.hpp>

#include "DBService.h"

namespace common{
	namespace db{

		template <class T>
		class DBServicePool : private boost::noncopyable
		{
		public:
			DBServicePool(unsigned int size,
				const std::string& hostname,
				unsigned int port,
				const std::string& username,
				const std::string& userkey,
				const std::string& dbname);
			virtual ~DBServicePool();

			// 停止所有DBService对象
			void Stop();

			DBService& GetDBService();

			// 获取IOService对象个数
			std::size_t Size();

			// 维持链接心跳
			void Ping();

		private:
			boost::mutex m_Lock;

			// DBService列表
			std::vector<boost::shared_ptr<DBService> > m_DBServices;

			// 顺序使用的下一个DBService下标
			unsigned int m_Next;
		};

		template <class T>
		DBServicePool<T>::DBServicePool(unsigned int size,
			const std::string& hostname,
			unsigned int port,
			const std::string& username,
			const std::string& userkey,
			const std::string& dbname)
		{
			for (unsigned int i = 0; i < size; i++)
			{
				boost::shared_ptr<DBService> dbService(new T());
				if (NULL != dbService)
				{
					dbService->Start(hostname, port, username, userkey, dbname);
					m_DBServices.push_back(dbService);
				}
			}

			m_Next = 0;
		}

		template <class T>
		DBServicePool<T>::~DBServicePool()
		{

		}

		template <class T>
		void DBServicePool<T>::Stop()
		{
			boost::mutex::scoped_lock lock(m_Lock);

			for (std::size_t i = 0; i < m_DBServices.size(); ++i)
			{
				m_DBServices[i]->Stop();
			}
		}

		template <class T>
		DBService& DBServicePool<T>::GetDBService()
		{
			boost::mutex::scoped_lock lock(m_Lock);

			if (0 < m_DBServices.size())
			{
				m_Next++;
				if (m_Next >= m_DBServices.size())
				{
					m_Next = 0;
				}

				return *m_DBServices[m_Next];
			}
			else
			{
				throw std::string("DBServicePool size is 0");
			}
		}

		template <class T>
		std::size_t DBServicePool<T>::Size()
		{
			boost::mutex::scoped_lock lock(m_Lock);

			return m_DBServices.size();
		}

		template <class T>
		void DBServicePool<T>::Ping()
		{
			boost::mutex::scoped_lock lock(m_Lock);

			for (std::size_t i = 0; i < m_DBServices.size(); ++i)
			{
				m_DBServices[i]->Ping();
			}
		}

	}
}

#endif