b0f5086e41
bitmap, if present). Per Tom Lane's suggestion the information whether a tuple has an oid or not is carried in the tuple descriptor. For debugging reasons tdhasoid is of type char, not bool. There are predefined values for WITHOID, WITHOUTOID and UNDEFOID. This patch has been generated against a cvs snapshot from last week and I don't expect it to apply cleanly to current sources. While I post it here for public review, I'm working on a new version against a current snapshot. (There's been heavy activity recently; hope to catch up some day ...) This is a long patch; if it is too hard to swallow, I can provide it in smaller pieces: Part 1: Accessor macros Part 2: tdhasoid in TupDesc Part 3: Regression test Part 4: Parameter withoid to heap_addheader Part 5: Eliminate t_oid from HeapTupleHeader Part 2 is the most hairy part because of changes in the executor and even in the parser; the other parts are straightforward. Up to part 4 the patched postmaster stays binary compatible to databases created with an unpatched version. Part 5 is small (100 lines) and finally breaks compatibility. Manfred Koizar
165 lines
3.2 KiB
C
165 lines
3.2 KiB
C
#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 = HeapTupleGetOid(tuple);
|
|
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);
|
|
}
|