summaryrefslogtreecommitdiff
path: root/contrib/dbsize/dbsize.c
diff options
context:
space:
mode:
Diffstat (limited to 'contrib/dbsize/dbsize.c')
-rw-r--r--contrib/dbsize/dbsize.c164
1 files changed, 0 insertions, 164 deletions
diff --git a/contrib/dbsize/dbsize.c b/contrib/dbsize/dbsize.c
deleted file mode 100644
index b072a86912f..00000000000
--- a/contrib/dbsize/dbsize.c
+++ /dev/null
@@ -1,164 +0,0 @@
-#include "postgres.h"
-
-#include <sys/types.h>
-#include <dirent.h>
-#include <sys/stat.h>
-#include <unistd.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 *
-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);
-
-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, SnapshotNow, 1, &scanKey);
- tuple = heap_getnext(scan, ForwardScanDirection);
-
- 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(text) returns bigint
- */
-
-PG_FUNCTION_INFO_V1(relation_size);
-
-Datum relation_size(PG_FUNCTION_ARGS);
-
-Datum
-relation_size(PG_FUNCTION_ARGS)
-{
- text *relname = PG_GETARG_TEXT_P(0);
-
- RangeVar *relrv;
- Relation relation;
- Oid relnode;
- int64 totalsize;
- unsigned int segcount;
-
- relrv = makeRangeVarFromNameList(textToQualifiedNameList(relname,
- "relation_size"));
- relation = relation_openrv(relrv, AccessShareLock);
-
- relnode = relation->rd_rel->relfilenode;
-
- 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: %m", fullname);
- }
- totalsize += statbuf.st_size;
- pfree(fullname);
- segcount++;
- }
-
- relation_close(relation, AccessShareLock);
-
- PG_RETURN_INT64(totalsize);
-}