#ifndef _QTL_ODBC_POOL_HPP_
#define _QTL_ODBC_POOL_HPP_

#include "qtl_database_pool.hpp"
#include "qtl_odbc.hpp"

namespace qtl
{

namespace odbc
{

class database_pool : public qtl::database_pool<database>
{
public:
	database_pool() { }
	virtual ~database_pool() { }
	virtual database* new_database() throw() override
	{
		database* db=NULL;
		try
		{
			db=new database(m_env);
			db->open(m_connection);
		}
		catch(error& e)
		{
			if(db)
			{
				delete db;
				db=NULL;
			}
		}
		return db;
	}

protected:
	std::string m_connection;
	environment m_env;
};

#ifdef QTL_ODBC_ENABLE_ASYNC_MODE

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) { }
	virtual ~async_pool() { }

	template<typename Handler>
	void new_connection(EventLoop& ev, Handler&& handler) throw()
	{
		async_connection* db = new async_connection(env);
		db->open(ev, [this, handler, db](const mysql::error& e) mutable {
			if (e)
			{
				delete db;
				db = nullptr;
			}
			handler(e, db);
		}, m_connection);
	}

protected:
	std::string m_connection;
	environment m_env;
};

#endif //QTL_ODBC_ENABLE_ASYNC_MODE

}

}

#endif //_QTL_ODBC_POOL_HPP_