Skip to content
Projects
Groups
Snippets
Help
Loading...
Help
Support
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
P
Postgres FD Implementation
Project overview
Project overview
Details
Activity
Releases
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Issues
0
Issues
0
List
Boards
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Analytics
Analytics
CI / CD
Repository
Value Stream
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
Abuhujair Javed
Postgres FD Implementation
Commits
676bb9a6
Commit
676bb9a6
authored
Aug 12, 2004
by
Bruce Momjian
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
dbsize modification to support tablespaces
Gavin Sherry
parent
a5836751
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
55 additions
and
11 deletions
+55
-11
contrib/dbsize/dbsize.c
contrib/dbsize/dbsize.c
+55
-11
No files found.
contrib/dbsize/dbsize.c
View file @
676bb9a6
...
...
@@ -6,13 +6,18 @@
#include "access/heapam.h"
#include "catalog/catalog.h"
#include "catalog/catname.h"
#include "catalog/namespace.h"
#include "catalog/pg_tablespace.h"
#include "commands/dbcommands.h"
#include "fmgr.h"
#include "storage/fd.h"
#include "utils/builtins.h"
static
int64
get_tablespace_size
(
Oid
dbid
,
Oid
spcid
,
bool
baddirOK
);
static
char
*
psnprintf
(
size_t
len
,
const
char
*
fmt
,...)
{
...
...
@@ -44,10 +49,12 @@ database_size(PG_FUNCTION_ARGS)
Name
dbname
=
PG_GETARG_NAME
(
0
);
Oid
dbid
;
char
*
dbpath
;
DIR
*
dirdesc
;
struct
dirent
*
direntry
;
int64
totalsize
;
#ifdef SYMLINK
Relation
dbrel
;
HeapScanDesc
scan
;
HeapTuple
tuple
;
#endif
dbid
=
get_database_oid
(
NameStr
(
*
dbname
));
if
(
!
OidIsValid
(
dbid
))
...
...
@@ -55,14 +62,55 @@ database_size(PG_FUNCTION_ARGS)
(
errcode
(
ERRCODE_UNDEFINED_DATABASE
),
errmsg
(
"database
\"
%s
\"
does not exist"
,
NameStr
(
*
dbname
))));
dbpath
=
GetDatabasePath
(
dbid
);
#ifdef SYMLINK
dbrel
=
heap_openr
(
TableSpaceRelationName
,
AccessShareLock
);
scan
=
heap_beginscan
(
dbrel
,
SnapshotNow
,
0
,
(
ScanKey
)
NULL
);
totalsize
=
0
;
while
((
tuple
=
heap_getnext
(
scan
,
ForwardScanDirection
)))
{
Oid
spcid
=
HeapTupleGetOid
(
tuple
);
if
(
spcid
!=
GLOBALTABLESPACE_OID
)
totalsize
+=
get_tablespace_size
(
dbid
,
spcid
,
true
);
}
heap_endscan
(
scan
);
heap_close
(
dbrel
,
AccessShareLock
);
#else
/* Same as always */
totalsize
=
get_tablespace_size
(
dbid
,
DEFAULTTABLESPACE_OID
,
false
);
#endif
/*
* We need to keep in mind that we may not be called from the database
* whose size we're reporting so, we need to look in every tablespace
* to see if our database has data in there
*/
PG_RETURN_INT64
(
totalsize
);
}
static
int64
get_tablespace_size
(
Oid
dbid
,
Oid
spcid
,
bool
baddirOK
)
{
char
*
dbpath
;
DIR
*
dirdesc
;
struct
dirent
*
direntry
;
int64
totalsize
;
dbpath
=
GetDatabasePath
(
dbid
,
spcid
);
dirdesc
=
AllocateDir
(
dbpath
);
if
(
!
dirdesc
)
{
if
(
baddirOK
)
return
0
;
else
ereport
(
ERROR
,
(
errcode_for_file_access
(),
errmsg
(
"could not open directory
\"
%s
\"
: %m"
,
dbpath
)));
}
totalsize
=
0
;
for
(;;)
{
...
...
@@ -87,18 +135,14 @@ database_size(PG_FUNCTION_ARGS)
ereport
(
ERROR
,
(
errcode_for_file_access
(),
errmsg
(
"could not stat
\"
%s
\"
: %m"
,
fullname
)));
totalsize
+=
statbuf
.
st_size
;
pfree
(
fullname
);
}
FreeDir
(
dirdesc
);
PG_RETURN_INT64
(
totalsize
);
return
(
totalsize
);
}
/*
* SQL function: relation_size(text) returns bigint
*/
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment