101 lines
2.4 KiB
C++
101 lines
2.4 KiB
C++
#include "TypeModel.h"
|
|
#include "catalog/PgDatabaseCatalog.h"
|
|
#include "catalog/PgNamespace.h"
|
|
#include "catalog/PgTypeContainer.h"
|
|
|
|
TypeModel::TypeModel(QObject * parent)
|
|
: QAbstractTableModel(parent)
|
|
{
|
|
}
|
|
|
|
QVariant TypeModel::headerData(int section, Qt::Orientation orientation, int role) const
|
|
{
|
|
if (orientation == Qt::Horizontal) {
|
|
if (role == Qt::DisplayRole) {
|
|
switch (section) {
|
|
case NameCol: return tr("Name");
|
|
case SchemaCol: return tr("Schema");
|
|
case OwnerCol: return tr("Owner");
|
|
}
|
|
}
|
|
}
|
|
return {};
|
|
}
|
|
|
|
void TypeModel::setCatalog(std::shared_ptr<const PgDatabaseCatalog> cat)
|
|
{
|
|
if (cat != m_catalog) {
|
|
m_catalog = cat;
|
|
refreshConnection = connect(m_catalog.get(), &PgDatabaseCatalog::refreshed,
|
|
this, &TypeModel::refresh);
|
|
}
|
|
refresh();
|
|
}
|
|
|
|
//void TypeModel::setNamespaceFilter(NamespaceFilter filter)
|
|
//{
|
|
// m_namespaceFilter = filter;
|
|
// refresh();
|
|
//}
|
|
|
|
void TypeModel::refresh()
|
|
{
|
|
if (!m_catalog)
|
|
return;
|
|
|
|
beginResetModel();
|
|
|
|
auto && typs = m_catalog->types();
|
|
m_types.clear();
|
|
for (auto&& s : *typs) {
|
|
// bool add = false;
|
|
// switch (m_namespaceFilter) {
|
|
// case NamespaceFilter::User:
|
|
// add = !s.ns().isSystemCatalog();
|
|
// break;
|
|
// case NamespaceFilter::PgCatalog:
|
|
// add = s.ns().objectName() == "pg_catalog";
|
|
// break;
|
|
// case NamespaceFilter::InformationSchema:
|
|
// add = s.ns().objectName() == "information_schema";
|
|
// break;
|
|
// }
|
|
// if (add)
|
|
m_types.push_back(s);
|
|
}
|
|
|
|
endResetModel();
|
|
}
|
|
|
|
PgType TypeModel::typ(int row) const
|
|
{
|
|
return m_types.at(static_cast<size_t>(row));
|
|
}
|
|
|
|
int TypeModel::rowCount(const QModelIndex &) const
|
|
{
|
|
return static_cast<int>(m_types.size());
|
|
}
|
|
|
|
int TypeModel::columnCount(const QModelIndex &) const
|
|
{
|
|
return colCount;
|
|
}
|
|
|
|
QVariant TypeModel::data(const QModelIndex &index, int role) const
|
|
{
|
|
if (m_types.empty())
|
|
return {};
|
|
|
|
int row = index.row();
|
|
auto && seq = m_types.at(static_cast<size_t>(row));
|
|
if (role == Qt::DisplayRole) {
|
|
switch (index.column()) {
|
|
case NameCol: return seq.objectName();
|
|
case SchemaCol: return seq.nsName();
|
|
case OwnerCol: return seq.ownerName();
|
|
}
|
|
}
|
|
|
|
return {};
|
|
}
|