mirror of
https://git.postgresql.org/git/postgresql.git
synced 2024-08-28 00:57:18 +02:00
* Portions Copyright (c) 1996-2000, PostgreSQL, Inc to all files copyright Regents of Berkeley. Man, that's a lot of files.
213 lines
4.4 KiB
C
213 lines
4.4 KiB
C
/*-------------------------------------------------------------------------
|
|
*
|
|
* vacuumlo.c
|
|
* This removes orphaned large objects from a database.
|
|
*
|
|
* Portions Copyright (c) 1996-2000, PostgreSQL, Inc
|
|
* Portions Copyright (c) 1994, Regents of the University of California
|
|
*
|
|
*
|
|
* IDENTIFICATION
|
|
* $Header: /cvsroot/pgsql/contrib/vacuumlo/vacuumlo.c,v 1.3 2000/01/26 05:55:52 momjian Exp $
|
|
*
|
|
*-------------------------------------------------------------------------
|
|
*/
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <strings.h>
|
|
|
|
#include <sys/types.h>
|
|
#include <sys/stat.h>
|
|
#include <fcntl.h>
|
|
#include <unistd.h>
|
|
|
|
#include "libpq-fe.h"
|
|
#include "libpq/libpq-fs.h"
|
|
|
|
#define BUFSIZE 1024
|
|
|
|
int vacuumlo(char *, int);
|
|
|
|
|
|
/*
|
|
* This vacuums a database. It returns 1 on success, -1 on failure.
|
|
*/
|
|
int
|
|
vacuumlo(char *database, int verbose)
|
|
{
|
|
PGconn *conn;
|
|
PGresult *res,
|
|
*res2;
|
|
char buf[BUFSIZE];
|
|
int matched = 0; /* Number matched per scan */
|
|
int i;
|
|
|
|
conn = PQsetdb(NULL, NULL, NULL, NULL, database);
|
|
|
|
/* check to see that the backend connection was successfully made */
|
|
if (PQstatus(conn) == CONNECTION_BAD)
|
|
{
|
|
fprintf(stderr, "Connection to database '%s' failed.\n", database);
|
|
fprintf(stderr, "%s", PQerrorMessage(conn));
|
|
return -1;
|
|
}
|
|
|
|
if (verbose)
|
|
fprintf(stdout, "Connected to %s\n", database);
|
|
|
|
/*
|
|
* First we create and populate the lo temp table
|
|
*/
|
|
buf[0] = '\0';
|
|
strcat(buf, "SELECT oid AS lo ");
|
|
strcat(buf, "INTO TEMP TABLE vacuum_l ");
|
|
strcat(buf, "FROM pg_class ");
|
|
strcat(buf, "WHERE relkind='l'");
|
|
if (!(res = PQexec(conn, buf)))
|
|
{
|
|
fprintf(stderr, "Failed to create temp table.\n");
|
|
PQfinish(conn);
|
|
return -1;
|
|
}
|
|
PQclear(res);
|
|
|
|
/*
|
|
* Now find any candidate tables who have columns of type oid (the
|
|
* column oid is ignored, as it has attnum < 1)
|
|
*/
|
|
buf[0] = '\0';
|
|
strcat(buf, "SELECT c.relname, a.attname ");
|
|
strcat(buf, "FROM pg_class c, pg_attribute a, pg_type t ");
|
|
strcat(buf, "WHERE a.attnum > 0 ");
|
|
strcat(buf, " AND a.attrelid = c.oid ");
|
|
strcat(buf, " AND a.atttypid = t.oid ");
|
|
strcat(buf, " AND t.typname = 'oid' ");
|
|
strcat(buf, " AND c.relname NOT LIKE 'pg_%'");
|
|
if (!(res = PQexec(conn, buf)))
|
|
{
|
|
fprintf(stderr, "Failed to create temp table.\n");
|
|
PQfinish(conn);
|
|
return -1;
|
|
}
|
|
for (i = 0; i < PQntuples(res); i++)
|
|
{
|
|
char *table,
|
|
*field;
|
|
|
|
table = PQgetvalue(res, i, 0);
|
|
field = PQgetvalue(res, i, 1);
|
|
|
|
if (verbose)
|
|
{
|
|
fprintf(stdout, "Checking %s in %s: ", field, table);
|
|
fflush(stdout);
|
|
}
|
|
|
|
res2 = PQexec(conn, "begin");
|
|
PQclear(res2);
|
|
|
|
buf[0] = '\0';
|
|
strcat(buf, "DELETE FROM vacuum_l ");
|
|
strcat(buf, "WHERE lo IN (");
|
|
strcat(buf, "SELECT ");
|
|
strcat(buf, field);
|
|
strcat(buf, " FROM ");
|
|
strcat(buf, table);
|
|
strcat(buf, ");");
|
|
if (!(res2 = PQexec(conn, buf)))
|
|
{
|
|
fprintf(stderr, "Failed to check %s in table %s\n", field, table);
|
|
PQclear(res);
|
|
PQfinish(conn);
|
|
return -1;
|
|
}
|
|
if (PQresultStatus(res2) != PGRES_COMMAND_OK)
|
|
{
|
|
fprintf(stderr,
|
|
"Failed to check %s in table %s\n%s\n",
|
|
field, table,
|
|
PQerrorMessage(conn)
|
|
);
|
|
PQclear(res2);
|
|
PQclear(res);
|
|
PQfinish(conn);
|
|
return -1;
|
|
}
|
|
PQclear(res2);
|
|
|
|
res2 = PQexec(conn, "end");
|
|
PQclear(res2);
|
|
|
|
}
|
|
PQclear(res);
|
|
|
|
/* Start the transaction */
|
|
res = PQexec(conn, "begin");
|
|
PQclear(res);
|
|
|
|
/*
|
|
* Finally, those entries remaining in vacuum_l are orphans.
|
|
*/
|
|
buf[0] = '\0';
|
|
strcat(buf, "SELECT lo ");
|
|
strcat(buf, "FROM vacuum_l");
|
|
if (!(res = PQexec(conn, buf)))
|
|
{
|
|
fprintf(stderr, "Failed to read temp table.\n");
|
|
PQfinish(conn);
|
|
return -1;
|
|
}
|
|
matched = PQntuples(res);
|
|
for (i = 0; i < matched; i++)
|
|
{
|
|
Oid lo = (Oid) atoi(PQgetvalue(res, i, 0));
|
|
|
|
if (verbose)
|
|
{
|
|
fprintf(stdout, "\rRemoving lo %6d \n", lo);
|
|
fflush(stdout);
|
|
}
|
|
|
|
if (lo_unlink(conn, lo) < 0)
|
|
fprintf(stderr, "Failed to remove lo %d\n", lo);
|
|
}
|
|
PQclear(res);
|
|
|
|
/*
|
|
* That's all folks!
|
|
*/
|
|
res = PQexec(conn, "end");
|
|
PQclear(res);
|
|
PQfinish(conn);
|
|
|
|
if (verbose)
|
|
fprintf(stdout, "\rRemoved %d large objects from %s.\n", matched, database);
|
|
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
main(int argc, char **argv)
|
|
{
|
|
int verbose = 0;
|
|
int arg;
|
|
int rc = 0;
|
|
|
|
if (argc < 2)
|
|
{
|
|
fprintf(stderr, "Usage: %s [-v] database_name [db2 ... dbn]\n",
|
|
argv[0]);
|
|
exit(1);
|
|
}
|
|
|
|
for (arg = 1; arg < argc; arg++)
|
|
{
|
|
if (strcmp("-v", argv[arg]) == 0)
|
|
verbose = !verbose;
|
|
else
|
|
rc += vacuumlo(argv[arg], verbose);
|
|
}
|
|
|
|
return rc;
|
|
}
|