Reorganization of pgLab project

pglab/catalog/models/TablesTableModel.h

@@ -0,0 +1,95 @@
+#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<TableSize>;
+
+    class Table {
+    public:
+        PgClass _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