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

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

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

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

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

const char *
PQ::Column::value() const
{
	return PQgetvalue(sc->execRes, (int)sc->tuple, (int)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(val));
			}
			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;
		}
	}
}