pgLab/pglablib/PgAttributeContainer.cpp

#include "PgAttributeContainer.h"
#include "Pgsql_Col.h"
#include "PgDatabaseCatalog.h"

//SELECT attname, pg_get_expr(adbin, adrelid) AS def_value
//FROM pg_attribute
// JOIN pg_attrdef ON attrelid=adrelid AND attnum=adnum
//WHERE atthasdef=true

std::string PgAttributeContainer::getLoadQuery() const
{
	std::string q = R"__(
SELECT attrelid, attname, atttypid, attstattarget,
	attnum, attndims, atttypmod, attnotnull, atthasdef, attisdropped,
	attcollation, attacl, attoptions, pg_get_expr(adbin, adrelid) AS def_value)__";
	if (m_catalog.serverVersion() >= 100000)
		q += ", attidentity";
	q +=
		"\n  FROM pg_catalog.pg_attribute \n"
		"  LEFT JOIN pg_attrdef ON attrelid=adrelid AND attnum=adnum";
	return q;
}

PgAttribute PgAttributeContainer::loadElem(const Pgsql::Row &row)
{
	Pgsql::Col col(row);
	PgAttribute v;
	col >> v.relid >> v.name >> v.typid >> v.stattarget
			>> v.num >> v.ndims >> v.typmod >> v.notnull >> v.hasdef >> v.isdropped
			>> v.collation >> v.acl >> v.options >> v.defaultValue;
	if (m_catalog.serverVersion() >= 100000)
		col >> v.identity;

	return v;
}

std::vector<PgAttribute> PgAttributeContainer::getColumnsForRelation(Oid oid) const
{
	std::vector<PgAttribute> result;
	for (const auto &e : m_container)
		if (e.relid == oid)
			result.push_back(e);
	return result;
}