数据库MySql类库系列(二)-DBService

第二个工具类：DBService

用于MySQL数据库服务类的基类。主要处理数据库连接的建立（Start），断开（Stop），维持心跳（Ping）。

基于之前的DBOperator实现。

该基类提供两个虚接口，ProcessStart，ProcessStop，交由子类去实现子类自己的初始化工作，和释放工作。

建立连接（Start）时，会断开现有连接（如果有的话，会先调用ProcessStop做清理工作），然后建立新连接，连接建立好之后调用ProcessStart。

断开连接（Stop）时，调用ProcessStop做清理工作，然后断开现有连接。

维持心跳（Ping）时，如果发现连接断开，则重新连接。如果连不上则放弃，等待下一次Ping，或逻辑主动调用Start重新建立连接。

DBService中保存一个建立好的数据库连接，并加锁，保证多线程同时访问DBService时线程安全。

具体会在这个连接上，做哪些数据库操作，做什么样的增删改查，由子类实现，它跟具体业务逻辑相关。

DBService.h

#ifndef __DBService_H__
#define	__DBService_H__

#include <boost/thread.hpp>

struct st_mysql;
typedef struct st_mysql MYSQL;

namespace common{
	namespace db{

		class DBService
		{
		public:
			DBService();
			virtual ~DBService();

		public:
			// 建立数据库连接
			bool Start(const std::string& hostname,
				unsigned int port,
				const std::string& username,
				const std::string& userkey,
				const std::string& dbname);
			virtual bool ProcessStart();

			// 维持数据库连接心跳，校验链接是否断开，断开则重连
			void Ping();

			// 断开数据库连接
			void Stop();
			virtual void ProcessStop();

		protected:
			// 数据库连接
			boost::mutex m_Lock;
			MYSQL *m_Connect;

		private:
			// 建立新链接
			bool Connect();
			// 终止链接
			void DisConnect();

			// 数据库连接参数
			std::string m_HostName;
			unsigned int m_Port;
			std::string m_UserName;
			std::string m_UserKey;
			std::string m_DBName;
		};

	}
}

#endif

DBService.cpp

#include "DBService.h"

#include "Logger.h"

using namespace common::tool;

#include "DBDefine.h"
#include "DBOperator.h"

namespace common{
	namespace db{

		DBService::DBService()
		{
			m_Connect = NULL;
		}

		DBService::~DBService()
		{
			DisConnect();
		}

		bool DBService::Start(const std::string& hostname,
			unsigned int port,
			const std::string& username,
			const std::string& userkey,
			const std::string& dbname)
		{
			boost::mutex::scoped_lock lock(m_Lock);

			DisConnect();

			m_HostName = hostname;
			m_Port = port;
			m_UserName = username;
			m_UserKey = userkey;
			m_DBName = dbname;

			return Connect();
		}

		bool DBService::ProcessStart()
		{
			return false;
		}

		void DBService::Ping()
		{
			boost::mutex::scoped_try_lock lock(m_Lock);
			if (lock.owns_lock())
			{
				if (!DBOperator::Ping(m_Connect))
				{
					DisConnect();
					Connect();
				}
			}
		}

		void DBService::Stop()
		{
			boost::mutex::scoped_lock lock(m_Lock);

			DisConnect();
		}

		void DBService::ProcessStop()
		{

		}

		bool DBService::Connect()
		{
			m_Connect = DBOperator::Connect(m_HostName.c_str(), m_Port, m_UserName.c_str(), m_UserKey.c_str(), m_DBName.c_str());
			if (NULL != m_Connect)
			{
				if (ProcessStart())
				{
					LOG_INFO(g_LibDBLog) << "connect  " << m_HostName << ":" << m_DBName << "@" << m_UserName << " success";
					return true;
				}
				else
				{
					LOG_INFO(g_LibDBLog) << "connect  " << m_HostName << ":" << m_DBName << "@" << m_UserName << " error";
					DisConnect();
					return false;
				}
			}
			else
			{
				LOG_INFO(g_LibDBLog) << "connect  " << m_HostName << ":" << m_DBName << "@" << m_UserName << " error";
				return false;
			}
		}

		void DBService::DisConnect()
		{
			if (NULL != m_Connect)
			{
				LOG_INFO(g_LibDBLog) << "close  " << m_HostName << ":" << m_DBName << "@" << m_UserName;

				ProcessStop();

				DBOperator::DisConnect(m_Connect);
				m_Connect = NULL;
			}
		}

	}
}