1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
|
#include "pq-column.h"
#include "column.h"
#include "dbTypes.h"
#include "pq-selectbase.h"
#include <boost/date_time/gregorian_calendar.hpp>
#include <boost/date_time/posix_time/posix_time.hpp>
#include <cmath>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <libpq-fe.h>
#include <memory>
#include <server/catalog/pg_type_d.h>
PQ::Column::Column(const SelectBase * s, unsigned int i) :
DB::Column(PQfname(s->execRes, static_cast<int>(i)), i), sc(s), oid(PQftype(sc->execRes, static_cast<int>(colNo))),
buf(nullptr)
{
}
PQ::Column::~Column()
{
if (buf) {
PQfreemem(buf);
}
}
bool
PQ::Column::isNull() const
{
return PQgetisnull(sc->execRes, static_cast<int>(sc->tuple), static_cast<int>(colNo));
}
std::size_t
PQ::Column::length() const
{
return static_cast<std::size_t>(PQgetlength(sc->execRes, static_cast<int>(sc->tuple), static_cast<int>(colNo)));
}
const char *
PQ::Column::value() const
{
return PQgetvalue(sc->execRes, static_cast<int>(sc->tuple), static_cast<int>(colNo));
}
void
PQ::Column::apply(DB::HandleField & h) const
{
if (isNull()) {
h.null();
return;
}
switch (oid) {
case CHAROID:
case VARCHAROID:
case TEXTOID:
case XMLOID:
default:
h.string({value(), length()});
break;
case BOOLOID:
h.boolean(value()[0] == 't');
break;
case INT2OID:
case INT4OID:
case INT8OID:
h.integer(atol(value()));
break;
case NUMERICOID:
case FLOAT4OID:
case FLOAT8OID:
h.floatingpoint(atof(value()));
break;
case TIMEOID:
case INTERVALOID: {
int days = 0, hours = 0, minutes = 0, seconds = 0, fractions = 0, flen1 = 0, flen2 = 0;
const char * val = value();
// NOLINTNEXTLINE(hicpp-vararg)
if (sscanf(val, "%d %*[days] %d:%d:%d.%n%d%n", &days, &hours, &minutes, &seconds, &flen1, &fractions,
&flen2)
>= 4) {
h.interval(boost::posix_time::time_duration((24 * days) + hours, minutes, seconds,
fractions
* static_cast<long>(pow(10,
boost::posix_time::time_res_traits::num_fractional_digits() + flen1 - flen2))));
}
else {
h.interval(boost::posix_time::duration_from_string(val));
}
break;
}
case DATEOID:
h.timestamp(boost::posix_time::ptime(boost::gregorian::from_string(value())));
break;
case TIMESTAMPOID:
case TIMESTAMPTZOID:
h.timestamp(boost::posix_time::time_from_string(value()));
break;
case BYTEAOID: {
if (buf) {
PQfreemem(buf);
}
size_t len;
buf = PQunescapeBytea(reinterpret_cast<const unsigned char *>(value()), &len);
h.blob(DB::Blob(buf, len));
break;
}
}
}
|
