mirror of https://github.com/postgres/postgres
http://webmail.postgresql.org/~petere/dbsize.html The tarball can be rolled into contrib -- now that I think of it I don't know why I never did that. Never imagined this would have anything to do with that TODO item, though. I figured oid2name accomplished that. Peter Eisentraut peter_e@gmx.netecpg_big_bison
parent
40bda4cbc4
commit
ad0787b2bd
@ -0,0 +1,9 @@ |
|||||||
|
subdir = contrib/dbsize
|
||||||
|
top_builddir = ../..
|
||||||
|
include $(top_builddir)/src/Makefile.global |
||||||
|
|
||||||
|
MODULES = dbsize
|
||||||
|
DATA_built = dbsize.sql
|
||||||
|
DOCS = README.dbsize
|
||||||
|
|
||||||
|
include $(top_srcdir)/contrib/contrib-global.mk |
||||||
@ -0,0 +1,11 @@ |
|||||||
|
This module contains two functions that report the size of a given |
||||||
|
database or relation. E.g., |
||||||
|
|
||||||
|
SELECT database_size('template1'); |
||||||
|
SELECT relation_size('pg_class'); |
||||||
|
|
||||||
|
These functions report the actual file system space. Thus, users can |
||||||
|
avoid digging through the details of the database directories. |
||||||
|
|
||||||
|
Copy this directory to contrib/dbsize in your PostgreSQL source tree. |
||||||
|
Then just run make; make install. |
||||||
@ -0,0 +1,169 @@ |
|||||||
|
#include "postgres.h" |
||||||
|
#include "fmgr.h" |
||||||
|
|
||||||
|
#include "access/heapam.h" |
||||||
|
#include "catalog/catalog.h" |
||||||
|
#include "catalog/catname.h" |
||||||
|
#include "catalog/pg_database.h" |
||||||
|
#include "utils/fmgroids.h" |
||||||
|
|
||||||
|
#include <stdlib.h> |
||||||
|
#include <sys/types.h> |
||||||
|
#include <dirent.h> |
||||||
|
#include <sys/stat.h> |
||||||
|
#include <unistd.h> |
||||||
|
#include <errno.h> |
||||||
|
|
||||||
|
|
||||||
|
static char * |
||||||
|
psnprintf(size_t len, const char *fmt,...) |
||||||
|
{ |
||||||
|
va_list ap; |
||||||
|
char *buf; |
||||||
|
|
||||||
|
buf = palloc(len); |
||||||
|
|
||||||
|
va_start(ap, fmt); |
||||||
|
vsnprintf(buf, len, fmt, ap); |
||||||
|
va_end(ap); |
||||||
|
|
||||||
|
return buf; |
||||||
|
} |
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
/*
|
||||||
|
* SQL function: database_size(name) returns bigint |
||||||
|
*/ |
||||||
|
|
||||||
|
PG_FUNCTION_INFO_V1(database_size); |
||||||
|
|
||||||
|
Datum |
||||||
|
database_size(PG_FUNCTION_ARGS) |
||||||
|
{ |
||||||
|
Name dbname = PG_GETARG_NAME(0); |
||||||
|
|
||||||
|
HeapTuple tuple; |
||||||
|
Relation relation; |
||||||
|
ScanKeyData scanKey; |
||||||
|
HeapScanDesc scan; |
||||||
|
Oid dbid; |
||||||
|
char *dbpath; |
||||||
|
DIR *dirdesc; |
||||||
|
struct dirent *direntry; |
||||||
|
int64 totalsize; |
||||||
|
|
||||||
|
relation = heap_openr(DatabaseRelationName, AccessShareLock); |
||||||
|
ScanKeyEntryInitialize(&scanKey, 0, Anum_pg_database_datname, |
||||||
|
F_NAMEEQ, NameGetDatum(dbname)); |
||||||
|
scan = heap_beginscan(relation, 0, SnapshotNow, 1, &scanKey); |
||||||
|
tuple = heap_getnext(scan, 0); |
||||||
|
|
||||||
|
if (!HeapTupleIsValid(tuple)) |
||||||
|
elog(ERROR, "database %s does not exist", NameStr(*dbname)); |
||||||
|
|
||||||
|
dbid = tuple->t_data->t_oid; |
||||||
|
if (dbid == InvalidOid) |
||||||
|
elog(ERROR, "invalid database id"); |
||||||
|
|
||||||
|
heap_endscan(scan); |
||||||
|
heap_close(relation, NoLock); |
||||||
|
|
||||||
|
dbpath = GetDatabasePath(dbid); |
||||||
|
|
||||||
|
dirdesc = opendir(dbpath); |
||||||
|
if (!dirdesc) |
||||||
|
elog(ERROR, "could not open directory %s: %s", dbpath, strerror(errno)); |
||||||
|
|
||||||
|
totalsize = 0; |
||||||
|
for (;;) |
||||||
|
{ |
||||||
|
char *fullname; |
||||||
|
struct stat statbuf; |
||||||
|
|
||||||
|
errno = 0; |
||||||
|
direntry = readdir(dirdesc); |
||||||
|
if (!direntry) |
||||||
|
{ |
||||||
|
if (errno) |
||||||
|
elog(ERROR, "error reading directory: %s", strerror(errno)); |
||||||
|
else |
||||||
|
break; |
||||||
|
} |
||||||
|
|
||||||
|
fullname = psnprintf(strlen(dbpath) + 1 + strlen(direntry->d_name) + 1, |
||||||
|
"%s/%s", dbpath, direntry->d_name); |
||||||
|
if (stat(fullname, &statbuf) == -1) |
||||||
|
elog(ERROR, "could not stat %s: %s", fullname, strerror(errno)); |
||||||
|
totalsize += statbuf.st_size; |
||||||
|
pfree(fullname); |
||||||
|
} |
||||||
|
|
||||||
|
closedir(dirdesc); |
||||||
|
|
||||||
|
PG_RETURN_INT64(totalsize); |
||||||
|
} |
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
/*
|
||||||
|
* SQL function: relation_size(name) returns bigint |
||||||
|
*/ |
||||||
|
|
||||||
|
PG_FUNCTION_INFO_V1(relation_size); |
||||||
|
|
||||||
|
Datum |
||||||
|
relation_size(PG_FUNCTION_ARGS) |
||||||
|
{ |
||||||
|
Name relname = PG_GETARG_NAME(0); |
||||||
|
|
||||||
|
HeapTuple tuple; |
||||||
|
Relation relation; |
||||||
|
ScanKeyData scanKey; |
||||||
|
HeapScanDesc scan; |
||||||
|
Oid relnode; |
||||||
|
int64 totalsize; |
||||||
|
unsigned int segcount; |
||||||
|
|
||||||
|
relation = heap_openr(RelationRelationName, AccessShareLock); |
||||||
|
ScanKeyEntryInitialize(&scanKey, 0, Anum_pg_class_relname, |
||||||
|
F_NAMEEQ, NameGetDatum(relname)); |
||||||
|
scan = heap_beginscan(relation, 0, SnapshotNow, 1, &scanKey); |
||||||
|
tuple = heap_getnext(scan, 0); |
||||||
|
|
||||||
|
if (!HeapTupleIsValid(tuple)) |
||||||
|
elog(ERROR, "relation %s does not exist", NameStr(*relname)); |
||||||
|
|
||||||
|
relnode = ((Form_pg_class) GETSTRUCT(tuple))->relfilenode; |
||||||
|
if (relnode == InvalidOid) |
||||||
|
elog(ERROR, "invalid relation node id"); |
||||||
|
|
||||||
|
heap_endscan(scan); |
||||||
|
heap_close(relation, NoLock); |
||||||
|
|
||||||
|
totalsize = 0; |
||||||
|
segcount = 0; |
||||||
|
for (;;) |
||||||
|
{ |
||||||
|
char *fullname; |
||||||
|
struct stat statbuf; |
||||||
|
|
||||||
|
if (segcount == 0) |
||||||
|
fullname = psnprintf(25, "%u", (unsigned) relnode); |
||||||
|
else |
||||||
|
fullname = psnprintf(50, "%u.%u", (unsigned) relnode, segcount); |
||||||
|
|
||||||
|
if (stat(fullname, &statbuf) == -1) |
||||||
|
{ |
||||||
|
if (errno == ENOENT) |
||||||
|
break; |
||||||
|
else |
||||||
|
elog(ERROR, "could not stat %s: %s", fullname, strerror(errno)); |
||||||
|
} |
||||||
|
totalsize += statbuf.st_size; |
||||||
|
pfree(fullname); |
||||||
|
segcount++; |
||||||
|
} |
||||||
|
|
||||||
|
PG_RETURN_INT64(totalsize); |
||||||
|
} |
||||||
@ -0,0 +1,7 @@ |
|||||||
|
CREATE FUNCTION database_size (name) RETURNS bigint |
||||||
|
AS '@MODULE_FILENAME@', 'database_size' |
||||||
|
LANGUAGE C WITH (isstrict); |
||||||
|
|
||||||
|
CREATE FUNCTION relation_size (name) RETURNS bigint |
||||||
|
AS '@MODULE_FILENAME@', 'relation_size' |
||||||
|
LANGUAGE C WITH (isstrict); |
||||||
Loading…
Reference in new issue