eelke/pgLab
1
0
Fork 0
Code Issues 28 Pull requests Projects Releases 1 Packages Wiki Activity Actions

lazy load the database sizes only when needed by the server inspector.

Browse source
This commit is contained in:
eelke 2022-01-19 19:10:05 +01:00
parent 39195153cd
commit b5a706a2a2
10 changed files with 165 additions and 70 deletions
Download patch file Download diff file Expand all files Collapse all files

180
pglab/DatabasesTableModel.cpp
View file

@ -1,15 +1,19 @@
#include "DatabasesTableModel.h"
#include "CustomDataRole.h"
#include "OpenDatabase.h"
#include "ScopeGuard.h"
#include "catalog/PgDatabaseCatalog.h"
#include "catalog/PgDatabaseContainer.h"
#include "catalog/PgAuthIdContainer.h"
#include "ResultTableModelUtil.h"
#include <QtConcurrent>
#include "Pgsql_Connection.h"
using namespace Pgsql;
DatabasesTableModel::DatabasesTableModel(QObject *parent)
DatabasesTableModel::DatabasesTableModel(std::shared_ptr<OpenDatabase> opendatabase, QObject *parent)
: BaseTableModel(parent)
, openDatabase(opendatabase)
{
}
@ -18,15 +22,27 @@ void DatabasesTableModel::setDatabaseList(std::shared_ptr<const PgDatabaseCatalo
beginResetModel();
SCOPE_EXIT { endResetModel(); };
m_catalog = cat;
m_databases = cat->databases();
databases.clear();
std::map<Oid, int> oidIndex;
m_catalog = cat;
auto dats = cat->databases();
databases.reserve(dats->count());
for (const auto& d : *dats)
{
databases.emplace_back(d);
oidIndex.emplace(d.oid(), databases.size() - 1);
}
StartLoadDatabaseSizes(std::move(oidIndex));
}
QVariant DatabasesTableModel::headerData(int section, Qt::Orientation orientation, int role) const
{
QVariant v;
if (orientation == Qt::Horizontal) {
if (role == Qt::DisplayRole) {
if (orientation == Qt::Horizontal)
{
if (role == Qt::DisplayRole)
{
switch (section) {
case NameCol:
v = tr("Name");
@ -74,9 +90,7 @@ QVariant DatabasesTableModel::headerData(int section, Qt::Orientation orientatio
int DatabasesTableModel::rowCount(const QModelIndex &) const
{
int result = 0;
if (m_databases) {
result = m_databases->count();
}
result = databases.size();
return result;
}
@ -90,7 +104,8 @@ int DatabasesTableModel::columnCount(const QModelIndex &) const
Oid DatabasesTableModel::getType(int column) const
{
Oid oid;
switch (column) {
switch (column)
{
case AllowConnCol:
case IsTemplateCol:
oid = Pgsql::bool_oid;
@ -120,58 +135,113 @@ Oid DatabasesTableModel::getType(int column) const
QVariant DatabasesTableModel::getData(const QModelIndex &index) const
{
QVariant v;
//if (!index.isValid())
if (m_databases) {
const PgDatabase &db = m_databases->getByIdx(index.row());
switch (index.column()) {
case NameCol:
v = db.objectName();
break;
case DbaCol:
v = getRoleDisplayString(*m_catalog, db.dba);
break;
case EncodingCol:
// todo lookup encoding name
v = db.encodingString;
break;
case CollateCol:
v = db.collate;
break;
case CTypeCol:
v = db.ctype;
break;
case IsTemplateCol:
v = db.isTemplate;
break;
case AllowConnCol:
v = db.allowConn;
break;
case ConnLimitCol:
v = db.connLimit;
break;
case TablespaceCol:
// todo lookup tablespace name
v = getTablespaceDisplayString(*m_catalog, db.tablespace);
break;
case CommentCol:
v = db.description;
break;
case SizeCol:
if (db.sizeBytes >= 0)
v = db.sizeBytes;
break;
case AclCol:
v = db.aclString();
break;
}
}
const Database &d = databases[index.row()];
const PgDatabase &db = d.database;
switch (index.column()) {
case NameCol:
v = db.objectName();
break;
case DbaCol:
v = getRoleDisplayString(*m_catalog, db.dba);
break;
case EncodingCol:
// todo lookup encoding name
v = db.encodingString;
break;
case CollateCol:
v = db.collate;
break;
case CTypeCol:
v = db.ctype;
break;
case IsTemplateCol:
v = db.isTemplate;
break;
case AllowConnCol:
v = db.allowConn;
break;
case ConnLimitCol:
v = db.connLimit;
break;
case TablespaceCol:
// todo lookup tablespace name
v = getTablespaceDisplayString(*m_catalog, db.tablespace);
break;
case CommentCol:
v = db.description;
break;
case SizeCol:
if (d.size.totalBytes >= 0)
v = d.size.totalBytes;
break;
case AclCol:
v = db.aclString();
break;
}
return v;
}
QVariant DatabasesTableModel::getDataMeaning(const QModelIndex &index) const
{
if (index.column() == SizeCol)
{
return static_cast<int>(DataMeaningBytes);
}
return BaseTableModel::getDataMeaning(index);
}
void DatabasesTableModel::StartLoadDatabaseSizes(std::map<Oid, int> oidIndex)
{
QPointer p(this);
QtConcurrent::run([this]
{
return QueryDatabaseSizes();
})
.then(qApp, [p, oi = std::move(oidIndex)] (DatabaseSizes sizes)
{
if (p)
{
p.data()->PopulateSizes(std::move(oi), std::move(sizes));
}
});
}
DatabasesTableModel::DatabaseSizes DatabasesTableModel::QueryDatabaseSizes()
{
std::string q =
"SELECT pg_database.oid, "
" case when has_database_privilege(current_role, pg_database.oid, \n"
" 'connect') then pg_database_size(pg_database.oid) else -1 end \n"
"FROM pg_database";
Pgsql::Connection connection;
connection.connect(openDatabase->config().connectionString());
Pgsql::Result sizesResult = connection.query(q.c_str());
DatabaseSizes result;
result.reserve(sizesResult.rows());
for (auto& row : sizesResult)
{
DatabaseSize ds;
ds.oid = row.get(0);
ds.totalBytes = row.get(1);
result.push_back(ds);
}
return result;
}
void DatabasesTableModel::PopulateSizes(std::map<Oid, int> oidIndex, DatabaseSizes sizes)
{
for (auto & s : sizes)
{
auto findIter = oidIndex.find(s.oid);
if (findIter != oidIndex.end())
{
databases[findIter->second].size = s;
}
}
emit dataChanged(
createIndex(0, SizeCol),
createIndex(static_cast<int>(databases.size()), SizeCol)
);
}

36
pglab/DatabasesTableModel.h
View file

@ -2,11 +2,11 @@
#define DATABASESTABLEMODEL_H
#include "BaseTableModel.h"
#include "catalog/PgDatabase.h"
#include <memory>
class PgDatabaseContainer;
class OpenDatabase;
class PgDatabaseCatalog;
class PgDatabase;
/** Class for displaying the list of databases of a server in a QTableView
*
@ -24,7 +24,7 @@ public:
explicit DatabasesTableModel(QObject *parent);
explicit DatabasesTableModel(std::shared_ptr<OpenDatabase> opendatabase, QObject *parent);
void setDatabaseList(std::shared_ptr<const PgDatabaseCatalog> cat);
@ -41,8 +41,36 @@ public:
protected:
virtual QVariant getDataMeaning(const QModelIndex &index) const override;
private:
class DatabaseSize {
public:
Oid oid;
int64_t totalBytes;
DatabaseSize()
: oid(InvalidOid)
, totalBytes(-1)
{}
};
using DatabaseSizes = std::vector<DatabaseSize>;
class Database {
public:
PgDatabase database;
DatabaseSize size;
Database(const PgDatabase &db)
: database(db)
{}
};
using Databases = std::vector<Database>;
std::shared_ptr<OpenDatabase> openDatabase;
std::shared_ptr<const PgDatabaseCatalog> m_catalog;
std::shared_ptr<const PgDatabaseContainer> m_databases;
Databases databases;
void StartLoadDatabaseSizes(std::map<Oid, int> oidIndex);
DatabaseSizes QueryDatabaseSizes();
void PopulateSizes(std::map<Oid, int> oidIndex, DatabaseSizes sizes);
};
#endif // DATABASESTABLEMODEL_H

2
pglab/TablesTableModel.cpp
View file

@ -188,7 +188,7 @@ void TablesTableModel::StartLoadTableSizes(std::map<Oid, int> oidIndex)
{
return QueryTableSizes();
})
.then(qApp, [p, oidIndex] (std::vector<TableSize> sizes)
.then(qApp, [p, oidIndex] (TableSizes sizes)
{
if (p)
{

4
pglab/TablesTableModel.h
View file

@ -60,12 +60,12 @@ private:
: oid(0)
{}
};
using TableSizes = std::vector<class TableSize>;
using TableSizes = std::vector<TableSize>;
class Table {
public:
PgClass _class;
class TableSize sizes;
TableSize sizes;
Table(const PgClass &cls)
: _class(cls)

4
pglab/serverinspector/DatabasesPage.cpp
View file

@ -5,9 +5,9 @@
#include "PgLabTableView.h"
DatabasesPage::DatabasesPage(QWidget * parent)
DatabasesPage::DatabasesPage(std::shared_ptr<OpenDatabase> opendatabase, QWidget * parent)
: QSplitter(Qt::Horizontal, parent)
, m_databasesTableView(this)
, m_databasesTableView(this, new DatabasesTableModel(opendatabase, this))
{
auto tv = m_databasesTableView.tableView();
tv->setSelectionMode(QAbstractItemView::SingleSelection);

2
pglab/serverinspector/DatabasesPage.h
View file

@ -12,7 +12,7 @@ class QSortFilterProxyModel;
class DatabasesPage: public QSplitter {
public:
explicit DatabasesPage(QWidget * parent = nullptr);
explicit DatabasesPage(std::shared_ptr<OpenDatabase> opendatabase, QWidget * parent = nullptr);
void setCatalog(std::shared_ptr<PgDatabaseCatalog> cat);

2
pglab/serverinspector/ServerInspector.cpp
View file

@ -9,7 +9,7 @@
ServerInspector::ServerInspector(std::shared_ptr<OpenDatabase> open_database, QWidget *parent)
: QWidget(parent)
{
m_databasesPage = new DatabasesPage(this);
m_databasesPage = new DatabasesPage(open_database, this);
m_rolesPage = new RolesPage(this);
m_tabWidget = new QTabWidget(this);