libpqpp/pq-column.h


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

#pragma once

#include "pq-helpers.h"
#include <column.h>
#include <cstring>
#include <libpq-fe.h>

namespace PQ {
	class SelectBase;

	class Column : public DB::Column {
	public:
		Column(const SelectBase *, unsigned int field);

		[[nodiscard]] bool isNull() const override;
		void apply(DB::HandleField &) const override;

	protected:
		const char * value() const;
		std::size_t length() const;

		const SelectBase * sc;
		const Oid oid;

		// Buffer for PQunescapeBytea
		struct Buffer {
			using BufPtr = std::unique_ptr<unsigned char, pq_deleter<PQfreemem>>;
			size_t row {};
			size_t length {};
			BufPtr data;
		};

		mutable Buffer buffer;
	};
}