95 lines
2.3 KiB
C++
95 lines
2.3 KiB
C++
#ifndef TABLESTABLEMODEL_H
|
|
#define TABLESTABLEMODEL_H
|
|
|
|
#include "BaseTableModel.h"
|
|
#include "NamespaceFilter.h"
|
|
#include "catalog/PgClass.h"
|
|
#include <memory>
|
|
#include <vector>
|
|
|
|
class OpenDatabase;
|
|
class PgClass;
|
|
class PgDatabaseCatalog;
|
|
|
|
class TablesTableModel: public QAbstractTableModel {
|
|
public:
|
|
using RowItem = PgClass;
|
|
enum e_Columns : int {
|
|
NameCol, ///< either table, ns.table or table (ns) depending on settings/filters
|
|
NamespaceCol,
|
|
KindCol,
|
|
OwnerCol,
|
|
TablespaceCol,
|
|
OptionsCol,
|
|
AclCol,
|
|
CommentCol,
|
|
TotalSizeCol,
|
|
TableSizeCol,
|
|
IndexSizeCol,
|
|
ToastSizeCol,
|
|
|
|
colCount };
|
|
|
|
TablesTableModel(std::shared_ptr<OpenDatabase> opendatabase, QObject *parent);
|
|
|
|
void setNamespaceFilter(NamespaceFilter nsf);
|
|
void setCatalog(std::shared_ptr<const PgDatabaseCatalog> cat);
|
|
|
|
QVariant headerData(int section, Qt::Orientation orientation, int role) const override;
|
|
|
|
int rowCount(const QModelIndex &parent) const override;
|
|
int columnCount(const QModelIndex &parent) const override;
|
|
|
|
virtual QVariant data(const QModelIndex &index, int role) const override;
|
|
PgClass getTable(int row) const;
|
|
RowItem rowItem(int row) const
|
|
{
|
|
return getTable(row);
|
|
}
|
|
Oid getTableOid(int row) const;
|
|
|
|
private:
|
|
class TableSize {
|
|
public:
|
|
int oid;
|
|
int64_t totalBytes = -1;
|
|
int64_t indexBytes = -1;
|
|
int64_t toastBytes = -1;
|
|
|
|
TableSize()
|
|
: oid(0)
|
|
{}
|
|
};
|
|
using TableSizes = std::vector<class TableSize>;
|
|
|
|
class Table {
|
|
public:
|
|
PgClass _class;
|
|
class TableSize sizes;
|
|
|
|
Table(const PgClass &cls)
|
|
: _class(cls)
|
|
{}
|
|
};
|
|
using t_Tables = std::vector<Table>;
|
|
|
|
std::shared_ptr<const PgDatabaseCatalog> m_catalog;
|
|
NamespaceFilter m_namespaceFilter = NamespaceFilter::User;
|
|
t_Tables m_tables;
|
|
QMetaObject::Connection refreshConnection;
|
|
std::shared_ptr<OpenDatabase> openDatabase;
|
|
|
|
|
|
Oid getType(int column) const;
|
|
QVariant getData(const QModelIndex &index) const;
|
|
|
|
void StartLoadTableSizes(std::map<Oid, int> oidIndex);
|
|
TableSizes QueryTableSizes() const;
|
|
void PopulateSizes(std::map<Oid, int> oidIndex, std::vector<TableSize> sizes);
|
|
|
|
private slots:
|
|
void refresh();
|
|
|
|
};
|
|
|
|
#endif // TABLESTABLEMODEL_H
|