f0c1035378
The enitities and containers of the catalog now go into catalog subfolder Models go into model
75 lines
2.3 KiB
C++
75 lines
2.3 KiB
C++
#include "PgConstraintContainer.h"
|
|
#include "Pgsql_Col.h"
|
|
#include <algorithm>
|
|
|
|
std::string PgConstraintContainer::getLoadQuery() const
|
|
{
|
|
std::string q = R"__(
|
|
SELECT oid, conname, connamespace, contype, condeferrable,
|
|
condeferred, convalidated, conrelid, contypid, conindid,
|
|
confrelid, confupdtype, confdeltype, confmatchtype,
|
|
conislocal, coninhcount, connoinherit, conkey, confkey,
|
|
conpfeqop, conppeqop, conffeqop, conexclop, conbin, consrc,
|
|
pg_get_constraintdef(oid)
|
|
FROM pg_constraint)__";
|
|
|
|
// auto cat = m_catalog.lock();
|
|
// if (cat && cat->serverVersion() >= 90400)
|
|
// q += ", indisreplident ";
|
|
// q += "\nFROM pg_index";
|
|
return q;
|
|
}
|
|
|
|
|
|
|
|
PgConstraint PgConstraintContainer::loadElem(const Pgsql::Row &row)
|
|
{
|
|
using namespace Pgsql;
|
|
Col col(row);
|
|
Oid oid = col.nextValue();
|
|
QString name = col.nextValue();
|
|
Oid ns_oid = col.nextValue();
|
|
PgConstraint v(m_catalog, oid, name, ns_oid);
|
|
col >> v.type >> v.deferrable
|
|
>> v.deferred >> v.validated >> v.relid >> v.typid >> v.indid
|
|
>> v.frelid >> v.fupdtype >> v.fdeltype >> v.fmatchtype
|
|
>> v.islocal >> v.inhcount >> v.noinherit
|
|
>> v.key >> v.fkey >> v.pfeqop >> v.ppeqop >> v.ffeqop >> v.exclop
|
|
>> v.bin >> v.src >> v.definition;
|
|
return v;
|
|
}
|
|
|
|
std::vector<PgConstraint> PgConstraintContainer::getFKeyForTableColumn(Oid relid, int16_t attnum) const
|
|
{
|
|
//const PgConstraint *result = nullptr;
|
|
std::vector<PgConstraint> result;
|
|
// WHat do we want to find here? On ly single column constraints or all contstraints.
|
|
auto res = std::copy_if(m_container.begin(), m_container.end(), std::back_inserter(result),
|
|
[relid, attnum] (const auto &c) {
|
|
// the find on v.key may not look super efficient but remember it in general only has one or two elements.
|
|
return c.type == ConstraintType::ForeignKey && relid == c.relid &&
|
|
(std::find(c.key.begin(), c.key.end(), attnum) != c.key.end());
|
|
});
|
|
return result;
|
|
}
|
|
|
|
std::vector<PgConstraint> PgConstraintContainer::getConstraintsForRelation(Oid relid) const
|
|
{
|
|
std::vector<PgConstraint> result;
|
|
for (const auto &e : m_container)
|
|
if (e.relid == relid)
|
|
result.push_back(e);
|
|
return result;
|
|
}
|
|
|
|
std::optional<PgConstraint> PgConstraintContainer::getPrimaryForRelation(Oid relid) const
|
|
{
|
|
std::optional<PgConstraint> result;
|
|
for (const auto &e : m_container) {
|
|
if (e.relid == relid && e.type == ConstraintType::PrimaryKey) {
|
|
result = e;
|
|
break;
|
|
}
|
|
}
|
|
return result;
|
|
}
|