PreVerify/lib/qtl/include/qtl_postgres.hpp

2899 lines
67 KiB
C++
Raw Normal View History

2024-09-09 19:18:28 +08:00
#ifndef _SQL_POSTGRES_H_
#define _SQL_POSTGRES_H_
#pragma once
#include <string>
#include <map>
#include <vector>
#include <array>
#include <exception>
#include <sstream>
#include <chrono>
#include <algorithm>
#include <assert.h>
#include "qtl_common.hpp"
#include "qtl_async.hpp"
#define FRONTEND
#include <libpq-fe.h>
#include <libpq/libpq-fs.h>
#include <pgtypes_error.h>
#include <pgtypes_interval.h>
#include <pgtypes_timestamp.h>
#include <pgtypes_numeric.h>
#include <pgtypes_date.h>
extern "C"
{
#include <c.h>
#include <catalog/pg_type.h>
}
#ifdef open
#undef open
#endif //open
#ifdef vsnprintf
#undef vsnprintf
#endif
#ifdef snprintf
#undef snprintf
#endif
#ifdef sprintf
#undef sprintf
#endif
#ifdef vfprintf
#undef vfprintf
#endif
#ifdef fprintf
#undef fprintf
#endif
#ifdef printf
#undef printf
#endif
#ifdef rename
#undef rename
#endif
#ifdef unlink
#undef unlink
#endif
#if defined(_WIN32) && _WIN32_WINNT < 0x0601
#ifdef _M_IX86
#define _WS2_32_WINSOCK_SWAP_LONGLONG(l) \
( ( ((l) >> 56) & 0x00000000000000FFLL ) | \
( ((l) >> 40) & 0x000000000000FF00LL ) | \
( ((l) >> 24) & 0x0000000000FF0000LL ) | \
( ((l) >> 8) & 0x00000000FF000000LL ) | \
( ((l) << 8) & 0x000000FF00000000LL ) | \
( ((l) << 24) & 0x0000FF0000000000LL ) | \
( ((l) << 40) & 0x00FF000000000000LL ) | \
( ((l) << 56) & 0xFF00000000000000LL ) )
#ifndef htonll
__inline unsigned __int64 htonll(unsigned __int64 Value)
{
const unsigned __int64 Retval = _WS2_32_WINSOCK_SWAP_LONGLONG(Value);
return Retval;
}
#endif /* htonll */
#ifndef ntohll
__inline unsigned __int64 ntohll(unsigned __int64 Value)
{
const unsigned __int64 Retval = _WS2_32_WINSOCK_SWAP_LONGLONG(Value);
return Retval;
}
#endif /* ntohll */
#endif
#endif
namespace qtl
{
namespace postgres
{
namespace detail
{
inline int16_t ntoh(int16_t v)
{
return static_cast<int16_t>(ntohs(v));
}
inline uint16_t ntoh(uint16_t v)
{
return ntohs(v);
}
inline int32_t ntoh(int32_t v)
{
return static_cast<int32_t>(ntohl(v));
}
inline uint32_t ntoh(uint32_t v)
{
return ntohl(v);
}
inline uint64_t ntoh(uint64_t v)
{
#ifdef _WIN32
return ntohll(v);
#else
return be64toh(v);
#endif
}
inline int64_t ntoh(int64_t v)
{
return ntoh(static_cast<uint64_t>(v));
}
template<typename T, typename = typename std::enable_if<std::is_integral<T>::value && !std::is_const<T>::value>::type>
inline T& ntoh_inplace(T& v)
{
v = ntoh(v);
return v;
}
inline int16_t hton(int16_t v)
{
return static_cast<int16_t>(htons(v));
}
inline uint16_t hton(uint16_t v)
{
return htons(v);
}
inline int32_t hton(int32_t v)
{
return static_cast<int32_t>(htonl(v));
}
inline uint32_t hton(uint32_t v)
{
return htonl(v);
}
inline uint64_t hton(uint64_t v)
{
#ifdef _WIN32
return htonll(v);
#else
return htobe64(v);
#endif
}
inline int64_t hton(int64_t v)
{
return hton(static_cast<uint64_t>(v));
}
template<typename T, typename = typename std::enable_if<std::is_integral<T>::value && !std::is_const<T>::value>::type>
inline T& hton_inplace(T& v)
{
v = hton(v);
return v;
}
template<typename T, typename = typename std::enable_if<std::is_integral<T>::value && !std::is_const<T>::value>::type>
std::pair<std::vector<char>::iterator, size_t> push(std::vector<char>& buffer, T v)
{
v = hton_inplace(v);
char* data = reinterpret_cast<char*>(&v);
auto it = buffer.insert(buffer.end(), data, data + sizeof(T));
return std::make_pair(it, sizeof(T));
}
template<typename T, typename = typename std::enable_if<std::is_integral<T>::value && !std::is_const<T>::value>::type>
const char* pop(const char* data, T& v)
{
v = ntoh(*reinterpret_cast<const T*>(data));
return data + sizeof(T);
}
}
class base_database;
class result;
class error : public std::exception
{
public:
error() : m_errmsg() { }
explicit error(PGconn* conn, PGVerbosity verbosity = PQERRORS_DEFAULT, PGContextVisibility show_context = PQSHOW_CONTEXT_ERRORS)
{
//PQsetErrorVerbosity(conn, verbosity);
//PQsetErrorContextVisibility(conn, show_context);
const char* errmsg = PQerrorMessage(conn);
if (errmsg) m_errmsg = errmsg;
else m_errmsg.clear();
}
explicit error(PGresult* res)
{
const char* errmsg = PQresultErrorMessage(res);
if (errmsg) m_errmsg = errmsg;
else m_errmsg.clear();
}
explicit error(const char* errmsg) : m_errmsg(errmsg) { }
virtual const char* what() const NOEXCEPT override { return m_errmsg.data(); }
operator bool() const { return !m_errmsg.empty(); }
protected:
std::string m_errmsg;
};
class timeout : public error
{
public:
timeout()
{
m_errmsg = "timeout";
}
};
inline void verify_pgtypes_error(int ret)
{
if(ret && errno != 0)
throw std::system_error(std::error_code(errno, std::generic_category()));
}
struct interval
{
::interval* value;
interval()
{
value = PGTYPESinterval_new();
}
explicit interval(char* str)
{
value = PGTYPESinterval_from_asc(str, nullptr);
}
interval(const interval& src) : interval()
{
verify_pgtypes_error(PGTYPESinterval_copy(src.value, value));
}
interval(interval&& src)
{
value = src.value;
src.value = PGTYPESinterval_new();
}
~interval()
{
PGTYPESinterval_free(value);
}
std::string to_string() const
{
return PGTYPESinterval_to_asc(value);
}
interval& operator=(const interval& src)
{
if(&src!=this)
verify_pgtypes_error(PGTYPESinterval_copy(src.value, value));
return *this;
}
};
struct timestamp
{
::timestamp value;
timestamp() = default;
static timestamp now()
{
timestamp result;
PGTYPEStimestamp_current(&result.value);
return result;
}
explicit timestamp(char* str)
{
value = PGTYPEStimestamp_from_asc(str, nullptr);
verify_pgtypes_error(1);
}
int format(char* str, int n, const char* format) const
{
timestamp temp = *this;
return PGTYPEStimestamp_fmt_asc(&temp.value, str, n, format);
}
static timestamp parse(char* str, const char* format)
{
timestamp result;
verify_pgtypes_error(PGTYPEStimestamp_defmt_asc(str, format, &result.value));
return result;
}
std::string to_string() const
{
char* str = PGTYPEStimestamp_to_asc(value);
std::string result = str;
PGTYPESchar_free(str);
return result;
}
timestamp& operator += (const interval& span)
{
verify_pgtypes_error(PGTYPEStimestamp_add_interval(&value, span.value, &value));
return *this;
}
timestamp& operator -= (const interval& span)
{
verify_pgtypes_error(PGTYPEStimestamp_sub_interval(&value, span.value, &value));
return *this;
}
};
inline timestamp operator+(const timestamp& a, const interval& b)
{
timestamp result=a;
return result+=b;
}
inline timestamp operator-(const timestamp& a, const interval& b)
{
timestamp result=a;
result -= b;
return result;
}
struct timestamptz
{
::TimestampTz value;
/*
timestamptz() = default;
explicit timestamptz(pg_time_t v)
{
value = (TimestampTz)v -
((POSTGRES_EPOCH_JDATE - UNIX_EPOCH_JDATE) * SECS_PER_DAY);
value *= USECS_PER_SEC;
}
static timestamptz now()
{
timestamptz result;
auto tp = std::chrono::system_clock::now();
int sec = tp.time_since_epoch().count()*std::nano::num/std::nano::den;
int usec = tp.time_since_epoch().count()*std::nano::num % std::nano::den;
result.value = (TimestampTz)sec -
((POSTGRES_EPOCH_JDATE - UNIX_EPOCH_JDATE) * SECS_PER_DAY);
result.value = (result.value * USECS_PER_SEC) + usec;
return result;
}
*/
};
struct date
{
::date value;
date() = default;
explicit date(timestamp dt)
{
value = PGTYPESdate_from_timestamp(dt.value);
}
explicit date(char* str)
{
value = PGTYPESdate_from_asc(str, nullptr);
verify_pgtypes_error(1);
}
explicit date(int year, int month, int day)
{
int mdy[3] = { month, day, year };
PGTYPESdate_mdyjul(mdy, &value);
}
std::string to_string() const
{
char* str = PGTYPESdate_to_asc(value);
std::string result = str;
PGTYPESchar_free(str);
return str;
}
static date now()
{
date result;
PGTYPESdate_today(&result.value);
return result;
}
static date parse(char* str, const char* format)
{
date result;
verify_pgtypes_error(PGTYPESdate_defmt_asc(&result.value, format, str));
return result;
}
std::string format(const char* format)
{
std::string result;
result.resize(128);
verify_pgtypes_error(PGTYPESdate_fmt_asc(value, format, const_cast<char*>(result.data())));
result.resize(strlen(result.data()));
return result;
}
std::tuple<int, int, int> get_date()
{
int mdy[3];
PGTYPESdate_julmdy(value, mdy);
return std::make_tuple(mdy[2], mdy[0], mdy[1]);
}
int dayofweek()
{
return PGTYPESdate_dayofweek(value);
}
};
struct decimal
{
::decimal value;
};
struct numeric
{
::numeric* value;
numeric()
{
value = PGTYPESnumeric_new();
}
numeric(int v) : numeric()
{
verify_pgtypes_error(PGTYPESnumeric_from_int(v, value));
}
numeric(long v) : numeric()
{
verify_pgtypes_error(PGTYPESnumeric_from_long(v, value));
}
numeric(double v) : numeric()
{
verify_pgtypes_error(PGTYPESnumeric_from_double(v, value));
}
numeric(const decimal& v) : numeric()
{
verify_pgtypes_error(PGTYPESnumeric_from_decimal(const_cast<::decimal*>(&v.value), value));
}
numeric(const numeric& src) : numeric()
{
verify_pgtypes_error(PGTYPESnumeric_copy(src.value, value));
}
explicit numeric(const char* str)
{
value = PGTYPESnumeric_from_asc(const_cast<char*>(str), nullptr);
}
~numeric()
{
PGTYPESnumeric_free(value);
}
operator double() const
{
double result;
verify_pgtypes_error(PGTYPESnumeric_to_double(value, &result));
return result;
}
operator int() const
{
int result;
verify_pgtypes_error(PGTYPESnumeric_to_int(value, &result));
return result;
}
operator long() const
{
long result;
verify_pgtypes_error(PGTYPESnumeric_to_long(value, &result));
return result;
}
operator decimal() const
{
decimal result;
verify_pgtypes_error(PGTYPESnumeric_to_decimal(value, &result.value));
return result;
}
int compare(const numeric& other) const
{
return PGTYPESnumeric_cmp(value, other.value);
}
inline numeric& operator+=(const numeric& b)
{
verify_pgtypes_error(PGTYPESnumeric_add(value, b.value, value));
return *this;
}
inline numeric& operator-=(const numeric& b)
{
verify_pgtypes_error(PGTYPESnumeric_sub(value, b.value, value));
return *this;
}
inline numeric& operator*=(const numeric& b)
{
verify_pgtypes_error(PGTYPESnumeric_mul(value, b.value, value));
return *this;
}
inline numeric& operator/=(const numeric& b)
{
verify_pgtypes_error(PGTYPESnumeric_div(value, b.value, value));
return *this;
}
std::string to_string(int dscale=-1) const
{
char* str = PGTYPESnumeric_to_asc(value, dscale);
std::string result = str;
PGTYPESchar_free(str);
return result;
}
};
inline numeric operator+(const numeric& a, const numeric& b)
{
numeric result;
verify_pgtypes_error(PGTYPESnumeric_add(a.value, b.value, result.value));
return result;
}
inline numeric operator-(const numeric& a, const numeric& b)
{
numeric result;
verify_pgtypes_error(PGTYPESnumeric_sub(a.value, b.value, result.value));
return result;
}
inline numeric operator*(const numeric& a, const numeric& b)
{
numeric result;
verify_pgtypes_error(PGTYPESnumeric_mul(a.value, b.value, result.value));
return result;
}
inline numeric operator/(const numeric& a, const numeric& b)
{
numeric result;
verify_pgtypes_error(PGTYPESnumeric_div(a.value, b.value, result.value));
return result;
}
inline bool operator==(const numeric& a, const numeric& b)
{
return a.compare(b) == 0;
}
inline bool operator<(const numeric& a, const numeric& b)
{
return a.compare(b) < 0;
}
inline bool operator>(const numeric& a, const numeric& b)
{
return a.compare(b) > 0;
}
inline bool operator<=(const numeric& a, const numeric& b)
{
return a.compare(b) <= 0;
}
inline bool operator>=(const numeric& a, const numeric& b)
{
return a.compare(b) >= 0;
}
inline bool operator!=(const numeric& a, const numeric& b)
{
return a.compare(b) != 0;
}
class large_object : public qtl::blobbuf
{
public:
large_object() : m_conn(nullptr), m_id(InvalidOid), m_fd(-1) { }
large_object(PGconn* conn, Oid loid, std::ios_base::openmode mode)
{
open(conn, loid, mode);
}
large_object(const large_object&) = delete;
large_object(large_object&& src)
{
swap(src);
src.m_conn = nullptr;
src.m_fd = -1;
}
~large_object()
{
close();
}
static large_object create(PGconn* conn, Oid loid = InvalidOid)
{
Oid oid = lo_create(conn, loid);
if (oid == InvalidOid)
throw error(conn);
return large_object(conn, oid, std::ios::in|std::ios::out|std::ios::binary);
}
static large_object load(PGconn* conn, const char* filename, Oid loid = InvalidOid)
{
Oid oid = lo_import_with_oid(conn, filename, loid);
if (oid == InvalidOid)
throw error(conn);
return large_object(conn, oid, std::ios::in | std::ios::out | std::ios::binary);
}
void save(const char* filename) const
{
if (lo_export(m_conn, m_id, filename) < 0)
throw error(m_conn);
}
void unlink()
{
close();
if (lo_unlink(m_conn, m_id) < 0)
throw error(m_conn);
}
large_object& operator=(const large_object&) = delete;
large_object& operator=(large_object&& src)
{
if (this != &src)
{
swap(src);
src.close();
}
return *this;
}
bool is_open() const { return m_fd >= 0; }
Oid oid() const { return m_id; }
void open(PGconn* conn, Oid loid, std::ios_base::openmode mode)
{
int lomode = 0;
if (mode&std::ios_base::in)
lomode |= INV_READ;
if (mode&std::ios_base::out)
lomode |= INV_WRITE;
m_conn = conn;
m_id = loid;
m_fd = lo_open(m_conn, loid, lomode);
if (m_fd < 0)
throw error(m_conn);
m_size = size();
init_buffer(mode);
if (mode&std::ios_base::trunc)
{
if (lo_truncate(m_conn, m_fd, 0) < 0)
throw error(m_conn);
}
}
void close()
{
if (m_fd >= 0)
{
overflow();
if (lo_close(m_conn, m_fd) < 0)
throw error(m_conn);
m_fd = -1;
}
}
void flush()
{
if (m_fd >= 0)
overflow();
}
size_t size() const
{
pg_int64 size = 0;
if (m_fd >= 0)
{
pg_int64 org = lo_tell64(m_conn, m_fd);
size = lo_lseek64(m_conn, m_fd, 0, SEEK_END);
lo_lseek64(m_conn, m_fd, org, SEEK_SET);
}
return size;
}
void resize(size_t n)
{
if (m_fd >= 0 && lo_truncate64(m_conn, m_fd, n) < 0)
throw error(m_conn);
}
void swap(large_object& other)
{
std::swap(m_conn, other.m_conn);
std::swap(m_id, other.m_id);
std::swap(m_fd, other.m_fd);
qtl::blobbuf::swap(other);
}
protected:
enum { default_buffer_size = 4096 };
virtual bool read_blob(char* buffer, off_type& count, pos_type position) override
{
return lo_lseek64(m_conn, m_fd, position, SEEK_SET) >= 0 && lo_read(m_conn, m_fd, buffer, count) > 0;
}
virtual void write_blob(const char* buffer, size_t count) override
{
if (lo_write(m_conn, m_fd, buffer, count) < 0)
throw error(m_conn);
}
private:
PGconn* m_conn;
Oid m_id;
int m_fd;
};
struct array_header
{
int32_t ndim;
int32_t flags;
int32_t elemtype;
struct dimension {
int32_t length;
int32_t lower_bound;
} dims[1];
};
/*
template<typename T>
struct oid_traits
{
typedef T value_type;
static Oid type_id;
static Oid array_type_id; //optional
static const char* get(value_type& result, const char* begin, const char* end);
static std::pair<const char*, size_t> data(const T& v, std::vector<char*>& buffer);
};
*/
template<typename T, Oid id>
struct base_object_traits
{
typedef T value_type;
enum { type_id = id };
static bool is_match(Oid v)
{
return v == type_id;
}
};
template<typename T>
struct object_traits;
#define QTL_POSTGRES_SIMPLE_TRAITS(T, oid, array_oid) \
template<> struct object_traits<T> : public base_object_traits<T, oid> { \
enum { array_type_id = array_oid }; \
static const char* get(value_type& result, const char* data, const char* end) \
{ \
result = *reinterpret_cast<const value_type*>(data); \
return data+sizeof(value_type); \
} \
static std::pair<const char*, size_t> data(const T& v, std::vector<char>& /*data*/) { \
return std::make_pair(reinterpret_cast<const char*>(&v), sizeof(T)); \
} \
};
QTL_POSTGRES_SIMPLE_TRAITS(bool, BOOLOID, 1000)
QTL_POSTGRES_SIMPLE_TRAITS(char, CHAROID, 1002)
QTL_POSTGRES_SIMPLE_TRAITS(float, FLOAT4OID, FLOAT4ARRAYOID)
QTL_POSTGRES_SIMPLE_TRAITS(double, FLOAT8OID, 1022)
template<typename T, Oid id, Oid array_id>
struct integral_traits : public base_object_traits<T, id>
{
enum { array_type_id = array_id };
typedef typename base_object_traits<T, id>::value_type value_type;
static const char* get(value_type& v, const char* data, const char* end)
{
return detail::pop(data, v);
}
static std::pair<const char*, size_t> data(value_type v, std::vector<char>& buffer)
{
size_t n = buffer.size();
detail::push(buffer, v);
return std::make_pair(buffer.data()+n, buffer.size()-n);
}
};
template<> struct object_traits<int16_t> : public integral_traits<int16_t, INT2OID, INT2ARRAYOID>
{
};
template<> struct object_traits<int32_t> : public integral_traits<int32_t, INT4OID, INT4ARRAYOID>
{
};
template<> struct object_traits<int64_t> : public integral_traits<int64_t, INT8OID, 1016>
{
};
template<> struct object_traits<Oid> : public integral_traits<Oid, OIDOID, OIDARRAYOID>
{
};
template<typename T>
struct text_traits : public base_object_traits<T, TEXTOID>
{
enum { array_type_id = TEXTARRAYOID };
};
template<> struct object_traits<const char*> : public text_traits<const char*>
{
static bool is_match(Oid v)
{
return v == TEXTOID || v == VARCHAROID || v == BPCHAROID;
}
static const char* get(const char*& result, const char* data, const char* end)
{
result = data;
return end;
}
static std::pair<const char*, size_t> data(const char* v, std::vector<char>& /*buffer*/)
{
return std::make_pair(v, strlen(v));
}
};
template<> struct object_traits<char*> : public object_traits<const char*>
{
};
template<> struct object_traits<std::string> : public text_traits<std::string>
{
static bool is_match(Oid v)
{
return v == TEXTOID || v == VARCHAROID || v == BPCHAROID;
}
static const char* get(value_type& result, const char* data, const char* end)
{
result.assign(data, end);
return end;
}
static std::pair<const char*, size_t> data(const std::string& v, std::vector<char>& /*buffer*/)
{
return std::make_pair(v.data(), v.size());
}
};
template<> struct object_traits<timestamp> : public base_object_traits<timestamp, TIMESTAMPOID>
{
enum { array_type_id = TIMESTAMPOID+1 };
static const char* get(value_type& result, const char* data, const char* end)
{
result = *reinterpret_cast<const timestamp*>(data);
result.value = detail::ntoh(result.value);
return data+sizeof(timestamp);
}
static std::pair<const char*, size_t> data(const timestamp& v, std::vector<char>& buffer)
{
size_t n = buffer.size();
detail::push(buffer, v.value);
return std::make_pair(buffer.data()+n, buffer.size()-n);
}
};
template<> struct object_traits<timestamptz> : public base_object_traits<timestamptz, TIMESTAMPTZOID>
{
enum { array_type_id = TIMESTAMPTZOID+1 };
static const char* get(value_type& result, const char* data, const char* end)
{
result = *reinterpret_cast<const timestamptz*>(data);
result.value = detail::ntoh(result.value);
return data+sizeof(timestamptz);
}
static std::pair<const char*, size_t> data(const timestamptz& v, std::vector<char>& buffer)
{
size_t n = buffer.size();
detail::push(buffer, v.value);
return std::make_pair(buffer.data() + n, buffer.size() - n);
}
};
template<> struct object_traits<interval> : public base_object_traits<interval, INTERVALOID>
{
enum { array_type_id = INTERVALOID+1 };
static const char* get(value_type& result, const char* data, const char* end)
{
const ::interval* value = reinterpret_cast<const ::interval*>(data);
result.value->time = detail::ntoh(value->time);
result.value->month = detail::ntoh(value->month);
return data+sizeof(interval);
}
static std::pair<const char*, size_t> data(const interval& v, std::vector<char>& buffer)
{
size_t n = buffer.size();
detail::push(buffer, v.value->time);
detail::push(buffer, v.value->month);
return std::make_pair(buffer.data()+n, buffer.size()-n);
}
};
template<> struct object_traits<date> : public base_object_traits<date, DATEOID>
{
enum { array_type_id = 1182 };
static const char* get(value_type& result, const char* data, const char* end)
{
result = *reinterpret_cast<const date*>(data);
result.value = detail::ntoh(result.value);
return data+sizeof(date);
}
static std::pair<const char*, size_t> data(const date& v, std::vector<char>& buffer)
{
size_t n=buffer.size();
detail::push(buffer, v.value);
return std::make_pair(buffer.data()+n, buffer.size()-n);
}
};
template<typename T>
struct bytea_traits : public base_object_traits<T, BYTEAOID>
{
enum { array_type_id = 1001 };
};
template<> struct object_traits<qtl::const_blob_data> : public bytea_traits<qtl::const_blob_data>
{
static const char* get(value_type& result, const char* data, const char* end)
{
result.data = data;
result.size = end-data;
return end;
}
static std::pair<const char*, size_t> data(const qtl::const_blob_data& v, std::vector<char>& /*buffer*/)
{
assert(v.size <= UINT32_MAX);
return std::make_pair(static_cast<const char*>(v.data), v.size);
}
};
template<> struct object_traits<qtl::blob_data> : public bytea_traits<qtl::blob_data>
{
static const char* get(qtl::blob_data& value, const char* data, const char* end)
{
if (value.size < end-data)
throw std::out_of_range("no enough buffer to receive blob data.");
memcpy(value.data, data, end-data);
return end;
}
static std::pair<const char*, size_t> data(const qtl::blob_data& v, std::vector<char>& /*buffer*/)
{
assert(v.size <= UINT32_MAX);
return std::make_pair(static_cast<char*>(v.data), v.size);
}
};
template<> struct object_traits<std::vector<uint8_t>> : public bytea_traits<std::vector<uint8_t>>
{
static const char* get(value_type& result, const char* data, const char* end)
{
result.assign(data, end);
return end;
}
static std::pair<const char*, size_t> data(const std::vector<uint8_t>& v, std::vector<char>& /*buffer*/)
{
assert(v.size() <= UINT32_MAX);
return std::make_pair(reinterpret_cast<const char*>(v.data()), v.size());
}
};
template<> struct object_traits<large_object> : public base_object_traits<large_object, OIDOID>
{
enum { array_type_id = OIDARRAYOID };
static value_type get(PGconn* conn, const char* data, const char* end)
{
int32_t oid;
object_traits<int32_t>::get(oid, data, end);
return large_object(conn, oid, std::ios::in | std::ios::out | std::ios::binary);
}
static std::pair<const char*, size_t> data(const large_object& v, std::vector<char>& buffer)
{
return object_traits<int32_t>::data(v.oid(), buffer);
}
};
template<typename T, Oid id>
struct vector_traits : public base_object_traits<std::vector<T>, id>
{
typedef typename base_object_traits<std::vector<T>, id>::value_type value_type;
static const char* get(value_type& result, const char* data, const char* end)
{
if (end - data < sizeof(array_header))
throw std::overflow_error("insufficient data left in message");
array_header header = *reinterpret_cast<const array_header*>(data);
detail::ntoh_inplace(header.ndim);
detail::ntoh_inplace(header.flags);
detail::ntoh_inplace(header.elemtype);
detail::ntoh_inplace(header.dims[0].length);
detail::ntoh_inplace(header.dims[0].lower_bound);
if (header.ndim != 1 || !object_traits<T>::is_match(header.elemtype))
throw std::bad_cast();
data += sizeof(array_header);
result.reserve(header.dims[0].length);
for (int32_t i = 0; i != header.dims[0].length; i++)
{
int32_t size;
T value;
data = detail::pop(data, size);
if (end - data < size)
throw std::overflow_error("insufficient data left in message");
data = object_traits<T>::get(value, data, data + size);
if (data > end)
throw std::overflow_error("insufficient data left in message");
result.push_back(value);
}
return data;
}
static std::pair<const char*, size_t> data(const std::vector<T>& v, std::vector<char>& buffer)
{
assert(v.size() <= INT32_MAX);
size_t n = buffer.size();
buffer.resize(n+sizeof(array_header));
array_header* header = reinterpret_cast<array_header*>(buffer.data()+n);
header->ndim = detail::hton(1);
header->flags = detail::hton(0);
header->elemtype = detail::hton(static_cast<int32_t>(object_traits<T>::type_id));
header->dims[0].length = detail::hton(static_cast<int32_t>(v.size()));
header->dims[0].lower_bound = detail::hton(1);
std::vector<char> temp;
for (const T& e : v)
{
std::pair<const char*, size_t> blob = object_traits<T>::data(e, temp);
detail::push(buffer, static_cast<int32_t>(blob.second));
buffer.insert(buffer.end(), blob.first, blob.first + blob.second);
}
return std::make_pair(buffer.data()+n, buffer.size()-n);
}
};
template<typename Iterator, Oid id>
struct iterator_traits : public base_object_traits<Iterator, id>
{
static const char* get(Iterator first, Iterator last, const char* data, const char* end)
{
if (end - data < sizeof(array_header))
throw std::overflow_error("insufficient data left in message");
array_header header = *reinterpret_cast<const array_header*>(data);
detail::ntoh_inplace(header.ndim);
detail::ntoh_inplace(header.flags);
detail::ntoh_inplace(header.elemtype);
detail::ntoh_inplace(header.dims[0].length);
detail::ntoh_inplace(header.dims[0].lower_bound);
if (header.ndim != 1 || !object_traits<typename std::iterator_traits<Iterator>::value_type>::is_match(header.elemtype))
throw std::bad_cast();
data += sizeof(array_header);
if (std::distance(first, last) < header.dims[0].length)
throw std::out_of_range("length of array out of range");
Iterator it = first;
for (int32_t i = 0; i != header.dims[0].length; i++, it++)
{
int32_t size;
data = detail::pop(data, size);
if (end - data < size)
throw std::overflow_error("insufficient data left in message");
data = object_traits<typename std::iterator_traits<Iterator>::value_type>::get(*it, data, data + size);
if (data >= end)
throw std::overflow_error("insufficient data left in message");
}
return data;
}
static std::pair<const char*, size_t> data(Iterator first, Iterator last, std::vector<char>& buffer)
{
assert(std::distance(first, last) <= INT32_MAX);
size_t n = buffer.size();
buffer.resize(n + sizeof(array_header));
array_header* header = reinterpret_cast<array_header*>(buffer.data() + n);
header->ndim = detail::hton(1);
header->flags = detail::hton(0);
header->elemtype = detail::hton(static_cast<int32_t>(object_traits<typename std::iterator_traits<Iterator>::value_type>::type_id));
header->dims[0].length = detail::hton(static_cast<int32_t>(std::distance(first, last)));
header->dims[0].lower_bound = detail::hton(1);
std::vector<char> temp;
for (Iterator it=first; it!=last; it++)
{
std::pair<const char*, size_t> blob = object_traits<typename std::iterator_traits<Iterator>::value_type>::data(*it, temp);
detail::push(buffer, static_cast<int32_t>(blob.second));
buffer.insert(buffer.end(), blob.first, blob.first + blob.second);
}
return std::make_pair(buffer.data() + n, buffer.size() - n);
}
};
template<typename Iterator, Oid id>
struct range_traits : public base_object_traits<std::pair<Iterator, Iterator>, id>
{
static const char* get(std::pair<Iterator, Iterator>& result, const char* data, const char* end)
{
return iterator_traits<Iterator, id>::get(result.first, result.second, data, end);
}
static std::pair<const char*, size_t> data(const std::pair<Iterator, Iterator>& v, std::vector<char>& buffer)
{
return iterator_traits<Iterator, id>::data(v.first, v.second, buffer);
}
};
template<typename T>
struct object_traits<std::vector<T>> : public vector_traits<T, object_traits<T>::array_type_id>
{
};
template<typename Iterator>
struct object_traits<std::pair<typename std::enable_if<std::is_object<typename std::iterator_traits<Iterator>::value_type>::value, Iterator>::type, Iterator>> :
public range_traits<Iterator, object_traits<typename std::iterator_traits<Iterator>::value_type>::array_type_id>
{
};
template<typename T, size_t N, Oid id>
struct carray_traits : public base_object_traits<T(&)[N], id>
{
static const char* get(T (&result)[N], const char* data, const char* end)
{
return iterator_traits<T*, id>::get(std::begin(result), std::end(result), data, end);
}
static std::pair<const char*, size_t> data(const T (&v)[N], std::vector<char>& buffer)
{
return iterator_traits<const T*, id>::data(std::begin(v), std::end(v), buffer);
}
};
template<typename T, size_t N, Oid id>
struct array_traits : public base_object_traits<std::array<T, N>, id>
{
static const char* get(std::array<T, N>& result, const char* data, const char* end)
{
return iterator_traits<T*, id>::get(std::begin(result), std::end(result), data, end);
}
static std::pair<const char*, size_t> data(const std::array<T, N>& v, std::vector<char>& buffer)
{
return iterator_traits<T*, id>::data(std::begin(v), std::end(v), buffer);
}
};
template<typename T, size_t N> struct object_traits<T (&)[N]> : public carray_traits<T, N, object_traits<T>::array_type_id>
{
};
template<typename T, size_t N> struct object_traits<std::array<T, N>> : public array_traits<T, N, object_traits<T>::array_type_id>
{
};
namespace detail
{
struct field_header
{
Oid type;
int32_t length;
};
template<typename Type>
static const char* get_field(Type& field, const char* data, const char* end)
{
field_header header = *reinterpret_cast<const field_header*>(data);
detail::ntoh_inplace(header.type);
detail::ntoh_inplace(header.length);
data += sizeof(field_header);
if (end - data < header.length)
throw std::overflow_error("insufficient data left in message");
return object_traits<Type>::get(field, data, data + header.length);
}
template<typename Tuple, size_t N>
struct get_field_helper
{
const char* operator()(Tuple& result, const char* data, const char* end)
{
if (end - data < sizeof(field_header))
throw std::overflow_error("insufficient data left in message");
auto& field = std::get<std::tuple_size<Tuple>::value - N>(result);
data = get_field(field, data, end);
get_field_helper<Tuple, N - 1>()(result, data, end);
return data;
}
};
template<typename Tuple>
struct get_field_helper<Tuple, 1>
{
const char* operator()(Tuple& result, const char* data, const char* end)
{
if (end - data < sizeof(field_header))
throw std::overflow_error("insufficient data left in message");
auto& field = std::get<std::tuple_size<Tuple>::value - 1>(result);
return get_field(field, data, end);
}
};
template<typename Type>
static void push_field(const Type& field, std::vector<char>& buffer)
{
std::vector<char> temp;
detail::push(buffer, static_cast<int32_t>(object_traits<Type>::type_id));
auto result = object_traits<Type>::data(field, temp);
detail::push(buffer, static_cast<int32_t>(result.second));
buffer.insert(buffer.end(), result.first, result.first + result.second);
}
template<typename Tuple, size_t N>
struct push_field_helper
{
void operator()(const Tuple& data, std::vector<char>& buffer)
{
const auto& field = std::get<std::tuple_size<Tuple>::value - N>(data);
push_field(field, buffer);
push_field_helper<Tuple, N - 1>()(data, buffer);
}
};
template<typename Tuple>
struct push_field_helper<Tuple, 1>
{
void operator()(const Tuple& data, std::vector<char>& buffer)
{
const auto& field = std::get<std::tuple_size<Tuple>::value - 1>(data);
push_field(field, buffer);
}
};
template<typename Tuple>
static const char* get_fields(Tuple& result, const char* data, const char* end)
{
return get_field_helper<Tuple, std::tuple_size<Tuple>::value>()(result, data, end);
}
template<typename Tuple>
static void push_fields(const Tuple& data, std::vector<char>& buffer)
{
push_field_helper<Tuple, std::tuple_size<Tuple>::value>()(data, buffer);
}
}
template<typename Tuple, Oid id>
struct tuple_traits : public base_object_traits<Tuple, id>
{
typedef typename base_object_traits<Tuple, id>::value_type value_type;
static const char* get(value_type& result, const char* data, const char* end)
{
int32_t count;
data = detail::pop(data, count);
if (data >= end)
throw std::overflow_error("insufficient data left in message");
if (std::tuple_size<Tuple>::value != count)
throw std::bad_cast();
return detail::get_fields(result, data, end);
}
static std::pair<const char*, size_t> data(const value_type& v, std::vector<char>& buffer)
{
size_t n = buffer.size();
detail::push(buffer, static_cast<int32_t>(std::tuple_size<value_type>::value));
detail::push_fields(v, buffer);
return std::make_pair(buffer.data()+n, buffer.size()-n);
}
};
template<typename... Types>
struct object_traits<std::tuple<Types...>> : public tuple_traits<std::tuple<Types...>, InvalidOid>
{
};
template<typename T1, typename T2>
struct object_traits<std::pair<T1, T2>> : public tuple_traits<std::pair<T1, T2>, InvalidOid>
{
};
struct binder
{
binder() = default;
template<typename T>
explicit binder(const T& v)
{
m_type = object_traits<T>::value();
auto pair = object_traits<T>::data(v);
m_value = pair.first;
m_length = pair.second;
}
binder(const char* data, size_t n, Oid oid)
{
m_type = oid;
m_value = data;
m_length = n;
}
Oid constexpr type() const { return m_type; }
size_t length() const { return m_length; }
const char* value() const { return m_value; }
template<typename T>
T get()
{
if (!object_traits<T>::is_match(m_type))
throw std::bad_cast();
T v;
object_traits<T>::get(v, m_value, m_value + m_length);
return v;
}
template<typename T>
T get(PGconn* conn)
{
if (!object_traits<T>::is_match(m_type))
throw std::bad_cast();
return object_traits<T>::get(conn, m_value, m_value + m_length);
}
template<typename T>
void get(T& v)
{
if (object_traits<T>::type_id!= InvalidOid && !object_traits<T>::is_match(m_type))
throw std::bad_cast();
object_traits<T>::get(v, m_value, m_value + m_length);
}
void bind(std::nullptr_t)
{
m_value = nullptr;
m_length = 0;
}
void bind(qtl::null)
{
bind(nullptr);
}
template<typename T, typename = typename std::enable_if<!std::is_array<T>::value>::type>
void bind(const T& v)
{
typedef typename std::decay<T>::type param_type;
if (m_type!=0 && !object_traits<param_type>::is_match(m_type))
throw std::bad_cast();
auto pair = object_traits<param_type>::data(v, m_data);
m_value = pair.first;
m_length = pair.second;
}
void bind(const char* data, size_t length=0)
{
m_value = data;
if(length>0) m_length = length;
else m_length = strlen(data);
}
template<typename T, size_t N>
void bind(const T(&v)[N])
{
if (m_type != 0 && !object_traits<T(&)[N]>::is_match(m_type))
throw std::bad_cast();
auto pair = object_traits<T(&)[N]>::data(v, m_data);
m_value = pair.first;
m_length = pair.second;
}
private:
Oid m_type;
const char* m_value;
size_t m_length;
std::vector<char> m_data;
};
template<size_t N, size_t I, typename Arg, typename... Other>
inline void make_binder_list_helper(std::array<binder, N>& binders, Arg&& arg, Other&&... other)
{
binders[I]=binder(arg);
make_binder_list_helper<N, I+1>(binders, std::forward<Other>(other)...);
}
template<typename... Args>
inline std::array<binder, sizeof...(Args)> make_binder_list(Args&&... args)
{
std::array<binder, sizeof...(Args)> binders;
binders.reserve(sizeof...(Args));
make_binder_list_helper<sizeof...(Args), 0>(binders, std::forward<Args>(args)...);
return binders;
}
template<typename T>
inline bool in_impl(const T& from, const T& to)
{
return std::equal_to<T>()(from, to);
}
template<typename T, typename... Ts >
inline bool in_impl(const T& from, const T& to, const Ts&... other)
{
return std::equal_to<T>()(from, to) || in_impl(from, other...);
}
template<typename T, T... values>
inline bool in(const T& v)
{
return in_impl(v, values...);
}
class result
{
public:
result(PGresult* res) : m_res(res) { }
result(const result&) = delete;
result(result&& src)
{
m_res = src.m_res;
src.m_res = nullptr;
}
result& operator=(const result&) = delete;
result& operator=(result&& src)
{
if (this != &src)
{
clear();
m_res = src.m_res;
src.m_res = nullptr;
}
return *this;
}
~result()
{
clear();
}
PGresult* handle() const { return m_res; }
operator bool() const { return m_res != nullptr; }
ExecStatusType status() const
{
return PQresultStatus(m_res);
}
long long affected_rows() const
{
char* result = PQcmdTuples(m_res);
if (result)
return strtoll(result, nullptr, 10);
else
return 0LL;
}
unsigned int get_column_count() const { return PQnfields(m_res); }
int get_param_count() const
{
return PQnparams(m_res);
}
Oid get_param_type(int col) const
{
return PQparamtype(m_res, col);
}
const char* get_column_name(int col) const
{
return PQfname(m_res, col);
}
int get_column_index(const char* name) const
{
return PQfnumber(m_res, name);
}
int get_column_length(int col) const
{
return PQfsize(m_res, col);
}
Oid get_column_type(int col) const
{
return PQftype(m_res, col);
}
const char* get_value(int row, int col) const
{
return PQgetvalue(m_res, row, col);
}
bool is_null(int row, int col) const
{
return PQgetisnull(m_res, row, col);
}
int length(int row, int col) const
{
return PQgetlength(m_res, row, col);
}
Oid insert_oid() const
{
return PQoidValue(m_res);
}
template<ExecStatusType... Excepted>
void verify_error()
{
if (m_res)
{
ExecStatusType got = status();
if (! in<ExecStatusType, Excepted...>(got))
throw error(m_res);
}
}
template<ExecStatusType... Excepted>
void verify_error(error& e)
{
if (m_res)
{
ExecStatusType got = status();
if (!in<ExecStatusType, Excepted...>(got))
e = error(m_res);
}
}
void clear()
{
if (m_res)
{
PQclear(m_res);
m_res = nullptr;
}
}
private:
PGresult* m_res;
};
class base_statement
{
friend class error;
public:
explicit base_statement(base_database& db);
~base_statement()
{
}
base_statement(const base_statement&) = delete;
base_statement(base_statement&& src)
: m_conn(src.m_conn), m_binders(std::move(src.m_binders)), m_res(std::move(src.m_res)), _name(std::move(src._name))
{
}
base_statement& operator=(const base_statement&) = delete;
base_statement& operator=(base_statement&& src)
{
if (this != &src)
{
close();
m_conn = src.m_conn;
m_binders = std::move(src.m_binders);
m_res = std::move(src.m_res);
}
return *this;
}
result& get_result() { return m_res; }
void close()
{
m_res = nullptr;
}
uint64_t affetced_rows() const
{
return m_res.affected_rows();
}
void bind_param(size_t index, const char* param, size_t length)
{
m_binders[index].bind(param, length);
}
template<class Param>
void bind_param(size_t index, const Param& param)
{
m_binders[index].bind(param);
}
template<class Type>
void bind_field(size_t index, Type&& value)
{
if (m_res.is_null(0, static_cast<int>(index)))
value = Type();
else
value = m_binders[index].get<typename std::remove_const<Type>::type>();
}
void bind_field(size_t index, char* value, size_t length)
{
memcpy(value, m_binders[index].value(), std::min<size_t>(length, m_binders[index].length()));
}
template<size_t N>
void bind_field(size_t index, std::array<char, N>&& value)
{
bind_field(index, value.data(), value.size());
}
template<typename T>
void bind_field(size_t index, bind_string_helper<T>&& value)
{
value.assign(m_binders[index].value(), m_binders[index].length());
}
template<typename Type>
void bind_field(size_t index, indicator<Type>&& value)
{
if (m_res)
{
qtl::bind_field(*this, index, value.data);
value.is_null = m_res.is_null(0, static_cast<int>(index));
value.length = m_res.length(0, static_cast<int>(index));
value.is_truncated = m_binders[index].length() < value.length;
}
}
void bind_field(size_t index, large_object&& value)
{
if (m_res.is_null(0, static_cast<int>(index)))
value.close();
else
value = m_binders[index].get<large_object>(m_conn);
}
void bind_field(size_t index, blob_data&& value)
{
if (m_res.is_null(0, static_cast<int>(index)))
{
value.data = nullptr;
value.size = 0;
}
else
{
m_binders[index].get(value);
}
}
template<typename... Types>
void bind_field(size_t index, std::tuple<Types...>&& value)
{
if (m_res.is_null(0, static_cast<int>(index)))
value = std::tuple<Types...>();
else
m_binders[index].get(value);
}
#ifdef _QTL_ENABLE_CPP17
template<typename T>
inline void bind_field(size_t index, std::optional<T>&& value)
{
if (m_res.is_null(0, static_cast<int>(index)))
{
value.reset();
}
else
{
T v;
bind_field(index, v);
value = std::move(v);
}
}
void bind_field(size_t index, std::any&& value)
{
if (m_res.is_null(0, static_cast<int>(index)))
{
value = nullptr;
}
else
{
Oid oid = m_res.get_column_type(index);
switch (oid)
{
case object_traits<bool>::type_id:
value = field_cast<bool>(index);
break;
case object_traits<char>::type_id:
value = field_cast<char>(index);
break;
case object_traits<float>::type_id:
value = field_cast<float>(index);
break;
case object_traits<double>::type_id:
value = field_cast<double>(index);
break;
case object_traits<int16_t>::type_id:
value = field_cast<int16_t>(index);
break;
case object_traits<int32_t>::type_id:
value = field_cast<int32_t>(index);
break;
case object_traits<int64_t>::type_id:
value = field_cast<int64_t>(index);
break;
case object_traits<Oid>::type_id:
value = field_cast<Oid>(index);
break;
case object_traits<std::string>::type_id:
value = field_cast<std::string>(index);
break;
case object_traits<timestamp>::type_id:
value = field_cast<timestamp>(index);
break;
case object_traits<interval>::type_id:
value = field_cast<interval>(index);
break;
case object_traits<date>::type_id:
value = field_cast<date>(index);
break;
case object_traits<std::vector<uint8_t>>::type_id:
value = field_cast<std::vector<uint8_t>>(index);
break;
case object_traits<bool>::array_type_id:
value = field_cast<std::vector<bool>>(index);
break;
case object_traits<char>::array_type_id:
value = field_cast<std::vector<char>>(index);
break;
case object_traits<float>::array_type_id:
value = field_cast<std::vector<float>>(index);
break;
case object_traits<double>::array_type_id:
value = field_cast<std::vector<double>>(index);
break;
case object_traits<int16_t>::array_type_id:
value = field_cast<std::vector<int16_t>>(index);
break;
case object_traits<int32_t>::array_type_id:
value = field_cast<std::vector<int32_t>>(index);
break;
case object_traits<int64_t>::array_type_id:
value = field_cast<std::vector<int64_t>>(index);
break;
case object_traits<Oid>::array_type_id:
value = field_cast<std::vector<Oid>>(index);
break;
case object_traits<std::string>::array_type_id:
value = field_cast<std::vector<std::string>>(index);
break;
case object_traits<timestamp>::array_type_id:
value = field_cast<std::vector<timestamp>>(index);
break;
case object_traits<interval>::array_type_id:
value = field_cast<std::vector<interval>>(index);
break;
case object_traits<date>::array_type_id:
value = field_cast<std::vector<date>>(index);
break;
default:
throw postgres::error("Unsupported field type");
}
}
}
#endif // C++17
protected:
PGconn* m_conn;
result m_res;
std::string _name;
std::vector<binder> m_binders;
template<ExecStatusType... Excepted>
void verify_error()
{
if (m_res)
m_res.verify_error<Excepted...>();
else
throw error(m_conn);
}
void finish(result& res)
{
while (res)
{
res = PQgetResult(m_conn);
}
}
template<typename T>
T field_cast(size_t index)
{
T v;
m_binders[index].get(v);
return v;
}
};
class statement : public base_statement
{
public:
explicit statement(base_database& db) : base_statement(db)
{
}
statement(const statement&) = delete;
statement(statement&& src) : base_statement(std::move(src))
{
}
~statement()
{
finish(m_res);
if (!_name.empty())
{
std::ostringstream oss;
oss << "DEALLOCATE " << _name << ";";
result res = PQexec(m_conn, oss.str().data());
error e(res.handle());
}
}
void open(const char* command, int nParams=0, const Oid *paramTypes=nullptr)
{
_name.resize(sizeof(intptr_t) * 2+1);
int n = sprintf(const_cast<char*>(_name.data()), "q%p", this);
_name.resize(n);
std::transform(_name.begin(), _name.end(), _name.begin(), tolower);
result res = PQprepare(m_conn, _name.data(), command, nParams, paramTypes);
res.verify_error<PGRES_COMMAND_OK>();
}
template<typename... Types>
void open(const char* command)
{
auto binder_list = make_binder_list(Types()...);
std::array<Oid, sizeof...(Types)> types;
std::transform(binder_list.begin(), binder_list.end(), types.begin(), [](const binder& b) {
return b.type();
});
open(command, types.size(), types.data());
}
void attach(const char* name)
{
result res = PQdescribePrepared(m_conn, name);
res.verify_error<PGRES_COMMAND_OK>();
_name = name;
}
void execute()
{
if(!PQsendQueryPrepared(m_conn, _name.data(), 0, nullptr, nullptr, nullptr, 1))
throw error(m_conn);
if (!PQsetSingleRowMode(m_conn))
throw error(m_conn);
m_res = PQgetResult(m_conn);
verify_error<PGRES_COMMAND_OK, PGRES_SINGLE_TUPLE>();
}
template<typename Types>
void execute(const Types& params)
{
const size_t count = qtl::params_binder<statement, Types>::size;
if (count > 0)
{
m_binders.resize(count);
qtl::bind_params(*this, params);
std::array<const char*, count> values;
std::array<int, count> lengths;
std::array<int, count> formats;
for (size_t i = 0; i != m_binders.size(); i++)
{
values[i] = m_binders[i].value();
lengths[i] = static_cast<int>(m_binders[i].length());
formats[i] = 1;
}
if (!PQsendQueryPrepared(m_conn, _name.data(), static_cast<int>(m_binders.size()), values.data(), lengths.data(), formats.data(), 1))
throw error(m_conn);
}
else
{
if (!PQsendQueryPrepared(m_conn, _name.data(), 0, nullptr, nullptr, nullptr, 1))
throw error(m_conn);
}
if (!PQsetSingleRowMode(m_conn))
throw error(m_conn);
m_res = PQgetResult(m_conn);
verify_error<PGRES_COMMAND_OK, PGRES_SINGLE_TUPLE, PGRES_TUPLES_OK>();
}
template<typename Types>
bool fetch(Types&& values)
{
if (m_res)
{
ExecStatusType status = m_res.status();
if (status == PGRES_SINGLE_TUPLE)
{
int count = m_res.get_column_count();
if (count > 0)
{
m_binders.resize(count);
for (int i = 0; i != count; i++)
{
m_binders[i]=binder(m_res.get_value(0, i), m_res.length(0, i),
m_res.get_column_type(i));
}
qtl::bind_record(*this, std::forward<Types>(values));
}
m_res = PQgetResult(m_conn);
return true;
}
else
{
verify_error<PGRES_TUPLES_OK>();
}
}
return false;
}
bool next_result()
{
m_res = PQgetResult(m_conn);
return m_res && m_res.status() == PGRES_SINGLE_TUPLE;
}
void reset()
{
finish(m_res);
m_res.clear();
}
};
class base_database
{
protected:
base_database()
{
m_conn = nullptr;
}
public:
typedef postgres::error exception_type;
base_database(const base_database&) = delete;
base_database(base_database&& src)
{
m_conn = src.m_conn;
src.m_conn = nullptr;
}
~base_database()
{
if (m_conn)
PQfinish(m_conn);
}
base_database& operator=(const base_database&) = delete;
base_database& operator=(base_database&& src)
{
if (this != &src)
{
if (m_conn)
PQfinish(m_conn);
m_conn = src.m_conn;
src.m_conn = nullptr;
}
return *this;
}
const char* errmsg() const
{
return PQerrorMessage(m_conn);
}
PGconn* handle() { return m_conn; }
const char* encoding() const
{
int encoding = PQclientEncoding(m_conn);
return (encoding >= 0) ? pg_encoding_to_char(encoding) : nullptr;
}
void encoding(const char* encoding)
{
if (PQsetClientEncoding(m_conn, encoding))
throw error(m_conn);
}
void trace(FILE* stream)
{
PQtrace(m_conn, stream);
}
void untrace()
{
PQuntrace(m_conn);
}
const char* current() const
{
return PQdb(m_conn);
}
const char* user() const
{
return PQuser(m_conn);
}
const char* host() const
{
return PQhost(m_conn);
}
const char* password() const
{
return PQpass(m_conn);
}
const char* port() const
{
return PQport(m_conn);
}
const char* options() const
{
return PQoptions(m_conn);
}
ConnStatusType status() const
{
return PQstatus(m_conn);
}
PGTransactionStatusType transactionStatus() const
{
return PQtransactionStatus(m_conn);
}
const char* parameterStatus(const char *paramName) const
{
return PQparameterStatus(m_conn, paramName);
}
void reset()
{
if(status() == CONNECTION_BAD)
PQreset(m_conn);
}
void close()
{
PQfinish(m_conn);
m_conn = nullptr;
}
protected:
PGconn* m_conn;
void throw_exception() { throw postgres::error(m_conn); }
};
class simple_statment : public base_statement
{
public:
simple_statment(base_database& db, qtl::postgres::result&& res) : base_statement(db)
{
m_res = std::move(res);
}
template<typename ValueProc>
void fetch_all(ValueProc& proc)
{
int row_count = PQntuples(m_res.handle());
if (row_count > 0)
{
int col_count = m_res.get_column_count();
m_binders.resize(col_count);
auto values = qtl::detail::make_values(proc);
for (int i = 0; i != row_count; i++)
{
for (int j = 0; j != col_count; j++)
{
m_binders[j] = binder(m_res.get_value(i, j), m_res.length(i, j),
m_res.get_column_type(j));
}
qtl::bind_record(*this, std::forward<decltype(values)>(values));
qtl::detail::apply(proc, std::forward<decltype(values)>(values));
}
}
}
};
class database : public base_database, public qtl::base_database<database, statement>
{
public:
database() = default;
bool open(const std::map<std::string, std::string>& params, bool expand_dbname = false)
{
std::vector<const char*> keywords(params.size()+1);
std::vector<const char*> values(params.size()+1);
for (auto& param : params)
{
keywords.push_back(param.first.data());
values.push_back(param.second.data());
}
keywords.push_back(nullptr);
values.push_back(nullptr);
m_conn = PQconnectdbParams(keywords.data(), values.data(), expand_dbname);
return m_conn != nullptr && status()== CONNECTION_OK;
}
bool open(const char * conninfo)
{
m_conn = PQconnectdb(conninfo);
return m_conn != nullptr && status() == CONNECTION_OK;
}
bool open(const char* host, const char* user, const char* password,
unsigned short port = 5432, const char* db = "postgres", const char* options = nullptr)
{
char port_text[16];
sprintf(port_text, "%u", port);
m_conn = PQsetdbLogin(host, port_text, options, nullptr, db, user, password);
return m_conn != nullptr && status() == CONNECTION_OK;
}
statement open_command(const char* query_text, size_t /*text_length*/)
{
statement stmt(*this);
stmt.open(query_text);
return stmt;
}
statement open_command(const char* query_text)
{
return open_command(query_text, 0);
}
statement open_command(const std::string& query_text)
{
return open_command(query_text.data());
}
void simple_execute(const char* query_text, uint64_t* paffected = nullptr)
{
qtl::postgres::result res(PQexec(m_conn, query_text));
if (!res) throw_exception();
res.verify_error<PGRES_COMMAND_OK, PGRES_TUPLES_OK>();
if (paffected) *paffected = res.affected_rows();
}
template<typename ValueProc>
void simple_query(const char* query_text, ValueProc&& proc)
{
qtl::postgres::result res(PQexec(m_conn, query_text));
if (!res) throw_exception();
res.verify_error<PGRES_COMMAND_OK, PGRES_TUPLES_OK>();
if (res.status() == PGRES_TUPLES_OK)
{
simple_statment stmt(*this, std::move(res));
stmt.fetch_all(std::forward<ValueProc>(proc));
}
}
void auto_commit(bool on)
{
if(on)
simple_execute("SET AUTOCOMMIT TO ON");
else
simple_execute("SET AUTOCOMMIT TO OFF");
}
void begin_transaction()
{
simple_execute("BEGIN");
}
void rollback()
{
simple_execute("ROLLBACK");
}
void commit()
{
simple_execute("COMMIT");
}
bool is_alive()
{
qtl::postgres::result res(PQexec(m_conn, ""));
return res && res.status() == PGRES_COMMAND_OK;
}
};
inline int event_flags(PostgresPollingStatusType status)
{
int flags = 0;
if (status == PGRES_POLLING_READING)
flags |= event::ef_read;
else if (status == PGRES_POLLING_WRITING)
flags |= event::ef_write;
else if (status == PGRES_POLLING_FAILED)
flags |= event::ef_exception;
return flags;
}
class async_connection;
template<typename Handler>
inline void async_wait(qtl::event* event, PGconn* conn, int timeout, Handler&& handler)
{
int flushed = PQflush(conn);
if (flushed < 0)
{
handler(error(conn));
return;
}
if (flushed == 1)
{
event->set_io_handler(qtl::event::ef_read | qtl::event::ef_write, timeout,
[event, conn, timeout, handler](int flags) mutable {
if (flags&qtl::event::ef_timeout)
{
handler(postgres::timeout());
return;
}
if (flags&qtl::event::ef_read)
{
if (!PQconsumeInput(conn))
{
handler(error(conn));
return;
}
}
if (flags&(qtl::event::ef_read | qtl::event::ef_write | event::ef_exception))
async_wait(event, conn, timeout, handler);
});
}
else
{
event->set_io_handler(qtl::event::ef_read, 10,
[event, conn, timeout, handler](int flags) mutable {
if (flags&qtl::event::ef_timeout)
{
handler(postgres::timeout());
}
else if (flags&(qtl::event::ef_read | qtl::event::ef_exception))
{
if (PQconsumeInput(conn))
{
if (!PQisBusy(conn))
handler(postgres::error());
else
async_wait(event, conn, timeout, handler);
}
else
{
handler(postgres::error(conn));
}
}
else
{
handler(postgres::error(conn));
}
});
}
}
class async_statement : public base_statement
{
public:
async_statement(async_connection& db);
async_statement(async_statement&& src)
: base_statement(std::move(src)), m_timeout(2)
{
m_event = src.m_event;
m_timeout = src.m_timeout;
src.m_event = nullptr;
}
async_statement& operator=(async_statement&& src)
{
if (this != &src)
{
base_statement::operator =(std::move(src));
m_event = src.m_event;
m_timeout = src.m_timeout;
src.m_event = nullptr;
}
return *this;
}
~async_statement()
{
close();
}
/*
Handler defiens as:
void handler(const qtl::mysql::error& e);
*/
template<typename Handler>
void open(Handler&& handler, const char* command, int nParams = 0, const Oid *paramTypes = nullptr)
{
_name.resize(sizeof(intptr_t) * 2 + 1);
int n = sprintf(const_cast<char*>(_name.data()), "q%p", this);
_name.resize(n);
std::transform(_name.begin(), _name.end(), _name.begin(), tolower);
if (PQsendPrepare(m_conn, _name.data(), command, nParams, paramTypes))
{
async_wait([this, handler](error e) mutable {
if (!e)
{
m_res = PQgetResult(m_conn);
if (m_res)
{
m_res.verify_error<PGRES_COMMAND_OK>(e);
while(m_res)
m_res = PQgetResult(m_conn);
}
}
handler(e);
});
}
else
{
_name.clear();
handler(error(m_conn));
}
}
template<typename Handler, typename... Types>
void open(Handler&& handler, const char* command)
{
auto binder_list = make_binder_list(Types()...);
std::array<Oid, sizeof...(Types)> types;
std::transform(binder_list.begin(), binder_list.end(), types.begin(), [](const binder& b) {
return b.type();
});
open(std::forward<Handler>(handler), command, types.size(), types.data());
}
void close()
{
while (m_res)
{
m_res = PQgetResult(m_conn);
}
if (!_name.empty())
{
std::ostringstream oss;
oss << "DEALLOCATE " << _name << ";";
result res = PQexec(m_conn, oss.str().data());
error e;
res.verify_error<PGRES_COMMAND_OK, PGRES_TUPLES_OK>(e);
finish(res);
if(e) throw e;
}
base_statement::close();
}
template<typename Handler>
void close(Handler&& handler)
{
while (m_res)
{
if(PQisBusy(m_conn))
{
async_wait([this, handler](const error& e) mutable {
close(handler);
});
}
else
{
m_res = PQgetResult(m_conn);
}
}
if (!_name.empty() && PQstatus(m_conn) == CONNECTION_OK)
{
std::ostringstream oss;
oss << "DEALLOCATE " << _name << ";";
bool ok = PQsendQuery(m_conn, oss.str().data());
if (ok)
{
async_wait([this, handler](postgres::error e) mutable {
if (PQstatus(m_conn) == CONNECTION_OK)
{
result res(PQgetResult(m_conn));
if (res)
res.verify_error<PGRES_COMMAND_OK, PGRES_TUPLES_OK>(e);
if (!e) _name.clear();
finish(res);
handler(e);
}
else
{
_name.clear();
handler(error());
}
});
}
else
{
handler(error(m_conn));
}
}
else
{
_name.clear();
}
}
/*
ExecuteHandler defiens as:
void handler(const qtl::mysql::error& e, uint64_t affected);
*/
template<typename ExecuteHandler>
void execute(ExecuteHandler&& handler)
{
if (PQsendQueryPrepared(m_conn, _name.data(), 0, nullptr, nullptr, nullptr, 1) &&
PQsetSingleRowMode(m_conn))
{
async_wait([this, handler](error e) {
if (!e)
{
m_res = PQgetResult(m_conn);
m_res.verify_error<PGRES_COMMAND_OK, PGRES_SINGLE_TUPLE>(e);
finish(m_res);
}
handler(e);
});
}
else
{
handler(error(m_conn));
}
}
template<typename Types, typename Handler>
void execute(const Types& params, Handler&& handler)
{
const size_t count = qtl::params_binder<statement, Types>::size;
if (count > 0)
{
m_binders.resize(count);
qtl::bind_params(*this, params);
std::array<const char*, count> values;
std::array<int, count> lengths;
std::array<int, count> formats;
for (size_t i = 0; i != m_binders.size(); i++)
{
values[i] = m_binders[i].value();
lengths[i] = static_cast<int>(m_binders[i].length());
formats[i] = 1;
}
if (!PQsendQueryPrepared(m_conn, _name.data(), static_cast<int>(m_binders.size()), values.data(), lengths.data(), formats.data(), 1))
{
handler(error(m_conn), 0);
return;
}
}
else
{
if (!PQsendQueryPrepared(m_conn, _name.data(), 0, nullptr, nullptr, nullptr, 1))
{
handler(error(m_conn), 0);
return;
}
}
if (!PQsetSingleRowMode(m_conn))
{
handler(error(m_conn), 0);
return;
}
if (PQisBusy(m_conn))
{
async_wait([this, handler](error e) mutable {
if (!e)
{
m_res = PQgetResult(m_conn);
m_res.verify_error<PGRES_COMMAND_OK, PGRES_SINGLE_TUPLE>(e);
int64_t affected = m_res.affected_rows();
finish(m_res);
handler(e, affected);
}
else
{
handler(e, 0);
}
});
}
}
template<typename Types, typename RowHandler, typename FinishHandler>
void fetch(Types&& values, RowHandler&& row_handler, FinishHandler&& finish_handler)
{
if (m_res)
{
ExecStatusType status = m_res.status();
if (status == PGRES_SINGLE_TUPLE)
{
int count = m_res.get_column_count();
if (count > 0)
{
m_binders.resize(count);
for (int i = 0; i != count; i++)
{
m_binders[i] = binder(m_res.get_value(0, i), m_res.length(0, i),
m_res.get_column_type(i));
}
qtl::bind_record(*this, std::forward<Types>(values));
}
row_handler();
if (PQisBusy(m_conn))
{
async_wait([this, &values, row_handler, finish_handler](const error& e) {
if (e)
{
finish_handler(e);
}
else
{
m_res = PQgetResult(m_conn);
fetch(std::forward<Types>(values), row_handler, finish_handler);
}
});
}
else
{
m_res = PQgetResult(m_conn);
fetch(std::forward<Types>(values), row_handler, finish_handler);
}
}
else
{
error e;
m_res.verify_error<PGRES_TUPLES_OK>(e);
finish_handler(e);
}
}
else
{
finish_handler(error());
}
}
template<typename Handler>
void next_result(Handler&& handler)
{
async_wait([this, handler](const error& e) {
if (e)
{
handler(e);
}
else
{
m_res = PQgetResult(m_conn);
handler(error());
}
});
}
private:
event* m_event;
int m_timeout;
template<typename Handler>
void async_wait(Handler&& handler)
{
qtl::postgres::async_wait(m_event, m_conn, m_timeout, std::forward<Handler>(handler));
}
};
class async_connection : public base_database, public qtl::async_connection<async_connection, async_statement>
{
public:
async_connection() : m_connect_timeout(2), m_query_timeout(2)
{
}
async_connection(async_connection&& src)
: base_database(std::move(src)), m_connect_timeout(src.m_connect_timeout), m_query_timeout(src.m_query_timeout)
{
}
async_connection& operator=(async_connection&& src)
{
if (this != &src)
{
base_database::operator=(std::move(src));
m_connect_timeout = src.m_connect_timeout;
m_query_timeout = src.m_query_timeout;
}
return *this;
}
/*
OpenHandler defines as:
void handler(const qtl::postgres::error& e) NOEXCEPT;
*/
template<typename EventLoop, typename OpenHandler>
void open(EventLoop& ev, OpenHandler&& handler, const std::map<std::string, std::string>& params, bool expand_dbname = false)
{
std::vector<const char*> keywords;
std::vector<const char*> values;
keywords.reserve(params.size());
values.reserve(params.size());
for (auto& param : params)
{
keywords.push_back(param.first.data());
values.push_back(param.second.data());
}
keywords.push_back(nullptr);
values.push_back(nullptr);
m_conn = PQconnectStartParams(keywords.data(), values.data(), expand_dbname);
if (m_conn == nullptr)
throw std::bad_alloc();
if (status() == CONNECTION_BAD)
{
handler(error(m_conn));
return;
}
if (PQsetnonblocking(m_conn, true)!=0)
handler(error(m_conn));
get_options();
bind(ev);
wait_connect(std::forward<OpenHandler>(handler));
}
template<typename EventLoop, typename OpenHandler>
void open(EventLoop& ev, OpenHandler&& handler, const char * conninfo)
{
m_conn = PQconnectStart(conninfo);
if (m_conn == nullptr)
throw std::bad_alloc();
if (status() == CONNECTION_BAD)
{
handler(error(m_conn));
return;
}
PQsetnonblocking(m_conn, true);
get_options();
bind(ev);
wait_connect(std::forward<OpenHandler>(handler));
}
template<typename OpenHandler>
void reset(OpenHandler&& handler)
{
PQresetStart(m_conn);
wait_reset(std::forward<OpenHandler>(handler));
}
/*
Handler defines as:
void handler(const qtl::mysql::error& e, uint64_t affected) NOEXCEPT;
*/
template<typename ExecuteHandler>
void simple_execute(ExecuteHandler&& handler, const char* query_text) NOEXCEPT
{
bool ok = PQsendQuery(m_conn, query_text);
if (ok)
{
async_wait([this, handler](postgres::error e) mutable {
result res(PQgetResult(m_conn));
res.verify_error<PGRES_COMMAND_OK, PGRES_TUPLES_OK>(e);
uint64_t affected = res.affected_rows();
handler(e, affected);
while (res)
res = PQgetResult(m_conn);
});
}
else
{
handler(error(m_conn), 0);
}
}
template<typename Handler>
void auto_commit(Handler&& handler, bool on) NOEXCEPT
{
simple_execute(std::forward<Handler>(handler),
on ? "SET AUTOCOMMIT TO ON" : "SET AUTOCOMMIT TO OFF");
}
template<typename Handler>
void begin_transaction(Handler&& handler) NOEXCEPT
{
simple_execute(std::forward<Handler>(handler), "BEGIN");
}
template<typename Handler>
void rollback(Handler&& handler) NOEXCEPT
{
simple_execute(std::forward<Handler>(handler), "ROLLBACK");
}
template<typename Handler>
void commit(Handler&& handler) NOEXCEPT
{
simple_execute(std::forward<Handler>(handler), "COMMIT");
}
/*
ResultHandler defines as:
void result_handler(const qtl::postgres::error& e) NOEXCEPT;
*/
template<typename RowHandler, typename ResultHandler>
void simple_query(const char* query, RowHandler&& row_handler, ResultHandler&& result_handler) NOEXCEPT
{
bool ok = PQsendQuery(m_conn, query);
if (ok)
{
async_wait([this, row_handler, result_handler](postgres::error e) mutable {
result res(PQgetResult(m_conn));
res.verify_error<PGRES_COMMAND_OK, PGRES_TUPLES_OK>(e);
if (e)
{
result_handler(e, 0);
return;
}
uint64_t affected = res.affected_rows();
while (res && res.status() == PGRES_TUPLES_OK)
{
simple_statment stmt(*this, std::move(res));
stmt.fetch_all(row_handler);
res = PQgetResult(m_conn);
}
result_handler(e, affected);
});
}
else
{
result_handler(error(m_conn), 0);
}
}
template<typename Handler>
void open_command(const char* query_text, size_t /*text_length*/, Handler&& handler)
{
std::shared_ptr<async_statement> stmt = std::make_shared<async_statement>(*this);
stmt->open([stmt, handler](const postgres::error& e) mutable {
handler(e, stmt);
}, query_text, 0);
}
template<typename Handler>
void is_alive(Handler&& handler) NOEXCEPT
{
simple_execute(std::forward<Handler>(handler), "");
}
socket_type socket() const NOEXCEPT { return PQsocket(m_conn); }
int connect_timeout() const { return m_connect_timeout; }
void connect_timeout(int timeout) { m_connect_timeout = timeout; }
int query_timeout() const { return m_query_timeout; }
void query_timeout(int timeout) { m_query_timeout = timeout; }
private:
int m_connect_timeout;
int m_query_timeout;
void get_options()
{
PQconninfoOption* options = PQconninfo(m_conn);
m_connect_timeout = 2;
for (PQconninfoOption* option = options; option; option++)
{
if (strcmp(option->keyword, "connect_timeout") == 0)
{
if (option->val)
m_connect_timeout = atoi(option->val);
break;
}
}
PQconninfoFree(options);
}
template<typename OpenHandler>
void wait_connect(OpenHandler&& handler) NOEXCEPT
{
PostgresPollingStatusType status = PQconnectPoll(m_conn);
switch (status)
{
case PGRES_POLLING_READING:
case PGRES_POLLING_WRITING:
m_event_handler->set_io_handler(event_flags(status), m_connect_timeout,
[this, handler](int flags) mutable {
if (flags&event::ef_timeout)
{
handler(postgres::timeout());
}
else if(flags&(event::ef_read|event::ef_write | event::ef_exception))
wait_connect(std::forward<OpenHandler>(handler));
});
break;
case PGRES_POLLING_FAILED:
handler(postgres::error(handle()));
break;
case PGRES_POLLING_OK:
//PQsetnonblocking(m_conn, true);
handler(postgres::error());
}
}
template<typename OpenHandler>
void wait_reset(OpenHandler&& handler) NOEXCEPT
{
PostgresPollingStatusType status = PQresetPoll(m_conn);
switch (status)
{
case PGRES_POLLING_READING:
case PGRES_POLLING_WRITING:
m_event_handler->set_io_handler(event_flags(status), m_connect_timeout,
[this, handler](int flags) mutable {
if (flags&event::ef_timeout)
{
handler(postgres::timeout());
}
else if (flags&(event::ef_read | event::ef_write | event::ef_exception))
wait_reset(std::forward<OpenHandler>(handler));
});
break;
case PGRES_POLLING_FAILED:
handler(postgres::error(m_conn));
break;
case PGRES_POLLING_OK:
handler(postgres::error());
}
}
template<typename Handler>
void async_wait(Handler&& handler)
{
qtl::postgres::async_wait(event(), m_conn, m_query_timeout, std::forward<Handler>(handler));
}
};
inline async_statement::async_statement(async_connection& db)
: base_statement(static_cast<base_database&>(db))
{
m_event = db.event();
m_timeout = db.query_timeout();
}
typedef qtl::transaction<database> transaction;
template<typename Record>
using query_iterator = qtl::query_iterator<statement, Record>;
template<typename Record>
using query_result = qtl::query_result<statement, Record>;
inline base_statement::base_statement(base_database& db) : m_res(nullptr)
{
m_conn = db.handle();
m_res = nullptr;
}
}
}
#endif //_SQL_POSTGRES_H_