Reorganization of pgLab project

2022-04-09 07:10:29 +02:00 · 2022-04-09 07:10:29 +02:00 · c71fdc4af7
commit c71fdc4af7
parent 7300865c77
78 changed files with 204 additions and 148 deletions
--- a/pglab/catalog/models/ProcTableModel.cpp
+++ b/pglab/catalog/models/ProcTableModel.cpp
@ -0,0 +1,131 @@
+#include "ProcTableModel.h"
+#include "catalog/PgDatabaseCatalog.h"
+#include "catalog/PgProcContainer.h"
+#include "catalog/PgLanguageContainer.h"
+#include "catalog/PgNamespace.h"
+#include "CustomDataRole.h"
+
+ProcTableModel::ProcTableModel(QObject *parent)
+	: QAbstractTableModel(parent)
+{
+}
+
+QVariant ProcTableModel::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 NamespaceCol: return tr("Schema");
+			case OwnerCol:     return tr("Owner");
+			case LangCol:      return tr("Language");
+			case AclCol:       return tr("ACL");
+			}
+		}
+	}
+	return QVariant();
+}
+
+void ProcTableModel::setCatalog(std::shared_ptr<const PgDatabaseCatalog> cat)
+{
+	if (cat != m_catalog) {
+		m_catalog = cat;
+		refreshConnection = connect(m_catalog.get(), &PgDatabaseCatalog::refreshed,
+										this, &ProcTableModel::refresh);
+	}
+	refresh();
+}
+
+void ProcTableModel::setNamespaceFilter(NamespaceFilter filter)
+{
+	m_namespaceFilter = filter;
+	refresh();
+}
+
+void ProcTableModel::refresh()
+{
+	if (!m_catalog)
+		return;
+
+	beginResetModel();
+
+	auto && procs = m_catalog->procs();
+	m_procs.clear();
+	for (auto&& p : *procs) {
+		bool add = false;
+		switch (m_namespaceFilter) {
+		case NamespaceFilter::User:
+			add = !p.ns().isSystemCatalog();
+			break;
+		case NamespaceFilter::PgCatalog:
+			add = p.ns().objectName() == "pg_catalog";
+			break;
+		case NamespaceFilter::InformationSchema:
+			add = p.ns().objectName() == "information_schema";
+			break;
+		}
+		if (add)
+			m_procs.push_back(p);
+	}
+
+	endResetModel();
+}
+
+int ProcTableModel::rowCount(const QModelIndex &) const
+{
+	return static_cast<int>(m_procs.size());
+}
+
+int ProcTableModel::columnCount(const QModelIndex &) const
+{
+	return colCount;
+}
+
+QVariant ProcTableModel::data(const QModelIndex &index, int role) const
+{
+	if (role == Qt::DisplayRole)
+		return getData(index);
+	else if (role == CustomDataTypeRole)
+		return getType(index.column());
+//	else if (role == FirstHiddenValue) {
+//		auto&& t = m_triggers->getByIdx(index.row());
+//		return t.relid; //
+//	}
+	return QVariant();
+}
+
+PgProc ProcTableModel::proc(int row) const
+{
+	return m_procs.at(static_cast<size_t>(row));
+}
+
+Oid ProcTableModel::getType(int ) const
+{
+//	switch (column) {
+//	case NameCol:      return tr("Name");
+//	case NamespaceCol: return tr("Schema");
+//	case OwnerCol:     return tr("Owner");
+//	case LangCol:      return tr("Language");
+//			return Pgsql::bool_oid;
+//	}
+	return Pgsql::varchar_oid;
+}
+
+QVariant ProcTableModel::getData(const QModelIndex &index) const
+{
+	auto&& t = m_procs.at(static_cast<size_t>(index.row()));
+	switch (index.column()) {
+	case NameCol: return t.objectName();
+	case NamespaceCol: return t.nsName();
+	case OwnerCol: return t.ownerName();
+	case LangCol: {
+		auto lan = m_catalog->languages()->getByKey(t.lang);
+		if (lan) return lan->objectName();
+		return t.lang;
+	}
+	case AclCol: return t.aclString();
+	}
+	return QVariant();
+}
+
+