142 lines
2.9 KiB
C++
142 lines
2.9 KiB
C++
#include "IndexModel.h"
|
|
#include "catalog/PgDatabaseCatalog.h"
|
|
#include "catalog/PgIndexContainer.h"
|
|
#include "Pgsql_oids.h"
|
|
#include "ScopeGuard.h"
|
|
#include "CustomDataRole.h"
|
|
|
|
void IndexModel::setData(std::shared_ptr<const PgDatabaseCatalog> cat, const std::optional<PgClass> &table)
|
|
{
|
|
if (cat != m_catalog) {
|
|
m_catalog = cat;
|
|
refreshConnection = connect(m_catalog.get(), &PgDatabaseCatalog::refreshed, this, &IndexModel::refresh);
|
|
}
|
|
m_table = table;
|
|
refresh();
|
|
}
|
|
|
|
void IndexModel::refresh()
|
|
{
|
|
beginResetModel();
|
|
SCOPE_EXIT { endResetModel(); };
|
|
|
|
if (m_table)
|
|
m_indexes = m_catalog->indexes()->getIndexesForTable(m_table->oid());
|
|
else
|
|
m_indexes.clear();
|
|
}
|
|
|
|
int IndexModel::rowCount(const QModelIndex &/*parent*/) const
|
|
{
|
|
return (int)m_indexes.size();
|
|
}
|
|
|
|
int IndexModel::columnCount(const QModelIndex &/*parent*/) const
|
|
{
|
|
return colCount;
|
|
}
|
|
|
|
Oid IndexModel::getType(int column) const
|
|
{
|
|
switch (column) {
|
|
case SizeCol:
|
|
return Pgsql::int8_oid;
|
|
case ExplicitIndexCol:
|
|
return Pgsql::bool_oid;
|
|
default:
|
|
return Pgsql::varchar_oid;
|
|
}
|
|
}
|
|
|
|
QVariant IndexModel::headerData(int section, Qt::Orientation orientation, int role) const
|
|
{
|
|
QVariant v;
|
|
if (orientation == Qt::Horizontal) {
|
|
if (role == Qt::DisplayRole) {
|
|
QString c;
|
|
switch (section) {
|
|
case TypeCol:
|
|
c = tr("Type");
|
|
break;
|
|
case NameCol:
|
|
c = tr("Name");
|
|
break;
|
|
case AmCol:
|
|
c = tr("AM");
|
|
break;
|
|
case ColumnsCol:
|
|
c = tr("On");
|
|
break;
|
|
case ConditionCol:
|
|
c = tr("Condition");
|
|
break;
|
|
case SizeCol:
|
|
c = tr("Size");
|
|
break;
|
|
case ExplicitIndexCol:
|
|
c = tr("Explicit");
|
|
break;
|
|
// case DefinitionCol:
|
|
// c = tr("Definition");
|
|
// break;
|
|
}
|
|
v = c;
|
|
}
|
|
}
|
|
return v;
|
|
|
|
}
|
|
|
|
QVariant IndexModel::getData(const QModelIndex &index) const
|
|
{
|
|
QVariant v;
|
|
int rij = index.row();
|
|
const auto &dat = m_indexes[rij];
|
|
switch (index.column()) {
|
|
case TypeCol:
|
|
if (dat.isprimary) v = ":/icons/constraints/primarykey.png";
|
|
else if (dat.isunique) v = ":/icons/constraints/unique.png";
|
|
else v = ":/icons/constraints/index.png";
|
|
break;
|
|
|
|
case NameCol:
|
|
v = dat.objectName(); // getIndexDisplayString(*m_catalog, dat.indexrelid);
|
|
break;
|
|
|
|
case AmCol:
|
|
v = dat.getAm();
|
|
break;
|
|
|
|
case ColumnsCol:
|
|
break;
|
|
|
|
case ConditionCol:
|
|
break;
|
|
|
|
case SizeCol:
|
|
v = dat.sizeBytes;
|
|
break;
|
|
case ExplicitIndexCol:
|
|
v = !dat.isSupportingIndex();
|
|
break;
|
|
}
|
|
return v;
|
|
}
|
|
|
|
QVariant IndexModel::data(const QModelIndex &index, int role) const
|
|
{
|
|
QVariant v;
|
|
if (role == Qt::DisplayRole)
|
|
v = getData(index);
|
|
else if (role == CustomDataTypeRole)
|
|
v = getType(index.column());
|
|
else if (role == CustomDataMeaningRole) {
|
|
switch (index.column()) {
|
|
case SizeCol:
|
|
return static_cast<int>(DataMeaning::Bytes);
|
|
default:
|
|
return static_cast<int>(DataMeaning::Normal);
|
|
}
|
|
}
|
|
return v;
|
|
}
|