#ifndef _QTL_MYSQL_POOL_H_
|
#define _QTL_MYSQL_POOL_H_
|
|
#include "qtl_database_pool.hpp"
|
#include "qtl_postgres.hpp"
|
|
namespace qtl
|
{
|
|
namespace postgres
|
{
|
|
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_port, m_database.data()))
|
{
|
delete db;
|
db = NULL;
|
}
|
else
|
{
|
PQsetClientEncoding(db->handle(), "UTF8");
|
}
|
return db;
|
}
|
|
protected:
|
std::string m_host;
|
unsigned short m_port;
|
std::string m_database;
|
std::string m_user;
|
std::string m_password;
|
};
|
|
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;
|
db->open(ev, [this, handler, db](const postgres::error &e) mutable
|
{
|
if (e)
|
{
|
delete db;
|
db = nullptr;
|
}
|
handler(e, db); }, m_params);
|
}
|
|
protected:
|
std::map<std::string, std::string> m_params;
|
};
|
|
}
|
|
}
|
|
#endif //_QTL_MYSQL_POOL_H_
|
