mirror of
https://git.postgresql.org/git/postgresql.git
synced 2024-12-09 08:10:09 +08:00
98 lines
1.7 KiB
C
98 lines
1.7 KiB
C
/* -------------------------------------------------------------------------
|
|
* pg_dumplo
|
|
*
|
|
* $PostgreSQL: pgsql/contrib/pg_dumplo/utils.c,v 1.9 2004/08/29 05:06:36 momjian Exp $
|
|
*
|
|
* Karel Zak 1999-2000
|
|
* -------------------------------------------------------------------------
|
|
*/
|
|
|
|
#include "postgres_fe.h"
|
|
|
|
#include <fcntl.h>
|
|
#include <errno.h>
|
|
#include <time.h>
|
|
#include <unistd.h>
|
|
#include <sys/stat.h>
|
|
|
|
#include "libpq-fe.h"
|
|
#include "libpq/libpq-fs.h"
|
|
|
|
#include "pg_dumplo.h"
|
|
|
|
extern int errno;
|
|
|
|
static void Dummy_NoticeProcessor(void *arg, const char *message);
|
|
static void Default_NoticeProcessor(void *arg, const char *message);
|
|
|
|
|
|
void
|
|
index_file(LODumpMaster * pgLO)
|
|
{
|
|
char path[BUFSIZ];
|
|
int sz;
|
|
|
|
if (pgLO->action == ACTION_SHOW)
|
|
return;
|
|
|
|
snprintf(path, BUFSIZ, "%s/%s", pgLO->space, pgLO->db);
|
|
|
|
if (pgLO->action == ACTION_EXPORT_ATTR ||
|
|
pgLO->action == ACTION_EXPORT_ALL)
|
|
{
|
|
|
|
if (mkdir(path, DIR_UMASK) == -1)
|
|
{
|
|
if (errno != EEXIST)
|
|
{
|
|
perror(path);
|
|
exit(RE_ERROR);
|
|
}
|
|
}
|
|
|
|
sz = strlen(path);
|
|
strncat(path, "/lo_dump.index", BUFSIZ - sz);
|
|
|
|
if ((pgLO->index = fopen(path, "w")) == NULL)
|
|
{
|
|
perror(path);
|
|
exit(RE_ERROR);
|
|
}
|
|
|
|
}
|
|
else if (pgLO->action != ACTION_NONE)
|
|
{
|
|
sz = strlen(path);
|
|
strncat(path, "/lo_dump.index", BUFSIZ - sz);
|
|
|
|
if ((pgLO->index = fopen(path, "r")) == NULL)
|
|
{
|
|
perror(path);
|
|
exit(RE_ERROR);
|
|
}
|
|
}
|
|
}
|
|
|
|
static
|
|
void
|
|
Dummy_NoticeProcessor(void *arg, const char *message)
|
|
{
|
|
;
|
|
}
|
|
|
|
static
|
|
void
|
|
Default_NoticeProcessor(void *arg, const char *message)
|
|
{
|
|
fprintf(stderr, "%s", message);
|
|
}
|
|
|
|
void
|
|
notice(LODumpMaster * pgLO, int set)
|
|
{
|
|
if (set)
|
|
PQsetNoticeProcessor(pgLO->conn, Default_NoticeProcessor, NULL);
|
|
else
|
|
PQsetNoticeProcessor(pgLO->conn, Dummy_NoticeProcessor, NULL);
|
|
}
|