From 717ebe719da520b399dc89ce4c25a16d7ab2adf0 Mon Sep 17 00:00:00 2001 From: heikki Date: Fri, 3 Oct 2008 07:33:10 +0000 Subject: [PATCH] Add relation fork support to pg_relation_size() function. You can now pass name of a fork ('main' or 'fsm', at the moment) to pg_relation_size() to get the size of a specific fork. Defaults to 'main', if none given. While we're at it, modify pg_relation_size to take a regclass as argument, instead of separate variants taking oid and name. This change is transparent to typical use where the table name is passed as a string literal, like pg_relation_size('table'), but will break queries like pg_relation_size(namecol), where namecol is of type name. text-type input still works, and using a non-schema-qualified table name is not very reliable anyway, so this is unlikely to break anyone's queries in practice. --- doc/src/sgml/func.sgml | 36 +++++++++---------- src/backend/utils/adt/dbsize.c | 73 ++++++++++++++++++++------------------- src/include/catalog/catversion.h | 2 +- src/include/catalog/pg_proc.h | 10 +++--- src/include/storage/relfilenode.h | 5 ++- src/include/utils/builtins.h | 6 ++-- 6 files changed, 66 insertions(+), 66 deletions(-) diff --git a/doc/src/sgml/func.sgml b/doc/src/sgml/func.sgml index bdb0d9f838..49147179cf 100644 --- a/doc/src/sgml/func.sgml +++ b/doc/src/sgml/func.sgml @@ -12417,7 +12417,7 @@ postgres=# select * from pg_xlogfile_name_offset(pg_stop_backup()); - pg_column_size(any) + pg_column_size(any) int Number of bytes used to store a particular value (possibly compressed) @@ -12437,19 +12437,22 @@ postgres=# select * from pg_xlogfile_name_offset(pg_stop_backup()); - pg_relation_size(oid) + pg_relation_size(relation regclass, fork text) bigint - Disk space used by the table or index with the specified OID + + Disk space used by the specified fork, 'main' or + 'fsm', of a table or index with the specified OID + or name. The table name can be qualified with a schema name + - pg_relation_size(text) + pg_relation_size(relation regclass) bigint - Disk space used by the table or index with the specified name. - The table name can be qualified with a schema name + Shorthand for pg_relation_size(..., 'main') @@ -12475,21 +12478,11 @@ postgres=# select * from pg_xlogfile_name_offset(pg_stop_backup()); - pg_total_relation_size(oid) - - bigint - - Total disk space used by the table with the specified OID, - including indexes and toasted data - - - - - pg_total_relation_size(text) + pg_total_relation_size(regclass) bigint - Total disk space used by the table with the specified name, + Total disk space used by the table with the specified OID or name, including indexes and toasted data. The table name can be qualified with a schema name @@ -12511,7 +12504,12 @@ postgres=# select * from pg_xlogfile_name_offset(pg_stop_backup()); pg_relation_size accepts the OID or name of a table, index or - toast table, and returns the size in bytes. + toast table, and returns the size in bytes. Specifying + 'main' or leaving out the second argument returns the + size of the main data fork of the relation. Specifying + 'fsm' returns the size of the + Free Space Map (see ) associated with the + relation. diff --git a/src/backend/utils/adt/dbsize.c b/src/backend/utils/adt/dbsize.c index e9b5f0c279..613cbc6a17 100644 --- a/src/backend/utils/adt/dbsize.c +++ b/src/backend/utils/adt/dbsize.c @@ -248,15 +248,14 @@ pg_tablespace_size_name(PG_FUNCTION_ARGS) * calculate size of a relation */ static int64 -calculate_relation_size(RelFileNode *rfn) +calculate_relation_size(RelFileNode *rfn, ForkNumber forknum) { int64 totalsize = 0; char *relationpath; char pathname[MAXPGPATH]; unsigned int segcount = 0; - /* XXX: This ignores the other forks. */ - relationpath = relpath(*rfn, MAIN_FORKNUM); + relationpath = relpath(*rfn, forknum); for (segcount = 0;; segcount++) { @@ -284,34 +283,47 @@ calculate_relation_size(RelFileNode *rfn) return totalsize; } -Datum -pg_relation_size_oid(PG_FUNCTION_ARGS) -{ - Oid relOid = PG_GETARG_OID(0); - Relation rel; - int64 size; - rel = relation_open(relOid, AccessShareLock); +/* + * XXX: Consider making this global and moving elsewhere. But currently + * there's no other users for this. + * + * Remember to also update the errhint below if you add entries, and the + * documentation for pg_relation_size(). + */ +static char *forkNames[] = { + "main", /* MAIN_FORKNUM */ + "fsm" /* FSM_FORKNUM */ +}; - size = calculate_relation_size(&(rel->rd_node)); +static ForkNumber +forkname_to_number(char *forkName) +{ + ForkNumber forkNum; - relation_close(rel, AccessShareLock); + for (forkNum = 0; forkNum <= MAX_FORKNUM; forkNum++) + if (strcmp(forkNames[forkNum], forkName) == 0) + return forkNum; - PG_RETURN_INT64(size); + ereport(ERROR, + (errcode(ERRCODE_INVALID_PARAMETER_VALUE), + errmsg("invalid fork name"), + errhint("Valid fork names are 'main' and 'fsm'"))); + return InvalidForkNumber; /* keep compiler quiet */ } Datum -pg_relation_size_name(PG_FUNCTION_ARGS) +pg_relation_size(PG_FUNCTION_ARGS) { - text *relname = PG_GETARG_TEXT_P(0); - RangeVar *relrv; + Oid relOid = PG_GETARG_OID(0); + text *forkName = PG_GETARG_TEXT_P(1); Relation rel; int64 size; - relrv = makeRangeVarFromNameList(textToQualifiedNameList(relname)); - rel = relation_openrv(relrv, AccessShareLock); + rel = relation_open(relOid, AccessShareLock); - size = calculate_relation_size(&(rel->rd_node)); + size = calculate_relation_size(&(rel->rd_node), + forkname_to_number(text_to_cstring(forkName))); relation_close(rel, AccessShareLock); @@ -330,12 +342,15 @@ calculate_total_relation_size(Oid Relid) Oid toastOid; int64 size; ListCell *cell; + ForkNumber forkNum; heapRel = relation_open(Relid, AccessShareLock); toastOid = heapRel->rd_rel->reltoastrelid; /* Get the heap size */ - size = calculate_relation_size(&(heapRel->rd_node)); + size = 0; + for (forkNum = 0; forkNum <= MAX_FORKNUM; forkNum++) + size += calculate_relation_size(&(heapRel->rd_node), forkNum); /* Include any dependent indexes */ if (heapRel->rd_rel->relhasindex) @@ -349,7 +364,8 @@ calculate_total_relation_size(Oid Relid) iRel = relation_open(idxOid, AccessShareLock); - size += calculate_relation_size(&(iRel->rd_node)); + for (forkNum = 0; forkNum <= MAX_FORKNUM; forkNum++) + size += calculate_relation_size(&(iRel->rd_node), forkNum); relation_close(iRel, AccessShareLock); } @@ -367,26 +383,13 @@ calculate_total_relation_size(Oid Relid) } Datum -pg_total_relation_size_oid(PG_FUNCTION_ARGS) +pg_total_relation_size(PG_FUNCTION_ARGS) { Oid relid = PG_GETARG_OID(0); PG_RETURN_INT64(calculate_total_relation_size(relid)); } -Datum -pg_total_relation_size_name(PG_FUNCTION_ARGS) -{ - text *relname = PG_GETARG_TEXT_P(0); - RangeVar *relrv; - Oid relid; - - relrv = makeRangeVarFromNameList(textToQualifiedNameList(relname)); - relid = RangeVarGetRelid(relrv, false); - - PG_RETURN_INT64(calculate_total_relation_size(relid)); -} - /* * formatting with size units */ diff --git a/src/include/catalog/catversion.h b/src/include/catalog/catversion.h index 31fad08049..0ca115843a 100644 --- a/src/include/catalog/catversion.h +++ b/src/include/catalog/catversion.h @@ -53,6 +53,6 @@ */ /* yyyymmddN */ -#define CATALOG_VERSION_NO 200809301 +#define CATALOG_VERSION_NO 200810031 #endif diff --git a/src/include/catalog/pg_proc.h b/src/include/catalog/pg_proc.h index f5df27d20d..3aad9372c0 100644 --- a/src/include/catalog/pg_proc.h +++ b/src/include/catalog/pg_proc.h @@ -3591,13 +3591,11 @@ DATA(insert OID = 2324 ( pg_database_size PGNSP PGUID 12 1 0 0 f f t f v 1 20 " DESCR("total disk space usage for the specified database"); DATA(insert OID = 2168 ( pg_database_size PGNSP PGUID 12 1 0 0 f f t f v 1 20 "19" _null_ _null_ _null_ pg_database_size_name _null_ _null_ _null_ )); DESCR("total disk space usage for the specified database"); -DATA(insert OID = 2325 ( pg_relation_size PGNSP PGUID 12 1 0 0 f f t f v 1 20 "26" _null_ _null_ _null_ pg_relation_size_oid _null_ _null_ _null_ )); +DATA(insert OID = 2325 ( pg_relation_size PGNSP PGUID 14 1 0 0 f f t f v 1 20 "2205" _null_ _null_ _null_ "select pg_catalog.pg_relation_size($1, ''main'')" _null_ _null_ _null_ )); DESCR("disk space usage for the specified table or index"); -DATA(insert OID = 2289 ( pg_relation_size PGNSP PGUID 12 1 0 0 f f t f v 1 20 "25" _null_ _null_ _null_ pg_relation_size_name _null_ _null_ _null_ )); -DESCR("disk space usage for the specified table or index"); -DATA(insert OID = 2286 ( pg_total_relation_size PGNSP PGUID 12 1 0 0 f f t f v 1 20 "26" _null_ _null_ _null_ pg_total_relation_size_oid _null_ _null_ _null_ )); -DESCR("total disk space usage for the specified table and associated indexes and toast tables"); -DATA(insert OID = 2287 ( pg_total_relation_size PGNSP PGUID 12 1 0 0 f f t f v 1 20 "25" _null_ _null_ _null_ pg_total_relation_size_name _null_ _null_ _null_ )); +DATA(insert OID = 2332 ( pg_relation_size PGNSP PGUID 12 1 0 0 f f t f v 2 20 "2205 25" _null_ _null_ _null_ pg_relation_size _null_ _null_ _null_ )); +DESCR("disk space usage for the specified fork of a table or index"); +DATA(insert OID = 2286 ( pg_total_relation_size PGNSP PGUID 12 1 0 0 f f t f v 1 20 "2205" _null_ _null_ _null_ pg_total_relation_size _null_ _null_ _null_ )); DESCR("total disk space usage for the specified table and associated indexes and toast tables"); DATA(insert OID = 2288 ( pg_size_pretty PGNSP PGUID 12 1 0 0 f f t f v 1 25 "20" _null_ _null_ _null_ pg_size_pretty _null_ _null_ _null_ )); DESCR("convert a long int to a human readable text using size units"); diff --git a/src/include/storage/relfilenode.h b/src/include/storage/relfilenode.h index 3a71e76fa6..595a7352b1 100644 --- a/src/include/storage/relfilenode.h +++ b/src/include/storage/relfilenode.h @@ -25,7 +25,10 @@ typedef enum ForkNumber InvalidForkNumber = -1, MAIN_FORKNUM = 0, FSM_FORKNUM - /* NOTE: change MAX_FORKNUM below when you add new forks */ + /* + * NOTE: if you add a new fork, change MAX_FORKNUM below and update the + * name to number mapping in utils/adt/dbsize.c + */ } ForkNumber; #define MAX_FORKNUM FSM_FORKNUM diff --git a/src/include/utils/builtins.h b/src/include/utils/builtins.h index ccf0d6869f..d5633eeb69 100644 --- a/src/include/utils/builtins.h +++ b/src/include/utils/builtins.h @@ -389,10 +389,8 @@ extern Datum pg_tablespace_size_oid(PG_FUNCTION_ARGS); extern Datum pg_tablespace_size_name(PG_FUNCTION_ARGS); extern Datum pg_database_size_oid(PG_FUNCTION_ARGS); extern Datum pg_database_size_name(PG_FUNCTION_ARGS); -extern Datum pg_relation_size_oid(PG_FUNCTION_ARGS); -extern Datum pg_relation_size_name(PG_FUNCTION_ARGS); -extern Datum pg_total_relation_size_oid(PG_FUNCTION_ARGS); -extern Datum pg_total_relation_size_name(PG_FUNCTION_ARGS); +extern Datum pg_relation_size(PG_FUNCTION_ARGS); +extern Datum pg_total_relation_size(PG_FUNCTION_ARGS); extern Datum pg_size_pretty(PG_FUNCTION_ARGS); /* genfile.c */ -- 2.11.4.GIT