pg_classes can be loaded now, used in TablesTableModel for overview of tables in database.

pglab/PgDatabaseCatalog.cpp

@@ -0,0 +1,151 @@
+#include "PgDatabaseCatalog.h"
+#include "PgTypeContainer.h"
+#include "PgDatabaseContainer.h"
+#include "PgAuthIdContainer.h"
+#include "PgClassContainer.h"
+#include "Pgsql_Connection.h"
+
+
+QString getRoleNameFromOid(const PgDatabaseCatalog &cat, Oid oid)
+{
+	QString name;
+	auto auth_ids = cat.authIds();
+	if (auth_ids) {
+		const PgAuthId& auth_id = auth_ids->getByOid(oid);
+		if (auth_id.valid()) {
+			name = auth_id.name;
+		}
+	}
+	return name;
+}
+
+QString getRoleDisplayString(const PgDatabaseCatalog &cat, Oid oid)
+{
+	QString name = getRoleNameFromOid(cat, oid);
+	return QString("%1 (%2)").arg(name).arg(oid);
+}
+
+QString getNamespaceDisplayString(const PgDatabaseCatalog &cat, Oid oid)
+{
+	//QString name;
+//	auto c = cat.();
+//	if (auth_ids) {
+//		const PgAuthId& auth_id = auth_ids->getByOid(oid);
+//		if (auth_id.valid()) {
+//			name = auth_id.name;
+//		}
+//	}
+	//return name;
+	// TODO load list and lookup name
+	return QString("ns %1").arg(oid);
+
+}
+
+QString getTablespaceDisplayString(const PgDatabaseCatalog &cat, Oid oid)
+{
+	// TODO load list and lookup name
+	return QString("ts %1").arg(oid);
+}
+
+
+PgDatabaseCatalog::PgDatabaseCatalog()
+{
+}
+
+PgDatabaseCatalog::~PgDatabaseCatalog()
+{
+}
+
+void PgDatabaseCatalog::loadAll(Pgsql::Connection &conn)
+{
+	loadTypes(conn);
+	loadDatabases(conn);
+	loadAuthIds(conn);
+	loadClasses(conn);
+}
+
+void PgDatabaseCatalog::loadInfo(Pgsql::Connection &conn)
+{
+	Pgsql::Result r = conn.query("SHOW server_version_num");
+	if (r && r.resultStatus() == PGRES_TUPLES_OK)
+		if (r.rows() == 1)
+			m_serverVersion << r.get(0, 0);
+
+	r = conn.query("SELECT version()");
+	if (r && r.resultStatus() == PGRES_TUPLES_OK)
+		if (r.rows() == 1)
+			m_serverVersionString = r.get(0, 0).asQString();
+}
+
+void load(Pgsql::Connection &conn, IPgContainter &pg_cont)
+{
+	std::string q = pg_cont.getLoadQuery();
+	Pgsql::Result result = conn.query(q.c_str());
+	if (result && result.resultStatus() == PGRES_TUPLES_OK)
+		pg_cont.load(result);
+	else
+		throw std::runtime_error("Query failed");
+
+}
+
+void PgDatabaseCatalog::loadTypes(Pgsql::Connection &conn)
+{
+	if (!m_types)
+		m_types = std::make_shared<PgTypeContainer>(shared_from_this());
+
+	load(conn, *m_types);
+}
+
+void PgDatabaseCatalog::loadDatabases(Pgsql::Connection &conn)
+{
+	if (!m_databases)
+		m_databases = std::make_shared<PgDatabaseContainer>(shared_from_this());
+
+	load(conn, *m_databases);
+}
+
+void PgDatabaseCatalog::loadAuthIds(Pgsql::Connection &conn)
+{
+	if (!m_authIds)
+		m_authIds = std::make_shared<PgAuthIdContainer>(shared_from_this());
+
+	load(conn, *m_authIds);
+}
+
+void PgDatabaseCatalog::loadClasses(Pgsql::Connection &conn)
+{
+	if (!m_classes)
+		m_classes = std::make_shared<PgClassContainer>(shared_from_this());
+
+	load(conn, *m_classes);
+}
+
+const QString& PgDatabaseCatalog::serverVersionString() const
+{
+	return m_serverVersionString;
+}
+
+int PgDatabaseCatalog::serverVersion() const
+{
+	return m_serverVersion;
+}
+
+std::shared_ptr<const PgTypeContainer> PgDatabaseCatalog::types() const
+{
+	return m_types;
+}
+
+std::shared_ptr<const PgDatabaseContainer> PgDatabaseCatalog::databases() const
+{
+	return m_databases;
+}
+
+std::shared_ptr<const PgAuthIdContainer> PgDatabaseCatalog::authIds() const
+{
+	return m_authIds;
+}
+
+std::shared_ptr<const PgClassContainer> PgDatabaseCatalog::classes() const
+{
+	return m_classes;
+}