Add tests for loading the catalog
This commit is contained in:
parent
591573b6b8
commit
a1ac545086
13 changed files with 203 additions and 32 deletions
25
pglablib/catalog/PgDatabaseSizes.cpp
Normal file
25
pglablib/catalog/PgDatabaseSizes.cpp
Normal file
|
|
@ -0,0 +1,25 @@
|
|||
#include "PgDatabaseSizes.h"
|
||||
|
||||
#include "Pgsql_Result.h"
|
||||
|
||||
PgDatabaseSizes loadDatabaseSizes(Pgsql::Connection &connection)
|
||||
{
|
||||
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::Result sizesResult = connection.query(q.c_str());
|
||||
|
||||
PgDatabaseSizes result;
|
||||
result.reserve(sizesResult.rows());
|
||||
for (auto& row : sizesResult)
|
||||
{
|
||||
PgDatabaseSize ds;
|
||||
ds.oid = row.get(0);
|
||||
ds.totalBytes = row.get(1);
|
||||
result.push_back(ds);
|
||||
}
|
||||
return result;
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue