#ifndef _QTL_MYSQL_POOL_H_
#define _QTL_MYSQL_POOL_H_

#include "qtl_database_pool.hpp"
#include "qtl_mysql.hpp"

namespace qtl
{

	namespace mysql
	{

		class database_pool : public qtl::database_pool<database>
		{
		public:
			database_pool() : m_port(0) { }
			virtual ~database_pool() { }
			virtual database* new_database() throw() override
			{
				database* db = new database;
				if (!db->open(m_host.data(), m_user.data(), m_password.data(), m_database.data(), 0, m_port))
				{
					delete db;
					db = NULL;
				}
				else
				{
					db->charset_name("utf8");
					//设置为自动重连
					db->reconnect(true);
				}
				return db;
			}

		protected:
			std::string m_host;
			unsigned short m_port;
			std::string m_database;
			std::string m_user;
			std::string m_password;
		};

#if MARIADB_VERSION_ID >= 050500

		template<typename EventLoop>
		class async_pool : public qtl::async_pool<async_pool<EventLoop>, EventLoop, async_connection>
		{
			typedef qtl::async_pool<async_pool<EventLoop>, EventLoop, async_connection> base_class;
		public:
			async_pool(EventLoop& ev) : base_class(ev), m_port(0) { }
			virtual ~async_pool() { }

			template<typename Handler>
			void new_connection(EventLoop& ev, Handler&& handler) throw()
			{
				async_connection* db = new async_connection;
				db->open(ev, [this, handler, db](const mysql::error& e) mutable {
					if (e)
					{
						delete db;
						db = nullptr;
					}
					else
					{
						db->charset_name("utf8");
					}
					handler(e, db);
					}, m_host.data(), m_user.data(), m_password.data(), m_database.data(), 0, m_port);
			}

		protected:
			std::string m_host;
			unsigned short m_port;
			std::string m_database;
			std::string m_user;
			std::string m_password;
		};

#endif //MariaDB

	}

}

#endif //_QTL_MYSQL_POOL_H_