summaryrefslogtreecommitdiff
path: root/libpqpp/pq-column.cpp
blob: e743e604be6f4c11bb7b7c258133c0e12e13adc8 (plain)
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
#include "pq-column.h"
#include "pq-selectbase.h"
#include "pq-error.h"
#include <cstring>
#include <boost/date_time/posix_time/posix_time.hpp>

PQ::Column::Column(const SelectBase * s, unsigned int i) :
	DB::Column(PQfname(s->execRes, i), i),
	sc(s),
	oid(PQftype(sc->execRes, colNo)),
	buf(nullptr)
{
}

PQ::Column::~Column()
{
	if (buf) {
		PQfreemem(buf);
	}
}

bool
PQ::Column::isNull() const
{
	return PQgetisnull(sc->execRes, sc->tuple, colNo);
}

std::size_t
PQ::Column::length() const
{
	return PQgetlength(sc->execRes, sc->tuple, colNo);
}

const char *
PQ::Column::value() const
{
	return PQgetvalue(sc->execRes, sc->tuple, colNo);
}

void
PQ::Column::apply(DB::HandleField & h) const
{
	if (isNull()) {
		h.null();
		return;
	}
	switch (oid) {
		case 18: //CHAROID:
		case 1043: //VARCHAROID:
		case 25: //TEXTOID:
		case 142: //XMLOID:
		default:
			h.string({ value(), length() });
			break;
		case 16: //BOOLOID:
			h.boolean(value()[0] == 't');
			break;
		case 21: //INT2OID:
		case 23: //INT4OID:
		case 20: //INT8OID:
			h.integer(atol(value()));
			break;
		case 1700: //NUMERICOID:
		case 700: //FLOAT4OID:
		case 701: //FLOAT8OID:
			h.floatingpoint(atof(value()));
			break;
		case 704: //TINTERVALOID
		case 1083: //TIMEOID:
		case 1186: //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 * (long)pow(10, boost::posix_time::time_res_traits::num_fractional_digits() + flen1 - flen2)));
				}
				else {
					h.interval(boost::posix_time::duration_from_string(value()));
				}
				break;
			}
		case 1082: //DATEOID:
			h.timestamp(boost::posix_time::ptime(
						boost::gregorian::from_string(value())));
			break;
		case 702: //ABSTIMEOID:
		case 1114: //TIMESTAMPOID:
		case 1184: //TIMESTAMPTZOID:
			h.timestamp(boost::posix_time::time_from_string(value()));
			break;
		case 17: //BYTEAOID
			{
				if (buf) {
					PQfreemem(buf);
				}
				size_t len;
				buf = PQunescapeBytea(valueAsPtr<unsigned char>(), &len);
				h.blob(DB::Blob(buf, len));
				break;
			}
	}
}