libpqpp/pq-selectbase.cpp


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

#include "pq-selectbase.h"
#include "command.h"
#include "pq-binarycolumn.h"
#include "pq-column.h"
#include "pq-command.h"
#include <libpq-fe.h>
#include <memory>
#include <selectcommand.h>
#include <string>

PQ::SelectBase::SelectBase(const std::string & sql, const PQ::CommandOptionsCPtr & pqco) :
	DB::Command(sql), DB::SelectCommand(sql), nTuples(0), tuple(0), execRes(nullptr),
	binary(pqco ? pqco->fetchBinary : false)
{
}

PQ::SelectBase::~SelectBase()
{
	if (execRes) {
		PQclear(execRes);
	}
}

void
PQ::SelectBase::createColumns(PGresult * execRes)
{
	auto nFields = PQnfields(execRes);
	for (decltype(nFields) f = 0; f < nFields; f += 1) {
		if (binary) {
			insertColumn(std::make_unique<BinaryColumn>(this, f));
		}
		else {
			insertColumn(std::make_unique<Column>(this, f));
		}
	}
}