mirrors/postgres
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix contrib/dbsize for schema-qualified table names.

Browse Source
This commit is contained in:
Tom Lane 2002-04-02 01:17:28 +00:00
parent 1dc43ea75f
commit 688781d450
2 changed files with 26 additions and 31 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

51
contrib/dbsize/dbsize.c
View File

@ -1,19 +1,20 @@
#include "postgres.h" #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 <sys/types.h>
#include <dirent.h> #include <dirent.h>
#include <sys/stat.h> #include <sys/stat.h>
#include <unistd.h> #include <unistd.h>
#include <errno.h> #include <errno.h>
#include "access/heapam.h"
#include "catalog/catalog.h"
#include "catalog/catname.h"
#include "catalog/namespace.h"
#include "catalog/pg_database.h"
#include "fmgr.h"
#include "utils/builtins.h"
#include "utils/fmgroids.h"
static char * static char *
psnprintf(size_t len, const char *fmt,...) psnprintf(size_t len, const char *fmt,...)
@ -38,6 +39,8 @@ psnprintf(size_t len, const char *fmt,...)
PG_FUNCTION_INFO_V1(database_size); PG_FUNCTION_INFO_V1(database_size);
Datum database_size(PG_FUNCTION_ARGS);
Datum Datum
database_size(PG_FUNCTION_ARGS) database_size(PG_FUNCTION_ARGS)
{ {
@ -107,39 +110,29 @@ database_size(PG_FUNCTION_ARGS)
/* /*
* SQL function: relation_size(name) returns bigint * SQL function: relation_size(text) returns bigint
*/ */
PG_FUNCTION_INFO_V1(relation_size); PG_FUNCTION_INFO_V1(relation_size);
Datum relation_size(PG_FUNCTION_ARGS);
Datum Datum
relation_size(PG_FUNCTION_ARGS) relation_size(PG_FUNCTION_ARGS)
{ {
Name relname = PG_GETARG_NAME(0); text *relname = PG_GETARG_TEXT_P(0);
HeapTuple tuple; RangeVar *relrv;
Relation relation; Relation relation;
ScanKeyData scanKey;
HeapScanDesc scan;
Oid relnode; Oid relnode;
int64 totalsize; int64 totalsize;
unsigned int segcount; unsigned int segcount;
relation = heap_openr(RelationRelationName, AccessShareLock); relrv = makeRangeVarFromNameList(textToQualifiedNameList(relname,
ScanKeyEntryInitialize(&scanKey, 0, Anum_pg_class_relname, "relation_size"));
F_NAMEEQ, NameGetDatum(relname)); relation = relation_openrv(relrv, AccessShareLock);
scan = heap_beginscan(relation, 0, SnapshotNow, 1, &scanKey);
tuple = heap_getnext(scan, 0);
if (!HeapTupleIsValid(tuple)) relnode = relation->rd_rel->relfilenode;
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; totalsize = 0;
segcount = 0; segcount = 0;
@ -158,12 +151,14 @@ relation_size(PG_FUNCTION_ARGS)
if (errno == ENOENT) if (errno == ENOENT)
break; break;
else else
elog(ERROR, "could not stat %s: %s", fullname, strerror(errno)); elog(ERROR, "could not stat %s: %m", fullname);
} }
totalsize += statbuf.st_size; totalsize += statbuf.st_size;
pfree(fullname); pfree(fullname);
segcount++; segcount++;
} }
relation_close(relation, AccessShareLock);
PG_RETURN_INT64(totalsize); PG_RETURN_INT64(totalsize);
} }

6
contrib/dbsize/dbsize.sql.in
View File

@ -1,7 +1,7 @@
CREATE FUNCTION database_size (name) RETURNS bigint CREATE FUNCTION database_size (name) RETURNS bigint
AS '@MODULE_FILENAME@', 'database_size' AS 'MODULE_PATHNAME', 'database_size'
LANGUAGE C WITH (isstrict); LANGUAGE C WITH (isstrict);
CREATE FUNCTION relation_size (name) RETURNS bigint CREATE FUNCTION relation_size (text) RETURNS bigint
AS '@MODULE_FILENAME@', 'relation_size' AS 'MODULE_PATHNAME', 'relation_size'
LANGUAGE C WITH (isstrict); LANGUAGE C WITH (isstrict);