/src/bin/pg_dump/pg_dump.c
C | 15699 lines | 11346 code | 1874 blank | 2479 comment | 1611 complexity | bb079b515b6be228674b0fcc00c8e61b MD5 | raw file
Possible License(s): AGPL-3.0
- /*-------------------------------------------------------------------------
- *
- * pg_dump.c
- * pg_dump is a utility for dumping out a postgres database
- * into a script file.
- *
- * Portions Copyright (c) 1996-2014, PostgreSQL Global Development Group
- * Portions Copyright (c) 1994, Regents of the University of California
- *
- * pg_dump will read the system catalogs in a database and dump out a
- * script that reproduces the schema in terms of SQL that is understood
- * by PostgreSQL
- *
- * Note that pg_dump runs in a transaction-snapshot mode transaction,
- * so it sees a consistent snapshot of the database including system
- * catalogs. However, it relies in part on various specialized backend
- * functions like pg_get_indexdef(), and those things tend to look at
- * the currently committed state. So it is possible to get 'cache
- * lookup failed' error if someone performs DDL changes while a dump is
- * happening. The window for this sort of thing is from the acquisition
- * of the transaction snapshot to getSchemaData() (when pg_dump acquires
- * AccessShareLock on every table it intends to dump). It isn't very large,
- * but it can happen.
- *
- * http://archives.postgresql.org/pgsql-bugs/2010-02/msg00187.php
- *
- * IDENTIFICATION
- * src/bin/pg_dump/pg_dump.c
- *
- *-------------------------------------------------------------------------
- */
- #include "postgres_fe.h"
- #include <unistd.h>
- #include <ctype.h>
- #ifdef ENABLE_NLS
- #include <locale.h>
- #endif
- #ifdef HAVE_TERMIOS_H
- #include <termios.h>
- #endif
- #include "getopt_long.h"
- #include "access/attnum.h"
- #include "access/sysattr.h"
- #include "access/transam.h"
- #include "catalog/pg_cast.h"
- #include "catalog/pg_class.h"
- #include "catalog/pg_default_acl.h"
- #include "catalog/pg_event_trigger.h"
- #include "catalog/pg_largeobject.h"
- #include "catalog/pg_largeobject_metadata.h"
- #include "catalog/pg_proc.h"
- #include "catalog/pg_trigger.h"
- #include "catalog/pg_type.h"
- #include "libpq/libpq-fs.h"
- #include "pg_backup_archiver.h"
- #include "pg_backup_db.h"
- #include "pg_backup_utils.h"
- #include "dumputils.h"
- #include "parallel.h"
- typedef struct
- {
- const char *descr; /* comment for an object */
- Oid classoid; /* object class (catalog OID) */
- Oid objoid; /* object OID */
- int objsubid; /* subobject (table column #) */
- } CommentItem;
- typedef struct
- {
- const char *provider; /* label provider of this security label */
- const char *label; /* security label for an object */
- Oid classoid; /* object class (catalog OID) */
- Oid objoid; /* object OID */
- int objsubid; /* subobject (table column #) */
- } SecLabelItem;
- /* global decls */
- bool g_verbose; /* User wants verbose narration of our
- * activities. */
- /* various user-settable parameters */
- static bool schemaOnly;
- static bool dataOnly;
- static int dumpSections; /* bitmask of chosen sections */
- static bool aclsSkip;
- static const char *lockWaitTimeout;
- /* subquery used to convert user ID (eg, datdba) to user name */
- static const char *username_subquery;
- /* obsolete as of 7.3: */
- static Oid g_last_builtin_oid; /* value of the last builtin oid */
- /*
- * Object inclusion/exclusion lists
- *
- * The string lists record the patterns given by command-line switches,
- * which we then convert to lists of OIDs of matching objects.
- */
- static SimpleStringList schema_include_patterns = {NULL, NULL};
- static SimpleOidList schema_include_oids = {NULL, NULL};
- static SimpleStringList schema_exclude_patterns = {NULL, NULL};
- static SimpleOidList schema_exclude_oids = {NULL, NULL};
- static SimpleStringList table_include_patterns = {NULL, NULL};
- static SimpleOidList table_include_oids = {NULL, NULL};
- static SimpleStringList table_exclude_patterns = {NULL, NULL};
- static SimpleOidList table_exclude_oids = {NULL, NULL};
- static SimpleStringList tabledata_exclude_patterns = {NULL, NULL};
- static SimpleOidList tabledata_exclude_oids = {NULL, NULL};
- /* default, if no "inclusion" switches appear, is to dump everything */
- static bool include_everything = true;
- char g_opaque_type[10]; /* name for the opaque type */
- /* placeholders for the delimiters for comments */
- char g_comment_start[10];
- char g_comment_end[10];
- static const CatalogId nilCatalogId = {0, 0};
- /* flags for various command-line long options */
- static int binary_upgrade = 0;
- static int disable_dollar_quoting = 0;
- static int dump_inserts = 0;
- static int column_inserts = 0;
- static int if_exists = 0;
- static int no_security_labels = 0;
- static int no_synchronized_snapshots = 0;
- static int no_unlogged_table_data = 0;
- static int serializable_deferrable = 0;
- static void help(const char *progname);
- static void setup_connection(Archive *AH, const char *dumpencoding,
- char *use_role);
- static ArchiveFormat parseArchiveFormat(const char *format, ArchiveMode *mode);
- static void expand_schema_name_patterns(Archive *fout,
- SimpleStringList *patterns,
- SimpleOidList *oids);
- static void expand_table_name_patterns(Archive *fout,
- SimpleStringList *patterns,
- SimpleOidList *oids);
- static NamespaceInfo *findNamespace(Archive *fout, Oid nsoid, Oid objoid);
- static void dumpTableData(Archive *fout, TableDataInfo *tdinfo);
- static void refreshMatViewData(Archive *fout, TableDataInfo *tdinfo);
- static void guessConstraintInheritance(TableInfo *tblinfo, int numTables);
- static void dumpComment(Archive *fout, const char *target,
- const char *namespace, const char *owner,
- CatalogId catalogId, int subid, DumpId dumpId);
- static int findComments(Archive *fout, Oid classoid, Oid objoid,
- CommentItem **items);
- static int collectComments(Archive *fout, CommentItem **items);
- static void dumpSecLabel(Archive *fout, const char *target,
- const char *namespace, const char *owner,
- CatalogId catalogId, int subid, DumpId dumpId);
- static int findSecLabels(Archive *fout, Oid classoid, Oid objoid,
- SecLabelItem **items);
- static int collectSecLabels(Archive *fout, SecLabelItem **items);
- static void dumpDumpableObject(Archive *fout, DumpableObject *dobj);
- static void dumpNamespace(Archive *fout, NamespaceInfo *nspinfo);
- static void dumpExtension(Archive *fout, ExtensionInfo *extinfo);
- static void dumpType(Archive *fout, TypeInfo *tyinfo);
- static void dumpBaseType(Archive *fout, TypeInfo *tyinfo);
- static void dumpEnumType(Archive *fout, TypeInfo *tyinfo);
- static void dumpRangeType(Archive *fout, TypeInfo *tyinfo);
- static void dumpDomain(Archive *fout, TypeInfo *tyinfo);
- static void dumpCompositeType(Archive *fout, TypeInfo *tyinfo);
- static void dumpCompositeTypeColComments(Archive *fout, TypeInfo *tyinfo);
- static void dumpShellType(Archive *fout, ShellTypeInfo *stinfo);
- static void dumpProcLang(Archive *fout, ProcLangInfo *plang);
- static void dumpFunc(Archive *fout, FuncInfo *finfo);
- static void dumpCast(Archive *fout, CastInfo *cast);
- static void dumpOpr(Archive *fout, OprInfo *oprinfo);
- static void dumpOpclass(Archive *fout, OpclassInfo *opcinfo);
- static void dumpOpfamily(Archive *fout, OpfamilyInfo *opfinfo);
- static void dumpCollation(Archive *fout, CollInfo *convinfo);
- static void dumpConversion(Archive *fout, ConvInfo *convinfo);
- static void dumpRule(Archive *fout, RuleInfo *rinfo);
- static void dumpAgg(Archive *fout, AggInfo *agginfo);
- static void dumpTrigger(Archive *fout, TriggerInfo *tginfo);
- static void dumpEventTrigger(Archive *fout, EventTriggerInfo *evtinfo);
- static void dumpTable(Archive *fout, TableInfo *tbinfo);
- static void dumpTableSchema(Archive *fout, TableInfo *tbinfo);
- static void dumpAttrDef(Archive *fout, AttrDefInfo *adinfo);
- static void dumpSequence(Archive *fout, TableInfo *tbinfo);
- static void dumpSequenceData(Archive *fout, TableDataInfo *tdinfo);
- static void dumpIndex(Archive *fout, IndxInfo *indxinfo);
- static void dumpConstraint(Archive *fout, ConstraintInfo *coninfo);
- static void dumpTableConstraintComment(Archive *fout, ConstraintInfo *coninfo);
- static void dumpTSParser(Archive *fout, TSParserInfo *prsinfo);
- static void dumpTSDictionary(Archive *fout, TSDictInfo *dictinfo);
- static void dumpTSTemplate(Archive *fout, TSTemplateInfo *tmplinfo);
- static void dumpTSConfig(Archive *fout, TSConfigInfo *cfginfo);
- static void dumpForeignDataWrapper(Archive *fout, FdwInfo *fdwinfo);
- static void dumpForeignServer(Archive *fout, ForeignServerInfo *srvinfo);
- static void dumpUserMappings(Archive *fout,
- const char *servername, const char *namespace,
- const char *owner, CatalogId catalogId, DumpId dumpId);
- static void dumpDefaultACL(Archive *fout, DefaultACLInfo *daclinfo);
- static void dumpACL(Archive *fout, CatalogId objCatId, DumpId objDumpId,
- const char *type, const char *name, const char *subname,
- const char *tag, const char *nspname, const char *owner,
- const char *acls);
- static void getDependencies(Archive *fout);
- static void BuildArchiveDependencies(Archive *fout);
- static void findDumpableDependencies(ArchiveHandle *AH, DumpableObject *dobj,
- DumpId **dependencies, int *nDeps, int *allocDeps);
- static DumpableObject *createBoundaryObjects(void);
- static void addBoundaryDependencies(DumpableObject **dobjs, int numObjs,
- DumpableObject *boundaryObjs);
- static void getDomainConstraints(Archive *fout, TypeInfo *tyinfo);
- static void getTableData(TableInfo *tblinfo, int numTables, bool oids);
- static void makeTableDataInfo(TableInfo *tbinfo, bool oids);
- static void buildMatViewRefreshDependencies(Archive *fout);
- static void getTableDataFKConstraints(void);
- static char *format_function_arguments(FuncInfo *finfo, char *funcargs,
- bool is_agg);
- static char *format_function_arguments_old(Archive *fout,
- FuncInfo *finfo, int nallargs,
- char **allargtypes,
- char **argmodes,
- char **argnames);
- static char *format_function_signature(Archive *fout,
- FuncInfo *finfo, bool honor_quotes);
- static char *convertRegProcReference(Archive *fout,
- const char *proc);
- static char *convertOperatorReference(Archive *fout, const char *opr);
- static const char *convertTSFunction(Archive *fout, Oid funcOid);
- static Oid findLastBuiltinOid_V71(Archive *fout, const char *);
- static Oid findLastBuiltinOid_V70(Archive *fout);
- static void selectSourceSchema(Archive *fout, const char *schemaName);
- static char *getFormattedTypeName(Archive *fout, Oid oid, OidOptions opts);
- static char *myFormatType(const char *typname, int32 typmod);
- static void getBlobs(Archive *fout);
- static void dumpBlob(Archive *fout, BlobInfo *binfo);
- static int dumpBlobs(Archive *fout, void *arg);
- static void dumpDatabase(Archive *AH);
- static void dumpEncoding(Archive *AH);
- static void dumpStdStrings(Archive *AH);
- static void binary_upgrade_set_type_oids_by_type_oid(Archive *fout,
- PQExpBuffer upgrade_buffer, Oid pg_type_oid);
- static bool binary_upgrade_set_type_oids_by_rel_oid(Archive *fout,
- PQExpBuffer upgrade_buffer, Oid pg_rel_oid);
- static void binary_upgrade_set_pg_class_oids(Archive *fout,
- PQExpBuffer upgrade_buffer,
- Oid pg_class_oid, bool is_index);
- static void binary_upgrade_extension_member(PQExpBuffer upgrade_buffer,
- DumpableObject *dobj,
- const char *objlabel);
- static const char *getAttrName(int attrnum, TableInfo *tblInfo);
- static const char *fmtCopyColumnList(const TableInfo *ti, PQExpBuffer buffer);
- static char *get_synchronized_snapshot(Archive *fout);
- static PGresult *ExecuteSqlQueryForSingleRow(Archive *fout, char *query);
- static void setupDumpWorker(Archive *AHX, RestoreOptions *ropt);
- int
- main(int argc, char **argv)
- {
- int c;
- const char *filename = NULL;
- const char *format = "p";
- const char *dbname = NULL;
- const char *pghost = NULL;
- const char *pgport = NULL;
- const char *username = NULL;
- const char *dumpencoding = NULL;
- bool oids = false;
- TableInfo *tblinfo;
- int numTables;
- DumpableObject **dobjs;
- int numObjs;
- DumpableObject *boundaryObjs;
- int i;
- int numWorkers = 1;
- enum trivalue prompt_password = TRI_DEFAULT;
- int compressLevel = -1;
- int plainText = 0;
- int outputClean = 0;
- int outputCreateDB = 0;
- bool outputBlobs = false;
- int outputNoOwner = 0;
- char *outputSuperuser = NULL;
- char *use_role = NULL;
- int optindex;
- RestoreOptions *ropt;
- ArchiveFormat archiveFormat = archUnknown;
- ArchiveMode archiveMode;
- Archive *fout; /* the script file */
- static int disable_triggers = 0;
- static int outputNoTablespaces = 0;
- static int use_setsessauth = 0;
- static struct option long_options[] = {
- {"data-only", no_argument, NULL, 'a'},
- {"blobs", no_argument, NULL, 'b'},
- {"clean", no_argument, NULL, 'c'},
- {"create", no_argument, NULL, 'C'},
- {"dbname", required_argument, NULL, 'd'},
- {"file", required_argument, NULL, 'f'},
- {"format", required_argument, NULL, 'F'},
- {"host", required_argument, NULL, 'h'},
- {"ignore-version", no_argument, NULL, 'i'},
- {"jobs", 1, NULL, 'j'},
- {"no-reconnect", no_argument, NULL, 'R'},
- {"oids", no_argument, NULL, 'o'},
- {"no-owner", no_argument, NULL, 'O'},
- {"port", required_argument, NULL, 'p'},
- {"schema", required_argument, NULL, 'n'},
- {"exclude-schema", required_argument, NULL, 'N'},
- {"schema-only", no_argument, NULL, 's'},
- {"superuser", required_argument, NULL, 'S'},
- {"table", required_argument, NULL, 't'},
- {"exclude-table", required_argument, NULL, 'T'},
- {"no-password", no_argument, NULL, 'w'},
- {"password", no_argument, NULL, 'W'},
- {"username", required_argument, NULL, 'U'},
- {"verbose", no_argument, NULL, 'v'},
- {"no-privileges", no_argument, NULL, 'x'},
- {"no-acl", no_argument, NULL, 'x'},
- {"compress", required_argument, NULL, 'Z'},
- {"encoding", required_argument, NULL, 'E'},
- {"help", no_argument, NULL, '?'},
- {"version", no_argument, NULL, 'V'},
- /*
- * the following options don't have an equivalent short option letter
- */
- {"attribute-inserts", no_argument, &column_inserts, 1},
- {"binary-upgrade", no_argument, &binary_upgrade, 1},
- {"column-inserts", no_argument, &column_inserts, 1},
- {"disable-dollar-quoting", no_argument, &disable_dollar_quoting, 1},
- {"disable-triggers", no_argument, &disable_triggers, 1},
- {"exclude-table-data", required_argument, NULL, 4},
- {"if-exists", no_argument, &if_exists, 1},
- {"inserts", no_argument, &dump_inserts, 1},
- {"lock-wait-timeout", required_argument, NULL, 2},
- {"no-tablespaces", no_argument, &outputNoTablespaces, 1},
- {"quote-all-identifiers", no_argument, "e_all_identifiers, 1},
- {"role", required_argument, NULL, 3},
- {"section", required_argument, NULL, 5},
- {"serializable-deferrable", no_argument, &serializable_deferrable, 1},
- {"use-set-session-authorization", no_argument, &use_setsessauth, 1},
- {"no-security-labels", no_argument, &no_security_labels, 1},
- {"no-synchronized-snapshots", no_argument, &no_synchronized_snapshots, 1},
- {"no-unlogged-table-data", no_argument, &no_unlogged_table_data, 1},
- {NULL, 0, NULL, 0}
- };
- set_pglocale_pgservice(argv[0], PG_TEXTDOMAIN("pg_dump"));
- /*
- * Initialize what we need for parallel execution, especially for thread
- * support on Windows.
- */
- init_parallel_dump_utils();
- g_verbose = false;
- strcpy(g_comment_start, "-- ");
- g_comment_end[0] = '\0';
- strcpy(g_opaque_type, "opaque");
- dataOnly = schemaOnly = false;
- dumpSections = DUMP_UNSECTIONED;
- lockWaitTimeout = NULL;
- progname = get_progname(argv[0]);
- /* Set default options based on progname */
- if (strcmp(progname, "pg_backup") == 0)
- format = "c";
- if (argc > 1)
- {
- if (strcmp(argv[1], "--help") == 0 || strcmp(argv[1], "-?") == 0)
- {
- help(progname);
- exit_nicely(0);
- }
- if (strcmp(argv[1], "--version") == 0 || strcmp(argv[1], "-V") == 0)
- {
- puts("pg_dump (PostgreSQL) " PG_VERSION);
- exit_nicely(0);
- }
- }
- while ((c = getopt_long(argc, argv, "abcCd:E:f:F:h:ij:n:N:oOp:RsS:t:T:U:vwWxZ:",
- long_options, &optindex)) != -1)
- {
- switch (c)
- {
- case 'a': /* Dump data only */
- dataOnly = true;
- break;
- case 'b': /* Dump blobs */
- outputBlobs = true;
- break;
- case 'c': /* clean (i.e., drop) schema prior to create */
- outputClean = 1;
- break;
- case 'C': /* Create DB */
- outputCreateDB = 1;
- break;
- case 'd': /* database name */
- dbname = pg_strdup(optarg);
- break;
- case 'E': /* Dump encoding */
- dumpencoding = pg_strdup(optarg);
- break;
- case 'f':
- filename = pg_strdup(optarg);
- break;
- case 'F':
- format = pg_strdup(optarg);
- break;
- case 'h': /* server host */
- pghost = pg_strdup(optarg);
- break;
- case 'i':
- /* ignored, deprecated option */
- break;
- case 'j': /* number of dump jobs */
- numWorkers = atoi(optarg);
- break;
- case 'n': /* include schema(s) */
- simple_string_list_append(&schema_include_patterns, optarg);
- include_everything = false;
- break;
- case 'N': /* exclude schema(s) */
- simple_string_list_append(&schema_exclude_patterns, optarg);
- break;
- case 'o': /* Dump oids */
- oids = true;
- break;
- case 'O': /* Don't reconnect to match owner */
- outputNoOwner = 1;
- break;
- case 'p': /* server port */
- pgport = pg_strdup(optarg);
- break;
- case 'R':
- /* no-op, still accepted for backwards compatibility */
- break;
- case 's': /* dump schema only */
- schemaOnly = true;
- break;
- case 'S': /* Username for superuser in plain text output */
- outputSuperuser = pg_strdup(optarg);
- break;
- case 't': /* include table(s) */
- simple_string_list_append(&table_include_patterns, optarg);
- include_everything = false;
- break;
- case 'T': /* exclude table(s) */
- simple_string_list_append(&table_exclude_patterns, optarg);
- break;
- case 'U':
- username = pg_strdup(optarg);
- break;
- case 'v': /* verbose */
- g_verbose = true;
- break;
- case 'w':
- prompt_password = TRI_NO;
- break;
- case 'W':
- prompt_password = TRI_YES;
- break;
- case 'x': /* skip ACL dump */
- aclsSkip = true;
- break;
- case 'Z': /* Compression Level */
- compressLevel = atoi(optarg);
- break;
- case 0:
- /* This covers the long options. */
- break;
- case 2: /* lock-wait-timeout */
- lockWaitTimeout = pg_strdup(optarg);
- break;
- case 3: /* SET ROLE */
- use_role = pg_strdup(optarg);
- break;
- case 4: /* exclude table(s) data */
- simple_string_list_append(&tabledata_exclude_patterns, optarg);
- break;
- case 5: /* section */
- set_dump_section(optarg, &dumpSections);
- break;
- default:
- fprintf(stderr, _("Try \"%s --help\" for more information.\n"), progname);
- exit_nicely(1);
- }
- }
- /*
- * Non-option argument specifies database name as long as it wasn't
- * already specified with -d / --dbname
- */
- if (optind < argc && dbname == NULL)
- dbname = argv[optind++];
- /* Complain if any arguments remain */
- if (optind < argc)
- {
- fprintf(stderr, _("%s: too many command-line arguments (first is \"%s\")\n"),
- progname, argv[optind]);
- fprintf(stderr, _("Try \"%s --help\" for more information.\n"),
- progname);
- exit_nicely(1);
- }
- /* --column-inserts implies --inserts */
- if (column_inserts)
- dump_inserts = 1;
- if (dataOnly && schemaOnly)
- {
- write_msg(NULL, "options -s/--schema-only and -a/--data-only cannot be used together\n");
- exit_nicely(1);
- }
- if (dataOnly && outputClean)
- {
- write_msg(NULL, "options -c/--clean and -a/--data-only cannot be used together\n");
- exit_nicely(1);
- }
- if (dump_inserts && oids)
- {
- write_msg(NULL, "options --inserts/--column-inserts and -o/--oids cannot be used together\n");
- write_msg(NULL, "(The INSERT command cannot set OIDs.)\n");
- exit_nicely(1);
- }
- if (if_exists && !outputClean)
- exit_horribly(NULL, "option --if-exists requires option -c/--clean\n");
- /* Identify archive format to emit */
- archiveFormat = parseArchiveFormat(format, &archiveMode);
- /* archiveFormat specific setup */
- if (archiveFormat == archNull)
- plainText = 1;
- /* Custom and directory formats are compressed by default, others not */
- if (compressLevel == -1)
- {
- if (archiveFormat == archCustom || archiveFormat == archDirectory)
- compressLevel = Z_DEFAULT_COMPRESSION;
- else
- compressLevel = 0;
- }
- /*
- * On Windows we can only have at most MAXIMUM_WAIT_OBJECTS (= 64 usually)
- * parallel jobs because that's the maximum limit for the
- * WaitForMultipleObjects() call.
- */
- if (numWorkers <= 0
- #ifdef WIN32
- || numWorkers > MAXIMUM_WAIT_OBJECTS
- #endif
- )
- exit_horribly(NULL, "%s: invalid number of parallel jobs\n", progname);
- /* Parallel backup only in the directory archive format so far */
- if (archiveFormat != archDirectory && numWorkers > 1)
- exit_horribly(NULL, "parallel backup only supported by the directory format\n");
- /* Open the output file */
- fout = CreateArchive(filename, archiveFormat, compressLevel, archiveMode,
- setupDumpWorker);
- /* Register the cleanup hook */
- on_exit_close_archive(fout);
- if (fout == NULL)
- exit_horribly(NULL, "could not open output file \"%s\" for writing\n", filename);
- /* Let the archiver know how noisy to be */
- fout->verbose = g_verbose;
- /*
- * We allow the server to be back to 7.0, and up to any minor release of
- * our own major version. (See also version check in pg_dumpall.c.)
- */
- fout->minRemoteVersion = 70000;
- fout->maxRemoteVersion = (PG_VERSION_NUM / 100) * 100 + 99;
- fout->numWorkers = numWorkers;
- /*
- * Open the database using the Archiver, so it knows about it. Errors mean
- * death.
- */
- ConnectDatabase(fout, dbname, pghost, pgport, username, prompt_password);
- setup_connection(fout, dumpencoding, use_role);
- /*
- * Disable security label support if server version < v9.1.x (prevents
- * access to nonexistent pg_seclabel catalog)
- */
- if (fout->remoteVersion < 90100)
- no_security_labels = 1;
- /*
- * When running against 9.0 or later, check if we are in recovery mode,
- * which means we are on a hot standby.
- */
- if (fout->remoteVersion >= 90000)
- {
- PGresult *res = ExecuteSqlQueryForSingleRow(fout, "SELECT pg_catalog.pg_is_in_recovery()");
- if (strcmp(PQgetvalue(res, 0, 0), "t") == 0)
- {
- /*
- * On hot standby slaves, never try to dump unlogged table data,
- * since it will just throw an error.
- */
- no_unlogged_table_data = true;
- }
- PQclear(res);
- }
- /* Select the appropriate subquery to convert user IDs to names */
- if (fout->remoteVersion >= 80100)
- username_subquery = "SELECT rolname FROM pg_catalog.pg_roles WHERE oid =";
- else if (fout->remoteVersion >= 70300)
- username_subquery = "SELECT usename FROM pg_catalog.pg_user WHERE usesysid =";
- else
- username_subquery = "SELECT usename FROM pg_user WHERE usesysid =";
- /* check the version for the synchronized snapshots feature */
- if (numWorkers > 1 && fout->remoteVersion < 90200
- && !no_synchronized_snapshots)
- exit_horribly(NULL,
- "Synchronized snapshots are not supported by this server version.\n"
- "Run with --no-synchronized-snapshots instead if you do not need\n"
- "synchronized snapshots.\n");
- /* Find the last built-in OID, if needed */
- if (fout->remoteVersion < 70300)
- {
- if (fout->remoteVersion >= 70100)
- g_last_builtin_oid = findLastBuiltinOid_V71(fout,
- PQdb(GetConnection(fout)));
- else
- g_last_builtin_oid = findLastBuiltinOid_V70(fout);
- if (g_verbose)
- write_msg(NULL, "last built-in OID is %u\n", g_last_builtin_oid);
- }
- /* Expand schema selection patterns into OID lists */
- if (schema_include_patterns.head != NULL)
- {
- expand_schema_name_patterns(fout, &schema_include_patterns,
- &schema_include_oids);
- if (schema_include_oids.head == NULL)
- exit_horribly(NULL, "No matching schemas were found\n");
- }
- expand_schema_name_patterns(fout, &schema_exclude_patterns,
- &schema_exclude_oids);
- /* non-matching exclusion patterns aren't an error */
- /* Expand table selection patterns into OID lists */
- if (table_include_patterns.head != NULL)
- {
- expand_table_name_patterns(fout, &table_include_patterns,
- &table_include_oids);
- if (table_include_oids.head == NULL)
- exit_horribly(NULL, "No matching tables were found\n");
- }
- expand_table_name_patterns(fout, &table_exclude_patterns,
- &table_exclude_oids);
- expand_table_name_patterns(fout, &tabledata_exclude_patterns,
- &tabledata_exclude_oids);
- /* non-matching exclusion patterns aren't an error */
- /*
- * Dumping blobs is now default unless we saw an inclusion switch or -s
- * ... but even if we did see one of these, -b turns it back on.
- */
- if (include_everything && !schemaOnly)
- outputBlobs = true;
- /*
- * Now scan the database and create DumpableObject structs for all the
- * objects we intend to dump.
- */
- tblinfo = getSchemaData(fout, &numTables);
- if (fout->remoteVersion < 80400)
- guessConstraintInheritance(tblinfo, numTables);
- if (!schemaOnly)
- {
- getTableData(tblinfo, numTables, oids);
- buildMatViewRefreshDependencies(fout);
- if (dataOnly)
- getTableDataFKConstraints();
- }
- if (outputBlobs)
- getBlobs(fout);
- /*
- * Collect dependency data to assist in ordering the objects.
- */
- getDependencies(fout);
- /* Lastly, create dummy objects to represent the section boundaries */
- boundaryObjs = createBoundaryObjects();
- /* Get pointers to all the known DumpableObjects */
- getDumpableObjects(&dobjs, &numObjs);
- /*
- * Add dummy dependencies to enforce the dump section ordering.
- */
- addBoundaryDependencies(dobjs, numObjs, boundaryObjs);
- /*
- * Sort the objects into a safe dump order (no forward references).
- *
- * In 7.3 or later, we can rely on dependency information to help us
- * determine a safe order, so the initial sort is mostly for cosmetic
- * purposes: we sort by name to ensure that logically identical schemas
- * will dump identically. Before 7.3 we don't have dependencies and we
- * use OID ordering as an (unreliable) guide to creation order.
- */
- if (fout->remoteVersion >= 70300)
- sortDumpableObjectsByTypeName(dobjs, numObjs);
- else
- sortDumpableObjectsByTypeOid(dobjs, numObjs);
- /* If we do a parallel dump, we want the largest tables to go first */
- if (archiveFormat == archDirectory && numWorkers > 1)
- sortDataAndIndexObjectsBySize(dobjs, numObjs);
- sortDumpableObjects(dobjs, numObjs,
- boundaryObjs[0].dumpId, boundaryObjs[1].dumpId);
- /*
- * Create archive TOC entries for all the objects to be dumped, in a safe
- * order.
- */
- /* First the special ENCODING and STDSTRINGS entries. */
- dumpEncoding(fout);
- dumpStdStrings(fout);
- /* The database item is always next, unless we don't want it at all */
- if (include_everything && !dataOnly)
- dumpDatabase(fout);
- /* Now the rearrangeable objects. */
- for (i = 0; i < numObjs; i++)
- dumpDumpableObject(fout, dobjs[i]);
- /*
- * Set up options info to ensure we dump what we want.
- */
- ropt = NewRestoreOptions();
- ropt->filename = filename;
- ropt->dropSchema = outputClean;
- ropt->dataOnly = dataOnly;
- ropt->schemaOnly = schemaOnly;
- ropt->if_exists = if_exists;
- ropt->dumpSections = dumpSections;
- ropt->aclsSkip = aclsSkip;
- ropt->superuser = outputSuperuser;
- ropt->createDB = outputCreateDB;
- ropt->noOwner = outputNoOwner;
- ropt->noTablespace = outputNoTablespaces;
- ropt->disable_triggers = disable_triggers;
- ropt->use_setsessauth = use_setsessauth;
- if (compressLevel == -1)
- ropt->compression = 0;
- else
- ropt->compression = compressLevel;
- ropt->suppressDumpWarnings = true; /* We've already shown them */
- SetArchiveRestoreOptions(fout, ropt);
- /*
- * The archive's TOC entries are now marked as to which ones will actually
- * be output, so we can set up their dependency lists properly. This isn't
- * necessary for plain-text output, though.
- */
- if (!plainText)
- BuildArchiveDependencies(fout);
- /*
- * And finally we can do the actual output.
- *
- * Note: for non-plain-text output formats, the output file is written
- * inside CloseArchive(). This is, um, bizarre; but not worth changing
- * right now.
- */
- if (plainText)
- RestoreArchive(fout);
- CloseArchive(fout);
- exit_nicely(0);
- }
- static void
- help(const char *progname)
- {
- printf(_("%s dumps a database as a text file or to other formats.\n\n"), progname);
- printf(_("Usage:\n"));
- printf(_(" %s [OPTION]... [DBNAME]\n"), progname);
- printf(_("\nGeneral options:\n"));
- printf(_(" -f, --file=FILENAME output file or directory name\n"));
- printf(_(" -F, --format=c|d|t|p output file format (custom, directory, tar,\n"
- " plain text (default))\n"));
- printf(_(" -j, --jobs=NUM use this many parallel jobs to dump\n"));
- printf(_(" -v, --verbose verbose mode\n"));
- printf(_(" -V, --version output version information, then exit\n"));
- printf(_(" -Z, --compress=0-9 compression level for compressed formats\n"));
- printf(_(" --lock-wait-timeout=TIMEOUT fail after waiting TIMEOUT for a table lock\n"));
- printf(_(" -?, --help show this help, then exit\n"));
- printf(_("\nOptions controlling the output content:\n"));
- printf(_(" -a, --data-only dump only the data, not the schema\n"));
- printf(_(" -b, --blobs include large objects in dump\n"));
- printf(_(" -c, --clean clean (drop) database objects before recreating\n"));
- printf(_(" -C, --create include commands to create database in dump\n"));
- printf(_(" -E, --encoding=ENCODING dump the data in encoding ENCODING\n"));
- printf(_(" -n, --schema=SCHEMA dump the named schema(s) only\n"));
- printf(_(" -N, --exclude-schema=SCHEMA do NOT dump the named schema(s)\n"));
- printf(_(" -o, --oids include OIDs in dump\n"));
- printf(_(" -O, --no-owner skip restoration of object ownership in\n"
- " plain-text format\n"));
- printf(_(" -s, --schema-only dump only the schema, no data\n"));
- printf(_(" -S, --superuser=NAME superuser user name to use in plain-text format\n"));
- printf(_(" -t, --table=TABLE dump the named table(s) only\n"));
- printf(_(" -T, --exclude-table=TABLE do NOT dump the named table(s)\n"));
- printf(_(" -x, --no-privileges do not dump privileges (grant/revoke)\n"));
- printf(_(" --binary-upgrade for use by upgrade utilities only\n"));
- printf(_(" --column-inserts dump data as INSERT commands with column names\n"));
- printf(_(" --disable-dollar-quoting disable dollar quoting, use SQL standard quoting\n"));
- printf(_(" --disable-triggers disable triggers during data-only restore\n"));
- printf(_(" --exclude-table-data=TABLE do NOT dump data for the named table(s)\n"));
- printf(_(" --if-exists use IF EXISTS when dropping objects\n"));
- printf(_(" --inserts dump data as INSERT commands, rather than COPY\n"));
- printf(_(" --no-security-labels do not dump security label assignments\n"));
- printf(_(" --no-synchronized-snapshots do not use synchronized snapshots in parallel jobs\n"));
- printf(_(" --no-tablespaces do not dump tablespace assignments\n"));
- printf(_(" --no-unlogged-table-data do not dump unlogged table data\n"));
- printf(_(" --quote-all-identifiers quote all identifiers, even if not key words\n"));
- printf(_(" --section=SECTION dump named section (pre-data, data, or post-data)\n"));
- printf(_(" --serializable-deferrable wait until the dump can run without anomalies\n"));
- printf(_(" --use-set-session-authorization\n"
- " use SET SESSION AUTHORIZATION commands instead of\n"
- " ALTER OWNER commands to set ownership\n"));
- printf(_("\nConnection options:\n"));
- printf(_(" -d, --dbname=DBNAME database to dump\n"));
- printf(_(" -h, --host=HOSTNAME database server host or socket directory\n"));
- printf(_(" -p, --port=PORT database server port number\n"));
- printf(_(" -U, --username=NAME connect as specified database user\n"));
- printf(_(" -w, --no-password never prompt for password\n"));
- printf(_(" -W, --password force password prompt (should happen automatically)\n"));
- printf(_(" --role=ROLENAME do SET ROLE before dump\n"));
- printf(_("\nIf no database name is supplied, then the PGDATABASE environment\n"
- "variable value is used.\n\n"));
- printf(_("Report bugs to <pgsql-bugs@postgresql.org>.\n"));
- }
- static void
- setup_connection(Archive *AH, const char *dumpencoding, char *use_role)
- {
- PGconn *conn = GetConnection(AH);
- const char *std_strings;
- /*
- * Set the client encoding if requested. If dumpencoding == NULL then
- * either it hasn't been requested or we're a cloned connection and then
- * this has already been set in CloneArchive according to the original
- * connection encoding.
- */
- if (dumpencoding)
- {
- if (PQsetClientEncoding(conn, dumpencoding) < 0)
- exit_horribly(NULL, "invalid client encoding \"%s\" specified\n",
- dumpencoding);
- }
- /*
- * Get the active encoding and the standard_conforming_strings setting, so
- * we know how to escape strings.
- */
- AH->encoding = PQclientEncoding(conn);
- std_strings = PQparameterStatus(conn, "standard_conforming_strings");
- AH->std_strings = (std_strings && strcmp(std_strings, "on") == 0);
- /* Set the role if requested */
- if (!use_role && AH->use_role)
- use_role = AH->use_role;
- /* Set the role if requested */
- if (use_role && AH->remoteVersion >= 80100)
- {
- PQExpBuffer query = createPQExpBuffer();
- appendPQExpBuffer(query, "SET ROLE %s", fmtId(use_role));
- ExecuteSqlStatement(AH, query->data);
- destroyPQExpBuffer(query);
- /* save this for later use on parallel connections */
- if (!AH->use_role)
- AH->use_role = strdup(use_role);
- }
- /* Set the datestyle to ISO to ensure the dump's portability */
- ExecuteSqlStatement(AH, "SET DATESTYLE = ISO");
- /* Likewise, avoid using sql_standard intervalstyle */
- if (AH->remoteVersion >= 80400)
- ExecuteSqlStatement(AH, "SET INTERVALSTYLE = POSTGRES");
- /*
- * If supported, set extra_float_digits so that we can dump float data
- * exactly (given correctly implemented float I/O code, anyway)
- */
- if (AH->remoteVersion >= 90000)
- ExecuteSqlStatement(AH, "SET extra_float_digits TO 3");
- else if (AH->remoteVersion >= 70400)
- ExecuteSqlStatement(AH, "SET extra_float_digits TO 2");
- /*
- * If synchronized scanning is supported, disable it, to prevent
- * unpredictable changes in row ordering across a dump and reload.
- */
- if (AH->remoteVersion >= 80300)
- ExecuteSqlStatement(AH, "SET synchronize_seqscans TO off");
- /*
- * Disable timeouts if supported.
- */
- if (AH->remoteVersion >= 70300)
- ExecuteSqlStatement(AH, "SET statement_timeout = 0");
- if (AH->remoteVersion >= 90300)
- ExecuteSqlStatement(AH, "SET lock_timeout = 0");
- /*
- * Quote all identifiers, if requested.
- */
- if (quote_all_identifiers && AH->remoteVersion >= 90100)
- ExecuteSqlStatement(AH, "SET quote_all_identifiers = true");
- /*
- * Start transaction-snapshot mode transaction to dump consistent data.
- */
- ExecuteSqlStatement(AH, "BEGIN");
- if (AH->remoteVersion >= 90100)
- {
- if (serializable_deferrable)
- ExecuteSqlStatement(AH,
- "SET TRANSACTION ISOLATION LEVEL "
- "SERIALIZABLE, READ ONLY, DEFERRABLE");
- else
- ExecuteSqlStatement(AH,
- "SET TRANSACTION ISOLATION LEVEL "
- "REPEATABLE READ, READ ONLY");
- }
- else if (AH->remoteVersion >= 70400)
- {
- /* note: comma was not accepted in SET TRANSACTION before 8.0 */
- ExecuteSqlStatement(AH,
- "SET TRANSACTION ISOLATION LEVEL "
- "SERIALIZABLE READ ONLY");
- }
- else
- ExecuteSqlStatement(AH,
- "SET TRANSACTION ISOLATION LEVEL SERIALIZABLE");
- if (AH->numWorkers > 1 && AH->remoteVersion >= 90200 && !no_synchronized_snapshots)
- {
- if (AH->sync_snapshot_id)
- {
- PQExpBuffer query = createPQExpBuffer();
- appendPQExpBufferStr(query, "SET TRANSACTION SNAPSHOT ");
- appendStringLiteralConn(query, AH->sync_snapshot_id, conn);
- ExecuteSqlStatement(AH, query->data);
- destroyPQExpBuffer(query);
- }
- else
- AH->sync_snapshot_id = get_synchronized_snapshot(AH);
- }
- }
- static void
- setupDumpWorker(Archive *AHX, RestoreOptions *ropt)
- {
- setup_connection(AHX, NULL, NULL);
- }
- static char *
- get_synchronized_snapshot(Archive *fout)
- {
- char *query = "SELECT pg_export_snapshot()";
- char *result;
- PGresult *res;
- res = ExecuteSqlQueryForSingleRow(fout, query);
- result = strdup(PQgetvalue(res, 0, 0));
- PQclear(res);
- return result;
- }
- static ArchiveFormat
- parseArchiveFormat(const char *format, ArchiveMode *mode)
- {
- ArchiveFormat archiveFormat;
- *mode = archModeWrite;
- if (pg_strcasecmp(format, "a") == 0 || pg_strcasecmp(format, "append") == 0)
- {
- /* This is used by pg_dumpall, and is not documented */
- archiveFormat = archNull;
- *mode = archModeAppend;
- }
- else if (pg_strcasecmp(format, "c") == 0)
- archiveFormat = archCustom;
- else if (pg_strcasecmp(format, "custom") == 0)
- archiveFormat = archCustom;
- else if (pg_strcasecmp(format, "d") == 0)
- archiveFormat = archDirectory;
- else if (pg_strcasecmp(format, "directory") == 0)
- archiveFormat = archDirectory;
- else if (pg_strcasecmp(format, "p") == 0)
- archiveFormat = archNull;
- else if (pg_strcasecmp(format, "plain") == 0)
- archiveFormat = archNull;
- else if (pg_strcasecmp(format, "t") == 0)
- archiveFormat = archTar;
- else if (pg_strcasecmp(format, "tar") == 0)
- archiveFormat = archTar;
- else
- exit_horribly(NULL, "invalid output format \"%s\" specified\n", format);
- return archiveFormat;
- }
- /*
- * Find the OIDs of all schemas matching the given list of patterns,
- * and append them to the given OID list.
- */
- static void
- expand_schema_name_patterns(Archive *fout,
- SimpleStringList *patterns,
- SimpleOidList *oids)
- {
- PQExpBuffer query;
- PGresult *res;
- SimpleStringListCell *cell;
- int i;
- if (patterns->head == NULL)
- return; /* nothing to do */
- if (fout->remoteVersion < 70300)
- exit_horribly(NULL, "server version must be at least 7.3 to use schema selection switches\n");
- query = createPQExpBuffer();
- /*
- * We use UNION ALL rather than UNION; this might sometimes result in
- * duplicate entries in the OID list, but we don't care.
- */
- for (cell = patterns->head; cell; cell = cell->next)
- {
- if (cell != patterns->head)
- appendPQExpBufferStr(query, "UNION ALL\n");
- appendPQExpBuffer(query,
- "SELECT oid FROM pg_catalog.pg_namespace n\n");
- processSQLNamePattern(GetConnection(fout), query, cell->val, false,
- false, NULL, "n.nspname", NULL, NULL);
- }
- res = ExecuteSqlQuery(fout, query->data, PGRES_TUPLES_OK);
- for (i = 0; i < PQntuples(res); i++)
- {
- simple_oid_list_append(oids, atooid(PQgetvalue(res, i, 0)));
- }
- PQclear(res);
- destroyPQExpBuffer(query);
- }
- /*
- * Find the OIDs of all tables matching the given list of patterns,
- * and append them to the given OID list.
- */
- static void
- expand_table_name_patterns(Archive *fout,
- SimpleStringList *patterns, SimpleOidList *oids)
- {
- PQExpBuffer query;
- PGresult *res;
- SimpleStringListCell *cell;
- int i;
- if (patterns->head == NULL)
- return; /* nothing to do */
- query = createPQExpBuffer();
- /*
- * We use UNION ALL rather than UNION; this might sometimes result in
- * duplicate entries in the OID list, but we don't care.
- */
- for (cell = patterns->head; cell; cell = cell->next)
- {
- if (cell != patterns->head)
- appendPQExpBufferStr(query, "UNION ALL\n");
- appendPQExpBuffer(query,
- "SELECT c.oid"
- "\nFROM pg_catalog.pg_class c"
- "\n LEFT JOIN pg_catalog.pg_namespace n ON n.oid = c.relnamespace"
- "\nWHERE c.relkind in ('%c', '%c', '%c', '%c', '%c')\n",
- RELKIND_RELATION, RELKIND_SEQUENCE, RELKIND_VIEW,
- RELKIND_MATVIEW, RELKIND_FOREIGN_TABLE);
- processSQLNamePattern(GetConnection(fout), query, cell->val, true,
- false, "n.nspname", "c.relname", NULL,
- "pg_catalog.pg_table_is_visible(c.oid)");
- }
- res = ExecuteSqlQuery(fout, query->data, PGRES_TUPLES_OK);
- for (i = 0; i < PQntuples(res); i++)
- {
- simple_oid_list_append(oids, atooid(PQgetvalue(res, i, 0)));
- }
- PQclear(res);
- destroyPQExpBuffer(query);
- }
- /*
- * selectDumpableNamespace: policy-setting subroutine
- * Mark a namespace as to be dumped or not
- */
- static void
- selectDumpableNamespace(NamespaceInfo *nsinfo)
- {
- /*
- * If specific tables are being dumped, do not dump any complete
- * namespaces. If specific namespaces are being dumped, dump just those
- * namespaces. Otherwise, dump all non-system namespaces.
- */
- if (table_include_oids.head != NULL)
- nsinfo->dobj.dump = false;
- else if (schema_include_oids.head != NULL)
- nsinfo->dobj.dump = simple_oid_list_member(&schema_include_oids,
- nsinfo->dobj.catId.oid);
- else if (strncmp(nsinfo->dobj.name, "pg_", 3) == 0 ||
- strcmp(nsinfo->dobj.name, "information_schema") == 0)
- nsinfo->dobj.dump = false;
- else
- nsinfo->dobj.dump = true;
- /*
- * In any case, a namespace can be excluded by an exclusion switch
- */
- if (nsinfo->dobj.dump &&
- simple_oid_list_member(&schema_exclude_oids,
- nsinfo->dobj.catId.oid))
- nsinfo->dobj.dump = false;
- }
- /*
- * selectDumpableTable: policy-setting subroutine
- * Mark a table as to be dumped or not
- */
- static void
- selectDumpableTable(TableInfo *tbinfo)
- {
- /*
- * If specific tables are being dumped, dump just those tables; else, dump
- * according to the parent namespace's dump flag.
- */
- if (table_include_oids.head != NULL)
- tbinfo->dobj.dump = simple_oid_list_member(&table_include_oids,
- tbinfo->dobj.catId.oid);
- else
- tbinfo->dobj.dump = tbinfo->dobj.namespace->dobj.dump;
- /*
- * In any case, a table can be excluded by an exclusion switch
- */
- if (tbinfo->dobj.dump &&
- simple_oid_list_member(&table_exclude_oids,
- tbinfo->dobj.catId.oid))
- tbinfo->dobj.dump = false;
- }
- /*
- * selectDumpableType: policy-setting subroutine
- * Mark a type as to be dumped or not
- *
- * If it's a table's rowtype or an autogenerated array type, we also apply a
- * special type code to facilitate sorting into the desired order. (We don't
- * want to consider those to be ordinary types because that would bring tables
- * up into the datatype part of the dump order.) We still set the object's
- * dump flag; that's not going to cause the dummy type to be dumped, but we
- * need it so that casts involving such types will be dumped correctly -- see
- * dumpCast. This means the flag should be set the same as for the underlying
- * object (the table or base type).
- */
- static void
- selectDumpableType(TypeInfo *tyinfo)
- {
- /* skip complex types, except for standalone composite types */
- if (OidIsValid(tyinfo->typrelid) &&
- tyinfo->typrelkind != RELKIND_COMPOSITE_TYPE)
- {
- TableInfo *tytable = findTableByOid(tyinfo->typrelid);
- tyinfo->dobj.objType = DO_DUMMY_TYPE;
- if (tytable != NULL)
- tyinfo->dobj.dump = tytable->dobj.dump;
- else
- tyinfo->dobj.dump = false;
- return;
- }
- /* skip auto-generated array types */
- if (tyinfo->isArray)
- {
- tyinfo->dobj.objType = DO_DUMMY_TYPE;
- /*
- * Fall through to set the dump flag; we assume that the subsequent
- * rules will do the same thing as they would for the array's base
- * type. (We cannot reliably look up the base type here, since
- * getTypes may not have processed it yet.)
- */
- }
- /* dump only types in dumpable namespaces */
- if (!tyinfo->dobj.namespace->dobj.dump)
- tyinfo->dobj.dump = false;
- /* skip undefined placeholder types */
- else if (!tyinfo->isDefined)
- tyinfo->dobj.dump = false;
- else
- tyinfo->dobj.dump = true;
- }
- /*
- * selectDumpableDefaultACL: policy-setting subroutine
- * Mark a default ACL as to be dumped or not
- *
- * For per-schema default ACLs, dump if the schema is to be dumped.
- * Otherwise dump if we are dumping "everything". Note that dataOnly
- * and aclsSkip are checked separately.
- */
- static void
- selectDumpableDefaultACL(DefaultACLInfo *dinfo)
- {
- if (dinfo->dobj.namespace)
- dinfo->dobj.dump = dinfo->dobj.namespace->dobj.dump;
- else
- dinfo->dobj.dump = include_everything;
- }
- /*
- * selectDumpableExtension: policy-setting subroutine
- * Mark an extension as to be dumped or not
- *
- * Normally, we dump all extensions, or none of them if include_everything
- * is false (i.e., a --schema or --table switch was given). However, in
- * binary-upgrade mode it's necessary to skip built-in extensions, since we
- * assume those will already be installed in the target database. We identify
- * such extensions by their having OIDs in the range reserved for initdb.
- */
- static void
- selectDumpableExtension(ExtensionInfo *extinfo)
- {
- if (binary_upgrade && extinfo->dobj.catId.oid < (Oid) FirstNormalObjectId)
- extinfo->dobj.dump = false;
- else
- extinfo->dobj.dump = include_everything;
- }
- /*
- * selectDumpableObject: policy-setting subroutine
- * Mark a generic dumpable object as to be dumped or not
- *
- * Use this only for object types without a special-case routine above.
- */
- static void
- selectDumpableObject(DumpableObject *dobj)
- {
- /*
- * Default policy is to dump if parent namespace is dumpable, or always
- * for non-namespace-associated items.
- */
- if (dobj->namespace)
- dobj->dump = dobj->namespace->dobj.dump;
- else
- dobj->dump = true;
- }
- /*
- * Dump a table's contents for loading using the COPY command
- * - this routine is called by the Archiver when it wants the table
- * to be dumped.
- */
- static int
- dumpTableData_copy(Archive *fout, void *dcontext)
- {
- TableDataInfo *tdinfo = (TableDataInfo *) dcontext;
- TableInfo *tbinfo = tdinfo->tdtable;
- const char *classname = tbinfo->dobj.name;
- const bool hasoids = tbinfo->hasoids;
- const bool oids = tdinfo->oids;
- PQExpBuffer q = createPQExpBuffer();
- /*
- * Note: can't use getThreadLocalPQExpBuffer() here, we're calling fmtId
- * which uses it already.
- */
- PQExpBuffer clistBuf = createPQExpBuffer();
- PGconn *conn = GetConnection(fout);
- PGresult *res;
- int ret;
- char *copybuf;
- const char *column_list;
- if (g_verbose)
- write_msg(NULL, "dumping contents of table %s\n", classname);
- /*
- * Make sure we are in proper schema. We will qualify the table name
- * below anyway (in case its name conflicts with a pg_catalog table); but
- * this ensures reproducible results in case the table contains regproc,
- * regclass, etc columns.
- */
- selectSourceSchema(fout, tbinfo->dobj.namespace->dobj.name);
- /*
- * If possible, specify the column list explicitly so that we have no
- * possibility of retrieving data in the wrong column order. (The default
- * column ordering of COPY will not be what we want in certain corner
- * cases involving ADD COLUMN and inheritance.)
- */
- if (fout->remoteVersion >= 70300)
- column_list = fmtCopyColumnList(tbinfo, clistBuf);
- else
- column_list = ""; /* can't select columns in COPY */
- if (oids && hasoids)
- {
- appendPQExpBuffer(q, "COPY %s %s WITH OIDS TO stdout;",
- fmtQualifiedId(fout->remoteVersion,
- tbinfo->dobj.namespace->dobj.name,
- classname),
- column_list);
- }
- else if (tdinfo->filtercond)
- {
- /* Note: this syntax is only supported in 8.2 and up */
- appendPQExpBufferStr(q, "COPY (SELECT ");
- /* klugery to get rid of parens in column list */
- if (strlen(column_list) > 2)
- {
- appendPQExpBufferStr(q, column_list + 1);
- q->data[q->len - 1] = ' ';
- }
- else
- appendPQExpBufferStr(q, "* ");
- appendPQExpBuffer(q, "FROM %s %s) TO stdout;",
- fmtQualifiedId(fout->remoteVersion,
- tbinfo->dobj.namespace->dobj.name,
- classname),
- tdinfo->filtercond);
- }
- else
- {
- appendPQExpBuffer(q, "COPY %s %s TO stdout;",
- fmtQualifiedId(fout->remoteVersion,
- tbinfo->dobj.namespace->dobj.name,
- classname),
- column_list);
- }
- res = ExecuteSqlQuery(fout, q->data, PGRES_COPY_OUT);
- PQclear(res);
- destroyPQExpBuffer(clistBuf);
- for (;;)
- {
- ret = PQgetCopyData(conn, ©buf, 0);
- if (ret < 0)
- break; /* done or error */
- if (copybuf)
- {
- WriteData(fout, copybuf, ret);
- PQfreemem(copybuf);
- }
- /* ----------
- * THROTTLE:
- *
- * There was considerable discussion in late July, 2000 regarding
- * slowing down pg_dump when backing up large tables. Users with both
- * slow & fast (multi-processor) machines experienced performance
- * degradation when doing a backup.
- *
- * Initial attempts based on sleeping for a number of ms for each ms
- * of work were deemed too complex, then a simple 'sleep in each loop'
- * implementation was suggested. The latter failed because the loop
- * was too tight. Finally, the following was implemented:
- *
- * If throttle is non-zero, then
- * See how long since the last sleep.
- * Work out how long to sleep (based on ratio).
- * If sleep is more than 100ms, then
- * sleep
- * reset timer
- * EndIf
- * EndIf
- *
- * where the throttle value was the number of ms to sleep per ms of
- * work. The calculation was done in each loop.
- *
- * Most of the hard work is done in the backend, and this solution
- * still did not work particularly well: on slow machines, the ratio
- * was 50:1, and on medium paced machines, 1:1, and on fast
- * multi-processor machines, it had little or no effect, for reasons
- * that were unclear.
- *
- * Further discussion ensued, and the proposal was dropped.
- *
- * For those people who want this feature, it can be implemented using
- * gettimeofday in each loop, calculating the time since last sleep,
- * multiplying that by the sleep ratio, then if the result is more
- * than a preset 'minimum sleep time' (say 100ms), call the 'select'
- * function to sleep for a subsecond period ie.
- *
- * select(0, NULL, NULL, NULL, &tvi);
- *
- * This will return after the interval specified in the structure tvi.
- * Finally, call gettimeofday again to save the 'last sleep time'.
- * ----------
- */
- }
- archprintf(fout, "\\.\n\n\n");
- if (ret == -2)
- {
- /* copy data transfer failed */
- write_msg(NULL, "Dumping the contents of table \"%s\" failed: PQgetCopyData() failed.\n", classname);
- write_msg(NULL, "Error message from server: %s", PQerrorMessage(conn));
- write_msg(NULL, "The command was: %s\n", q->data);
- exit_nicely(1);
- }
- /* Check command status and return to normal libpq state */
- res = PQgetResult(conn);
- if (PQresultStatus(res) != PGRES_COMMAND_OK)
- {
- write_msg(NULL, "Dumping the contents of table \"%s\" failed: PQgetResult() failed.\n", classname);
- write_msg(NULL, "Error message from server: %s", PQerrorMessage(conn));
- write_msg(NULL, "The command was: %s\n", q->data);
- exit_nicely(1);
- }
- PQclear(res);
- destroyPQExpBuffer(q);
- return 1;
- }
- /*
- * Dump table data using INSERT commands.
- *
- * Caution: when we restore from an archive file direct to database, the
- * INSERT commands emitted by this function have to be parsed by
- * pg_backup_db.c's ExecuteSimpleCommands(), which will not handle comments,
- * E'' strings, or dollar-quoted strings. So don't emit anything like that.
- */
- static int
- dumpTableData_insert(Archive *fout, void *dcontext)
- {
- TableDataInfo *tdinfo = (TableDataInfo *) dcontext;
- TableInfo *tbinfo = tdinfo->tdtable;
- const char *classname = tbinfo->dobj.name;
- PQExpBuffer q = createPQExpBuffer();
- PQExpBuffer insertStmt = NULL;
- PGresult *res;
- int tuple;
- int nfields;
- int field;
- /*
- * Make sure we are in proper schema. We will qualify the table name
- * below anyway (in case its name conflicts with a pg_catalog table); but
- * this ensures reproducible results in case the table contains regproc,
- * regclass, etc columns.
- */
- selectSourceSchema(fout, tbinfo->dobj.namespace->dobj.name);
- if (fout->remoteVersion >= 70100)
- {
- appendPQExpBuffer(q, "DECLARE _pg_dump_cursor CURSOR FOR "
- "SELECT * FROM ONLY %s",
- fmtQualifiedId(fout->remoteVersion,
- tbinfo->dobj.namespace->dobj.name,
- classname));
- }
- else
- {
- appendPQExpBuffer(q, "DECLARE _pg_dump_cursor CURSOR FOR "
- "SELECT * FROM %s",
- fmtQualifiedId(fout->remoteVersion,
- tbinfo->dobj.namespace->dobj.name,
- classname));
- }
- if (tdinfo->filtercond)
- appendPQExpBuffer(q, " %s", tdinfo->filtercond);
- ExecuteSqlStatement(fout, q->data);
- while (1)
- {
- res = ExecuteSqlQuery(fout, "FETCH 100 FROM _pg_dump_cursor",
- PGRES_TUPLES_OK);
- nfields = PQnfields(res);
- for (tuple = 0; tuple < PQntuples(res); tuple++)
- {
- /*
- * First time through, we build as much of the INSERT statement as
- * possible in "insertStmt", which we can then just print for each
- * line. If the table happens to have zero columns then this will
- * be a complete statement, otherwise it will end in "VALUES(" and
- * be ready to have the row's column values appended.
- */
- if (insertStmt == NULL)
- {
- insertStmt = createPQExpBuffer();
- appendPQExpBuffer(insertStmt, "INSERT INTO %s ",
- fmtId(classname));
- /* corner case for zero-column table */
- if (nfields == 0)
- {
- appendPQExpBufferStr(insertStmt, "DEFAULT VALUES;\n");
- }
- else
- {
- /* append the list of column names if required */
- if (column_inserts)
- {
- appendPQExpBufferStr(insertStmt, "(");
- for (field = 0; field < nfields; field++)
- {
- if (field > 0)
- appendPQExpBufferStr(insertStmt, ", ");
- appendPQExpBufferStr(insertStmt,
- fmtId(PQfname(res, field)));
- }
- appendPQExpBufferStr(insertStmt, ") ");
- }
- appendPQExpBufferStr(insertStmt, "VALUES (");
- }
- }
- archputs(insertStmt->data, fout);
- /* if it is zero-column table then we're done */
- if (nfields == 0)
- continue;
- for (field = 0; field < nfields; field++)
- {
- if (field > 0)
- archputs(", ", fout);
- if (PQgetisnull(res, tuple, field))
- {
- archputs("NULL", fout);
- continue;
- }
- /* XXX This code is partially duplicated in ruleutils.c */
- switch (PQftype(res, field))
- {
- case INT2OID:
- case INT4OID:
- case INT8OID:
- case OIDOID:
- case FLOAT4OID:
- case FLOAT8OID:
- case NUMERICOID:
- {
- /*
- * These types are printed without quotes unless
- * they contain values that aren't accepted by the
- * scanner unquoted (e.g., 'NaN'). Note that
- * strtod() and friends might accept NaN, so we
- * can't use that to test.
- *
- * In reality we only need to defend against
- * infinity and NaN, so we need not get too crazy
- * about pattern matching here.
- */
- const char *s = PQgetvalue(res, tuple, field);
- if (strspn(s, "0123456789 +-eE.") == strlen(s))
- archputs(s, fout);
- else
- archprintf(fout, "'%s'", s);
- }
- break;
- case BITOID:
- case VARBITOID:
- archprintf(fout, "B'%s'",
- PQgetvalue(res, tuple, field));
- break;
- case BOOLOID:
- if (strcmp(PQgetvalue(res, tuple, field), "t") == 0)
- archputs("true", fout);
- else
- archputs("false", fout);
- break;
- default:
- /* All other types are printed as string literals. */
- resetPQExpBuffer(q);
- appendStringLiteralAH(q,
- PQgetvalue(res, tuple, field),
- fout);
- archputs(q->data, fout);
- break;
- }
- }
- archputs(");\n", fout);
- }
- if (PQntuples(res) <= 0)
- {
- PQclear(res);
- break;
- }
- PQclear(res);
- }
- archputs("\n\n", fout);
- ExecuteSqlStatement(fout, "CLOSE _pg_dump_cursor");
- destroyPQExpBuffer(q);
- if (insertStmt != NULL)
- destroyPQExpBuffer(insertStmt);
- return 1;
- }
- /*
- * dumpTableData -
- * dump the contents of a single table
- *
- * Actually, this just makes an ArchiveEntry for the table contents.
- */
- static void
- dumpTableData(Archive *fout, TableDataInfo *tdinfo)
- {
- TableInfo *tbinfo = tdinfo->tdtable;
- PQExpBuffer copyBuf = createPQExpBuffer();
- PQExpBuffer clistBuf = createPQExpBuffer();
- DataDumperPtr dumpFn;
- char *copyStmt;
- if (!dump_inserts)
- {
- /* Dump/restore using COPY */
- dumpFn = dumpTableData_copy;
- /* must use 2 steps here 'cause fmtId is nonreentrant */
- appendPQExpBuffer(copyBuf, "COPY %s ",
- fmtId(tbinfo->dobj.name));
- appendPQExpBuffer(copyBuf, "%s %sFROM stdin;\n",
- fmtCopyColumnList(tbinfo, clistBuf),
- (tdinfo->oids && tbinfo->hasoids) ? "WITH OIDS " : "");
- copyStmt = copyBuf->data;
- }
- else
- {
- /* Restore using INSERT */
- dumpFn = dumpTableData_insert;
- copyStmt = NULL;
- }
- /*
- * Note: although the TableDataInfo is a full DumpableObject, we treat its
- * dependency on its table as "special" and pass it to ArchiveEntry now.
- * See comments for BuildArchiveDependencies.
- */
- ArchiveEntry(fout, tdinfo->dobj.catId, tdinfo->dobj.dumpId,
- tbinfo->dobj.name, tbinfo->dobj.namespace->dobj.name,
- NULL, tbinfo->rolname,
- false, "TABLE DATA", SECTION_DATA,
- "", "", copyStmt,
- &(tbinfo->dobj.dumpId), 1,
- dumpFn, tdinfo);
- destroyPQExpBuffer(copyBuf);
- destroyPQExpBuffer(clistBuf);
- }
- /*
- * refreshMatViewData -
- * load or refresh the contents of a single materialized view
- *
- * Actually, this just makes an ArchiveEntry for the REFRESH MATERIALIZED VIEW
- * statement.
- */
- static void
- refreshMatViewData(Archive *fout, TableDataInfo *tdinfo)
- {
- TableInfo *tbinfo = tdinfo->tdtable;
- PQExpBuffer q;
- /* If the materialized view is not flagged as populated, skip this. */
- if (!tbinfo->relispopulated)
- return;
- q = createPQExpBuffer();
- appendPQExpBuffer(q, "REFRESH MATERIALIZED VIEW %s;\n",
- fmtId(tbinfo->dobj.name));
- ArchiveEntry(fout,
- tdinfo->dobj.catId, /* catalog ID */
- tdinfo->dobj.dumpId, /* dump ID */
- tbinfo->dobj.name, /* Name */
- tbinfo->dobj.namespace->dobj.name, /* Namespace */
- NULL, /* Tablespace */
- tbinfo->rolname, /* Owner */
- false, /* with oids */
- "MATERIALIZED VIEW DATA", /* Desc */
- SECTION_POST_DATA, /* Section */
- q->data, /* Create */
- "", /* Del */
- NULL, /* Copy */
- tdinfo->dobj.dependencies, /* Deps */
- tdinfo->dobj.nDeps, /* # Deps */
- NULL, /* Dumper */
- NULL); /* Dumper Arg */
- destroyPQExpBuffer(q);
- }
- /*
- * getTableData -
- * set up dumpable objects representing the contents of tables
- */
- static void
- getTableData(TableInfo *tblinfo, int numTables, bool oids)
- {
- int i;
- for (i = 0; i < numTables; i++)
- {
- if (tblinfo[i].dobj.dump)
- makeTableDataInfo(&(tblinfo[i]), oids);
- }
- }
- /*
- * Make a dumpable object for the data of this specific table
- *
- * Note: we make a TableDataInfo if and only if we are going to dump the
- * table data; the "dump" flag in such objects isn't used.
- */
- static void
- makeTableDataInfo(TableInfo *tbinfo, bool oids)
- {
- TableDataInfo *tdinfo;
- /*
- * Nothing to do if we already decided to dump the table. This will
- * happen for "config" tables.
- */
- if (tbinfo->dataObj != NULL)
- return;
- /* Skip VIEWs (no data to dump) */
- if (tbinfo->relkind == RELKIND_VIEW)
- return;
- /* Skip FOREIGN TABLEs (no data to dump) */
- if (tbinfo->relkind == RELKIND_FOREIGN_TABLE)
- return;
- /* Don't dump data in unlogged tables, if so requested */
- if (tbinfo->relpersistence == RELPERSISTENCE_UNLOGGED &&
- no_unlogged_table_data)
- return;
- /* Check that the data is not explicitly excluded */
- if (simple_oid_list_member(&tabledata_exclude_oids,
- tbinfo->dobj.catId.oid))
- return;
- /* OK, let's dump it */
- tdinfo = (TableDataInfo *) pg_malloc(sizeof(TableDataInfo));
- if (tbinfo->relkind == RELKIND_MATVIEW)
- tdinfo->dobj.objType = DO_REFRESH_MATVIEW;
- else
- tdinfo->dobj.objType = DO_TABLE_DATA;
- /*
- * Note: use tableoid 0 so that this object won't be mistaken for
- * something that pg_depend entries apply to.
- */
- tdinfo->dobj.catId.tableoid = 0;
- tdinfo->dobj.catId.oid = tbinfo->dobj.catId.oid;
- AssignDumpId(&tdinfo->dobj);
- tdinfo->dobj.name = tbinfo->dobj.name;
- tdinfo->dobj.namespace = tbinfo->dobj.namespace;
- tdinfo->tdtable = tbinfo;
- tdinfo->oids = oids;
- tdinfo->filtercond = NULL; /* might get set later */
- addObjectDependency(&tdinfo->dobj, tbinfo->dobj.dumpId);
- tbinfo->dataObj = tdinfo;
- }
- /*
- * The refresh for a materialized view must be dependent on the refresh for
- * any materialized view that this one is dependent on.
- *
- * This must be called after all the objects are created, but before they are
- * sorted.
- */
- static void
- buildMatViewRefreshDependencies(Archive *fout)
- {
- PQExpBuffer query;
- PGresult *res;
- int ntups,
- i;
- int i_classid,
- i_objid,
- i_refobjid;
- /* No Mat Views before 9.3. */
- if (fout->remoteVersion < 90300)
- return;
- /* Make sure we are in proper schema */
- selectSourceSchema(fout, "pg_catalog");
- query = createPQExpBuffer();
- appendPQExpBufferStr(query, "WITH RECURSIVE w AS "
- "( "
- "SELECT d1.objid, d2.refobjid, c2.relkind AS refrelkind "
- "FROM pg_depend d1 "
- "JOIN pg_class c1 ON c1.oid = d1.objid "
- "AND c1.relkind = 'm' "
- "JOIN pg_rewrite r1 ON r1.ev_class = d1.objid "
- "JOIN pg_depend d2 ON d2.classid = 'pg_rewrite'::regclass "
- "AND d2.objid = r1.oid "
- "AND d2.refobjid <> d1.objid "
- "JOIN pg_class c2 ON c2.oid = d2.refobjid "
- "AND c2.relkind IN ('m','v') "
- "WHERE d1.classid = 'pg_class'::regclass "
- "UNION "
- "SELECT w.objid, d3.refobjid, c3.relkind "
- "FROM w "
- "JOIN pg_rewrite r3 ON r3.ev_class = w.refobjid "
- "JOIN pg_depend d3 ON d3.classid = 'pg_rewrite'::regclass "
- "AND d3.objid = r3.oid "
- "AND d3.refobjid <> w.refobjid "
- "JOIN pg_class c3 ON c3.oid = d3.refobjid "
- "AND c3.relkind IN ('m','v') "
- ") "
- "SELECT 'pg_class'::regclass::oid AS classid, objid, refobjid "
- "FROM w "
- "WHERE refrelkind = 'm'");
- res = ExecuteSqlQuery(fout, query->data, PGRES_TUPLES_OK);
- ntups = PQntuples(res);
- i_classid = PQfnumber(res, "classid");
- i_objid = PQfnumber(res, "objid");
- i_refobjid = PQfnumber(res, "refobjid");
- for (i = 0; i < ntups; i++)
- {
- CatalogId objId;
- CatalogId refobjId;
- DumpableObject *dobj;
- DumpableObject *refdobj;
- TableInfo *tbinfo;
- TableInfo *reftbinfo;
- objId.tableoid = atooid(PQgetvalue(res, i, i_classid));
- objId.oid = atooid(PQgetvalue(res, i, i_objid));
- refobjId.tableoid = objId.tableoid;
- refobjId.oid = atooid(PQgetvalue(res, i, i_refobjid));
- dobj = findObjectByCatalogId(objId);
- if (dobj == NULL)
- continue;
- Assert(dobj->objType == DO_TABLE);
- tbinfo = (TableInfo *) dobj;
- Assert(tbinfo->relkind == RELKIND_MATVIEW);
- dobj = (DumpableObject *) tbinfo->dataObj;
- if (dobj == NULL)
- continue;
- Assert(dobj->objType == DO_REFRESH_MATVIEW);
- refdobj = findObjectByCatalogId(refobjId);
- if (refdobj == NULL)
- continue;
- Assert(refdobj->objType == DO_TABLE);
- reftbinfo = (TableInfo *) refdobj;
- Assert(reftbinfo->relkind == RELKIND_MATVIEW);
- refdobj = (DumpableObject *) reftbinfo->dataObj;
- if (refdobj == NULL)
- continue;
- Assert(refdobj->objType == DO_REFRESH_MATVIEW);
- addObjectDependency(dobj, refdobj->dumpId);
- if (!reftbinfo->relispopulated)
- tbinfo->relispopulated = false;
- }
- PQclear(res);
- destroyPQExpBuffer(query);
- }
- /*
- * getTableDataFKConstraints -
- * add dump-order dependencies reflecting foreign key constraints
- *
- * This code is executed only in a data-only dump --- in schema+data dumps
- * we handle foreign key issues by not creating the FK constraints until
- * after the data is loaded. In a data-only dump, however, we want to
- * order the table data objects in such a way that a table's referenced
- * tables are restored first. (In the presence of circular references or
- * self-references this may be impossible; we'll detect and complain about
- * that during the dependency sorting step.)
- */
- static void
- getTableDataFKConstraints(void)
- {
- DumpableObject **dobjs;
- int numObjs;
- int i;
- /* Search through all the dumpable objects for FK constraints */
- getDumpableObjects(&dobjs, &numObjs);
- for (i = 0; i < numObjs; i++)
- {
- if (dobjs[i]->objType == DO_FK_CONSTRAINT)
- {
- ConstraintInfo *cinfo = (ConstraintInfo *) dobjs[i];
- TableInfo *ftable;
- /* Not interesting unless both tables are to be dumped */
- if (cinfo->contable == NULL ||
- cinfo->contable->dataObj == NULL)
- continue;
- ftable = findTableByOid(cinfo->confrelid);
- if (ftable == NULL ||
- ftable->dataObj == NULL)
- continue;
- /*
- * Okay, make referencing table's TABLE_DATA object depend on the
- * referenced table's TABLE_DATA object.
- */
- addObjectDependency(&cinfo->contable->dataObj->dobj,
- ftable->dataObj->dobj.dumpId);
- }
- }
- free(dobjs);
- }
- /*
- * guessConstraintInheritance:
- * In pre-8.4 databases, we can't tell for certain which constraints
- * are inherited. We assume a CHECK constraint is inherited if its name
- * matches the name of any constraint in the parent. Originally this code
- * tried to compare the expression texts, but that can fail for various
- * reasons --- for example, if the parent and child tables are in different
- * schemas, reverse-listing of function calls may produce different text
- * (schema-qualified or not) depending on search path.
- *
- * In 8.4 and up we can rely on the conislocal field to decide which
- * constraints must be dumped; much safer.
- *
- * This function assumes all conislocal flags were initialized to TRUE.
- * It clears the flag on anything that seems to be inherited.
- */
- static void
- guessConstraintInheritance(TableInfo *tblinfo, int numTables)
- {
- int i,
- j,
- k;
- for (i = 0; i < numTables; i++)
- {
- TableInfo *tbinfo = &(tblinfo[i]);
- int numParents;
- TableInfo **parents;
- TableInfo *parent;
- /* Sequences and views never have parents */
- if (tbinfo->relkind == RELKIND_SEQUENCE ||
- tbinfo->relkind == RELKIND_VIEW)
- continue;
- /* Don't bother computing anything for non-target tables, either */
- if (!tbinfo->dobj.dump)
- continue;
- numParents = tbinfo->numParents;
- parents = tbinfo->parents;
- if (numParents == 0)
- continue; /* nothing to see here, move along */
- /* scan for inherited CHECK constraints */
- for (j = 0; j < tbinfo->ncheck; j++)
- {
- ConstraintInfo *constr;
- constr = &(tbinfo->checkexprs[j]);
- for (k = 0; k < numParents; k++)
- {
- int l;
- parent = parents[k];
- for (l = 0; l < parent->ncheck; l++)
- {
- ConstraintInfo *pconstr = &(parent->checkexprs[l]);
- if (strcmp(pconstr->dobj.name, constr->dobj.name) == 0)
- {
- constr->conislocal = false;
- break;
- }
- }
- if (!constr->conislocal)
- break;
- }
- }
- }
- }
- /*
- * dumpDatabase:
- * dump the database definition
- */
- static void
- dumpDatabase(Archive *fout)
- {
- PQExpBuffer dbQry = createPQExpBuffer();
- PQExpBuffer delQry = createPQExpBuffer();
- PQExpBuffer creaQry = createPQExpBuffer();
- PGconn *conn = GetConnection(fout);
- PGresult *res;
- int i_tableoid,
- i_oid,
- i_dba,
- i_encoding,
- i_collate,
- i_ctype,
- i_frozenxid,
- i_minmxid,
- i_tablespace;
- CatalogId dbCatId;
- DumpId dbDumpId;
- const char *datname,
- *dba,
- *encoding,
- *collate,
- *ctype,
- *tablespace;
- uint32 frozenxid, minmxid;
- datname = PQdb(conn);
- if (g_verbose)
- write_msg(NULL, "saving database definition\n");
- /* Make sure we are in proper schema */
- selectSourceSchema(fout, "pg_catalog");
- /* Get the database owner and parameters from pg_database */
- if (fout->remoteVersion >= 90300)
- {
- appendPQExpBuffer(dbQry, "SELECT tableoid, oid, "
- "(%s datdba) AS dba, "
- "pg_encoding_to_char(encoding) AS encoding, "
- "datcollate, datctype, datfrozenxid, datminmxid, "
- "(SELECT spcname FROM pg_tablespace t WHERE t.oid = dattablespace) AS tablespace, "
- "shobj_description(oid, 'pg_database') AS description "
- "FROM pg_database "
- "WHERE datname = ",
- username_subquery);
- appendStringLiteralAH(dbQry, datname, fout);
- }
- else if (fout->remoteVersion >= 80400)
- {
- appendPQExpBuffer(dbQry, "SELECT tableoid, oid, "
- "(%s datdba) AS dba, "
- "pg_encoding_to_char(encoding) AS encoding, "
- "datcollate, datctype, datfrozenxid, 0 AS datminmxid, "
- "(SELECT spcname FROM pg_tablespace t WHERE t.oid = dattablespace) AS tablespace, "
- "shobj_description(oid, 'pg_database') AS description "
- "FROM pg_database "
- "WHERE datname = ",
- username_subquery);
- appendStringLiteralAH(dbQry, datname, fout);
- }
- else if (fout->remoteVersion >= 80200)
- {
- appendPQExpBuffer(dbQry, "SELECT tableoid, oid, "
- "(%s datdba) AS dba, "
- "pg_encoding_to_char(encoding) AS encoding, "
- "NULL AS datcollate, NULL AS datctype, datfrozenxid, 0 AS datminmxid, "
- "(SELECT spcname FROM pg_tablespace t WHERE t.oid = dattablespace) AS tablespace, "
- "shobj_description(oid, 'pg_database') AS description "
- "FROM pg_database "
- "WHERE datname = ",
- username_subquery);
- appendStringLiteralAH(dbQry, datname, fout);
- }
- else if (fout->remoteVersion >= 80000)
- {
- appendPQExpBuffer(dbQry, "SELECT tableoid, oid, "
- "(%s datdba) AS dba, "
- "pg_encoding_to_char(encoding) AS encoding, "
- "NULL AS datcollate, NULL AS datctype, datfrozenxid, 0 AS datminmxid, "
- "(SELECT spcname FROM pg_tablespace t WHERE t.oid = dattablespace) AS tablespace "
- "FROM pg_database "
- "WHERE datname = ",
- username_subquery);
- appendStringLiteralAH(dbQry, datname, fout);
- }
- else if (fout->remoteVersion >= 70100)
- {
- appendPQExpBuffer(dbQry, "SELECT tableoid, oid, "
- "(%s datdba) AS dba, "
- "pg_encoding_to_char(encoding) AS encoding, "
- "NULL AS datcollate, NULL AS datctype, "
- "0 AS datfrozenxid, 0 AS datminmxid, "
- "NULL AS tablespace "
- "FROM pg_database "
- "WHERE datname = ",
- username_subquery);
- appendStringLiteralAH(dbQry, datname, fout);
- }
- else
- {
- appendPQExpBuffer(dbQry, "SELECT "
- "(SELECT oid FROM pg_class WHERE relname = 'pg_database') AS tableoid, "
- "oid, "
- "(%s datdba) AS dba, "
- "pg_encoding_to_char(encoding) AS encoding, "
- "NULL AS datcollate, NULL AS datctype, "
- "0 AS datfrozenxid, 0 AS datminmxid, "
- "NULL AS tablespace "
- "FROM pg_database "
- "WHERE datname = ",
- username_subquery);
- appendStringLiteralAH(dbQry, datname, fout);
- }
- res = ExecuteSqlQueryForSingleRow(fout, dbQry->data);
- i_tableoid = PQfnumber(res, "tableoid");
- i_oid = PQfnumber(res, "oid");
- i_dba = PQfnumber(res, "dba");
- i_encoding = PQfnumber(res, "encoding");
- i_collate = PQfnumber(res, "datcollate");
- i_ctype = PQfnumber(res, "datctype");
- i_frozenxid = PQfnumber(res, "datfrozenxid");
- i_minmxid = PQfnumber(res, "datminmxid");
- i_tablespace = PQfnumber(res, "tablespace");
- dbCatId.tableoid = atooid(PQgetvalue(res, 0, i_tableoid));
- dbCatId.oid = atooid(PQgetvalue(res, 0, i_oid));
- dba = PQgetvalue(res, 0, i_dba);
- encoding = PQgetvalue(res, 0, i_encoding);
- collate = PQgetvalue(res, 0, i_collate);
- ctype = PQgetvalue(res, 0, i_ctype);
- frozenxid = atooid(PQgetvalue(res, 0, i_frozenxid));
- minmxid = atooid(PQgetvalue(res, 0, i_minmxid));
- tablespace = PQgetvalue(res, 0, i_tablespace);
- appendPQExpBuffer(creaQry, "CREATE DATABASE %s WITH TEMPLATE = template0",
- fmtId(datname));
- if (strlen(encoding) > 0)
- {
- appendPQExpBufferStr(creaQry, " ENCODING = ");
- appendStringLiteralAH(creaQry, encoding, fout);
- }
- if (strlen(collate) > 0)
- {
- appendPQExpBufferStr(creaQry, " LC_COLLATE = ");
- appendStringLiteralAH(creaQry, collate, fout);
- }
- if (strlen(ctype) > 0)
- {
- appendPQExpBufferStr(creaQry, " LC_CTYPE = ");
- appendStringLiteralAH(creaQry, ctype, fout);
- }
- if (strlen(tablespace) > 0 && strcmp(tablespace, "pg_default") != 0)
- appendPQExpBuffer(creaQry, " TABLESPACE = %s",
- fmtId(tablespace));
- appendPQExpBufferStr(creaQry, ";\n");
- if (binary_upgrade)
- {
- appendPQExpBufferStr(creaQry, "\n-- For binary upgrade, set datfrozenxid and datminmxid.\n");
- appendPQExpBuffer(creaQry, "UPDATE pg_catalog.pg_database\n"
- "SET datfrozenxid = '%u', datminmxid = '%u'\n"
- "WHERE datname = ",
- frozenxid, minmxid);
- appendStringLiteralAH(creaQry, datname, fout);
- appendPQExpBufferStr(creaQry, ";\n");
- }
- appendPQExpBuffer(delQry, "DROP DATABASE %s;\n",
- fmtId(datname));
- dbDumpId = createDumpId();
- ArchiveEntry(fout,
- dbCatId, /* catalog ID */
- dbDumpId, /* dump ID */
- datname, /* Name */
- NULL, /* Namespace */
- NULL, /* Tablespace */
- dba, /* Owner */
- false, /* with oids */
- "DATABASE", /* Desc */
- SECTION_PRE_DATA, /* Section */
- creaQry->data, /* Create */
- delQry->data, /* Del */
- NULL, /* Copy */
- NULL, /* Deps */
- 0, /* # Deps */
- NULL, /* Dumper */
- NULL); /* Dumper Arg */
- /*
- * pg_largeobject and pg_largeobject_metadata come from the old system
- * intact, so set their relfrozenxids and relminmxids.
- */
- if (binary_upgrade)
- {
- PGresult *lo_res;
- PQExpBuffer loFrozenQry = createPQExpBuffer();
- PQExpBuffer loOutQry = createPQExpBuffer();
- int i_relfrozenxid, i_relminmxid;
- /*
- * pg_largeobject
- */
- if (fout->remoteVersion >= 90300)
- appendPQExpBuffer(loFrozenQry, "SELECT relfrozenxid, relminmxid\n"
- "FROM pg_catalog.pg_class\n"
- "WHERE oid = %u;\n",
- LargeObjectRelationId);
- else
- appendPQExpBuffer(loFrozenQry, "SELECT relfrozenxid, 0 AS relminmxid\n"
- "FROM pg_catalog.pg_class\n"
- "WHERE oid = %u;\n",
- LargeObjectRelationId);
- lo_res = ExecuteSqlQueryForSingleRow(fout, loFrozenQry->data);
- i_relfrozenxid = PQfnumber(lo_res, "relfrozenxid");
- i_relminmxid = PQfnumber(lo_res, "relminmxid");
- appendPQExpBufferStr(loOutQry, "\n-- For binary upgrade, set pg_largeobject relfrozenxid and relminmxid\n");
- appendPQExpBuffer(loOutQry, "UPDATE pg_catalog.pg_class\n"
- "SET relfrozenxid = '%u', relminmxid = '%u'\n"
- "WHERE oid = %u;\n",
- atoi(PQgetvalue(lo_res, 0, i_relfrozenxid)),
- atoi(PQgetvalue(lo_res, 0, i_relminmxid)),
- LargeObjectRelationId);
- ArchiveEntry(fout, nilCatalogId, createDumpId(),
- "pg_largeobject", NULL, NULL, "",
- false, "pg_largeobject", SECTION_PRE_DATA,
- loOutQry->data, "", NULL,
- NULL, 0,
- NULL, NULL);
- PQclear(lo_res);
- /*
- * pg_largeobject_metadata
- */
- if (fout->remoteVersion >= 90000)
- {
- resetPQExpBuffer(loFrozenQry);
- resetPQExpBuffer(loOutQry);
- if (fout->remoteVersion >= 90300)
- appendPQExpBuffer(loFrozenQry, "SELECT relfrozenxid, relminmxid\n"
- "FROM pg_catalog.pg_class\n"
- "WHERE oid = %u;\n",
- LargeObjectMetadataRelationId);
- else
- appendPQExpBuffer(loFrozenQry, "SELECT relfrozenxid, 0 AS relminmxid\n"
- "FROM pg_catalog.pg_class\n"
- "WHERE oid = %u;\n",
- LargeObjectMetadataRelationId);
- lo_res = ExecuteSqlQueryForSingleRow(fout, loFrozenQry->data);
- i_relfrozenxid = PQfnumber(lo_res, "relfrozenxid");
- i_relminmxid = PQfnumber(lo_res, "relminmxid");
- appendPQExpBufferStr(loOutQry, "\n-- For binary upgrade, set pg_largeobject_metadata relfrozenxid and relminmxid\n");
- appendPQExpBuffer(loOutQry, "UPDATE pg_catalog.pg_class\n"
- "SET relfrozenxid = '%u', relminmxid = '%u'\n"
- "WHERE oid = %u;\n",
- atoi(PQgetvalue(lo_res, 0, i_relfrozenxid)),
- atoi(PQgetvalue(lo_res, 0, i_relminmxid)),
- LargeObjectMetadataRelationId);
- ArchiveEntry(fout, nilCatalogId, createDumpId(),
- "pg_largeobject_metadata", NULL, NULL, "",
- false, "pg_largeobject_metadata", SECTION_PRE_DATA,
- loOutQry->data, "", NULL,
- NULL, 0,
- NULL, NULL);
- PQclear(lo_res);
- }
- destroyPQExpBuffer(loFrozenQry);
- destroyPQExpBuffer(loOutQry);
- }
- /* Dump DB comment if any */
- if (fout->remoteVersion >= 80200)
- {
- /*
- * 8.2 keeps comments on shared objects in a shared table, so we
- * cannot use the dumpComment used for other database objects.
- */
- char *comment = PQgetvalue(res, 0, PQfnumber(res, "description"));
- if (comment && strlen(comment))
- {
- resetPQExpBuffer(dbQry);
- /*
- * Generates warning when loaded into a differently-named
- * database.
- */
- appendPQExpBuffer(dbQry, "COMMENT ON DATABASE %s IS ", fmtId(datname));
- appendStringLiteralAH(dbQry, comment, fout);
- appendPQExpBufferStr(dbQry, ";\n");
- ArchiveEntry(fout, dbCatId, createDumpId(), datname, NULL, NULL,
- dba, false, "COMMENT", SECTION_NONE,
- dbQry->data, "", NULL,
- &dbDumpId, 1, NULL, NULL);
- }
- }
- else
- {
- resetPQExpBuffer(dbQry);
- appendPQExpBuffer(dbQry, "DATABASE %s", fmtId(datname));
- dumpComment(fout, dbQry->data, NULL, "",
- dbCatId, 0, dbDumpId);
- }
- PQclear(res);
- /* Dump shared security label. */
- if (!no_security_labels && fout->remoteVersion >= 90200)
- {
- PQExpBuffer seclabelQry = createPQExpBuffer();
- buildShSecLabelQuery(conn, "pg_database", dbCatId.oid, seclabelQry);
- res = ExecuteSqlQuery(fout, seclabelQry->data, PGRES_TUPLES_OK);
- resetPQExpBuffer(seclabelQry);
- emitShSecLabels(conn, res, seclabelQry, "DATABASE", datname);
- if (strlen(seclabelQry->data))
- ArchiveEntry(fout, dbCatId, createDumpId(), datname, NULL, NULL,
- dba, false, "SECURITY LABEL", SECTION_NONE,
- seclabelQry->data, "", NULL,
- &dbDumpId, 1, NULL, NULL);
- destroyPQExpBuffer(seclabelQry);
- }
- destroyPQExpBuffer(dbQry);
- destroyPQExpBuffer(delQry);
- destroyPQExpBuffer(creaQry);
- }
- /*
- * dumpEncoding: put the correct encoding into the archive
- */
- static void
- dumpEncoding(Archive *AH)
- {
- const char *encname = pg_encoding_to_char(AH->encoding);
- PQExpBuffer qry = createPQExpBuffer();
- if (g_verbose)
- write_msg(NULL, "saving encoding = %s\n", encname);
- appendPQExpBufferStr(qry, "SET client_encoding = ");
- appendStringLiteralAH(qry, encname, AH);
- appendPQExpBufferStr(qry, ";\n");
- ArchiveEntry(AH, nilCatalogId, createDumpId(),
- "ENCODING", NULL, NULL, "",
- false, "ENCODING", SECTION_PRE_DATA,
- qry->data, "", NULL,
- NULL, 0,
- NULL, NULL);
- destroyPQExpBuffer(qry);
- }
- /*
- * dumpStdStrings: put the correct escape string behavior into the archive
- */
- static void
- dumpStdStrings(Archive *AH)
- {
- const char *stdstrings = AH->std_strings ? "on" : "off";
- PQExpBuffer qry = createPQExpBuffer();
- if (g_verbose)
- write_msg(NULL, "saving standard_conforming_strings = %s\n",
- stdstrings);
- appendPQExpBuffer(qry, "SET standard_conforming_strings = '%s';\n",
- stdstrings);
- ArchiveEntry(AH, nilCatalogId, createDumpId(),
- "STDSTRINGS", NULL, NULL, "",
- false, "STDSTRINGS", SECTION_PRE_DATA,
- qry->data, "", NULL,
- NULL, 0,
- NULL, NULL);
- destroyPQExpBuffer(qry);
- }
- /*
- * getBlobs:
- * Collect schema-level data about large objects
- */
- static void
- getBlobs(Archive *fout)
- {
- PQExpBuffer blobQry = createPQExpBuffer();
- BlobInfo *binfo;
- DumpableObject *bdata;
- PGresult *res;
- int ntups;
- int i;
- /* Verbose message */
- if (g_verbose)
- write_msg(NULL, "reading large objects\n");
- /* Make sure we are in proper schema */
- selectSourceSchema(fout, "pg_catalog");
- /* Fetch BLOB OIDs, and owner/ACL data if >= 9.0 */
- if (fout->remoteVersion >= 90000)
- appendPQExpBuffer(blobQry,
- "SELECT oid, (%s lomowner) AS rolname, lomacl"
- " FROM pg_largeobject_metadata",
- username_subquery);
- else if (fout->remoteVersion >= 70100)
- appendPQExpBufferStr(blobQry,
- "SELECT DISTINCT loid, NULL::oid, NULL::oid"
- " FROM pg_largeobject");
- else
- appendPQExpBufferStr(blobQry,
- "SELECT oid, NULL::oid, NULL::oid"
- " FROM pg_class WHERE relkind = 'l'");
- res = ExecuteSqlQuery(fout, blobQry->data, PGRES_TUPLES_OK);
- ntups = PQntuples(res);
- if (ntups > 0)
- {
- /*
- * Each large object has its own BLOB archive entry.
- */
- binfo = (BlobInfo *) pg_malloc(ntups * sizeof(BlobInfo));
- for (i = 0; i < ntups; i++)
- {
- binfo[i].dobj.objType = DO_BLOB;
- binfo[i].dobj.catId.tableoid = LargeObjectRelationId;
- binfo[i].dobj.catId.oid = atooid(PQgetvalue(res, i, 0));
- AssignDumpId(&binfo[i].dobj);
- binfo[i].dobj.name = pg_strdup(PQgetvalue(res, i, 0));
- if (!PQgetisnull(res, i, 1))
- binfo[i].rolname = pg_strdup(PQgetvalue(res, i, 1));
- else
- binfo[i].rolname = "";
- if (!PQgetisnull(res, i, 2))
- binfo[i].blobacl = pg_strdup(PQgetvalue(res, i, 2));
- else
- binfo[i].blobacl = NULL;
- }
- /*
- * If we have any large objects, a "BLOBS" archive entry is needed.
- * This is just a placeholder for sorting; it carries no data now.
- */
- bdata = (DumpableObject *) pg_malloc(sizeof(DumpableObject));
- bdata->objType = DO_BLOB_DATA;
- bdata->catId = nilCatalogId;
- AssignDumpId(bdata);
- bdata->name = pg_strdup("BLOBS");
- }
- PQclear(res);
- destroyPQExpBuffer(blobQry);
- }
- /*
- * dumpBlob
- *
- * dump the definition (metadata) of the given large object
- */
- static void
- dumpBlob(Archive *fout, BlobInfo *binfo)
- {
- PQExpBuffer cquery = createPQExpBuffer();
- PQExpBuffer dquery = createPQExpBuffer();
- appendPQExpBuffer(cquery,
- "SELECT pg_catalog.lo_create('%s');\n",
- binfo->dobj.name);
- appendPQExpBuffer(dquery,
- "SELECT pg_catalog.lo_unlink('%s');\n",
- binfo->dobj.name);
- ArchiveEntry(fout, binfo->dobj.catId, binfo->dobj.dumpId,
- binfo->dobj.name,
- NULL, NULL,
- binfo->rolname, false,
- "BLOB", SECTION_PRE_DATA,
- cquery->data, dquery->data, NULL,
- NULL, 0,
- NULL, NULL);
- /* set up tag for comment and/or ACL */
- resetPQExpBuffer(cquery);
- appendPQExpBuffer(cquery, "LARGE OBJECT %s", binfo->dobj.name);
- /* Dump comment if any */
- dumpComment(fout, cquery->data,
- NULL, binfo->rolname,
- binfo->dobj.catId, 0, binfo->dobj.dumpId);
- /* Dump security label if any */
- dumpSecLabel(fout, cquery->data,
- NULL, binfo->rolname,
- binfo->dobj.catId, 0, binfo->dobj.dumpId);
- /* Dump ACL if any */
- if (binfo->blobacl)
- dumpACL(fout, binfo->dobj.catId, binfo->dobj.dumpId, "LARGE OBJECT",
- binfo->dobj.name, NULL, cquery->data,
- NULL, binfo->rolname, binfo->blobacl);
- destroyPQExpBuffer(cquery);
- destroyPQExpBuffer(dquery);
- }
- /*
- * dumpBlobs:
- * dump the data contents of all large objects
- */
- static int
- dumpBlobs(Archive *fout, void *arg)
- {
- const char *blobQry;
- const char *blobFetchQry;
- PGconn *conn = GetConnection(fout);
- PGresult *res;
- char buf[LOBBUFSIZE];
- int ntups;
- int i;
- int cnt;
- if (g_verbose)
- write_msg(NULL, "saving large objects\n");
- /* Make sure we are in proper schema */
- selectSourceSchema(fout, "pg_catalog");
- /*
- * Currently, we re-fetch all BLOB OIDs using a cursor. Consider scanning
- * the already-in-memory dumpable objects instead...
- */
- if (fout->remoteVersion >= 90000)
- blobQry = "DECLARE bloboid CURSOR FOR SELECT oid FROM pg_largeobject_metadata";
- else if (fout->remoteVersion >= 70100)
- blobQry = "DECLARE bloboid CURSOR FOR SELECT DISTINCT loid FROM pg_largeobject";
- else
- blobQry = "DECLARE bloboid CURSOR FOR SELECT oid FROM pg_class WHERE relkind = 'l'";
- ExecuteSqlStatement(fout, blobQry);
- /* Command to fetch from cursor */
- blobFetchQry = "FETCH 1000 IN bloboid";
- do
- {
- /* Do a fetch */
- res = ExecuteSqlQuery(fout, blobFetchQry, PGRES_TUPLES_OK);
- /* Process the tuples, if any */
- ntups = PQntuples(res);
- for (i = 0; i < ntups; i++)
- {
- Oid blobOid;
- int loFd;
- blobOid = atooid(PQgetvalue(res, i, 0));
- /* Open the BLOB */
- loFd = lo_open(conn, blobOid, INV_READ);
- if (loFd == -1)
- exit_horribly(NULL, "could not open large object %u: %s",
- blobOid, PQerrorMessage(conn));
- StartBlob(fout, blobOid);
- /* Now read it in chunks, sending data to archive */
- do
- {
- cnt = lo_read(conn, loFd, buf, LOBBUFSIZE);
- if (cnt < 0)
- exit_horribly(NULL, "error reading large object %u: %s",
- blobOid, PQerrorMessage(conn));
- WriteData(fout, buf, cnt);
- } while (cnt > 0);
- lo_close(conn, loFd);
- EndBlob(fout, blobOid);
- }
- PQclear(res);
- } while (ntups > 0);
- return 1;
- }
- static void
- binary_upgrade_set_type_oids_by_type_oid(Archive *fout,
- PQExpBuffer upgrade_buffer,
- Oid pg_type_oid)
- {
- PQExpBuffer upgrade_query = createPQExpBuffer();
- PGresult *upgrade_res;
- Oid pg_type_array_oid;
- appendPQExpBufferStr(upgrade_buffer, "\n-- For binary upgrade, must preserve pg_type oid\n");
- appendPQExpBuffer(upgrade_buffer,
- "SELECT binary_upgrade.set_next_pg_type_oid('%u'::pg_catalog.oid);\n\n",
- pg_type_oid);
- /* we only support old >= 8.3 for binary upgrades */
- appendPQExpBuffer(upgrade_query,
- "SELECT typarray "
- "FROM pg_catalog.pg_type "
- "WHERE pg_type.oid = '%u'::pg_catalog.oid;",
- pg_type_oid);
- upgrade_res = ExecuteSqlQueryForSingleRow(fout, upgrade_query->data);
- pg_type_array_oid = atooid(PQgetvalue(upgrade_res, 0, PQfnumber(upgrade_res, "typarray")));
- if (OidIsValid(pg_type_array_oid))
- {
- appendPQExpBufferStr(upgrade_buffer,
- "\n-- For binary upgrade, must preserve pg_type array oid\n");
- appendPQExpBuffer(upgrade_buffer,
- "SELECT binary_upgrade.set_next_array_pg_type_oid('%u'::pg_catalog.oid);\n\n",
- pg_type_array_oid);
- }
- PQclear(upgrade_res);
- destroyPQExpBuffer(upgrade_query);
- }
- static bool
- binary_upgrade_set_type_oids_by_rel_oid(Archive *fout,
- PQExpBuffer upgrade_buffer,
- Oid pg_rel_oid)
- {
- PQExpBuffer upgrade_query = createPQExpBuffer();
- PGresult *upgrade_res;
- Oid pg_type_oid;
- bool toast_set = false;
- /* we only support old >= 8.3 for binary upgrades */
- appendPQExpBuffer(upgrade_query,
- "SELECT c.reltype AS crel, t.reltype AS trel "
- "FROM pg_catalog.pg_class c "
- "LEFT JOIN pg_catalog.pg_class t ON "
- " (c.reltoastrelid = t.oid) "
- "WHERE c.oid = '%u'::pg_catalog.oid;",
- pg_rel_oid);
- upgrade_res = ExecuteSqlQueryForSingleRow(fout, upgrade_query->data);
- pg_type_oid = atooid(PQgetvalue(upgrade_res, 0, PQfnumber(upgrade_res, "crel")));
- binary_upgrade_set_type_oids_by_type_oid(fout, upgrade_buffer,
- pg_type_oid);
- if (!PQgetisnull(upgrade_res, 0, PQfnumber(upgrade_res, "trel")))
- {
- /* Toast tables do not have pg_type array rows */
- Oid pg_type_toast_oid = atooid(PQgetvalue(upgrade_res, 0,
- PQfnumber(upgrade_res, "trel")));
- appendPQExpBufferStr(upgrade_buffer, "\n-- For binary upgrade, must preserve pg_type toast oid\n");
- appendPQExpBuffer(upgrade_buffer,
- "SELECT binary_upgrade.set_next_toast_pg_type_oid('%u'::pg_catalog.oid);\n\n",
- pg_type_toast_oid);
- toast_set = true;
- }
- PQclear(upgrade_res);
- destroyPQExpBuffer(upgrade_query);
- return toast_set;
- }
- static void
- binary_upgrade_set_pg_class_oids(Archive *fout,
- PQExpBuffer upgrade_buffer, Oid pg_class_oid,
- bool is_index)
- {
- PQExpBuffer upgrade_query = createPQExpBuffer();
- PGresult *upgrade_res;
- Oid pg_class_reltoastrelid;
- Oid pg_index_indexrelid;
- appendPQExpBuffer(upgrade_query,
- "SELECT c.reltoastrelid, i.indexrelid "
- "FROM pg_catalog.pg_class c LEFT JOIN "
- "pg_catalog.pg_index i ON (c.reltoastrelid = i.indrelid AND i.indisvalid) "
- "WHERE c.oid = '%u'::pg_catalog.oid;",
- pg_class_oid);
- upgrade_res = ExecuteSqlQueryForSingleRow(fout, upgrade_query->data);
- pg_class_reltoastrelid = atooid(PQgetvalue(upgrade_res, 0, PQfnumber(upgrade_res, "reltoastrelid")));
- pg_index_indexrelid = atooid(PQgetvalue(upgrade_res, 0, PQfnumber(upgrade_res, "indexrelid")));
- appendPQExpBufferStr(upgrade_buffer,
- "\n-- For binary upgrade, must preserve pg_class oids\n");
- if (!is_index)
- {
- appendPQExpBuffer(upgrade_buffer,
- "SELECT binary_upgrade.set_next_heap_pg_class_oid('%u'::pg_catalog.oid);\n",
- pg_class_oid);
- /* only tables have toast tables, not indexes */
- if (OidIsValid(pg_class_reltoastrelid))
- {
- /*
- * One complexity is that the table definition might not require
- * the creation of a TOAST table, and the TOAST table might have
- * been created long after table creation, when the table was
- * loaded with wide data. By setting the TOAST oid we force
- * creation of the TOAST heap and TOAST index by the backend so we
- * can cleanly copy the files during binary upgrade.
- */
- appendPQExpBuffer(upgrade_buffer,
- "SELECT binary_upgrade.set_next_toast_pg_class_oid('%u'::pg_catalog.oid);\n",
- pg_class_reltoastrelid);
- /* every toast table has an index */
- appendPQExpBuffer(upgrade_buffer,
- "SELECT binary_upgrade.set_next_index_pg_class_oid('%u'::pg_catalog.oid);\n",
- pg_index_indexrelid);
- }
- }
- else
- appendPQExpBuffer(upgrade_buffer,
- "SELECT binary_upgrade.set_next_index_pg_class_oid('%u'::pg_catalog.oid);\n",
- pg_class_oid);
- appendPQExpBufferChar(upgrade_buffer, '\n');
- PQclear(upgrade_res);
- destroyPQExpBuffer(upgrade_query);
- }
- /*
- * If the DumpableObject is a member of an extension, add a suitable
- * ALTER EXTENSION ADD command to the creation commands in upgrade_buffer.
- */
- static void
- binary_upgrade_extension_member(PQExpBuffer upgrade_buffer,
- DumpableObject *dobj,
- const char *objlabel)
- {
- DumpableObject *extobj = NULL;
- int i;
- if (!dobj->ext_member)
- return;
- /*
- * Find the parent extension. We could avoid this search if we wanted to
- * add a link field to DumpableObject, but the space costs of that would
- * be considerable. We assume that member objects could only have a
- * direct dependency on their own extension, not any others.
- */
- for (i = 0; i < dobj->nDeps; i++)
- {
- extobj = findObjectByDumpId(dobj->dependencies[i]);
- if (extobj && extobj->objType == DO_EXTENSION)
- break;
- extobj = NULL;
- }
- if (extobj == NULL)
- exit_horribly(NULL, "could not find parent extension for %s\n", objlabel);
- appendPQExpBufferStr(upgrade_buffer,
- "\n-- For binary upgrade, handle extension membership the hard way\n");
- appendPQExpBuffer(upgrade_buffer, "ALTER EXTENSION %s ADD %s;\n",
- fmtId(extobj->name),
- objlabel);
- }
- /*
- * getNamespaces:
- * read all namespaces in the system catalogs and return them in the
- * NamespaceInfo* structure
- *
- * numNamespaces is set to the number of namespaces read in
- */
- NamespaceInfo *
- getNamespaces(Archive *fout, int *numNamespaces)
- {
- PGresult *res;
- int ntups;
- int i;
- PQExpBuffer query;
- NamespaceInfo *nsinfo;
- int i_tableoid;
- int i_oid;
- int i_nspname;
- int i_rolname;
- int i_nspacl;
- /*
- * Before 7.3, there are no real namespaces; create two dummy entries, one
- * for user stuff and one for system stuff.
- */
- if (fout->remoteVersion < 70300)
- {
- nsinfo = (NamespaceInfo *) pg_malloc(2 * sizeof(NamespaceInfo));
- nsinfo[0].dobj.objType = DO_NAMESPACE;
- nsinfo[0].dobj.catId.tableoid = 0;
- nsinfo[0].dobj.catId.oid = 0;
- AssignDumpId(&nsinfo[0].dobj);
- nsinfo[0].dobj.name = pg_strdup("public");
- nsinfo[0].rolname = pg_strdup("");
- nsinfo[0].nspacl = pg_strdup("");
- selectDumpableNamespace(&nsinfo[0]);
- nsinfo[1].dobj.objType = DO_NAMESPACE;
- nsinfo[1].dobj.catId.tableoid = 0;
- nsinfo[1].dobj.catId.oid = 1;
- AssignDumpId(&nsinfo[1].dobj);
- nsinfo[1].dobj.name = pg_strdup("pg_catalog");
- nsinfo[1].rolname = pg_strdup("");
- nsinfo[1].nspacl = pg_strdup("");
- selectDumpableNamespace(&nsinfo[1]);
- *numNamespaces = 2;
- return nsinfo;
- }
- query = createPQExpBuffer();
- /* Make sure we are in proper schema */
- selectSourceSchema(fout, "pg_catalog");
- /*
- * we fetch all namespaces including system ones, so that every object we
- * read in can be linked to a containing namespace.
- */
- appendPQExpBuffer(query, "SELECT tableoid, oid, nspname, "
- "(%s nspowner) AS rolname, "
- "nspacl FROM pg_namespace",
- username_subquery);
- res = ExecuteSqlQuery(fout, query->data, PGRES_TUPLES_OK);
- ntups = PQntuples(res);
- nsinfo = (NamespaceInfo *) pg_malloc(ntups * sizeof(NamespaceInfo));
- i_tableoid = PQfnumber(res, "tableoid");
- i_oid = PQfnumber(res, "oid");
- i_nspname = PQfnumber(res, "nspname");
- i_rolname = PQfnumber(res, "rolname");
- i_nspacl = PQfnumber(res, "nspacl");
- for (i = 0; i < ntups; i++)
- {
- nsinfo[i].dobj.objType = DO_NAMESPACE;
- nsinfo[i].dobj.catId.tableoid = atooid(PQgetvalue(res, i, i_tableoid));
- nsinfo[i].dobj.catId.oid = atooid(PQgetvalue(res, i, i_oid));
- AssignDumpId(&nsinfo[i].dobj);
- nsinfo[i].dobj.name = pg_strdup(PQgetvalue(res, i, i_nspname));
- nsinfo[i].rolname = pg_strdup(PQgetvalue(res, i, i_rolname));
- nsinfo[i].nspacl = pg_strdup(PQgetvalue(res, i, i_nspacl));
- /* Decide whether to dump this namespace */
- selectDumpableNamespace(&nsinfo[i]);
- if (strlen(nsinfo[i].rolname) == 0)
- write_msg(NULL, "WARNING: owner of schema \"%s\" appears to be invalid\n",
- nsinfo[i].dobj.name);
- }
- PQclear(res);
- destroyPQExpBuffer(query);
- *numNamespaces = ntups;
- return nsinfo;
- }
- /*
- * findNamespace:
- * given a namespace OID and an object OID, look up the info read by
- * getNamespaces
- *
- * NB: for pre-7.3 source database, we use object OID to guess whether it's
- * a system object or not. In 7.3 and later there is no guessing, and we
- * don't use objoid at all.
- */
- static NamespaceInfo *
- findNamespace(Archive *fout, Oid nsoid, Oid objoid)
- {
- NamespaceInfo *nsinfo;
- if (fout->remoteVersion >= 70300)
- {
- nsinfo = findNamespaceByOid(nsoid);
- }
- else
- {
- /* This code depends on the dummy objects set up by getNamespaces. */
- Oid i;
- if (objoid > g_last_builtin_oid)
- i = 0; /* user object */
- else
- i = 1; /* system object */
- nsinfo = findNamespaceByOid(i);
- }
- if (nsinfo == NULL)
- exit_horribly(NULL, "schema with OID %u does not exist\n", nsoid);
- return nsinfo;
- }
- /*
- * getExtensions:
- * read all extensions in the system catalogs and return them in the
- * ExtensionInfo* structure
- *
- * numExtensions is set to the number of extensions read in
- */
- ExtensionInfo *
- getExtensions(Archive *fout, int *numExtensions)
- {
- PGresult *res;
- int ntups;
- int i;
- PQExpBuffer query;
- ExtensionInfo *extinfo;
- int i_tableoid;
- int i_oid;
- int i_extname;
- int i_nspname;
- int i_extrelocatable;
- int i_extversion;
- int i_extconfig;
- int i_extcondition;
- /*
- * Before 9.1, there are no extensions.
- */
- if (fout->remoteVersion < 90100)
- {
- *numExtensions = 0;
- return NULL;
- }
- query = createPQExpBuffer();
- /* Make sure we are in proper schema */
- selectSourceSchema(fout, "pg_catalog");
- appendPQExpBufferStr(query, "SELECT x.tableoid, x.oid, "
- "x.extname, n.nspname, x.extrelocatable, x.extversion, x.extconfig, x.extcondition "
- "FROM pg_extension x "
- "JOIN pg_namespace n ON n.oid = x.extnamespace");
- res = ExecuteSqlQuery(fout, query->data, PGRES_TUPLES_OK);
- ntups = PQntuples(res);
- extinfo = (ExtensionInfo *) pg_malloc(ntups * sizeof(ExtensionInfo));
- i_tableoid = PQfnumber(res, "tableoid");
- i_oid = PQfnumber(res, "oid");
- i_extname = PQfnumber(res, "extname");
- i_nspname = PQfnumber(res, "nspname");
- i_extrelocatable = PQfnumber(res, "extrelocatable");
- i_extversion = PQfnumber(res, "extversion");
- i_extconfig = PQfnumber(res, "extconfig");
- i_extcondition = PQfnumber(res, "extcondition");
- for (i = 0; i < ntups; i++)
- {
- extinfo[i].dobj.objType = DO_EXTENSION;
- extinfo[i].dobj.catId.tableoid = atooid(PQgetvalue(res, i, i_tableoid));
- extinfo[i].dobj.catId.oid = atooid(PQgetvalue(res, i, i_oid));
- AssignDumpId(&extinfo[i].dobj);
- extinfo[i].dobj.name = pg_strdup(PQgetvalue(res, i, i_extname));
- extinfo[i].namespace = pg_strdup(PQgetvalue(res, i, i_nspname));
- extinfo[i].relocatable = *(PQgetvalue(res, i, i_extrelocatable)) == 't';
- extinfo[i].extversion = pg_strdup(PQgetvalue(res, i, i_extversion));
- extinfo[i].extconfig = pg_strdup(PQgetvalue(res, i, i_extconfig));
- extinfo[i].extcondition = pg_strdup(PQgetvalue(res, i, i_extcondition));
- /* Decide whether we want to dump it */
- selectDumpableExtension(&(extinfo[i]));
- }
- PQclear(res);
- destroyPQExpBuffer(query);
- *numExtensions = ntups;
- return extinfo;
- }
- /*
- * getTypes:
- * read all types in the system catalogs and return them in the
- * TypeInfo* structure
- *
- * numTypes is set to the number of types read in
- *
- * NB: this must run after getFuncs() because we assume we can do
- * findFuncByOid().
- */
- TypeInfo *
- getTypes(Archive *fout, int *numTypes)
- {
- PGresult *res;
- int ntups;
- int i;
- PQExpBuffer query = createPQExpBuffer();
- TypeInfo *tyinfo;
- ShellTypeInfo *stinfo;
- int i_tableoid;
- int i_oid;
- int i_typname;
- int i_typnamespace;
- int i_typacl;
- int i_rolname;
- int i_typinput;
- int i_typoutput;
- int i_typelem;
- int i_typrelid;
- int i_typrelkind;
- int i_typtype;
- int i_typisdefined;
- int i_isarray;
- /*
- * we include even the built-in types because those may be used as array
- * elements by user-defined types
- *
- * we filter out the built-in types when we dump out the types
- *
- * same approach for undefined (shell) types and array types
- *
- * Note: as of 8.3 we can reliably detect whether a type is an
- * auto-generated array type by checking the element type's typarray.
- * (Before that the test is capable of generating false positives.) We
- * still check for name beginning with '_', though, so as to avoid the
- * cost of the subselect probe for all standard types. This would have to
- * be revisited if the backend ever allows renaming of array types.
- */
- /* Make sure we are in proper schema */
- selectSourceSchema(fout, "pg_catalog");
- if (fout->remoteVersion >= 90200)
- {
- appendPQExpBuffer(query, "SELECT tableoid, oid, typname, "
- "typnamespace, typacl, "
- "(%s typowner) AS rolname, "
- "typinput::oid AS typinput, "
- "typoutput::oid AS typoutput, typelem, typrelid, "
- "CASE WHEN typrelid = 0 THEN ' '::\"char\" "
- "ELSE (SELECT relkind FROM pg_class WHERE oid = typrelid) END AS typrelkind, "
- "typtype, typisdefined, "
- "typname[0] = '_' AND typelem != 0 AND "
- "(SELECT typarray FROM pg_type te WHERE oid = pg_type.typelem) = oid AS isarray "
- "FROM pg_type",
- username_subquery);
- }
- else if (fout->remoteVersion >= 80300)
- {
- appendPQExpBuffer(query, "SELECT tableoid, oid, typname, "
- "typnamespace, '{=U}' AS typacl, "
- "(%s typowner) AS rolname, "
- "typinput::oid AS typinput, "
- "typoutput::oid AS typoutput, typelem, typrelid, "
- "CASE WHEN typrelid = 0 THEN ' '::\"char\" "
- "ELSE (SELECT relkind FROM pg_class WHERE oid = typrelid) END AS typrelkind, "
- "typtype, typisdefined, "
- "typname[0] = '_' AND typelem != 0 AND "
- "(SELECT typarray FROM pg_type te WHERE oid = pg_type.typelem) = oid AS isarray "
- "FROM pg_type",
- username_subquery);
- }
- else if (fout->remoteVersion >= 70300)
- {
- appendPQExpBuffer(query, "SELECT tableoid, oid, typname, "
- "typnamespace, '{=U}' AS typacl, "
- "(%s typowner) AS rolname, "
- "typinput::oid AS typinput, "
- "typoutput::oid AS typoutput, typelem, typrelid, "
- "CASE WHEN typrelid = 0 THEN ' '::\"char\" "
- "ELSE (SELECT relkind FROM pg_class WHERE oid = typrelid) END AS typrelkind, "
- "typtype, typisdefined, "
- "typname[0] = '_' AND typelem != 0 AS isarray "
- "FROM pg_type",
- username_subquery);
- }
- else if (fout->remoteVersion >= 70100)
- {
- appendPQExpBuffer(query, "SELECT tableoid, oid, typname, "
- "0::oid AS typnamespace, '{=U}' AS typacl, "
- "(%s typowner) AS rolname, "
- "typinput::oid AS typinput, "
- "typoutput::oid AS typoutput, typelem, typrelid, "
- "CASE WHEN typrelid = 0 THEN ' '::\"char\" "
- "ELSE (SELECT relkind FROM pg_class WHERE oid = typrelid) END AS typrelkind, "
- "typtype, typisdefined, "
- "typname[0] = '_' AND typelem != 0 AS isarray "
- "FROM pg_type",
- username_subquery);
- }
- else
- {
- appendPQExpBuffer(query, "SELECT "
- "(SELECT oid FROM pg_class WHERE relname = 'pg_type') AS tableoid, "
- "oid, typname, "
- "0::oid AS typnamespace, '{=U}' AS typacl, "
- "(%s typowner) AS rolname, "
- "typinput::oid AS typinput, "
- "typoutput::oid AS typoutput, typelem, typrelid, "
- "CASE WHEN typrelid = 0 THEN ' '::\"char\" "
- "ELSE (SELECT relkind FROM pg_class WHERE oid = typrelid) END AS typrelkind, "
- "typtype, typisdefined, "
- "typname[0] = '_' AND typelem != 0 AS isarray "
- "FROM pg_type",
- username_subquery);
- }
- res = ExecuteSqlQuery(fout, query->data, PGRES_TUPLES_OK);
- ntups = PQntuples(res);
- tyinfo = (TypeInfo *) pg_malloc(ntups * sizeof(TypeInfo));
- i_tableoid = PQfnumber(res, "tableoid");
- i_oid = PQfnumber(res, "oid");
- i_typname = PQfnumber(res, "typname");
- i_typnamespace = PQfnumber(res, "typnamespace");
- i_typacl = PQfnumber(res, "typacl");
- i_rolname = PQfnumber(res, "rolname");
- i_typinput = PQfnumber(res, "typinput");
- i_typoutput = PQfnumber(res, "typoutput");
- i_typelem = PQfnumber(res, "typelem");
- i_typrelid = PQfnumber(res, "typrelid");
- i_typrelkind = PQfnumber(res, "typrelkind");
- i_typtype = PQfnumber(res, "typtype");
- i_typisdefined = PQfnumber(res, "typisdefined");
- i_isarray = PQfnumber(res, "isarray");
- for (i = 0; i < ntups; i++)
- {
- tyinfo[i].dobj.objType = DO_TYPE;
- tyinfo[i].dobj.catId.tableoid = atooid(PQgetvalue(res, i, i_tableoid));
- tyinfo[i].dobj.catId.oid = atooid(PQgetvalue(res, i, i_oid));
- AssignDumpId(&tyinfo[i].dobj);
- tyinfo[i].dobj.name = pg_strdup(PQgetvalue(res, i, i_typname));
- tyinfo[i].dobj.namespace =
- findNamespace(fout,
- atooid(PQgetvalue(res, i, i_typnamespace)),
- tyinfo[i].dobj.catId.oid);
- tyinfo[i].rolname = pg_strdup(PQgetvalue(res, i, i_rolname));
- tyinfo[i].typacl = pg_strdup(PQgetvalue(res, i, i_typacl));
- tyinfo[i].typelem = atooid(PQgetvalue(res, i, i_typelem));
- tyinfo[i].typrelid = atooid(PQgetvalue(res, i, i_typrelid));
- tyinfo[i].typrelkind = *PQgetvalue(res, i, i_typrelkind);
- tyinfo[i].typtype = *PQgetvalue(res, i, i_typtype);
- tyinfo[i].shellType = NULL;
- if (strcmp(PQgetvalue(res, i, i_typisdefined), "t") == 0)
- tyinfo[i].isDefined = true;
- else
- tyinfo[i].isDefined = false;
- if (strcmp(PQgetvalue(res, i, i_isarray), "t") == 0)
- tyinfo[i].isArray = true;
- else
- tyinfo[i].isArray = false;
- /* Decide whether we want to dump it */
- selectDumpableType(&tyinfo[i]);
- /*
- * If it's a domain, fetch info about its constraints, if any
- */
- tyinfo[i].nDomChecks = 0;
- tyinfo[i].domChecks = NULL;
- if (tyinfo[i].dobj.dump && tyinfo[i].typtype == TYPTYPE_DOMAIN)
- getDomainConstraints(fout, &(tyinfo[i]));
- /*
- * If it's a base type, make a DumpableObject representing a shell
- * definition of the type. We will need to dump that ahead of the I/O
- * functions for the type. Similarly, range types need a shell
- * definition in case they have a canonicalize function.
- *
- * Note: the shell type doesn't have a catId. You might think it
- * should copy the base type's catId, but then it might capture the
- * pg_depend entries for the type, which we don't want.
- */
- if (tyinfo[i].dobj.dump && (tyinfo[i].typtype == TYPTYPE_BASE ||
- tyinfo[i].typtype == TYPTYPE_RANGE))
- {
- stinfo = (ShellTypeInfo *) pg_malloc(sizeof(ShellTypeInfo));
- stinfo->dobj.objType = DO_SHELL_TYPE;
- stinfo->dobj.catId = nilCatalogId;
- AssignDumpId(&stinfo->dobj);
- stinfo->dobj.name = pg_strdup(tyinfo[i].dobj.name);
- stinfo->dobj.namespace = tyinfo[i].dobj.namespace;
- stinfo->baseType = &(tyinfo[i]);
- tyinfo[i].shellType = stinfo;
- /*
- * Initially mark the shell type as not to be dumped. We'll only
- * dump it if the I/O or canonicalize functions need to be dumped;
- * this is taken care of while sorting dependencies.
- */
- stinfo->dobj.dump = false;
- /*
- * However, if dumping from pre-7.3, there will be no dependency
- * info so we have to fake it here. We only need to worry about
- * typinput and typoutput since the other functions only exist
- * post-7.3.
- */
- if (fout->remoteVersion < 70300)
- {
- Oid typinput;
- Oid typoutput;
- FuncInfo *funcInfo;
- typinput = atooid(PQgetvalue(res, i, i_typinput));
- typoutput = atooid(PQgetvalue(res, i, i_typoutput));
- funcInfo = findFuncByOid(typinput);
- if (funcInfo && funcInfo->dobj.dump)
- {
- /* base type depends on function */
- addObjectDependency(&tyinfo[i].dobj,
- funcInfo->dobj.dumpId);
- /* function depends on shell type */
- addObjectDependency(&funcInfo->dobj,
- stinfo->dobj.dumpId);
- /* mark shell type as to be dumped */
- stinfo->dobj.dump = true;
- }
- funcInfo = findFuncByOid(typoutput);
- if (funcInfo && funcInfo->dobj.dump)
- {
- /* base type depends on function */
- addObjectDependency(&tyinfo[i].dobj,
- funcInfo->dobj.dumpId);
- /* function depends on shell type */
- addObjectDependency(&funcInfo->dobj,
- stinfo->dobj.dumpId);
- /* mark shell type as to be dumped */
- stinfo->dobj.dump = true;
- }
- }
- }
- if (strlen(tyinfo[i].rolname) == 0 && tyinfo[i].isDefined)
- write_msg(NULL, "WARNING: owner of data type \"%s\" appears to be invalid\n",
- tyinfo[i].dobj.name);
- }
- *numTypes = ntups;
- PQclear(res);
- destroyPQExpBuffer(query);
- return tyinfo;
- }
- /*
- * getOperators:
- * read all operators in the system catalogs and return them in the
- * OprInfo* structure
- *
- * numOprs is set to the number of operators read in
- */
- OprInfo *
- getOperators(Archive *fout, int *numOprs)
- {
- PGresult *res;
- int ntups;
- int i;
- PQExpBuffer query = createPQExpBuffer();
- OprInfo *oprinfo;
- int i_tableoid;
- int i_oid;
- int i_oprname;
- int i_oprnamespace;
- int i_rolname;
- int i_oprkind;
- int i_oprcode;
- /*
- * find all operators, including builtin operators; we filter out
- * system-defined operators at dump-out time.
- */
- /* Make sure we are in proper schema */
- selectSourceSchema(fout, "pg_catalog");
- if (fout->remoteVersion >= 70300)
- {
- appendPQExpBuffer(query, "SELECT tableoid, oid, oprname, "
- "oprnamespace, "
- "(%s oprowner) AS rolname, "
- "oprkind, "
- "oprcode::oid AS oprcode "
- "FROM pg_operator",
- username_subquery);
- }
- else if (fout->remoteVersion >= 70100)
- {
- appendPQExpBuffer(query, "SELECT tableoid, oid, oprname, "
- "0::oid AS oprnamespace, "
- "(%s oprowner) AS rolname, "
- "oprkind, "
- "oprcode::oid AS oprcode "
- "FROM pg_operator",
- username_subquery);
- }
- else
- {
- appendPQExpBuffer(query, "SELECT "
- "(SELECT oid FROM pg_class WHERE relname = 'pg_operator') AS tableoid, "
- "oid, oprname, "
- "0::oid AS oprnamespace, "
- "(%s oprowner) AS rolname, "
- "oprkind, "
- "oprcode::oid AS oprcode "
- "FROM pg_operator",
- username_subquery);
- }
- res = ExecuteSqlQuery(fout, query->data, PGRES_TUPLES_OK);
- ntups = PQntuples(res);
- *numOprs = ntups;
- oprinfo = (OprInfo *) pg_malloc(ntups * sizeof(OprInfo));
- i_tableoid = PQfnumber(res, "tableoid");
- i_oid = PQfnumber(res, "oid");
- i_oprname = PQfnumber(res, "oprname");
- i_oprnamespace = PQfnumber(res, "oprnamespace");
- i_rolname = PQfnumber(res, "rolname");
- i_oprkind = PQfnumber(res, "oprkind");
- i_oprcode = PQfnumber(res, "oprcode");
- for (i = 0; i < ntups; i++)
- {
- oprinfo[i].dobj.objType = DO_OPERATOR;
- oprinfo[i].dobj.catId.tableoid = atooid(PQgetvalue(res, i, i_tableoid));
- oprinfo[i].dobj.catId.oid = atooid(PQgetvalue(res, i, i_oid));
- AssignDumpId(&oprinfo[i].dobj);
- oprinfo[i].dobj.name = pg_strdup(PQgetvalue(res, i, i_oprname));
- oprinfo[i].dobj.namespace =
- findNamespace(fout,
- atooid(PQgetvalue(res, i, i_oprnamespace)),
- oprinfo[i].dobj.catId.oid);
- oprinfo[i].rolname = pg_strdup(PQgetvalue(res, i, i_rolname));
- oprinfo[i].oprkind = (PQgetvalue(res, i, i_oprkind))[0];
- oprinfo[i].oprcode = atooid(PQgetvalue(res, i, i_oprcode));
- /* Decide whether we want to dump it */
- selectDumpableObject(&(oprinfo[i].dobj));
- if (strlen(oprinfo[i].rolname) == 0)
- write_msg(NULL, "WARNING: owner of operator \"%s\" appears to be invalid\n",
- oprinfo[i].dobj.name);
- }
- PQclear(res);
- destroyPQExpBuffer(query);
- return oprinfo;
- }
- /*
- * getCollations:
- * read all collations in the system catalogs and return them in the
- * CollInfo* structure
- *
- * numCollations is set to the number of collations read in
- */
- CollInfo *
- getCollations(Archive *fout, int *numCollations)
- {
- PGresult *res;
- int ntups;
- int i;
- PQExpBuffer query;
- CollInfo *collinfo;
- int i_tableoid;
- int i_oid;
- int i_collname;
- int i_collnamespace;
- int i_rolname;
- /* Collations didn't exist pre-9.1 */
- if (fout->remoteVersion < 90100)
- {
- *numCollations = 0;
- return NULL;
- }
- query = createPQExpBuffer();
- /*
- * find all collations, including builtin collations; we filter out
- * system-defined collations at dump-out time.
- */
- /* Make sure we are in proper schema */
- selectSourceSchema(fout, "pg_catalog");
- appendPQExpBuffer(query, "SELECT tableoid, oid, collname, "
- "collnamespace, "
- "(%s collowner) AS rolname "
- "FROM pg_collation",
- username_subquery);
- res = ExecuteSqlQuery(fout, query->data, PGRES_TUPLES_OK);
- ntups = PQntuples(res);
- *numCollations = ntups;
- collinfo = (CollInfo *) pg_malloc(ntups * sizeof(CollInfo));
- i_tableoid = PQfnumber(res, "tableoid");
- i_oid = PQfnumber(res, "oid");
- i_collname = PQfnumber(res, "collname");
- i_collnamespace = PQfnumber(res, "collnamespace");
- i_rolname = PQfnumber(res, "rolname");
- for (i = 0; i < ntups; i++)
- {
- collinfo[i].dobj.objType = DO_COLLATION;
- collinfo[i].dobj.catId.tableoid = atooid(PQgetvalue(res, i, i_tableoid));
- collinfo[i].dobj.catId.oid = atooid(PQgetvalue(res, i, i_oid));
- AssignDumpId(&collinfo[i].dobj);
- collinfo[i].dobj.name = pg_strdup(PQgetvalue(res, i, i_collname));
- collinfo[i].dobj.namespace =
- findNamespace(fout,
- atooid(PQgetvalue(res, i, i_collnamespace)),
- collinfo[i].dobj.catId.oid);
- collinfo[i].rolname = pg_strdup(PQgetvalue(res, i, i_rolname));
- /* Decide whether we want to dump it */
- selectDumpableObject(&(collinfo[i].dobj));
- }
- PQclear(res);
- destroyPQExpBuffer(query);
- return collinfo;
- }
- /*
- * getConversions:
- * read all conversions in the system catalogs and return them in the
- * ConvInfo* structure
- *
- * numConversions is set to the number of conversions read in
- */
- ConvInfo *
- getConversions(Archive *fout, int *numConversions)
- {
- PGresult *res;
- int ntups;
- int i;
- PQExpBuffer query;
- ConvInfo *convinfo;
- int i_tableoid;
- int i_oid;
- int i_conname;
- int i_connamespace;
- int i_rolname;
- /* Conversions didn't exist pre-7.3 */
- if (fout->remoteVersion < 70300)
- {
- *numConversions = 0;
- return NULL;
- }
- query = createPQExpBuffer();
- /*
- * find all conversions, including builtin conversions; we filter out
- * system-defined conversions at dump-out time.
- */
- /* Make sure we are in proper schema */
- selectSourceSchema(fout, "pg_catalog");
- appendPQExpBuffer(query, "SELECT tableoid, oid, conname, "
- "connamespace, "
- "(%s conowner) AS rolname "
- "FROM pg_conversion",
- username_subquery);
- res = ExecuteSqlQuery(fout, query->data, PGRES_TUPLES_OK);
- ntups = PQntuples(res);
- *numConversions = ntups;
- convinfo = (ConvInfo *) pg_malloc(ntups * sizeof(ConvInfo));
- i_tableoid = PQfnumber(res, "tableoid");
- i_oid = PQfnumber(res, "oid");
- i_conname = PQfnumber(res, "conname");
- i_connamespace = PQfnumber(res, "connamespace");
- i_rolname = PQfnumber(res, "rolname");
- for (i = 0; i < ntups; i++)
- {
- convinfo[i].dobj.objType = DO_CONVERSION;
- convinfo[i].dobj.catId.tableoid = atooid(PQgetvalue(res, i, i_tableoid));
- convinfo[i].dobj.catId.oid = atooid(PQgetvalue(res, i, i_oid));
- AssignDumpId(&convinfo[i].dobj);
- convinfo[i].dobj.name = pg_strdup(PQgetvalue(res, i, i_conname));
- convinfo[i].dobj.namespace =
- findNamespace(fout,
- atooid(PQgetvalue(res, i, i_connamespace)),
- convinfo[i].dobj.catId.oid);
- convinfo[i].rolname = pg_strdup(PQgetvalue(res, i, i_rolname));
- /* Decide whether we want to dump it */
- selectDumpableObject(&(convinfo[i].dobj));
- }
- PQclear(res);
- destroyPQExpBuffer(query);
- return convinfo;
- }
- /*
- * getOpclasses:
- * read all opclasses in the system catalogs and return them in the
- * OpclassInfo* structure
- *
- * numOpclasses is set to the number of opclasses read in
- */
- OpclassInfo *
- getOpclasses(Archive *fout, int *numOpclasses)
- {
- PGresult *res;
- int ntups;
- int i;
- PQExpBuffer query = createPQExpBuffer();
- OpclassInfo *opcinfo;
- int i_tableoid;
- int i_oid;
- int i_opcname;
- int i_opcnamespace;
- int i_rolname;
- /*
- * find all opclasses, including builtin opclasses; we filter out
- * system-defined opclasses at dump-out time.
- */
- /* Make sure we are in proper schema */
- selectSourceSchema(fout, "pg_catalog");
- if (fout->remoteVersion >= 70300)
- {
- appendPQExpBuffer(query, "SELECT tableoid, oid, opcname, "
- "opcnamespace, "
- "(%s opcowner) AS rolname "
- "FROM pg_opclass",
- username_subquery);
- }
- else if (fout->remoteVersion >= 70100)
- {
- appendPQExpBufferStr(query, "SELECT tableoid, oid, opcname, "
- "0::oid AS opcnamespace, "
- "''::name AS rolname "
- "FROM pg_opclass");
- }
- else
- {
- appendPQExpBufferStr(query, "SELECT "
- "(SELECT oid FROM pg_class WHERE relname = 'pg_opclass') AS tableoid, "
- "oid, opcname, "
- "0::oid AS opcnamespace, "
- "''::name AS rolname "
- "FROM pg_opclass");
- }
- res = ExecuteSqlQuery(fout, query->data, PGRES_TUPLES_OK);
- ntups = PQntuples(res);
- *numOpclasses = ntups;
- opcinfo = (OpclassInfo *) pg_malloc(ntups * sizeof(OpclassInfo));
- i_tableoid = PQfnumber(res, "tableoid");
- i_oid = PQfnumber(res, "oid");
- i_opcname = PQfnumber(res, "opcname");
- i_opcnamespace = PQfnumber(res, "opcnamespace");
- i_rolname = PQfnumber(res, "rolname");
- for (i = 0; i < ntups; i++)
- {
- opcinfo[i].dobj.objType = DO_OPCLASS;
- opcinfo[i].dobj.catId.tableoid = atooid(PQgetvalue(res, i, i_tableoid));
- opcinfo[i].dobj.catId.oid = atooid(PQgetvalue(res, i, i_oid));
- AssignDumpId(&opcinfo[i].dobj);
- opcinfo[i].dobj.name = pg_strdup(PQgetvalue(res, i, i_opcname));
- opcinfo[i].dobj.namespace =
- findNamespace(fout,
- atooid(PQgetvalue(res, i, i_opcnamespace)),
- opcinfo[i].dobj.catId.oid);
- opcinfo[i].rolname = pg_strdup(PQgetvalue(res, i, i_rolname));
- /* Decide whether we want to dump it */
- selectDumpableObject(&(opcinfo[i].dobj));
- if (fout->remoteVersion >= 70300)
- {
- if (strlen(opcinfo[i].rolname) == 0)
- write_msg(NULL, "WARNING: owner of operator class \"%s\" appears to be invalid\n",
- opcinfo[i].dobj.name);
- }
- }
- PQclear(res);
- destroyPQExpBuffer(query);
- return opcinfo;
- }
- /*
- * getOpfamilies:
- * read all opfamilies in the system catalogs and return them in the
- * OpfamilyInfo* structure
- *
- * numOpfamilies is set to the number of opfamilies read in
- */
- OpfamilyInfo *
- getOpfamilies(Archive *fout, int *numOpfamilies)
- {
- PGresult *res;
- int ntups;
- int i;
- PQExpBuffer query;
- OpfamilyInfo *opfinfo;
- int i_tableoid;
- int i_oid;
- int i_opfname;
- int i_opfnamespace;
- int i_rolname;
- /* Before 8.3, there is no separate concept of opfamilies */
- if (fout->remoteVersion < 80300)
- {
- *numOpfamilies = 0;
- return NULL;
- }
- query = createPQExpBuffer();
- /*
- * find all opfamilies, including builtin opfamilies; we filter out
- * system-defined opfamilies at dump-out time.
- */
- /* Make sure we are in proper schema */
- selectSourceSchema(fout, "pg_catalog");
- appendPQExpBuffer(query, "SELECT tableoid, oid, opfname, "
- "opfnamespace, "
- "(%s opfowner) AS rolname "
- "FROM pg_opfamily",
- username_subquery);
- res = ExecuteSqlQuery(fout, query->data, PGRES_TUPLES_OK);
- ntups = PQntuples(res);
- *numOpfamilies = ntups;
- opfinfo = (OpfamilyInfo *) pg_malloc(ntups * sizeof(OpfamilyInfo));
- i_tableoid = PQfnumber(res, "tableoid");
- i_oid = PQfnumber(res, "oid");
- i_opfname = PQfnumber(res, "opfname");
- i_opfnamespace = PQfnumber(res, "opfnamespace");
- i_rolname = PQfnumber(res, "rolname");
- for (i = 0; i < ntups; i++)
- {
- opfinfo[i].dobj.objType = DO_OPFAMILY;
- opfinfo[i].dobj.catId.tableoid = atooid(PQgetvalue(res, i, i_tableoid));
- opfinfo[i].dobj.catId.oid = atooid(PQgetvalue(res, i, i_oid));
- AssignDumpId(&opfinfo[i].dobj);
- opfinfo[i].dobj.name = pg_strdup(PQgetvalue(res, i, i_opfname));
- opfinfo[i].dobj.namespace =
- findNamespace(fout,
- atooid(PQgetvalue(res, i, i_opfnamespace)),
- opfinfo[i].dobj.catId.oid);
- opfinfo[i].rolname = pg_strdup(PQgetvalue(res, i, i_rolname));
- /* Decide whether we want to dump it */
- selectDumpableObject(&(opfinfo[i].dobj));
- if (fout->remoteVersion >= 70300)
- {
- if (strlen(opfinfo[i].rolname) == 0)
- write_msg(NULL, "WARNING: owner of operator family \"%s\" appears to be invalid\n",
- opfinfo[i].dobj.name);
- }
- }
- PQclear(res);
- destroyPQExpBuffer(query);
- return opfinfo;
- }
- /*
- * getAggregates:
- * read all the user-defined aggregates in the system catalogs and
- * return them in the AggInfo* structure
- *
- * numAggs is set to the number of aggregates read in
- */
- AggInfo *
- getAggregates(Archive *fout, int *numAggs)
- {
- PGresult *res;
- int ntups;
- int i;
- PQExpBuffer query = createPQExpBuffer();
- AggInfo *agginfo;
- int i_tableoid;
- int i_oid;
- int i_aggname;
- int i_aggnamespace;
- int i_pronargs;
- int i_proargtypes;
- int i_rolname;
- int i_aggacl;
- int i_proiargs;
- /* Make sure we are in proper schema */
- selectSourceSchema(fout, "pg_catalog");
- /*
- * Find all user-defined aggregates. See comment in getFuncs() for the
- * rationale behind the filtering logic.
- */
- if (fout->remoteVersion >= 80400)
- {
- appendPQExpBuffer(query, "SELECT tableoid, oid, proname AS aggname, "
- "pronamespace AS aggnamespace, "
- "pronargs, proargtypes, "
- "pg_catalog.pg_get_function_identity_arguments(oid) AS proiargs,"
- "(%s proowner) AS rolname, "
- "proacl AS aggacl "
- "FROM pg_proc p "
- "WHERE proisagg AND ("
- "pronamespace != "
- "(SELECT oid FROM pg_namespace "
- "WHERE nspname = 'pg_catalog')",
- username_subquery);
- if (binary_upgrade && fout->remoteVersion >= 90100)
- appendPQExpBufferStr(query,
- " OR EXISTS(SELECT 1 FROM pg_depend WHERE "
- "classid = 'pg_proc'::regclass AND "
- "objid = p.oid AND "
- "refclassid = 'pg_extension'::regclass AND "
- "deptype = 'e')");
- appendPQExpBufferChar(query, ')');
- }
- else if (fout->remoteVersion >= 80200)
- {
- appendPQExpBuffer(query, "SELECT tableoid, oid, proname AS aggname, "
- "pronamespace AS aggnamespace, "
- "pronargs, proargtypes, "
- "NULL::text AS proiargs,"
- "(%s proowner) AS rolname, "
- "proacl AS aggacl "
- "FROM pg_proc p "
- "WHERE proisagg AND ("
- "pronamespace != "
- "(SELECT oid FROM pg_namespace "
- "WHERE nspname = 'pg_catalog'))",
- username_subquery);
- }
- else if (fout->remoteVersion >= 70300)
- {
- appendPQExpBuffer(query, "SELECT tableoid, oid, proname AS aggname, "
- "pronamespace AS aggnamespace, "
- "CASE WHEN proargtypes[0] = 'pg_catalog.\"any\"'::pg_catalog.regtype THEN 0 ELSE 1 END AS pronargs, "
- "proargtypes, "
- "NULL::text AS proiargs, "
- "(%s proowner) AS rolname, "
- "proacl AS aggacl "
- "FROM pg_proc "
- "WHERE proisagg "
- "AND pronamespace != "
- "(SELECT oid FROM pg_namespace WHERE nspname = 'pg_catalog')",
- username_subquery);
- }
- else if (fout->remoteVersion >= 70100)
- {
- appendPQExpBuffer(query, "SELECT tableoid, oid, aggname, "
- "0::oid AS aggnamespace, "
- "CASE WHEN aggbasetype = 0 THEN 0 ELSE 1 END AS pronargs, "
- "aggbasetype AS proargtypes, "
- "NULL::text AS proiargs, "
- "(%s aggowner) AS rolname, "
- "'{=X}' AS aggacl "
- "FROM pg_aggregate "
- "where oid > '%u'::oid",
- username_subquery,
- g_last_builtin_oid);
- }
- else
- {
- appendPQExpBuffer(query, "SELECT "
- "(SELECT oid FROM pg_class WHERE relname = 'pg_aggregate') AS tableoid, "
- "oid, aggname, "
- "0::oid AS aggnamespace, "
- "CASE WHEN aggbasetype = 0 THEN 0 ELSE 1 END AS pronargs, "
- "aggbasetype AS proargtypes, "
- "NULL::text AS proiargs, "
- "(%s aggowner) AS rolname, "
- "'{=X}' AS aggacl "
- "FROM pg_aggregate "
- "where oid > '%u'::oid",
- username_subquery,
- g_last_builtin_oid);
- }
- res = ExecuteSqlQuery(fout, query->data, PGRES_TUPLES_OK);
- ntups = PQntuples(res);
- *numAggs = ntups;
- agginfo = (AggInfo *) pg_malloc(ntups * sizeof(AggInfo));
- i_tableoid = PQfnumber(res, "tableoid");
- i_oid = PQfnumber(res, "oid");
- i_aggname = PQfnumber(res, "aggname");
- i_aggnamespace = PQfnumber(res, "aggnamespace");
- i_pronargs = PQfnumber(res, "pronargs");
- i_proargtypes = PQfnumber(res, "proargtypes");
- i_rolname = PQfnumber(res, "rolname");
- i_aggacl = PQfnumber(res, "aggacl");
- i_proiargs = PQfnumber(res, "proiargs");
- for (i = 0; i < ntups; i++)
- {
- agginfo[i].aggfn.dobj.objType = DO_AGG;
- agginfo[i].aggfn.dobj.catId.tableoid = atooid(PQgetvalue(res, i, i_tableoid));
- agginfo[i].aggfn.dobj.catId.oid = atooid(PQgetvalue(res, i, i_oid));
- AssignDumpId(&agginfo[i].aggfn.dobj);
- agginfo[i].aggfn.dobj.name = pg_strdup(PQgetvalue(res, i, i_aggname));
- agginfo[i].aggfn.dobj.namespace =
- findNamespace(fout,
- atooid(PQgetvalue(res, i, i_aggnamespace)),
- agginfo[i].aggfn.dobj.catId.oid);
- agginfo[i].aggfn.rolname = pg_strdup(PQgetvalue(res, i, i_rolname));
- if (strlen(agginfo[i].aggfn.rolname) == 0)
- write_msg(NULL, "WARNING: owner of aggregate function \"%s\" appears to be invalid\n",
- agginfo[i].aggfn.dobj.name);
- agginfo[i].aggfn.lang = InvalidOid; /* not currently interesting */
- agginfo[i].aggfn.prorettype = InvalidOid; /* not saved */
- agginfo[i].aggfn.proacl = pg_strdup(PQgetvalue(res, i, i_aggacl));
- agginfo[i].aggfn.proiargs = pg_strdup(PQgetvalue(res, i, i_proiargs));
- agginfo[i].aggfn.nargs = atoi(PQgetvalue(res, i, i_pronargs));
- if (agginfo[i].aggfn.nargs == 0)
- agginfo[i].aggfn.argtypes = NULL;
- else
- {
- agginfo[i].aggfn.argtypes = (Oid *) pg_malloc(agginfo[i].aggfn.nargs * sizeof(Oid));
- if (fout->remoteVersion >= 70300)
- parseOidArray(PQgetvalue(res, i, i_proargtypes),
- agginfo[i].aggfn.argtypes,
- agginfo[i].aggfn.nargs);
- else
- /* it's just aggbasetype */
- agginfo[i].aggfn.argtypes[0] = atooid(PQgetvalue(res, i, i_proargtypes));
- }
- /* Decide whether we want to dump it */
- selectDumpableObject(&(agginfo[i].aggfn.dobj));
- }
- PQclear(res);
- destroyPQExpBuffer(query);
- return agginfo;
- }
- /*
- * getFuncs:
- * read all the user-defined functions in the system catalogs and
- * return them in the FuncInfo* structure
- *
- * numFuncs is set to the number of functions read in
- */
- FuncInfo *
- getFuncs(Archive *fout, int *numFuncs)
- {
- PGresult *res;
- int ntups;
- int i;
- PQExpBuffer query = createPQExpBuffer();
- FuncInfo *finfo;
- int i_tableoid;
- int i_oid;
- int i_proname;
- int i_pronamespace;
- int i_rolname;
- int i_prolang;
- int i_pronargs;
- int i_proargtypes;
- int i_prorettype;
- int i_proacl;
- int i_proiargs;
- /* Make sure we are in proper schema */
- selectSourceSchema(fout, "pg_catalog");
- /*
- * Find all user-defined functions. Normally we can exclude functions in
- * pg_catalog, which is worth doing since there are several thousand of
- * 'em. However, there are some extensions that create functions in
- * pg_catalog. In normal dumps we can still ignore those --- but in
- * binary-upgrade mode, we must dump the member objects of the extension,
- * so be sure to fetch any such functions.
- *
- * Also, in 9.2 and up, exclude functions that are internally dependent on
- * something else, since presumably those will be created as a result of
- * creating the something else. This currently only acts to suppress
- * constructor functions for range types. Note that this is OK only
- * because the constructors don't have any dependencies the range type
- * doesn't have; otherwise we might not get creation ordering correct.
- */
- if (fout->remoteVersion >= 80400)
- {
- appendPQExpBuffer(query,
- "SELECT tableoid, oid, proname, prolang, "
- "pronargs, proargtypes, prorettype, proacl, "
- "pronamespace, "
- "pg_catalog.pg_get_function_identity_arguments(oid) AS proiargs,"
- "(%s proowner) AS rolname "
- "FROM pg_proc p "
- "WHERE NOT proisagg AND ("
- "pronamespace != "
- "(SELECT oid FROM pg_namespace "
- "WHERE nspname = 'pg_catalog')",
- username_subquery);
- if (fout->remoteVersion >= 90200)
- appendPQExpBufferStr(query,
- "\n AND NOT EXISTS (SELECT 1 FROM pg_depend "
- "WHERE classid = 'pg_proc'::regclass AND "
- "objid = p.oid AND deptype = 'i')");
- if (binary_upgrade && fout->remoteVersion >= 90100)
- appendPQExpBufferStr(query,
- "\n OR EXISTS(SELECT 1 FROM pg_depend WHERE "
- "classid = 'pg_proc'::regclass AND "
- "objid = p.oid AND "
- "refclassid = 'pg_extension'::regclass AND "
- "deptype = 'e')");
- appendPQExpBufferChar(query, ')');
- }
- else if (fout->remoteVersion >= 70300)
- {
- appendPQExpBuffer(query,
- "SELECT tableoid, oid, proname, prolang, "
- "pronargs, proargtypes, prorettype, proacl, "
- "pronamespace, "
- "NULL::text AS proiargs,"
- "(%s proowner) AS rolname "
- "FROM pg_proc p "
- "WHERE NOT proisagg AND ("
- "pronamespace != "
- "(SELECT oid FROM pg_namespace "
- "WHERE nspname = 'pg_catalog'))",
- username_subquery);
- }
- else if (fout->remoteVersion >= 70100)
- {
- appendPQExpBuffer(query,
- "SELECT tableoid, oid, proname, prolang, "
- "pronargs, proargtypes, prorettype, "
- "'{=X}' AS proacl, "
- "0::oid AS pronamespace, "
- "NULL::text AS proiargs,"
- "(%s proowner) AS rolname "
- "FROM pg_proc "
- "WHERE pg_proc.oid > '%u'::oid",
- username_subquery,
- g_last_builtin_oid);
- }
- else
- {
- appendPQExpBuffer(query,
- "SELECT "
- "(SELECT oid FROM pg_class "
- " WHERE relname = 'pg_proc') AS tableoid, "
- "oid, proname, prolang, "
- "pronargs, proargtypes, prorettype, "
- "'{=X}' AS proacl, "
- "0::oid AS pronamespace, "
- "NULL::text AS proiargs,"
- "(%s proowner) AS rolname "
- "FROM pg_proc "
- "where pg_proc.oid > '%u'::oid",
- username_subquery,
- g_last_builtin_oid);
- }
- res = ExecuteSqlQuery(fout, query->data, PGRES_TUPLES_OK);
- ntups = PQntuples(res);
- *numFuncs = ntups;
- finfo = (FuncInfo *) pg_malloc0(ntups * sizeof(FuncInfo));
- i_tableoid = PQfnumber(res, "tableoid");
- i_oid = PQfnumber(res, "oid");
- i_proname = PQfnumber(res, "proname");
- i_pronamespace = PQfnumber(res, "pronamespace");
- i_rolname = PQfnumber(res, "rolname");
- i_prolang = PQfnumber(res, "prolang");
- i_pronargs = PQfnumber(res, "pronargs");
- i_proargtypes = PQfnumber(res, "proargtypes");
- i_prorettype = PQfnumber(res, "prorettype");
- i_proacl = PQfnumber(res, "proacl");
- i_proiargs = PQfnumber(res, "proiargs");
- for (i = 0; i < ntups; i++)
- {
- finfo[i].dobj.objType = DO_FUNC;
- finfo[i].dobj.catId.tableoid = atooid(PQgetvalue(res, i, i_tableoid));
- finfo[i].dobj.catId.oid = atooid(PQgetvalue(res, i, i_oid));
- AssignDumpId(&finfo[i].dobj);
- finfo[i].dobj.name = pg_strdup(PQgetvalue(res, i, i_proname));
- finfo[i].dobj.namespace =
- findNamespace(fout,
- atooid(PQgetvalue(res, i, i_pronamespace)),
- finfo[i].dobj.catId.oid);
- finfo[i].rolname = pg_strdup(PQgetvalue(res, i, i_rolname));
- finfo[i].lang = atooid(PQgetvalue(res, i, i_prolang));
- finfo[i].prorettype = atooid(PQgetvalue(res, i, i_prorettype));
- finfo[i].proiargs = pg_strdup(PQgetvalue(res, i, i_proiargs));
- finfo[i].proacl = pg_strdup(PQgetvalue(res, i, i_proacl));
- finfo[i].nargs = atoi(PQgetvalue(res, i, i_pronargs));
- if (finfo[i].nargs == 0)
- finfo[i].argtypes = NULL;
- else
- {
- finfo[i].argtypes = (Oid *) pg_malloc(finfo[i].nargs * sizeof(Oid));
- parseOidArray(PQgetvalue(res, i, i_proargtypes),
- finfo[i].argtypes, finfo[i].nargs);
- }
- /* Decide whether we want to dump it */
- selectDumpableObject(&(finfo[i].dobj));
- if (strlen(finfo[i].rolname) == 0)
- write_msg(NULL,
- "WARNING: owner of function \"%s\" appears to be invalid\n",
- finfo[i].dobj.name);
- }
- PQclear(res);
- destroyPQExpBuffer(query);
- return finfo;
- }
- /*
- * getTables
- * read all the user-defined tables (no indexes, no catalogs)
- * in the system catalogs return them in the TableInfo* structure
- *
- * numTables is set to the number of tables read in
- */
- TableInfo *
- getTables(Archive *fout, int *numTables)
- {
- PGresult *res;
- int ntups;
- int i;
- PQExpBuffer query = createPQExpBuffer();
- TableInfo *tblinfo;
- int i_reltableoid;
- int i_reloid;
- int i_relname;
- int i_relnamespace;
- int i_relkind;
- int i_relacl;
- int i_rolname;
- int i_relchecks;
- int i_relhastriggers;
- int i_relhasindex;
- int i_relhasrules;
- int i_relhasoids;
- int i_relfrozenxid;
- int i_relminmxid;
- int i_toastoid;
- int i_toastfrozenxid;
- int i_toastminmxid;
- int i_relpersistence;
- int i_relispopulated;
- int i_relreplident;
- int i_owning_tab;
- int i_owning_col;
- int i_reltablespace;
- int i_reloptions;
- int i_checkoption;
- int i_toastreloptions;
- int i_reloftype;
- int i_relpages;
- /* Make sure we are in proper schema */
- selectSourceSchema(fout, "pg_catalog");
- /*
- * Find all the tables and table-like objects.
- *
- * We include system catalogs, so that we can work if a user table is
- * defined to inherit from a system catalog (pretty weird, but...)
- *
- * We ignore relations that are not ordinary tables, sequences, views,
- * materialized views, composite types, or foreign tables.
- *
- * Composite-type table entries won't be dumped as such, but we have to
- * make a DumpableObject for them so that we can track dependencies of the
- * composite type (pg_depend entries for columns of the composite type
- * link to the pg_class entry not the pg_type entry).
- *
- * Note: in this phase we should collect only a minimal amount of
- * information about each table, basically just enough to decide if it is
- * interesting. We must fetch all tables in this phase because otherwise
- * we cannot correctly identify inherited columns, owned sequences, etc.
- */
- if (fout->remoteVersion >= 90400)
- {
- /*
- * Left join to pick up dependency info linking sequences to their
- * owning column, if any (note this dependency is AUTO as of 8.2)
- */
- appendPQExpBuffer(query,
- "SELECT c.tableoid, c.oid, c.relname, "
- "c.relacl, c.relkind, c.relnamespace, "
- "(%s c.relowner) AS rolname, "
- "c.relchecks, c.relhastriggers, "
- "c.relhasindex, c.relhasrules, c.relhasoids, "
- "c.relfrozenxid, c.relminmxid, tc.oid AS toid, "
- "tc.relfrozenxid AS tfrozenxid, "
- "tc.relminmxid AS tminmxid, "
- "c.relpersistence, c.relispopulated, "
- "c.relreplident, c.relpages, "
- "CASE WHEN c.reloftype <> 0 THEN c.reloftype::pg_catalog.regtype ELSE NULL END AS reloftype, "
- "d.refobjid AS owning_tab, "
- "d.refobjsubid AS owning_col, "
- "(SELECT spcname FROM pg_tablespace t WHERE t.oid = c.reltablespace) AS reltablespace, "
- "array_to_string(array_remove(array_remove(c.reloptions,'check_option=local'),'check_option=cascaded'), ', ') AS reloptions, "
- "CASE WHEN 'check_option=local' = ANY (c.reloptions) THEN 'LOCAL'::text "
- "WHEN 'check_option=cascaded' = ANY (c.reloptions) THEN 'CASCADED'::text ELSE NULL END AS checkoption, "
- "array_to_string(array(SELECT 'toast.' || x FROM unnest(tc.reloptions) x), ', ') AS toast_reloptions "
- "FROM pg_class c "
- "LEFT JOIN pg_depend d ON "
- "(c.relkind = '%c' AND "
- "d.classid = c.tableoid AND d.objid = c.oid AND "
- "d.objsubid = 0 AND "
- "d.refclassid = c.tableoid AND d.deptype = 'a') "
- "LEFT JOIN pg_class tc ON (c.reltoastrelid = tc.oid) "
- "WHERE c.relkind in ('%c', '%c', '%c', '%c', '%c', '%c') "
- "ORDER BY c.oid",
- username_subquery,
- RELKIND_SEQUENCE,
- RELKIND_RELATION, RELKIND_SEQUENCE,
- RELKIND_VIEW, RELKIND_COMPOSITE_TYPE,
- RELKIND_MATVIEW, RELKIND_FOREIGN_TABLE);
- }
- else if (fout->remoteVersion >= 90300)
- {
- /*
- * Left join to pick up dependency info linking sequences to their
- * owning column, if any (note this dependency is AUTO as of 8.2)
- */
- appendPQExpBuffer(query,
- "SELECT c.tableoid, c.oid, c.relname, "
- "c.relacl, c.relkind, c.relnamespace, "
- "(%s c.relowner) AS rolname, "
- "c.relchecks, c.relhastriggers, "
- "c.relhasindex, c.relhasrules, c.relhasoids, "
- "c.relfrozenxid, c.relminmxid, tc.oid AS toid, "
- "tc.relfrozenxid AS tfrozenxid, "
- "tc.relminmxid AS tminmxid, "
- "c.relpersistence, c.relispopulated, "
- "'d' AS relreplident, c.relpages, "
- "CASE WHEN c.reloftype <> 0 THEN c.reloftype::pg_catalog.regtype ELSE NULL END AS reloftype, "
- "d.refobjid AS owning_tab, "
- "d.refobjsubid AS owning_col, "
- "(SELECT spcname FROM pg_tablespace t WHERE t.oid = c.reltablespace) AS reltablespace, "
- "array_to_string(array_remove(array_remove(c.reloptions,'check_option=local'),'check_option=cascaded'), ', ') AS reloptions, "
- "CASE WHEN 'check_option=local' = ANY (c.reloptions) THEN 'LOCAL'::text "
- "WHEN 'check_option=cascaded' = ANY (c.reloptions) THEN 'CASCADED'::text ELSE NULL END AS checkoption, "
- "array_to_string(array(SELECT 'toast.' || x FROM unnest(tc.reloptions) x), ', ') AS toast_reloptions "
- "FROM pg_class c "
- "LEFT JOIN pg_depend d ON "
- "(c.relkind = '%c' AND "
- "d.classid = c.tableoid AND d.objid = c.oid AND "
- "d.objsubid = 0 AND "
- "d.refclassid = c.tableoid AND d.deptype = 'a') "
- "LEFT JOIN pg_class tc ON (c.reltoastrelid = tc.oid) "
- "WHERE c.relkind in ('%c', '%c', '%c', '%c', '%c', '%c') "
- "ORDER BY c.oid",
- username_subquery,
- RELKIND_SEQUENCE,
- RELKIND_RELATION, RELKIND_SEQUENCE,
- RELKIND_VIEW, RELKIND_COMPOSITE_TYPE,
- RELKIND_MATVIEW, RELKIND_FOREIGN_TABLE);
- }
- else if (fout->remoteVersion >= 90100)
- {
- /*
- * Left join to pick up dependency info linking sequences to their
- * owning column, if any (note this dependency is AUTO as of 8.2)
- */
- appendPQExpBuffer(query,
- "SELECT c.tableoid, c.oid, c.relname, "
- "c.relacl, c.relkind, c.relnamespace, "
- "(%s c.relowner) AS rolname, "
- "c.relchecks, c.relhastriggers, "
- "c.relhasindex, c.relhasrules, c.relhasoids, "
- "c.relfrozenxid, 0 AS relminmxid, tc.oid AS toid, "
- "tc.relfrozenxid AS tfrozenxid, "
- "0 AS tminmxid, "
- "c.relpersistence, 't' as relispopulated, "
- "'d' AS relreplident, c.relpages, "
- "CASE WHEN c.reloftype <> 0 THEN c.reloftype::pg_catalog.regtype ELSE NULL END AS reloftype, "
- "d.refobjid AS owning_tab, "
- "d.refobjsubid AS owning_col, "
- "(SELECT spcname FROM pg_tablespace t WHERE t.oid = c.reltablespace) AS reltablespace, "
- "array_to_string(c.reloptions, ', ') AS reloptions, "
- "array_to_string(array(SELECT 'toast.' || x FROM unnest(tc.reloptions) x), ', ') AS toast_reloptions "
- "FROM pg_class c "
- "LEFT JOIN pg_depend d ON "
- "(c.relkind = '%c' AND "
- "d.classid = c.tableoid AND d.objid = c.oid AND "
- "d.objsubid = 0 AND "
- "d.refclassid = c.tableoid AND d.deptype = 'a') "
- "LEFT JOIN pg_class tc ON (c.reltoastrelid = tc.oid) "
- "WHERE c.relkind in ('%c', '%c', '%c', '%c', '%c', '%c') "
- "ORDER BY c.oid",
- username_subquery,
- RELKIND_SEQUENCE,
- RELKIND_RELATION, RELKIND_SEQUENCE,
- RELKIND_VIEW, RELKIND_COMPOSITE_TYPE,
- RELKIND_MATVIEW, RELKIND_FOREIGN_TABLE);
- }
- else if (fout->remoteVersion >= 90000)
- {
- /*
- * Left join to pick up dependency info linking sequences to their
- * owning column, if any (note this dependency is AUTO as of 8.2)
- */
- appendPQExpBuffer(query,
- "SELECT c.tableoid, c.oid, c.relname, "
- "c.relacl, c.relkind, c.relnamespace, "
- "(%s c.relowner) AS rolname, "
- "c.relchecks, c.relhastriggers, "
- "c.relhasindex, c.relhasrules, c.relhasoids, "
- "c.relfrozenxid, 0 AS relminmxid, tc.oid AS toid, "
- "tc.relfrozenxid AS tfrozenxid, "
- "0 AS tminmxid, "
- "'p' AS relpersistence, 't' as relispopulated, "
- "'d' AS relreplident, c.relpages, "
- "CASE WHEN c.reloftype <> 0 THEN c.reloftype::pg_catalog.regtype ELSE NULL END AS reloftype, "
- "d.refobjid AS owning_tab, "
- "d.refobjsubid AS owning_col, "
- "(SELECT spcname FROM pg_tablespace t WHERE t.oid = c.reltablespace) AS reltablespace, "
- "array_to_string(c.reloptions, ', ') AS reloptions, "
- "array_to_string(array(SELECT 'toast.' || x FROM unnest(tc.reloptions) x), ', ') AS toast_reloptions "
- "FROM pg_class c "
- "LEFT JOIN pg_depend d ON "
- "(c.relkind = '%c' AND "
- "d.classid = c.tableoid AND d.objid = c.oid AND "
- "d.objsubid = 0 AND "
- "d.refclassid = c.tableoid AND d.deptype = 'a') "
- "LEFT JOIN pg_class tc ON (c.reltoastrelid = tc.oid) "
- "WHERE c.relkind in ('%c', '%c', '%c', '%c') "
- "ORDER BY c.oid",
- username_subquery,
- RELKIND_SEQUENCE,
- RELKIND_RELATION, RELKIND_SEQUENCE,
- RELKIND_VIEW, RELKIND_COMPOSITE_TYPE);
- }
- else if (fout->remoteVersion >= 80400)
- {
- /*
- * Left join to pick up dependency info linking sequences to their
- * owning column, if any (note this dependency is AUTO as of 8.2)
- */
- appendPQExpBuffer(query,
- "SELECT c.tableoid, c.oid, c.relname, "
- "c.relacl, c.relkind, c.relnamespace, "
- "(%s c.relowner) AS rolname, "
- "c.relchecks, c.relhastriggers, "
- "c.relhasindex, c.relhasrules, c.relhasoids, "
- "c.relfrozenxid, 0 AS relminmxid, tc.oid AS toid, "
- "tc.relfrozenxid AS tfrozenxid, "
- "0 AS tminmxid, "
- "'p' AS relpersistence, 't' as relispopulated, "
- "'d' AS relreplident, c.relpages, "
- "NULL AS reloftype, "
- "d.refobjid AS owning_tab, "
- "d.refobjsubid AS owning_col, "
- "(SELECT spcname FROM pg_tablespace t WHERE t.oid = c.reltablespace) AS reltablespace, "
- "array_to_string(c.reloptions, ', ') AS reloptions, "
- "array_to_string(array(SELECT 'toast.' || x FROM unnest(tc.reloptions) x), ', ') AS toast_reloptions "
- "FROM pg_class c "
- "LEFT JOIN pg_depend d ON "
- "(c.relkind = '%c' AND "
- "d.classid = c.tableoid AND d.objid = c.oid AND "
- "d.objsubid = 0 AND "
- "d.refclassid = c.tableoid AND d.deptype = 'a') "
- "LEFT JOIN pg_class tc ON (c.reltoastrelid = tc.oid) "
- "WHERE c.relkind in ('%c', '%c', '%c', '%c') "
- "ORDER BY c.oid",
- username_subquery,
- RELKIND_SEQUENCE,
- RELKIND_RELATION, RELKIND_SEQUENCE,
- RELKIND_VIEW, RELKIND_COMPOSITE_TYPE);
- }
- else if (fout->remoteVersion >= 80200)
- {
- /*
- * Left join to pick up dependency info linking sequences to their
- * owning column, if any (note this dependency is AUTO as of 8.2)
- */
- appendPQExpBuffer(query,
- "SELECT c.tableoid, c.oid, c.relname, "
- "c.relacl, c.relkind, c.relnamespace, "
- "(%s c.relowner) AS rolname, "
- "c.relchecks, (c.reltriggers <> 0) AS relhastriggers, "
- "c.relhasindex, c.relhasrules, c.relhasoids, "
- "c.relfrozenxid, 0 AS relminmxid, tc.oid AS toid, "
- "tc.relfrozenxid AS tfrozenxid, "
- "0 AS tminmxid, "
- "'p' AS relpersistence, 't' as relispopulated, "
- "'d' AS relreplident, c.relpages, "
- "NULL AS reloftype, "
- "d.refobjid AS owning_tab, "
- "d.refobjsubid AS owning_col, "
- "(SELECT spcname FROM pg_tablespace t WHERE t.oid = c.reltablespace) AS reltablespace, "
- "array_to_string(c.reloptions, ', ') AS reloptions, "
- "NULL AS toast_reloptions "
- "FROM pg_class c "
- "LEFT JOIN pg_depend d ON "
- "(c.relkind = '%c' AND "
- "d.classid = c.tableoid AND d.objid = c.oid AND "
- "d.objsubid = 0 AND "
- "d.refclassid = c.tableoid AND d.deptype = 'a') "
- "LEFT JOIN pg_class tc ON (c.reltoastrelid = tc.oid) "
- "WHERE c.relkind in ('%c', '%c', '%c', '%c') "
- "ORDER BY c.oid",
- username_subquery,
- RELKIND_SEQUENCE,
- RELKIND_RELATION, RELKIND_SEQUENCE,
- RELKIND_VIEW, RELKIND_COMPOSITE_TYPE);
- }
- else if (fout->remoteVersion >= 80000)
- {
- /*
- * Left join to pick up dependency info linking sequences to their
- * owning column, if any
- */
- appendPQExpBuffer(query,
- "SELECT c.tableoid, c.oid, relname, "
- "relacl, relkind, relnamespace, "
- "(%s relowner) AS rolname, "
- "relchecks, (reltriggers <> 0) AS relhastriggers, "
- "relhasindex, relhasrules, relhasoids, "
- "0 AS relfrozenxid, 0 AS relminmxid,"
- "0 AS toid, "
- "0 AS tfrozenxid, 0 AS tminmxid,"
- "'p' AS relpersistence, 't' as relispopulated, "
- "'d' AS relreplident, relpages, "
- "NULL AS reloftype, "
- "d.refobjid AS owning_tab, "
- "d.refobjsubid AS owning_col, "
- "(SELECT spcname FROM pg_tablespace t WHERE t.oid = c.reltablespace) AS reltablespace, "
- "NULL AS reloptions, "
- "NULL AS toast_reloptions "
- "FROM pg_class c "
- "LEFT JOIN pg_depend d ON "
- "(c.relkind = '%c' AND "
- "d.classid = c.tableoid AND d.objid = c.oid AND "
- "d.objsubid = 0 AND "
- "d.refclassid = c.tableoid AND d.deptype = 'i') "
- "WHERE relkind in ('%c', '%c', '%c', '%c') "
- "ORDER BY c.oid",
- username_subquery,
- RELKIND_SEQUENCE,
- RELKIND_RELATION, RELKIND_SEQUENCE,
- RELKIND_VIEW, RELKIND_COMPOSITE_TYPE);
- }
- else if (fout->remoteVersion >= 70300)
- {
- /*
- * Left join to pick up dependency info linking sequences to their
- * owning column, if any
- */
- appendPQExpBuffer(query,
- "SELECT c.tableoid, c.oid, relname, "
- "relacl, relkind, relnamespace, "
- "(%s relowner) AS rolname, "
- "relchecks, (reltriggers <> 0) AS relhastriggers, "
- "relhasindex, relhasrules, relhasoids, "
- "0 AS relfrozenxid, 0 AS relminmxid,"
- "0 AS toid, "
- "0 AS tfrozenxid, 0 AS tminmxid,"
- "'p' AS relpersistence, 't' as relispopulated, "
- "'d' AS relreplident, relpages, "
- "NULL AS reloftype, "
- "d.refobjid AS owning_tab, "
- "d.refobjsubid AS owning_col, "
- "NULL AS reltablespace, "
- "NULL AS reloptions, "
- "NULL AS toast_reloptions "
- "FROM pg_class c "
- "LEFT JOIN pg_depend d ON "
- "(c.relkind = '%c' AND "
- "d.classid = c.tableoid AND d.objid = c.oid AND "
- "d.objsubid = 0 AND "
- "d.refclassid = c.tableoid AND d.deptype = 'i') "
- "WHERE relkind IN ('%c', '%c', '%c', '%c') "
- "ORDER BY c.oid",
- username_subquery,
- RELKIND_SEQUENCE,
- RELKIND_RELATION, RELKIND_SEQUENCE,
- RELKIND_VIEW, RELKIND_COMPOSITE_TYPE);
- }
- else if (fout->remoteVersion >= 70200)
- {
- appendPQExpBuffer(query,
- "SELECT tableoid, oid, relname, relacl, relkind, "
- "0::oid AS relnamespace, "
- "(%s relowner) AS rolname, "
- "relchecks, (reltriggers <> 0) AS relhastriggers, "
- "relhasindex, relhasrules, relhasoids, "
- "0 AS relfrozenxid, 0 AS relminmxid,"
- "0 AS toid, "
- "0 AS tfrozenxid, 0 AS tminmxid,"
- "'p' AS relpersistence, 't' as relispopulated, "
- "'d' AS relreplident, relpages, "
- "NULL AS reloftype, "
- "NULL::oid AS owning_tab, "
- "NULL::int4 AS owning_col, "
- "NULL AS reltablespace, "
- "NULL AS reloptions, "
- "NULL AS toast_reloptions "
- "FROM pg_class "
- "WHERE relkind IN ('%c', '%c', '%c') "
- "ORDER BY oid",
- username_subquery,
- RELKIND_RELATION, RELKIND_SEQUENCE, RELKIND_VIEW);
- }
- else if (fout->remoteVersion >= 70100)
- {
- /* all tables have oids in 7.1 */
- appendPQExpBuffer(query,
- "SELECT tableoid, oid, relname, relacl, relkind, "
- "0::oid AS relnamespace, "
- "(%s relowner) AS rolname, "
- "relchecks, (reltriggers <> 0) AS relhastriggers, "
- "relhasindex, relhasrules, "
- "'t'::bool AS relhasoids, "
- "0 AS relfrozenxid, 0 AS relminmxid,"
- "0 AS toid, "
- "0 AS tfrozenxid, 0 AS tminmxid,"
- "'p' AS relpersistence, 't' as relispopulated, "
- "'d' AS relreplident, relpages, "
- "NULL AS reloftype, "
- "NULL::oid AS owning_tab, "
- "NULL::int4 AS owning_col, "
- "NULL AS reltablespace, "
- "NULL AS reloptions, "
- "NULL AS toast_reloptions "
- "FROM pg_class "
- "WHERE relkind IN ('%c', '%c', '%c') "
- "ORDER BY oid",
- username_subquery,
- RELKIND_RELATION, RELKIND_SEQUENCE, RELKIND_VIEW);
- }
- else
- {
- /*
- * Before 7.1, view relkind was not set to 'v', so we must check if we
- * have a view by looking for a rule in pg_rewrite.
- */
- appendPQExpBuffer(query,
- "SELECT "
- "(SELECT oid FROM pg_class WHERE relname = 'pg_class') AS tableoid, "
- "oid, relname, relacl, "
- "CASE WHEN relhasrules and relkind = 'r' "
- " and EXISTS(SELECT rulename FROM pg_rewrite r WHERE "
- " r.ev_class = c.oid AND r.ev_type = '1') "
- "THEN '%c'::\"char\" "
- "ELSE relkind END AS relkind,"
- "0::oid AS relnamespace, "
- "(%s relowner) AS rolname, "
- "relchecks, (reltriggers <> 0) AS relhastriggers, "
- "relhasindex, relhasrules, "
- "'t'::bool AS relhasoids, "
- "0 AS relfrozenxid, 0 AS relminmxid,"
- "0 AS toid, "
- "0 AS tfrozenxid, 0 AS tminmxid,"
- "'p' AS relpersistence, 't' as relispopulated, "
- "'d' AS relreplident, 0 AS relpages, "
- "NULL AS reloftype, "
- "NULL::oid AS owning_tab, "
- "NULL::int4 AS owning_col, "
- "NULL AS reltablespace, "
- "NULL AS reloptions, "
- "NULL AS toast_reloptions "
- "FROM pg_class c "
- "WHERE relkind IN ('%c', '%c') "
- "ORDER BY oid",
- RELKIND_VIEW,
- username_subquery,
- RELKIND_RELATION, RELKIND_SEQUENCE);
- }
- res = ExecuteSqlQuery(fout, query->data, PGRES_TUPLES_OK);
- ntups = PQntuples(res);
- *numTables = ntups;
- /*
- * Extract data from result and lock dumpable tables. We do the locking
- * before anything else, to minimize the window wherein a table could
- * disappear under us.
- *
- * Note that we have to save info about all tables here, even when dumping
- * only one, because we don't yet know which tables might be inheritance
- * ancestors of the target table.
- */
- tblinfo = (TableInfo *) pg_malloc0(ntups * sizeof(TableInfo));
- i_reltableoid = PQfnumber(res, "tableoid");
- i_reloid = PQfnumber(res, "oid");
- i_relname = PQfnumber(res, "relname");
- i_relnamespace = PQfnumber(res, "relnamespace");
- i_relacl = PQfnumber(res, "relacl");
- i_relkind = PQfnumber(res, "relkind");
- i_rolname = PQfnumber(res, "rolname");
- i_relchecks = PQfnumber(res, "relchecks");
- i_relhastriggers = PQfnumber(res, "relhastriggers");
- i_relhasindex = PQfnumber(res, "relhasindex");
- i_relhasrules = PQfnumber(res, "relhasrules");
- i_relhasoids = PQfnumber(res, "relhasoids");
- i_relfrozenxid = PQfnumber(res, "relfrozenxid");
- i_relminmxid = PQfnumber(res, "relminmxid");
- i_toastoid = PQfnumber(res, "toid");
- i_toastfrozenxid = PQfnumber(res, "tfrozenxid");
- i_toastminmxid = PQfnumber(res, "tminmxid");
- i_relpersistence = PQfnumber(res, "relpersistence");
- i_relispopulated = PQfnumber(res, "relispopulated");
- i_relreplident = PQfnumber(res, "relreplident");
- i_relpages = PQfnumber(res, "relpages");
- i_owning_tab = PQfnumber(res, "owning_tab");
- i_owning_col = PQfnumber(res, "owning_col");
- i_reltablespace = PQfnumber(res, "reltablespace");
- i_reloptions = PQfnumber(res, "reloptions");
- i_checkoption = PQfnumber(res, "checkoption");
- i_toastreloptions = PQfnumber(res, "toast_reloptions");
- i_reloftype = PQfnumber(res, "reloftype");
- if (lockWaitTimeout && fout->remoteVersion >= 70300)
- {
- /*
- * Arrange to fail instead of waiting forever for a table lock.
- *
- * NB: this coding assumes that the only queries issued within the
- * following loop are LOCK TABLEs; else the timeout may be undesirably
- * applied to other things too.
- */
- resetPQExpBuffer(query);
- appendPQExpBufferStr(query, "SET statement_timeout = ");
- appendStringLiteralConn(query, lockWaitTimeout, GetConnection(fout));
- ExecuteSqlStatement(fout, query->data);
- }
- for (i = 0; i < ntups; i++)
- {
- tblinfo[i].dobj.objType = DO_TABLE;
- tblinfo[i].dobj.catId.tableoid = atooid(PQgetvalue(res, i, i_reltableoid));
- tblinfo[i].dobj.catId.oid = atooid(PQgetvalue(res, i, i_reloid));
- AssignDumpId(&tblinfo[i].dobj);
- tblinfo[i].dobj.name = pg_strdup(PQgetvalue(res, i, i_relname));
- tblinfo[i].dobj.namespace =
- findNamespace(fout,
- atooid(PQgetvalue(res, i, i_relnamespace)),
- tblinfo[i].dobj.catId.oid);
- tblinfo[i].rolname = pg_strdup(PQgetvalue(res, i, i_rolname));
- tblinfo[i].relacl = pg_strdup(PQgetvalue(res, i, i_relacl));
- tblinfo[i].relkind = *(PQgetvalue(res, i, i_relkind));
- tblinfo[i].relpersistence = *(PQgetvalue(res, i, i_relpersistence));
- tblinfo[i].hasindex = (strcmp(PQgetvalue(res, i, i_relhasindex), "t") == 0);
- tblinfo[i].hasrules = (strcmp(PQgetvalue(res, i, i_relhasrules), "t") == 0);
- tblinfo[i].hastriggers = (strcmp(PQgetvalue(res, i, i_relhastriggers), "t") == 0);
- tblinfo[i].hasoids = (strcmp(PQgetvalue(res, i, i_relhasoids), "t") == 0);
- tblinfo[i].relispopulated = (strcmp(PQgetvalue(res, i, i_relispopulated), "t") == 0);
- tblinfo[i].relreplident = *(PQgetvalue(res, i, i_relreplident));
- tblinfo[i].relpages = atoi(PQgetvalue(res, i, i_relpages));
- tblinfo[i].frozenxid = atooid(PQgetvalue(res, i, i_relfrozenxid));
- tblinfo[i].minmxid = atooid(PQgetvalue(res, i, i_relminmxid));
- tblinfo[i].toast_oid = atooid(PQgetvalue(res, i, i_toastoid));
- tblinfo[i].toast_frozenxid = atooid(PQgetvalue(res, i, i_toastfrozenxid));
- tblinfo[i].toast_minmxid = atooid(PQgetvalue(res, i, i_toastminmxid));
- if (PQgetisnull(res, i, i_reloftype))
- tblinfo[i].reloftype = NULL;
- else
- tblinfo[i].reloftype = pg_strdup(PQgetvalue(res, i, i_reloftype));
- tblinfo[i].ncheck = atoi(PQgetvalue(res, i, i_relchecks));
- if (PQgetisnull(res, i, i_owning_tab))
- {
- tblinfo[i].owning_tab = InvalidOid;
- tblinfo[i].owning_col = 0;
- }
- else
- {
- tblinfo[i].owning_tab = atooid(PQgetvalue(res, i, i_owning_tab));
- tblinfo[i].owning_col = atoi(PQgetvalue(res, i, i_owning_col));
- }
- tblinfo[i].reltablespace = pg_strdup(PQgetvalue(res, i, i_reltablespace));
- tblinfo[i].reloptions = pg_strdup(PQgetvalue(res, i, i_reloptions));
- if (i_checkoption == -1 || PQgetisnull(res, i, i_checkoption))
- tblinfo[i].checkoption = NULL;
- else
- tblinfo[i].checkoption = pg_strdup(PQgetvalue(res, i, i_checkoption));
- tblinfo[i].toast_reloptions = pg_strdup(PQgetvalue(res, i, i_toastreloptions));
- /* other fields were zeroed above */
- /*
- * Decide whether we want to dump this table.
- */
- if (tblinfo[i].relkind == RELKIND_COMPOSITE_TYPE)
- tblinfo[i].dobj.dump = false;
- else
- selectDumpableTable(&tblinfo[i]);
- tblinfo[i].interesting = tblinfo[i].dobj.dump;
- tblinfo[i].postponed_def = false; /* might get set during sort */
- /*
- * Read-lock target tables to make sure they aren't DROPPED or altered
- * in schema before we get around to dumping them.
- *
- * Note that we don't explicitly lock parents of the target tables; we
- * assume our lock on the child is enough to prevent schema
- * alterations to parent tables.
- *
- * NOTE: it'd be kinda nice to lock other relations too, not only
- * plain tables, but the backend doesn't presently allow that.
- */
- if (tblinfo[i].dobj.dump && tblinfo[i].relkind == RELKIND_RELATION)
- {
- resetPQExpBuffer(query);
- appendPQExpBuffer(query,
- "LOCK TABLE %s IN ACCESS SHARE MODE",
- fmtQualifiedId(fout->remoteVersion,
- tblinfo[i].dobj.namespace->dobj.name,
- tblinfo[i].dobj.name));
- ExecuteSqlStatement(fout, query->data);
- }
- /* Emit notice if join for owner failed */
- if (strlen(tblinfo[i].rolname) == 0)
- write_msg(NULL, "WARNING: owner of table \"%s\" appears to be invalid\n",
- tblinfo[i].dobj.name);
- }
- if (lockWaitTimeout && fout->remoteVersion >= 70300)
- {
- ExecuteSqlStatement(fout, "SET statement_timeout = 0");
- }
- PQclear(res);
- destroyPQExpBuffer(query);
- return tblinfo;
- }
- /*
- * getOwnedSeqs
- * identify owned sequences and mark them as dumpable if owning table is
- *
- * We used to do this in getTables(), but it's better to do it after the
- * index used by findTableByOid() has been set up.
- */
- void
- getOwnedSeqs(Archive *fout, TableInfo tblinfo[], int numTables)
- {
- int i;
- /*
- * Force sequences that are "owned" by table columns to be dumped whenever
- * their owning table is being dumped.
- */
- for (i = 0; i < numTables; i++)
- {
- TableInfo *seqinfo = &tblinfo[i];
- TableInfo *owning_tab;
- if (!OidIsValid(seqinfo->owning_tab))
- continue; /* not an owned sequence */
- if (seqinfo->dobj.dump)
- continue; /* no need to search */
- owning_tab = findTableByOid(seqinfo->owning_tab);
- if (owning_tab && owning_tab->dobj.dump)
- {
- seqinfo->interesting = true;
- seqinfo->dobj.dump = true;
- }
- }
- }
- /*
- * getInherits
- * read all the inheritance information
- * from the system catalogs return them in the InhInfo* structure
- *
- * numInherits is set to the number of pairs read in
- */
- InhInfo *
- getInherits(Archive *fout, int *numInherits)
- {
- PGresult *res;
- int ntups;
- int i;
- PQExpBuffer query = createPQExpBuffer();
- InhInfo *inhinfo;
- int i_inhrelid;
- int i_inhparent;
- /* Make sure we are in proper schema */
- selectSourceSchema(fout, "pg_catalog");
- /* find all the inheritance information */
- appendPQExpBufferStr(query, "SELECT inhrelid, inhparent FROM pg_inherits");
- res = ExecuteSqlQuery(fout, query->data, PGRES_TUPLES_OK);
- ntups = PQntuples(res);
- *numInherits = ntups;
- inhinfo = (InhInfo *) pg_malloc(ntups * sizeof(InhInfo));
- i_inhrelid = PQfnumber(res, "inhrelid");
- i_inhparent = PQfnumber(res, "inhparent");
- for (i = 0; i < ntups; i++)
- {
- inhinfo[i].inhrelid = atooid(PQgetvalue(res, i, i_inhrelid));
- inhinfo[i].inhparent = atooid(PQgetvalue(res, i, i_inhparent));
- }
- PQclear(res);
- destroyPQExpBuffer(query);
- return inhinfo;
- }
- /*
- * getIndexes
- * get information about every index on a dumpable table
- *
- * Note: index data is not returned directly to the caller, but it
- * does get entered into the DumpableObject tables.
- */
- void
- getIndexes(Archive *fout, TableInfo tblinfo[], int numTables)
- {
- int i,
- j;
- PQExpBuffer query = createPQExpBuffer();
- PGresult *res;
- IndxInfo *indxinfo;
- ConstraintInfo *constrinfo;
- int i_tableoid,
- i_oid,
- i_indexname,
- i_indexdef,
- i_indnkeys,
- i_indkey,
- i_indisclustered,
- i_indisreplident,
- i_contype,
- i_conname,
- i_condeferrable,
- i_condeferred,
- i_contableoid,
- i_conoid,
- i_condef,
- i_tablespace,
- i_options,
- i_relpages;
- int ntups;
- for (i = 0; i < numTables; i++)
- {
- TableInfo *tbinfo = &tblinfo[i];
- /* Only plain tables and materialized views have indexes. */
- if (tbinfo->relkind != RELKIND_RELATION &&
- tbinfo->relkind != RELKIND_MATVIEW)
- continue;
- if (!tbinfo->hasindex)
- continue;
- /* Ignore indexes of tables not to be dumped */
- if (!tbinfo->dobj.dump)
- continue;
- if (g_verbose)
- write_msg(NULL, "reading indexes for table \"%s\"\n",
- tbinfo->dobj.name);
- /* Make sure we are in proper schema so indexdef is right */
- selectSourceSchema(fout, tbinfo->dobj.namespace->dobj.name);
- /*
- * The point of the messy-looking outer join is to find a constraint
- * that is related by an internal dependency link to the index. If we
- * find one, create a CONSTRAINT entry linked to the INDEX entry. We
- * assume an index won't have more than one internal dependency.
- *
- * As of 9.0 we don't need to look at pg_depend but can check for a
- * match to pg_constraint.conindid. The check on conrelid is
- * redundant but useful because that column is indexed while conindid
- * is not.
- */
- resetPQExpBuffer(query);
- if (fout->remoteVersion >= 90400)
- {
- /*
- * the test on indisready is necessary in 9.2, and harmless in
- * earlier/later versions
- */
- appendPQExpBuffer(query,
- "SELECT t.tableoid, t.oid, "
- "t.relname AS indexname, "
- "pg_catalog.pg_get_indexdef(i.indexrelid) AS indexdef, "
- "t.relnatts AS indnkeys, "
- "i.indkey, i.indisclustered, "
- "i.indisreplident, t.relpages, "
- "c.contype, c.conname, "
- "c.condeferrable, c.condeferred, "
- "c.tableoid AS contableoid, "
- "c.oid AS conoid, "
- "pg_catalog.pg_get_constraintdef(c.oid, false) AS condef, "
- "(SELECT spcname FROM pg_catalog.pg_tablespace s WHERE s.oid = t.reltablespace) AS tablespace, "
- "array_to_string(t.reloptions, ', ') AS options "
- "FROM pg_catalog.pg_index i "
- "JOIN pg_catalog.pg_class t ON (t.oid = i.indexrelid) "
- "LEFT JOIN pg_catalog.pg_constraint c "
- "ON (i.indrelid = c.conrelid AND "
- "i.indexrelid = c.conindid AND "
- "c.contype IN ('p','u','x')) "
- "WHERE i.indrelid = '%u'::pg_catalog.oid "
- "AND i.indisvalid AND i.indisready "
- "ORDER BY indexname",
- tbinfo->dobj.catId.oid);
- }
- else if (fout->remoteVersion >= 90000)
- {
- /*
- * the test on indisready is necessary in 9.2, and harmless in
- * earlier/later versions
- */
- appendPQExpBuffer(query,
- "SELECT t.tableoid, t.oid, "
- "t.relname AS indexname, "
- "pg_catalog.pg_get_indexdef(i.indexrelid) AS indexdef, "
- "t.relnatts AS indnkeys, "
- "i.indkey, i.indisclustered, "
- "false AS indisreplident, t.relpages, "
- "c.contype, c.conname, "
- "c.condeferrable, c.condeferred, "
- "c.tableoid AS contableoid, "
- "c.oid AS conoid, "
- "pg_catalog.pg_get_constraintdef(c.oid, false) AS condef, "
- "(SELECT spcname FROM pg_catalog.pg_tablespace s WHERE s.oid = t.reltablespace) AS tablespace, "
- "array_to_string(t.reloptions, ', ') AS options "
- "FROM pg_catalog.pg_index i "
- "JOIN pg_catalog.pg_class t ON (t.oid = i.indexrelid) "
- "LEFT JOIN pg_catalog.pg_constraint c "
- "ON (i.indrelid = c.conrelid AND "
- "i.indexrelid = c.conindid AND "
- "c.contype IN ('p','u','x')) "
- "WHERE i.indrelid = '%u'::pg_catalog.oid "
- "AND i.indisvalid AND i.indisready "
- "ORDER BY indexname",
- tbinfo->dobj.catId.oid);
- }
- else if (fout->remoteVersion >= 80200)
- {
- appendPQExpBuffer(query,
- "SELECT t.tableoid, t.oid, "
- "t.relname AS indexname, "
- "pg_catalog.pg_get_indexdef(i.indexrelid) AS indexdef, "
- "t.relnatts AS indnkeys, "
- "i.indkey, i.indisclustered, "
- "false AS indisreplident, t.relpages, "
- "c.contype, c.conname, "
- "c.condeferrable, c.condeferred, "
- "c.tableoid AS contableoid, "
- "c.oid AS conoid, "
- "null AS condef, "
- "(SELECT spcname FROM pg_catalog.pg_tablespace s WHERE s.oid = t.reltablespace) AS tablespace, "
- "array_to_string(t.reloptions, ', ') AS options "
- "FROM pg_catalog.pg_index i "
- "JOIN pg_catalog.pg_class t ON (t.oid = i.indexrelid) "
- "LEFT JOIN pg_catalog.pg_depend d "
- "ON (d.classid = t.tableoid "
- "AND d.objid = t.oid "
- "AND d.deptype = 'i') "
- "LEFT JOIN pg_catalog.pg_constraint c "
- "ON (d.refclassid = c.tableoid "
- "AND d.refobjid = c.oid) "
- "WHERE i.indrelid = '%u'::pg_catalog.oid "
- "AND i.indisvalid "
- "ORDER BY indexname",
- tbinfo->dobj.catId.oid);
- }
- else if (fout->remoteVersion >= 80000)
- {
- appendPQExpBuffer(query,
- "SELECT t.tableoid, t.oid, "
- "t.relname AS indexname, "
- "pg_catalog.pg_get_indexdef(i.indexrelid) AS indexdef, "
- "t.relnatts AS indnkeys, "
- "i.indkey, i.indisclustered, "
- "false AS indisreplident, t.relpages, "
- "c.contype, c.conname, "
- "c.condeferrable, c.condeferred, "
- "c.tableoid AS contableoid, "
- "c.oid AS conoid, "
- "null AS condef, "
- "(SELECT spcname FROM pg_catalog.pg_tablespace s WHERE s.oid = t.reltablespace) AS tablespace, "
- "null AS options "
- "FROM pg_catalog.pg_index i "
- "JOIN pg_catalog.pg_class t ON (t.oid = i.indexrelid) "
- "LEFT JOIN pg_catalog.pg_depend d "
- "ON (d.classid = t.tableoid "
- "AND d.objid = t.oid "
- "AND d.deptype = 'i') "
- "LEFT JOIN pg_catalog.pg_constraint c "
- "ON (d.refclassid = c.tableoid "
- "AND d.refobjid = c.oid) "
- "WHERE i.indrelid = '%u'::pg_catalog.oid "
- "ORDER BY indexname",
- tbinfo->dobj.catId.oid);
- }
- else if (fout->remoteVersion >= 70300)
- {
- appendPQExpBuffer(query,
- "SELECT t.tableoid, t.oid, "
- "t.relname AS indexname, "
- "pg_catalog.pg_get_indexdef(i.indexrelid) AS indexdef, "
- "t.relnatts AS indnkeys, "
- "i.indkey, i.indisclustered, "
- "false AS indisreplident, t.relpages, "
- "c.contype, c.conname, "
- "c.condeferrable, c.condeferred, "
- "c.tableoid AS contableoid, "
- "c.oid AS conoid, "
- "null AS condef, "
- "NULL AS tablespace, "
- "null AS options "
- "FROM pg_catalog.pg_index i "
- "JOIN pg_catalog.pg_class t ON (t.oid = i.indexrelid) "
- "LEFT JOIN pg_catalog.pg_depend d "
- "ON (d.classid = t.tableoid "
- "AND d.objid = t.oid "
- "AND d.deptype = 'i') "
- "LEFT JOIN pg_catalog.pg_constraint c "
- "ON (d.refclassid = c.tableoid "
- "AND d.refobjid = c.oid) "
- "WHERE i.indrelid = '%u'::pg_catalog.oid "
- "ORDER BY indexname",
- tbinfo->dobj.catId.oid);
- }
- else if (fout->remoteVersion >= 70100)
- {
- appendPQExpBuffer(query,
- "SELECT t.tableoid, t.oid, "
- "t.relname AS indexname, "
- "pg_get_indexdef(i.indexrelid) AS indexdef, "
- "t.relnatts AS indnkeys, "
- "i.indkey, false AS indisclustered, "
- "false AS indisreplident, t.relpages, "
- "CASE WHEN i.indisprimary THEN 'p'::char "
- "ELSE '0'::char END AS contype, "
- "t.relname AS conname, "
- "false AS condeferrable, "
- "false AS condeferred, "
- "0::oid AS contableoid, "
- "t.oid AS conoid, "
- "null AS condef, "
- "NULL AS tablespace, "
- "null AS options "
- "FROM pg_index i, pg_class t "
- "WHERE t.oid = i.indexrelid "
- "AND i.indrelid = '%u'::oid "
- "ORDER BY indexname",
- tbinfo->dobj.catId.oid);
- }
- else
- {
- appendPQExpBuffer(query,
- "SELECT "
- "(SELECT oid FROM pg_class WHERE relname = 'pg_class') AS tableoid, "
- "t.oid, "
- "t.relname AS indexname, "
- "pg_get_indexdef(i.indexrelid) AS indexdef, "
- "t.relnatts AS indnkeys, "
- "i.indkey, false AS indisclustered, "
- "false AS indisreplident, t.relpages, "
- "CASE WHEN i.indisprimary THEN 'p'::char "
- "ELSE '0'::char END AS contype, "
- "t.relname AS conname, "
- "false AS condeferrable, "
- "false AS condeferred, "
- "0::oid AS contableoid, "
- "t.oid AS conoid, "
- "null AS condef, "
- "NULL AS tablespace, "
- "null AS options "
- "FROM pg_index i, pg_class t "
- "WHERE t.oid = i.indexrelid "
- "AND i.indrelid = '%u'::oid "
- "ORDER BY indexname",
- tbinfo->dobj.catId.oid);
- }
- res = ExecuteSqlQuery(fout, query->data, PGRES_TUPLES_OK);
- ntups = PQntuples(res);
- i_tableoid = PQfnumber(res, "tableoid");
- i_oid = PQfnumber(res, "oid");
- i_indexname = PQfnumber(res, "indexname");
- i_indexdef = PQfnumber(res, "indexdef");
- i_indnkeys = PQfnumber(res, "indnkeys");
- i_indkey = PQfnumber(res, "indkey");
- i_indisclustered = PQfnumber(res, "indisclustered");
- i_indisreplident = PQfnumber(res, "indisreplident");
- i_relpages = PQfnumber(res, "relpages");
- i_contype = PQfnumber(res, "contype");
- i_conname = PQfnumber(res, "conname");
- i_condeferrable = PQfnumber(res, "condeferrable");
- i_condeferred = PQfnumber(res, "condeferred");
- i_contableoid = PQfnumber(res, "contableoid");
- i_conoid = PQfnumber(res, "conoid");
- i_condef = PQfnumber(res, "condef");
- i_tablespace = PQfnumber(res, "tablespace");
- i_options = PQfnumber(res, "options");
- indxinfo = (IndxInfo *) pg_malloc(ntups * sizeof(IndxInfo));
- constrinfo = (ConstraintInfo *) pg_malloc(ntups * sizeof(ConstraintInfo));
- for (j = 0; j < ntups; j++)
- {
- char contype;
- indxinfo[j].dobj.objType = DO_INDEX;
- indxinfo[j].dobj.catId.tableoid = atooid(PQgetvalue(res, j, i_tableoid));
- indxinfo[j].dobj.catId.oid = atooid(PQgetvalue(res, j, i_oid));
- AssignDumpId(&indxinfo[j].dobj);
- indxinfo[j].dobj.name = pg_strdup(PQgetvalue(res, j, i_indexname));
- indxinfo[j].dobj.namespace = tbinfo->dobj.namespace;
- indxinfo[j].indextable = tbinfo;
- indxinfo[j].indexdef = pg_strdup(PQgetvalue(res, j, i_indexdef));
- indxinfo[j].indnkeys = atoi(PQgetvalue(res, j, i_indnkeys));
- indxinfo[j].tablespace = pg_strdup(PQgetvalue(res, j, i_tablespace));
- indxinfo[j].options = pg_strdup(PQgetvalue(res, j, i_options));
- /*
- * In pre-7.4 releases, indkeys may contain more entries than
- * indnkeys says (since indnkeys will be 1 for a functional
- * index). We don't actually care about this case since we don't
- * examine indkeys except for indexes associated with PRIMARY and
- * UNIQUE constraints, which are never functional indexes. But we
- * have to allocate enough space to keep parseOidArray from
- * complaining.
- */
- indxinfo[j].indkeys = (Oid *) pg_malloc(INDEX_MAX_KEYS * sizeof(Oid));
- parseOidArray(PQgetvalue(res, j, i_indkey),
- indxinfo[j].indkeys, INDEX_MAX_KEYS);
- indxinfo[j].indisclustered = (PQgetvalue(res, j, i_indisclustered)[0] == 't');
- indxinfo[j].indisreplident = (PQgetvalue(res, j, i_indisreplident)[0] == 't');
- indxinfo[j].relpages = atoi(PQgetvalue(res, j, i_relpages));
- contype = *(PQgetvalue(res, j, i_contype));
- if (contype == 'p' || contype == 'u' || contype == 'x')
- {
- /*
- * If we found a constraint matching the index, create an
- * entry for it.
- *
- * In a pre-7.3 database, we take this path iff the index was
- * marked indisprimary.
- */
- constrinfo[j].dobj.objType = DO_CONSTRAINT;
- constrinfo[j].dobj.catId.tableoid = atooid(PQgetvalue(res, j, i_contableoid));
- constrinfo[j].dobj.catId.oid = atooid(PQgetvalue(res, j, i_conoid));
- AssignDumpId(&constrinfo[j].dobj);
- constrinfo[j].dobj.name = pg_strdup(PQgetvalue(res, j, i_conname));
- constrinfo[j].dobj.namespace = tbinfo->dobj.namespace;
- constrinfo[j].contable = tbinfo;
- constrinfo[j].condomain = NULL;
- constrinfo[j].contype = contype;
- if (contype == 'x')
- constrinfo[j].condef = pg_strdup(PQgetvalue(res, j, i_condef));
- else
- constrinfo[j].condef = NULL;
- constrinfo[j].confrelid = InvalidOid;
- constrinfo[j].conindex = indxinfo[j].dobj.dumpId;
- constrinfo[j].condeferrable = *(PQgetvalue(res, j, i_condeferrable)) == 't';
- constrinfo[j].condeferred = *(PQgetvalue(res, j, i_condeferred)) == 't';
- constrinfo[j].conislocal = true;
- constrinfo[j].separate = true;
- indxinfo[j].indexconstraint = constrinfo[j].dobj.dumpId;
- /* If pre-7.3 DB, better make sure table comes first */
- addObjectDependency(&constrinfo[j].dobj,
- tbinfo->dobj.dumpId);
- }
- else
- {
- /* Plain secondary index */
- indxinfo[j].indexconstraint = 0;
- }
- }
- PQclear(res);
- }
- destroyPQExpBuffer(query);
- }
- /*
- * getConstraints
- *
- * Get info about constraints on dumpable tables.
- *
- * Currently handles foreign keys only.
- * Unique and primary key constraints are handled with indexes,
- * while check constraints are processed in getTableAttrs().
- */
- void
- getConstraints(Archive *fout, TableInfo tblinfo[], int numTables)
- {
- int i,
- j;
- ConstraintInfo *constrinfo;
- PQExpBuffer query;
- PGresult *res;
- int i_contableoid,
- i_conoid,
- i_conname,
- i_confrelid,
- i_condef;
- int ntups;
- /* pg_constraint was created in 7.3, so nothing to do if older */
- if (fout->remoteVersion < 70300)
- return;
- query = createPQExpBuffer();
- for (i = 0; i < numTables; i++)
- {
- TableInfo *tbinfo = &tblinfo[i];
- if (!tbinfo->hastriggers || !tbinfo->dobj.dump)
- continue;
- if (g_verbose)
- write_msg(NULL, "reading foreign key constraints for table \"%s\"\n",
- tbinfo->dobj.name);
- /*
- * select table schema to ensure constraint expr is qualified if
- * needed
- */
- selectSourceSchema(fout, tbinfo->dobj.namespace->dobj.name);
- resetPQExpBuffer(query);
- appendPQExpBuffer(query,
- "SELECT tableoid, oid, conname, confrelid, "
- "pg_catalog.pg_get_constraintdef(oid) AS condef "
- "FROM pg_catalog.pg_constraint "
- "WHERE conrelid = '%u'::pg_catalog.oid "
- "AND contype = 'f'",
- tbinfo->dobj.catId.oid);
- res = ExecuteSqlQuery(fout, query->data, PGRES_TUPLES_OK);
- ntups = PQntuples(res);
- i_contableoid = PQfnumber(res, "tableoid");
- i_conoid = PQfnumber(res, "oid");
- i_conname = PQfnumber(res, "conname");
- i_confrelid = PQfnumber(res, "confrelid");
- i_condef = PQfnumber(res, "condef");
- constrinfo = (ConstraintInfo *) pg_malloc(ntups * sizeof(ConstraintInfo));
- for (j = 0; j < ntups; j++)
- {
- constrinfo[j].dobj.objType = DO_FK_CONSTRAINT;
- constrinfo[j].dobj.catId.tableoid = atooid(PQgetvalue(res, j, i_contableoid));
- constrinfo[j].dobj.catId.oid = atooid(PQgetvalue(res, j, i_conoid));
- AssignDumpId(&constrinfo[j].dobj);
- constrinfo[j].dobj.name = pg_strdup(PQgetvalue(res, j, i_conname));
- constrinfo[j].dobj.namespace = tbinfo->dobj.namespace;
- constrinfo[j].contable = tbinfo;
- constrinfo[j].condomain = NULL;
- constrinfo[j].contype = 'f';
- constrinfo[j].condef = pg_strdup(PQgetvalue(res, j, i_condef));
- constrinfo[j].confrelid = atooid(PQgetvalue(res, j, i_confrelid));
- constrinfo[j].conindex = 0;
- constrinfo[j].condeferrable = false;
- constrinfo[j].condeferred = false;
- constrinfo[j].conislocal = true;
- constrinfo[j].separate = true;
- }
- PQclear(res);
- }
- destroyPQExpBuffer(query);
- }
- /*
- * getDomainConstraints
- *
- * Get info about constraints on a domain.
- */
- static void
- getDomainConstraints(Archive *fout, TypeInfo *tyinfo)
- {
- int i;
- ConstraintInfo *constrinfo;
- PQExpBuffer query;
- PGresult *res;
- int i_tableoid,
- i_oid,
- i_conname,
- i_consrc;
- int ntups;
- /* pg_constraint was created in 7.3, so nothing to do if older */
- if (fout->remoteVersion < 70300)
- return;
- /*
- * select appropriate schema to ensure names in constraint are properly
- * qualified
- */
- selectSourceSchema(fout, tyinfo->dobj.namespace->dobj.name);
- query = createPQExpBuffer();
- if (fout->remoteVersion >= 90100)
- appendPQExpBuffer(query, "SELECT tableoid, oid, conname, "
- "pg_catalog.pg_get_constraintdef(oid) AS consrc, "
- "convalidated "
- "FROM pg_catalog.pg_constraint "
- "WHERE contypid = '%u'::pg_catalog.oid "
- "ORDER BY conname",
- tyinfo->dobj.catId.oid);
- else if (fout->remoteVersion >= 70400)
- appendPQExpBuffer(query, "SELECT tableoid, oid, conname, "
- "pg_catalog.pg_get_constraintdef(oid) AS consrc, "
- "true as convalidated "
- "FROM pg_catalog.pg_constraint "
- "WHERE contypid = '%u'::pg_catalog.oid "
- "ORDER BY conname",
- tyinfo->dobj.catId.oid);
- else
- appendPQExpBuffer(query, "SELECT tableoid, oid, conname, "
- "'CHECK (' || consrc || ')' AS consrc, "
- "true as convalidated "
- "FROM pg_catalog.pg_constraint "
- "WHERE contypid = '%u'::pg_catalog.oid "
- "ORDER BY conname",
- tyinfo->dobj.catId.oid);
- res = ExecuteSqlQuery(fout, query->data, PGRES_TUPLES_OK);
- ntups = PQntuples(res);
- i_tableoid = PQfnumber(res, "tableoid");
- i_oid = PQfnumber(res, "oid");
- i_conname = PQfnumber(res, "conname");
- i_consrc = PQfnumber(res, "consrc");
- constrinfo = (ConstraintInfo *) pg_malloc(ntups * sizeof(ConstraintInfo));
- tyinfo->nDomChecks = ntups;
- tyinfo->domChecks = constrinfo;
- for (i = 0; i < ntups; i++)
- {
- bool validated = PQgetvalue(res, i, 4)[0] == 't';
- constrinfo[i].dobj.objType = DO_CONSTRAINT;
- constrinfo[i].dobj.catId.tableoid = atooid(PQgetvalue(res, i, i_tableoid));
- constrinfo[i].dobj.catId.oid = atooid(PQgetvalue(res, i, i_oid));
- AssignDumpId(&constrinfo[i].dobj);
- constrinfo[i].dobj.name = pg_strdup(PQgetvalue(res, i, i_conname));
- constrinfo[i].dobj.namespace = tyinfo->dobj.namespace;
- constrinfo[i].contable = NULL;
- constrinfo[i].condomain = tyinfo;
- constrinfo[i].contype = 'c';
- constrinfo[i].condef = pg_strdup(PQgetvalue(res, i, i_consrc));
- constrinfo[i].confrelid = InvalidOid;
- constrinfo[i].conindex = 0;
- constrinfo[i].condeferrable = false;
- constrinfo[i].condeferred = false;
- constrinfo[i].conislocal = true;
- constrinfo[i].separate = !validated;
- /*
- * Make the domain depend on the constraint, ensuring it won't be
- * output till any constraint dependencies are OK. If the constraint
- * has not been validated, it's going to be dumped after the domain
- * anyway, so this doesn't matter.
- */
- if (validated)
- addObjectDependency(&tyinfo->dobj,
- constrinfo[i].dobj.dumpId);
- }
- PQclear(res);
- destroyPQExpBuffer(query);
- }
- /*
- * getRules
- * get basic information about every rule in the system
- *
- * numRules is set to the number of rules read in
- */
- RuleInfo *
- getRules(Archive *fout, int *numRules)
- {
- PGresult *res;
- int ntups;
- int i;
- PQExpBuffer query = createPQExpBuffer();
- RuleInfo *ruleinfo;
- int i_tableoid;
- int i_oid;
- int i_rulename;
- int i_ruletable;
- int i_ev_type;
- int i_is_instead;
- int i_ev_enabled;
- /* Make sure we are in proper schema */
- selectSourceSchema(fout, "pg_catalog");
- if (fout->remoteVersion >= 80300)
- {
- appendPQExpBufferStr(query, "SELECT "
- "tableoid, oid, rulename, "
- "ev_class AS ruletable, ev_type, is_instead, "
- "ev_enabled "
- "FROM pg_rewrite "
- "ORDER BY oid");
- }
- else if (fout->remoteVersion >= 70100)
- {
- appendPQExpBufferStr(query, "SELECT "
- "tableoid, oid, rulename, "
- "ev_class AS ruletable, ev_type, is_instead, "
- "'O'::char AS ev_enabled "
- "FROM pg_rewrite "
- "ORDER BY oid");
- }
- else
- {
- appendPQExpBufferStr(query, "SELECT "
- "(SELECT oid FROM pg_class WHERE relname = 'pg_rewrite') AS tableoid, "
- "oid, rulename, "
- "ev_class AS ruletable, ev_type, is_instead, "
- "'O'::char AS ev_enabled "
- "FROM pg_rewrite "
- "ORDER BY oid");
- }
- res = ExecuteSqlQuery(fout, query->data, PGRES_TUPLES_OK);
- ntups = PQntuples(res);
- *numRules = ntups;
- ruleinfo = (RuleInfo *) pg_malloc(ntups * sizeof(RuleInfo));
- i_tableoid = PQfnumber(res, "tableoid");
- i_oid = PQfnumber(res, "oid");
- i_rulename = PQfnumber(res, "rulename");
- i_ruletable = PQfnumber(res, "ruletable");
- i_ev_type = PQfnumber(res, "ev_type");
- i_is_instead = PQfnumber(res, "is_instead");
- i_ev_enabled = PQfnumber(res, "ev_enabled");
- for (i = 0; i < ntups; i++)
- {
- Oid ruletableoid;
- ruleinfo[i].dobj.objType = DO_RULE;
- ruleinfo[i].dobj.catId.tableoid = atooid(PQgetvalue(res, i, i_tableoid));
- ruleinfo[i].dobj.catId.oid = atooid(PQgetvalue(res, i, i_oid));
- AssignDumpId(&ruleinfo[i].dobj);
- ruleinfo[i].dobj.name = pg_strdup(PQgetvalue(res, i, i_rulename));
- ruletableoid = atooid(PQgetvalue(res, i, i_ruletable));
- ruleinfo[i].ruletable = findTableByOid(ruletableoid);
- if (ruleinfo[i].ruletable == NULL)
- exit_horribly(NULL, "failed sanity check, parent table OID %u of pg_rewrite entry OID %u not found\n",
- ruletableoid, ruleinfo[i].dobj.catId.oid);
- ruleinfo[i].dobj.namespace = ruleinfo[i].ruletable->dobj.namespace;
- ruleinfo[i].dobj.dump = ruleinfo[i].ruletable->dobj.dump;
- ruleinfo[i].ev_type = *(PQgetvalue(res, i, i_ev_type));
- ruleinfo[i].is_instead = *(PQgetvalue(res, i, i_is_instead)) == 't';
- ruleinfo[i].ev_enabled = *(PQgetvalue(res, i, i_ev_enabled));
- if (ruleinfo[i].ruletable)
- {
- /*
- * If the table is a view or materialized view, force its ON
- * SELECT rule to be sorted before the view itself --- this
- * ensures that any dependencies for the rule affect the table's
- * positioning. Other rules are forced to appear after their
- * table.
- */
- if ((ruleinfo[i].ruletable->relkind == RELKIND_VIEW ||
- ruleinfo[i].ruletable->relkind == RELKIND_MATVIEW) &&
- ruleinfo[i].ev_type == '1' && ruleinfo[i].is_instead)
- {
- addObjectDependency(&ruleinfo[i].ruletable->dobj,
- ruleinfo[i].dobj.dumpId);
- /* We'll merge the rule into CREATE VIEW, if possible */
- ruleinfo[i].separate = false;
- }
- else
- {
- addObjectDependency(&ruleinfo[i].dobj,
- ruleinfo[i].ruletable->dobj.dumpId);
- ruleinfo[i].separate = true;
- }
- }
- else
- ruleinfo[i].separate = true;
- /*
- * If we're forced to break a dependency loop by dumping a view as a
- * table and separate _RETURN rule, we'll move the view's reloptions
- * to the rule. (This is necessary because tables and views have
- * different valid reloptions, so we can't apply the options until the
- * backend knows it's a view.) Otherwise the rule's reloptions stay
- * NULL.
- */
- ruleinfo[i].reloptions = NULL;
- }
- PQclear(res);
- destroyPQExpBuffer(query);
- return ruleinfo;
- }
- /*
- * getTriggers
- * get information about every trigger on a dumpable table
- *
- * Note: trigger data is not returned directly to the caller, but it
- * does get entered into the DumpableObject tables.
- */
- void
- getTriggers(Archive *fout, TableInfo tblinfo[], int numTables)
- {
- int i,
- j;
- PQExpBuffer query = createPQExpBuffer();
- PGresult *res;
- TriggerInfo *tginfo;
- int i_tableoid,
- i_oid,
- i_tgname,
- i_tgfname,
- i_tgtype,
- i_tgnargs,
- i_tgargs,
- i_tgisconstraint,
- i_tgconstrname,
- i_tgconstrrelid,
- i_tgconstrrelname,
- i_tgenabled,
- i_tgdeferrable,
- i_tginitdeferred,
- i_tgdef;
- int ntups;
- for (i = 0; i < numTables; i++)
- {
- TableInfo *tbinfo = &tblinfo[i];
- if (!tbinfo->hastriggers || !tbinfo->dobj.dump)
- continue;
- if (g_verbose)
- write_msg(NULL, "reading triggers for table \"%s\"\n",
- tbinfo->dobj.name);
- /*
- * select table schema to ensure regproc name is qualified if needed
- */
- selectSourceSchema(fout, tbinfo->dobj.namespace->dobj.name);
- resetPQExpBuffer(query);
- if (fout->remoteVersion >= 90000)
- {
- /*
- * NB: think not to use pretty=true in pg_get_triggerdef. It
- * could result in non-forward-compatible dumps of WHEN clauses
- * due to under-parenthesization.
- */
- appendPQExpBuffer(query,
- "SELECT tgname, "
- "tgfoid::pg_catalog.regproc AS tgfname, "
- "pg_catalog.pg_get_triggerdef(oid, false) AS tgdef, "
- "tgenabled, tableoid, oid "
- "FROM pg_catalog.pg_trigger t "
- "WHERE tgrelid = '%u'::pg_catalog.oid "
- "AND NOT tgisinternal",
- tbinfo->dobj.catId.oid);
- }
- else if (fout->remoteVersion >= 80300)
- {
- /*
- * We ignore triggers that are tied to a foreign-key constraint
- */
- appendPQExpBuffer(query,
- "SELECT tgname, "
- "tgfoid::pg_catalog.regproc AS tgfname, "
- "tgtype, tgnargs, tgargs, tgenabled, "
- "tgisconstraint, tgconstrname, tgdeferrable, "
- "tgconstrrelid, tginitdeferred, tableoid, oid, "
- "tgconstrrelid::pg_catalog.regclass AS tgconstrrelname "
- "FROM pg_catalog.pg_trigger t "
- "WHERE tgrelid = '%u'::pg_catalog.oid "
- "AND tgconstraint = 0",
- tbinfo->dobj.catId.oid);
- }
- else if (fout->remoteVersion >= 70300)
- {
- /*
- * We ignore triggers that are tied to a foreign-key constraint,
- * but in these versions we have to grovel through pg_constraint
- * to find out
- */
- appendPQExpBuffer(query,
- "SELECT tgname, "
- "tgfoid::pg_catalog.regproc AS tgfname, "
- "tgtype, tgnargs, tgargs, tgenabled, "
- "tgisconstraint, tgconstrname, tgdeferrable, "
- "tgconstrrelid, tginitdeferred, tableoid, oid, "
- "tgconstrrelid::pg_catalog.regclass AS tgconstrrelname "
- "FROM pg_catalog.pg_trigger t "
- "WHERE tgrelid = '%u'::pg_catalog.oid "
- "AND (NOT tgisconstraint "
- " OR NOT EXISTS"
- " (SELECT 1 FROM pg_catalog.pg_depend d "
- " JOIN pg_catalog.pg_constraint c ON (d.refclassid = c.tableoid AND d.refobjid = c.oid) "
- " WHERE d.classid = t.tableoid AND d.objid = t.oid AND d.deptype = 'i' AND c.contype = 'f'))",
- tbinfo->dobj.catId.oid);
- }
- else if (fout->remoteVersion >= 70100)
- {
- appendPQExpBuffer(query,
- "SELECT tgname, tgfoid::regproc AS tgfname, "
- "tgtype, tgnargs, tgargs, tgenabled, "
- "tgisconstraint, tgconstrname, tgdeferrable, "
- "tgconstrrelid, tginitdeferred, tableoid, oid, "
- "(SELECT relname FROM pg_class WHERE oid = tgconstrrelid) "
- " AS tgconstrrelname "
- "FROM pg_trigger "
- "WHERE tgrelid = '%u'::oid",
- tbinfo->dobj.catId.oid);
- }
- else
- {
- appendPQExpBuffer(query,
- "SELECT tgname, tgfoid::regproc AS tgfname, "
- "tgtype, tgnargs, tgargs, tgenabled, "
- "tgisconstraint, tgconstrname, tgdeferrable, "
- "tgconstrrelid, tginitdeferred, "
- "(SELECT oid FROM pg_class WHERE relname = 'pg_trigger') AS tableoid, "
- "oid, "
- "(SELECT relname FROM pg_class WHERE oid = tgconstrrelid) "
- " AS tgconstrrelname "
- "FROM pg_trigger "
- "WHERE tgrelid = '%u'::oid",
- tbinfo->dobj.catId.oid);
- }
- res = ExecuteSqlQuery(fout, query->data, PGRES_TUPLES_OK);
- ntups = PQntuples(res);
- i_tableoid = PQfnumber(res, "tableoid");
- i_oid = PQfnumber(res, "oid");
- i_tgname = PQfnumber(res, "tgname");
- i_tgfname = PQfnumber(res, "tgfname");
- i_tgtype = PQfnumber(res, "tgtype");
- i_tgnargs = PQfnumber(res, "tgnargs");
- i_tgargs = PQfnumber(res, "tgargs");
- i_tgisconstraint = PQfnumber(res, "tgisconstraint");
- i_tgconstrname = PQfnumber(res, "tgconstrname");
- i_tgconstrrelid = PQfnumber(res, "tgconstrrelid");
- i_tgconstrrelname = PQfnumber(res, "tgconstrrelname");
- i_tgenabled = PQfnumber(res, "tgenabled");
- i_tgdeferrable = PQfnumber(res, "tgdeferrable");
- i_tginitdeferred = PQfnumber(res, "tginitdeferred");
- i_tgdef = PQfnumber(res, "tgdef");
- tginfo = (TriggerInfo *) pg_malloc(ntups * sizeof(TriggerInfo));
- for (j = 0; j < ntups; j++)
- {
- tginfo[j].dobj.objType = DO_TRIGGER;
- tginfo[j].dobj.catId.tableoid = atooid(PQgetvalue(res, j, i_tableoid));
- tginfo[j].dobj.catId.oid = atooid(PQgetvalue(res, j, i_oid));
- AssignDumpId(&tginfo[j].dobj);
- tginfo[j].dobj.name = pg_strdup(PQgetvalue(res, j, i_tgname));
- tginfo[j].dobj.namespace = tbinfo->dobj.namespace;
- tginfo[j].tgtable = tbinfo;
- tginfo[j].tgenabled = *(PQgetvalue(res, j, i_tgenabled));
- if (i_tgdef >= 0)
- {
- tginfo[j].tgdef = pg_strdup(PQgetvalue(res, j, i_tgdef));
- /* remaining fields are not valid if we have tgdef */
- tginfo[j].tgfname = NULL;
- tginfo[j].tgtype = 0;
- tginfo[j].tgnargs = 0;
- tginfo[j].tgargs = NULL;
- tginfo[j].tgisconstraint = false;
- tginfo[j].tgdeferrable = false;
- tginfo[j].tginitdeferred = false;
- tginfo[j].tgconstrname = NULL;
- tginfo[j].tgconstrrelid = InvalidOid;
- tginfo[j].tgconstrrelname = NULL;
- }
- else
- {
- tginfo[j].tgdef = NULL;
- tginfo[j].tgfname = pg_strdup(PQgetvalue(res, j, i_tgfname));
- tginfo[j].tgtype = atoi(PQgetvalue(res, j, i_tgtype));
- tginfo[j].tgnargs = atoi(PQgetvalue(res, j, i_tgnargs));
- tginfo[j].tgargs = pg_strdup(PQgetvalue(res, j, i_tgargs));
- tginfo[j].tgisconstraint = *(PQgetvalue(res, j, i_tgisconstraint)) == 't';
- tginfo[j].tgdeferrable = *(PQgetvalue(res, j, i_tgdeferrable)) == 't';
- tginfo[j].tginitdeferred = *(PQgetvalue(res, j, i_tginitdeferred)) == 't';
- if (tginfo[j].tgisconstraint)
- {
- tginfo[j].tgconstrname = pg_strdup(PQgetvalue(res, j, i_tgconstrname));
- tginfo[j].tgconstrrelid = atooid(PQgetvalue(res, j, i_tgconstrrelid));
- if (OidIsValid(tginfo[j].tgconstrrelid))
- {
- if (PQgetisnull(res, j, i_tgconstrrelname))
- exit_horribly(NULL, "query produced null referenced table name for foreign key trigger \"%s\" on table \"%s\" (OID of table: %u)\n",
- tginfo[j].dobj.name,
- tbinfo->dobj.name,
- tginfo[j].tgconstrrelid);
- tginfo[j].tgconstrrelname = pg_strdup(PQgetvalue(res, j, i_tgconstrrelname));
- }
- else
- tginfo[j].tgconstrrelname = NULL;
- }
- else
- {
- tginfo[j].tgconstrname = NULL;
- tginfo[j].tgconstrrelid = InvalidOid;
- tginfo[j].tgconstrrelname = NULL;
- }
- }
- }
- PQclear(res);
- }
- destroyPQExpBuffer(query);
- }
- /*
- * getEventTriggers
- * get information about event triggers
- */
- EventTriggerInfo *
- getEventTriggers(Archive *fout, int *numEventTriggers)
- {
- int i;
- PQExpBuffer query;
- PGresult *res;
- EventTriggerInfo *evtinfo;
- int i_tableoid,
- i_oid,
- i_evtname,
- i_evtevent,
- i_evtowner,
- i_evttags,
- i_evtfname,
- i_evtenabled;
- int ntups;
- /* Before 9.3, there are no event triggers */
- if (fout->remoteVersion < 90300)
- {
- *numEventTriggers = 0;
- return NULL;
- }
- query = createPQExpBuffer();
- /* Make sure we are in proper schema */
- selectSourceSchema(fout, "pg_catalog");
- appendPQExpBuffer(query,
- "SELECT e.tableoid, e.oid, evtname, evtenabled, "
- "evtevent, (%s evtowner) AS evtowner, "
- "array_to_string(array("
- "select quote_literal(x) "
- " from unnest(evttags) as t(x)), ', ') as evttags, "
- "e.evtfoid::regproc as evtfname "
- "FROM pg_event_trigger e "
- "ORDER BY e.oid",
- username_subquery);
- res = ExecuteSqlQuery(fout, query->data, PGRES_TUPLES_OK);
- ntups = PQntuples(res);
- *numEventTriggers = ntups;
- evtinfo = (EventTriggerInfo *) pg_malloc(ntups * sizeof(EventTriggerInfo));
- i_tableoid = PQfnumber(res, "tableoid");
- i_oid = PQfnumber(res, "oid");
- i_evtname = PQfnumber(res, "evtname");
- i_evtevent = PQfnumber(res, "evtevent");
- i_evtowner = PQfnumber(res, "evtowner");
- i_evttags = PQfnumber(res, "evttags");
- i_evtfname = PQfnumber(res, "evtfname");
- i_evtenabled = PQfnumber(res, "evtenabled");
- for (i = 0; i < ntups; i++)
- {
- evtinfo[i].dobj.objType = DO_EVENT_TRIGGER;
- evtinfo[i].dobj.catId.tableoid = atooid(PQgetvalue(res, i, i_tableoid));
- evtinfo[i].dobj.catId.oid = atooid(PQgetvalue(res, i, i_oid));
- AssignDumpId(&evtinfo[i].dobj);
- evtinfo[i].dobj.name = pg_strdup(PQgetvalue(res, i, i_evtname));
- evtinfo[i].evtname = pg_strdup(PQgetvalue(res, i, i_evtname));
- evtinfo[i].evtevent = pg_strdup(PQgetvalue(res, i, i_evtevent));
- evtinfo[i].evtowner = pg_strdup(PQgetvalue(res, i, i_evtowner));
- evtinfo[i].evttags = pg_strdup(PQgetvalue(res, i, i_evttags));
- evtinfo[i].evtfname = pg_strdup(PQgetvalue(res, i, i_evtfname));
- evtinfo[i].evtenabled = *(PQgetvalue(res, i, i_evtenabled));
- }
- PQclear(res);
- destroyPQExpBuffer(query);
- return evtinfo;
- }
- /*
- * getProcLangs
- * get basic information about every procedural language in the system
- *
- * numProcLangs is set to the number of langs read in
- *
- * NB: this must run after getFuncs() because we assume we can do
- * findFuncByOid().
- */
- ProcLangInfo *
- getProcLangs(Archive *fout, int *numProcLangs)
- {
- PGresult *res;
- int ntups;
- int i;
- PQExpBuffer query = createPQExpBuffer();
- ProcLangInfo *planginfo;
- int i_tableoid;
- int i_oid;
- int i_lanname;
- int i_lanpltrusted;
- int i_lanplcallfoid;
- int i_laninline;
- int i_lanvalidator;
- int i_lanacl;
- int i_lanowner;
- /* Make sure we are in proper schema */
- selectSourceSchema(fout, "pg_catalog");
- if (fout->remoteVersion >= 90000)
- {
- /* pg_language has a laninline column */
- appendPQExpBuffer(query, "SELECT tableoid, oid, "
- "lanname, lanpltrusted, lanplcallfoid, "
- "laninline, lanvalidator, lanacl, "
- "(%s lanowner) AS lanowner "
- "FROM pg_language "
- "WHERE lanispl "
- "ORDER BY oid",
- username_subquery);
- }
- else if (fout->remoteVersion >= 80300)
- {
- /* pg_language has a lanowner column */
- appendPQExpBuffer(query, "SELECT tableoid, oid, "
- "lanname, lanpltrusted, lanplcallfoid, "
- "lanvalidator, lanacl, "
- "(%s lanowner) AS lanowner "
- "FROM pg_language "
- "WHERE lanispl "
- "ORDER BY oid",
- username_subquery);
- }
- else if (fout->remoteVersion >= 80100)
- {
- /* Languages are owned by the bootstrap superuser, OID 10 */
- appendPQExpBuffer(query, "SELECT tableoid, oid, *, "
- "(%s '10') AS lanowner "
- "FROM pg_language "
- "WHERE lanispl "
- "ORDER BY oid",
- username_subquery);
- }
- else if (fout->remoteVersion >= 70400)
- {
- /* Languages are owned by the bootstrap superuser, sysid 1 */
- appendPQExpBuffer(query, "SELECT tableoid, oid, *, "
- "(%s '1') AS lanowner "
- "FROM pg_language "
- "WHERE lanispl "
- "ORDER BY oid",
- username_subquery);
- }
- else if (fout->remoteVersion >= 70100)
- {
- /* No clear notion of an owner at all before 7.4 ... */
- appendPQExpBufferStr(query, "SELECT tableoid, oid, * FROM pg_language "
- "WHERE lanispl "
- "ORDER BY oid");
- }
- else
- {
- appendPQExpBufferStr(query, "SELECT "
- "(SELECT oid FROM pg_class WHERE relname = 'pg_language') AS tableoid, "
- "oid, * FROM pg_language "
- "WHERE lanispl "
- "ORDER BY oid");
- }
- res = ExecuteSqlQuery(fout, query->data, PGRES_TUPLES_OK);
- ntups = PQntuples(res);
- *numProcLangs = ntups;
- planginfo = (ProcLangInfo *) pg_malloc(ntups * sizeof(ProcLangInfo));
- i_tableoid = PQfnumber(res, "tableoid");
- i_oid = PQfnumber(res, "oid");
- i_lanname = PQfnumber(res, "lanname");
- i_lanpltrusted = PQfnumber(res, "lanpltrusted");
- i_lanplcallfoid = PQfnumber(res, "lanplcallfoid");
- /* these may fail and return -1: */
- i_laninline = PQfnumber(res, "laninline");
- i_lanvalidator = PQfnumber(res, "lanvalidator");
- i_lanacl = PQfnumber(res, "lanacl");
- i_lanowner = PQfnumber(res, "lanowner");
- for (i = 0; i < ntups; i++)
- {
- planginfo[i].dobj.objType = DO_PROCLANG;
- planginfo[i].dobj.catId.tableoid = atooid(PQgetvalue(res, i, i_tableoid));
- planginfo[i].dobj.catId.oid = atooid(PQgetvalue(res, i, i_oid));
- AssignDumpId(&planginfo[i].dobj);
- planginfo[i].dobj.name = pg_strdup(PQgetvalue(res, i, i_lanname));
- planginfo[i].lanpltrusted = *(PQgetvalue(res, i, i_lanpltrusted)) == 't';
- planginfo[i].lanplcallfoid = atooid(PQgetvalue(res, i, i_lanplcallfoid));
- if (i_laninline >= 0)
- planginfo[i].laninline = atooid(PQgetvalue(res, i, i_laninline));
- else
- planginfo[i].laninline = InvalidOid;
- if (i_lanvalidator >= 0)
- planginfo[i].lanvalidator = atooid(PQgetvalue(res, i, i_lanvalidator));
- else
- planginfo[i].lanvalidator = InvalidOid;
- if (i_lanacl >= 0)
- planginfo[i].lanacl = pg_strdup(PQgetvalue(res, i, i_lanacl));
- else
- planginfo[i].lanacl = pg_strdup("{=U}");
- if (i_lanowner >= 0)
- planginfo[i].lanowner = pg_strdup(PQgetvalue(res, i, i_lanowner));
- else
- planginfo[i].lanowner = pg_strdup("");
- if (fout->remoteVersion < 70300)
- {
- /*
- * We need to make a dependency to ensure the function will be
- * dumped first. (In 7.3 and later the regular dependency
- * mechanism will handle this for us.)
- */
- FuncInfo *funcInfo = findFuncByOid(planginfo[i].lanplcallfoid);
- if (funcInfo)
- addObjectDependency(&planginfo[i].dobj,
- funcInfo->dobj.dumpId);
- }
- }
- PQclear(res);
- destroyPQExpBuffer(query);
- return planginfo;
- }
- /*
- * getCasts
- * get basic information about every cast in the system
- *
- * numCasts is set to the number of casts read in
- */
- CastInfo *
- getCasts(Archive *fout, int *numCasts)
- {
- PGresult *res;
- int ntups;
- int i;
- PQExpBuffer query = createPQExpBuffer();
- CastInfo *castinfo;
- int i_tableoid;
- int i_oid;
- int i_castsource;
- int i_casttarget;
- int i_castfunc;
- int i_castcontext;
- int i_castmethod;
- /* Make sure we are in proper schema */
- selectSourceSchema(fout, "pg_catalog");
- if (fout->remoteVersion >= 80400)
- {
- appendPQExpBufferStr(query, "SELECT tableoid, oid, "
- "castsource, casttarget, castfunc, castcontext, "
- "castmethod "
- "FROM pg_cast ORDER BY 3,4");
- }
- else if (fout->remoteVersion >= 70300)
- {
- appendPQExpBufferStr(query, "SELECT tableoid, oid, "
- "castsource, casttarget, castfunc, castcontext, "
- "CASE WHEN castfunc = 0 THEN 'b' ELSE 'f' END AS castmethod "
- "FROM pg_cast ORDER BY 3,4");
- }
- else
- {
- appendPQExpBufferStr(query, "SELECT 0 AS tableoid, p.oid, "
- "t1.oid AS castsource, t2.oid AS casttarget, "
- "p.oid AS castfunc, 'e' AS castcontext, "
- "'f' AS castmethod "
- "FROM pg_type t1, pg_type t2, pg_proc p "
- "WHERE p.pronargs = 1 AND "
- "p.proargtypes[0] = t1.oid AND "
- "p.prorettype = t2.oid AND p.proname = t2.typname "
- "ORDER BY 3,4");
- }
- res = ExecuteSqlQuery(fout, query->data, PGRES_TUPLES_OK);
- ntups = PQntuples(res);
- *numCasts = ntups;
- castinfo = (CastInfo *) pg_malloc(ntups * sizeof(CastInfo));
- i_tableoid = PQfnumber(res, "tableoid");
- i_oid = PQfnumber(res, "oid");
- i_castsource = PQfnumber(res, "castsource");
- i_casttarget = PQfnumber(res, "casttarget");
- i_castfunc = PQfnumber(res, "castfunc");
- i_castcontext = PQfnumber(res, "castcontext");
- i_castmethod = PQfnumber(res, "castmethod");
- for (i = 0; i < ntups; i++)
- {
- PQExpBufferData namebuf;
- TypeInfo *sTypeInfo;
- TypeInfo *tTypeInfo;
- castinfo[i].dobj.objType = DO_CAST;
- castinfo[i].dobj.catId.tableoid = atooid(PQgetvalue(res, i, i_tableoid));
- castinfo[i].dobj.catId.oid = atooid(PQgetvalue(res, i, i_oid));
- AssignDumpId(&castinfo[i].dobj);
- castinfo[i].castsource = atooid(PQgetvalue(res, i, i_castsource));
- castinfo[i].casttarget = atooid(PQgetvalue(res, i, i_casttarget));
- castinfo[i].castfunc = atooid(PQgetvalue(res, i, i_castfunc));
- castinfo[i].castcontext = *(PQgetvalue(res, i, i_castcontext));
- castinfo[i].castmethod = *(PQgetvalue(res, i, i_castmethod));
- /*
- * Try to name cast as concatenation of typnames. This is only used
- * for purposes of sorting. If we fail to find either type, the name
- * will be an empty string.
- */
- initPQExpBuffer(&namebuf);
- sTypeInfo = findTypeByOid(castinfo[i].castsource);
- tTypeInfo = findTypeByOid(castinfo[i].casttarget);
- if (sTypeInfo && tTypeInfo)
- appendPQExpBuffer(&namebuf, "%s %s",
- sTypeInfo->dobj.name, tTypeInfo->dobj.name);
- castinfo[i].dobj.name = namebuf.data;
- if (OidIsValid(castinfo[i].castfunc))
- {
- /*
- * We need to make a dependency to ensure the function will be
- * dumped first. (In 7.3 and later the regular dependency
- * mechanism will handle this for us.)
- */
- FuncInfo *funcInfo;
- funcInfo = findFuncByOid(castinfo[i].castfunc);
- if (funcInfo)
- addObjectDependency(&castinfo[i].dobj,
- funcInfo->dobj.dumpId);
- }
- }
- PQclear(res);
- destroyPQExpBuffer(query);
- return castinfo;
- }
- /*
- * getTableAttrs -
- * for each interesting table, read info about its attributes
- * (names, types, default values, CHECK constraints, etc)
- *
- * This is implemented in a very inefficient way right now, looping
- * through the tblinfo and doing a join per table to find the attrs and their
- * types. However, because we want type names and so forth to be named
- * relative to the schema of each table, we couldn't do it in just one
- * query. (Maybe one query per schema?)
- *
- * modifies tblinfo
- */
- void
- getTableAttrs(Archive *fout, TableInfo *tblinfo, int numTables)
- {
- int i,
- j;
- PQExpBuffer q = createPQExpBuffer();
- int i_attnum;
- int i_attname;
- int i_atttypname;
- int i_atttypmod;
- int i_attstattarget;
- int i_attstorage;
- int i_typstorage;
- int i_attnotnull;
- int i_atthasdef;
- int i_attisdropped;
- int i_attlen;
- int i_attalign;
- int i_attislocal;
- int i_attoptions;
- int i_attcollation;
- int i_attfdwoptions;
- PGresult *res;
- int ntups;
- bool hasdefaults;
- for (i = 0; i < numTables; i++)
- {
- TableInfo *tbinfo = &tblinfo[i];
- /* Don't bother to collect info for sequences */
- if (tbinfo->relkind == RELKIND_SEQUENCE)
- continue;
- /* Don't bother with uninteresting tables, either */
- if (!tbinfo->interesting)
- continue;
- /*
- * Make sure we are in proper schema for this table; this allows
- * correct retrieval of formatted type names and default exprs
- */
- selectSourceSchema(fout, tbinfo->dobj.namespace->dobj.name);
- /* find all the user attributes and their types */
- /*
- * we must read the attribute names in attribute number order! because
- * we will use the attnum to index into the attnames array later. We
- * actually ask to order by "attrelid, attnum" because (at least up to
- * 7.3) the planner is not smart enough to realize it needn't re-sort
- * the output of an indexscan on pg_attribute_relid_attnum_index.
- */
- if (g_verbose)
- write_msg(NULL, "finding the columns and types of table \"%s\"\n",
- tbinfo->dobj.name);
- resetPQExpBuffer(q);
- if (fout->remoteVersion >= 90200)
- {
- /*
- * attfdwoptions is new in 9.2.
- */
- appendPQExpBuffer(q, "SELECT a.attnum, a.attname, a.atttypmod, "
- "a.attstattarget, a.attstorage, t.typstorage, "
- "a.attnotnull, a.atthasdef, a.attisdropped, "
- "a.attlen, a.attalign, a.attislocal, "
- "pg_catalog.format_type(t.oid,a.atttypmod) AS atttypname, "
- "array_to_string(a.attoptions, ', ') AS attoptions, "
- "CASE WHEN a.attcollation <> t.typcollation "
- "THEN a.attcollation ELSE 0 END AS attcollation, "
- "pg_catalog.array_to_string(ARRAY("
- "SELECT pg_catalog.quote_ident(option_name) || "
- "' ' || pg_catalog.quote_literal(option_value) "
- "FROM pg_catalog.pg_options_to_table(attfdwoptions) "
- "ORDER BY option_name"
- "), E',\n ') AS attfdwoptions "
- "FROM pg_catalog.pg_attribute a LEFT JOIN pg_catalog.pg_type t "
- "ON a.atttypid = t.oid "
- "WHERE a.attrelid = '%u'::pg_catalog.oid "
- "AND a.attnum > 0::pg_catalog.int2 "
- "ORDER BY a.attrelid, a.attnum",
- tbinfo->dobj.catId.oid);
- }
- else if (fout->remoteVersion >= 90100)
- {
- /*
- * attcollation is new in 9.1. Since we only want to dump COLLATE
- * clauses for attributes whose collation is different from their
- * type's default, we use a CASE here to suppress uninteresting
- * attcollations cheaply.
- */
- appendPQExpBuffer(q, "SELECT a.attnum, a.attname, a.atttypmod, "
- "a.attstattarget, a.attstorage, t.typstorage, "
- "a.attnotnull, a.atthasdef, a.attisdropped, "
- "a.attlen, a.attalign, a.attislocal, "
- "pg_catalog.format_type(t.oid,a.atttypmod) AS atttypname, "
- "array_to_string(a.attoptions, ', ') AS attoptions, "
- "CASE WHEN a.attcollation <> t.typcollation "
- "THEN a.attcollation ELSE 0 END AS attcollation, "
- "NULL AS attfdwoptions "
- "FROM pg_catalog.pg_attribute a LEFT JOIN pg_catalog.pg_type t "
- "ON a.atttypid = t.oid "
- "WHERE a.attrelid = '%u'::pg_catalog.oid "
- "AND a.attnum > 0::pg_catalog.int2 "
- "ORDER BY a.attrelid, a.attnum",
- tbinfo->dobj.catId.oid);
- }
- else if (fout->remoteVersion >= 90000)
- {
- /* attoptions is new in 9.0 */
- appendPQExpBuffer(q, "SELECT a.attnum, a.attname, a.atttypmod, "
- "a.attstattarget, a.attstorage, t.typstorage, "
- "a.attnotnull, a.atthasdef, a.attisdropped, "
- "a.attlen, a.attalign, a.attislocal, "
- "pg_catalog.format_type(t.oid,a.atttypmod) AS atttypname, "
- "array_to_string(a.attoptions, ', ') AS attoptions, "
- "0 AS attcollation, "
- "NULL AS attfdwoptions "
- "FROM pg_catalog.pg_attribute a LEFT JOIN pg_catalog.pg_type t "
- "ON a.atttypid = t.oid "
- "WHERE a.attrelid = '%u'::pg_catalog.oid "
- "AND a.attnum > 0::pg_catalog.int2 "
- "ORDER BY a.attrelid, a.attnum",
- tbinfo->dobj.catId.oid);
- }
- else if (fout->remoteVersion >= 70300)
- {
- /* need left join here to not fail on dropped columns ... */
- appendPQExpBuffer(q, "SELECT a.attnum, a.attname, a.atttypmod, "
- "a.attstattarget, a.attstorage, t.typstorage, "
- "a.attnotnull, a.atthasdef, a.attisdropped, "
- "a.attlen, a.attalign, a.attislocal, "
- "pg_catalog.format_type(t.oid,a.atttypmod) AS atttypname, "
- "'' AS attoptions, 0 AS attcollation, "
- "NULL AS attfdwoptions "
- "FROM pg_catalog.pg_attribute a LEFT JOIN pg_catalog.pg_type t "
- "ON a.atttypid = t.oid "
- "WHERE a.attrelid = '%u'::pg_catalog.oid "
- "AND a.attnum > 0::pg_catalog.int2 "
- "ORDER BY a.attrelid, a.attnum",
- tbinfo->dobj.catId.oid);
- }
- else if (fout->remoteVersion >= 70100)
- {
- /*
- * attstattarget doesn't exist in 7.1. It does exist in 7.2, but
- * we don't dump it because we can't tell whether it's been
- * explicitly set or was just a default.
- *
- * attislocal doesn't exist before 7.3, either; in older databases
- * we assume it's TRUE, else we'd fail to dump non-inherited atts.
- */
- appendPQExpBuffer(q, "SELECT a.attnum, a.attname, a.atttypmod, "
- "-1 AS attstattarget, a.attstorage, "
- "t.typstorage, a.attnotnull, a.atthasdef, "
- "false AS attisdropped, a.attlen, "
- "a.attalign, true AS attislocal, "
- "format_type(t.oid,a.atttypmod) AS atttypname, "
- "'' AS attoptions, 0 AS attcollation, "
- "NULL AS attfdwoptions "
- "FROM pg_attribute a LEFT JOIN pg_type t "
- "ON a.atttypid = t.oid "
- "WHERE a.attrelid = '%u'::oid "
- "AND a.attnum > 0::int2 "
- "ORDER BY a.attrelid, a.attnum",
- tbinfo->dobj.catId.oid);
- }
- else
- {
- /* format_type not available before 7.1 */
- appendPQExpBuffer(q, "SELECT attnum, attname, atttypmod, "
- "-1 AS attstattarget, "
- "attstorage, attstorage AS typstorage, "
- "attnotnull, atthasdef, false AS attisdropped, "
- "attlen, attalign, "
- "true AS attislocal, "
- "(SELECT typname FROM pg_type WHERE oid = atttypid) AS atttypname, "
- "'' AS attoptions, 0 AS attcollation, "
- "NULL AS attfdwoptions "
- "FROM pg_attribute a "
- "WHERE attrelid = '%u'::oid "
- "AND attnum > 0::int2 "
- "ORDER BY attrelid, attnum",
- tbinfo->dobj.catId.oid);
- }
- res = ExecuteSqlQuery(fout, q->data, PGRES_TUPLES_OK);
- ntups = PQntuples(res);
- i_attnum = PQfnumber(res, "attnum");
- i_attname = PQfnumber(res, "attname");
- i_atttypname = PQfnumber(res, "atttypname");
- i_atttypmod = PQfnumber(res, "atttypmod");
- i_attstattarget = PQfnumber(res, "attstattarget");
- i_attstorage = PQfnumber(res, "attstorage");
- i_typstorage = PQfnumber(res, "typstorage");
- i_attnotnull = PQfnumber(res, "attnotnull");
- i_atthasdef = PQfnumber(res, "atthasdef");
- i_attisdropped = PQfnumber(res, "attisdropped");
- i_attlen = PQfnumber(res, "attlen");
- i_attalign = PQfnumber(res, "attalign");
- i_attislocal = PQfnumber(res, "attislocal");
- i_attoptions = PQfnumber(res, "attoptions");
- i_attcollation = PQfnumber(res, "attcollation");
- i_attfdwoptions = PQfnumber(res, "attfdwoptions");
- tbinfo->numatts = ntups;
- tbinfo->attnames = (char **) pg_malloc(ntups * sizeof(char *));
- tbinfo->atttypnames = (char **) pg_malloc(ntups * sizeof(char *));
- tbinfo->atttypmod = (int *) pg_malloc(ntups * sizeof(int));
- tbinfo->attstattarget = (int *) pg_malloc(ntups * sizeof(int));
- tbinfo->attstorage = (char *) pg_malloc(ntups * sizeof(char));
- tbinfo->typstorage = (char *) pg_malloc(ntups * sizeof(char));
- tbinfo->attisdropped = (bool *) pg_malloc(ntups * sizeof(bool));
- tbinfo->attlen = (int *) pg_malloc(ntups * sizeof(int));
- tbinfo->attalign = (char *) pg_malloc(ntups * sizeof(char));
- tbinfo->attislocal = (bool *) pg_malloc(ntups * sizeof(bool));
- tbinfo->attoptions = (char **) pg_malloc(ntups * sizeof(char *));
- tbinfo->attcollation = (Oid *) pg_malloc(ntups * sizeof(Oid));
- tbinfo->attfdwoptions = (char **) pg_malloc(ntups * sizeof(char *));
- tbinfo->notnull = (bool *) pg_malloc(ntups * sizeof(bool));
- tbinfo->inhNotNull = (bool *) pg_malloc(ntups * sizeof(bool));
- tbinfo->attrdefs = (AttrDefInfo **) pg_malloc(ntups * sizeof(AttrDefInfo *));
- hasdefaults = false;
- for (j = 0; j < ntups; j++)
- {
- if (j + 1 != atoi(PQgetvalue(res, j, i_attnum)))
- exit_horribly(NULL,
- "invalid column numbering in table \"%s\"\n",
- tbinfo->dobj.name);
- tbinfo->attnames[j] = pg_strdup(PQgetvalue(res, j, i_attname));
- tbinfo->atttypnames[j] = pg_strdup(PQgetvalue(res, j, i_atttypname));
- tbinfo->atttypmod[j] = atoi(PQgetvalue(res, j, i_atttypmod));
- tbinfo->attstattarget[j] = atoi(PQgetvalue(res, j, i_attstattarget));
- tbinfo->attstorage[j] = *(PQgetvalue(res, j, i_attstorage));
- tbinfo->typstorage[j] = *(PQgetvalue(res, j, i_typstorage));
- tbinfo->attisdropped[j] = (PQgetvalue(res, j, i_attisdropped)[0] == 't');
- tbinfo->attlen[j] = atoi(PQgetvalue(res, j, i_attlen));
- tbinfo->attalign[j] = *(PQgetvalue(res, j, i_attalign));
- tbinfo->attislocal[j] = (PQgetvalue(res, j, i_attislocal)[0] == 't');
- tbinfo->notnull[j] = (PQgetvalue(res, j, i_attnotnull)[0] == 't');
- tbinfo->attoptions[j] = pg_strdup(PQgetvalue(res, j, i_attoptions));
- tbinfo->attcollation[j] = atooid(PQgetvalue(res, j, i_attcollation));
- tbinfo->attfdwoptions[j] = pg_strdup(PQgetvalue(res, j, i_attfdwoptions));
- tbinfo->attrdefs[j] = NULL; /* fix below */
- if (PQgetvalue(res, j, i_atthasdef)[0] == 't')
- hasdefaults = true;
- /* these flags will be set in flagInhAttrs() */
- tbinfo->inhNotNull[j] = false;
- }
- PQclear(res);
- /*
- * Get info about column defaults
- */
- if (hasdefaults)
- {
- AttrDefInfo *attrdefs;
- int numDefaults;
- if (g_verbose)
- write_msg(NULL, "finding default expressions of table \"%s\"\n",
- tbinfo->dobj.name);
- resetPQExpBuffer(q);
- if (fout->remoteVersion >= 70300)
- {
- appendPQExpBuffer(q, "SELECT tableoid, oid, adnum, "
- "pg_catalog.pg_get_expr(adbin, adrelid) AS adsrc "
- "FROM pg_catalog.pg_attrdef "
- "WHERE adrelid = '%u'::pg_catalog.oid",
- tbinfo->dobj.catId.oid);
- }
- else if (fout->remoteVersion >= 70200)
- {
- /* 7.2 did not have OIDs in pg_attrdef */
- appendPQExpBuffer(q, "SELECT tableoid, 0 AS oid, adnum, "
- "pg_get_expr(adbin, adrelid) AS adsrc "
- "FROM pg_attrdef "
- "WHERE adrelid = '%u'::oid",
- tbinfo->dobj.catId.oid);
- }
- else if (fout->remoteVersion >= 70100)
- {
- /* no pg_get_expr, so must rely on adsrc */
- appendPQExpBuffer(q, "SELECT tableoid, oid, adnum, adsrc "
- "FROM pg_attrdef "
- "WHERE adrelid = '%u'::oid",
- tbinfo->dobj.catId.oid);
- }
- else
- {
- /* no pg_get_expr, no tableoid either */
- appendPQExpBuffer(q, "SELECT "
- "(SELECT oid FROM pg_class WHERE relname = 'pg_attrdef') AS tableoid, "
- "oid, adnum, adsrc "
- "FROM pg_attrdef "
- "WHERE adrelid = '%u'::oid",
- tbinfo->dobj.catId.oid);
- }
- res = ExecuteSqlQuery(fout, q->data, PGRES_TUPLES_OK);
- numDefaults = PQntuples(res);
- attrdefs = (AttrDefInfo *) pg_malloc(numDefaults * sizeof(AttrDefInfo));
- for (j = 0; j < numDefaults; j++)
- {
- int adnum;
- adnum = atoi(PQgetvalue(res, j, 2));
- if (adnum <= 0 || adnum > ntups)
- exit_horribly(NULL,
- "invalid adnum value %d for table \"%s\"\n",
- adnum, tbinfo->dobj.name);
- /*
- * dropped columns shouldn't have defaults, but just in case,
- * ignore 'em
- */
- if (tbinfo->attisdropped[adnum - 1])
- continue;
- attrdefs[j].dobj.objType = DO_ATTRDEF;
- attrdefs[j].dobj.catId.tableoid = atooid(PQgetvalue(res, j, 0));
- attrdefs[j].dobj.catId.oid = atooid(PQgetvalue(res, j, 1));
- AssignDumpId(&attrdefs[j].dobj);
- attrdefs[j].adtable = tbinfo;
- attrdefs[j].adnum = adnum;
- attrdefs[j].adef_expr = pg_strdup(PQgetvalue(res, j, 3));
- attrdefs[j].dobj.name = pg_strdup(tbinfo->dobj.name);
- attrdefs[j].dobj.namespace = tbinfo->dobj.namespace;
- attrdefs[j].dobj.dump = tbinfo->dobj.dump;
- /*
- * Defaults on a VIEW must always be dumped as separate ALTER
- * TABLE commands. Defaults on regular tables are dumped as
- * part of the CREATE TABLE if possible, which it won't be if
- * the column is not going to be emitted explicitly.
- */
- if (tbinfo->relkind == RELKIND_VIEW)
- {
- attrdefs[j].separate = true;
- /* needed in case pre-7.3 DB: */
- addObjectDependency(&attrdefs[j].dobj,
- tbinfo->dobj.dumpId);
- }
- else if (!shouldPrintColumn(tbinfo, adnum - 1))
- {
- /* column will be suppressed, print default separately */
- attrdefs[j].separate = true;
- /* needed in case pre-7.3 DB: */
- addObjectDependency(&attrdefs[j].dobj,
- tbinfo->dobj.dumpId);
- }
- else
- {
- attrdefs[j].separate = false;
- /*
- * Mark the default as needing to appear before the table,
- * so that any dependencies it has must be emitted before
- * the CREATE TABLE. If this is not possible, we'll
- * change to "separate" mode while sorting dependencies.
- */
- addObjectDependency(&tbinfo->dobj,
- attrdefs[j].dobj.dumpId);
- }
- tbinfo->attrdefs[adnum - 1] = &attrdefs[j];
- }
- PQclear(res);
- }
- /*
- * Get info about table CHECK constraints
- */
- if (tbinfo->ncheck > 0)
- {
- ConstraintInfo *constrs;
- int numConstrs;
- if (g_verbose)
- write_msg(NULL, "finding check constraints for table \"%s\"\n",
- tbinfo->dobj.name);
- resetPQExpBuffer(q);
- if (fout->remoteVersion >= 90200)
- {
- /*
- * convalidated is new in 9.2 (actually, it is there in 9.1,
- * but it wasn't ever false for check constraints until 9.2).
- */
- appendPQExpBuffer(q, "SELECT tableoid, oid, conname, "
- "pg_catalog.pg_get_constraintdef(oid) AS consrc, "
- "conislocal, convalidated "
- "FROM pg_catalog.pg_constraint "
- "WHERE conrelid = '%u'::pg_catalog.oid "
- " AND contype = 'c' "
- "ORDER BY conname",
- tbinfo->dobj.catId.oid);
- }
- else if (fout->remoteVersion >= 80400)
- {
- /* conislocal is new in 8.4 */
- appendPQExpBuffer(q, "SELECT tableoid, oid, conname, "
- "pg_catalog.pg_get_constraintdef(oid) AS consrc, "
- "conislocal, true AS convalidated "
- "FROM pg_catalog.pg_constraint "
- "WHERE conrelid = '%u'::pg_catalog.oid "
- " AND contype = 'c' "
- "ORDER BY conname",
- tbinfo->dobj.catId.oid);
- }
- else if (fout->remoteVersion >= 70400)
- {
- appendPQExpBuffer(q, "SELECT tableoid, oid, conname, "
- "pg_catalog.pg_get_constraintdef(oid) AS consrc, "
- "true AS conislocal, true AS convalidated "
- "FROM pg_catalog.pg_constraint "
- "WHERE conrelid = '%u'::pg_catalog.oid "
- " AND contype = 'c' "
- "ORDER BY conname",
- tbinfo->dobj.catId.oid);
- }
- else if (fout->remoteVersion >= 70300)
- {
- /* no pg_get_constraintdef, must use consrc */
- appendPQExpBuffer(q, "SELECT tableoid, oid, conname, "
- "'CHECK (' || consrc || ')' AS consrc, "
- "true AS conislocal, true AS convalidated "
- "FROM pg_catalog.pg_constraint "
- "WHERE conrelid = '%u'::pg_catalog.oid "
- " AND contype = 'c' "
- "ORDER BY conname",
- tbinfo->dobj.catId.oid);
- }
- else if (fout->remoteVersion >= 70200)
- {
- /* 7.2 did not have OIDs in pg_relcheck */
- appendPQExpBuffer(q, "SELECT tableoid, 0 AS oid, "
- "rcname AS conname, "
- "'CHECK (' || rcsrc || ')' AS consrc, "
- "true AS conislocal, true AS convalidated "
- "FROM pg_relcheck "
- "WHERE rcrelid = '%u'::oid "
- "ORDER BY rcname",
- tbinfo->dobj.catId.oid);
- }
- else if (fout->remoteVersion >= 70100)
- {
- appendPQExpBuffer(q, "SELECT tableoid, oid, "
- "rcname AS conname, "
- "'CHECK (' || rcsrc || ')' AS consrc, "
- "true AS conislocal, true AS convalidated "
- "FROM pg_relcheck "
- "WHERE rcrelid = '%u'::oid "
- "ORDER BY rcname",
- tbinfo->dobj.catId.oid);
- }
- else
- {
- /* no tableoid in 7.0 */
- appendPQExpBuffer(q, "SELECT "
- "(SELECT oid FROM pg_class WHERE relname = 'pg_relcheck') AS tableoid, "
- "oid, rcname AS conname, "
- "'CHECK (' || rcsrc || ')' AS consrc, "
- "true AS conislocal, true AS convalidated "
- "FROM pg_relcheck "
- "WHERE rcrelid = '%u'::oid "
- "ORDER BY rcname",
- tbinfo->dobj.catId.oid);
- }
- res = ExecuteSqlQuery(fout, q->data, PGRES_TUPLES_OK);
- numConstrs = PQntuples(res);
- if (numConstrs != tbinfo->ncheck)
- {
- write_msg(NULL, ngettext("expected %d check constraint on table \"%s\" but found %d\n",
- "expected %d check constraints on table \"%s\" but found %d\n",
- tbinfo->ncheck),
- tbinfo->ncheck, tbinfo->dobj.name, numConstrs);
- write_msg(NULL, "(The system catalogs might be corrupted.)\n");
- exit_nicely(1);
- }
- constrs = (ConstraintInfo *) pg_malloc(numConstrs * sizeof(ConstraintInfo));
- tbinfo->checkexprs = constrs;
- for (j = 0; j < numConstrs; j++)
- {
- bool validated = PQgetvalue(res, j, 5)[0] == 't';
- constrs[j].dobj.objType = DO_CONSTRAINT;
- constrs[j].dobj.catId.tableoid = atooid(PQgetvalue(res, j, 0));
- constrs[j].dobj.catId.oid = atooid(PQgetvalue(res, j, 1));
- AssignDumpId(&constrs[j].dobj);
- constrs[j].dobj.name = pg_strdup(PQgetvalue(res, j, 2));
- constrs[j].dobj.namespace = tbinfo->dobj.namespace;
- constrs[j].contable = tbinfo;
- constrs[j].condomain = NULL;
- constrs[j].contype = 'c';
- constrs[j].condef = pg_strdup(PQgetvalue(res, j, 3));
- constrs[j].confrelid = InvalidOid;
- constrs[j].conindex = 0;
- constrs[j].condeferrable = false;
- constrs[j].condeferred = false;
- constrs[j].conislocal = (PQgetvalue(res, j, 4)[0] == 't');
- /*
- * An unvalidated constraint needs to be dumped separately, so
- * that potentially-violating existing data is loaded before
- * the constraint.
- */
- constrs[j].separate = !validated;
- constrs[j].dobj.dump = tbinfo->dobj.dump;
- /*
- * Mark the constraint as needing to appear before the table
- * --- this is so that any other dependencies of the
- * constraint will be emitted before we try to create the
- * table. If the constraint is to be dumped separately, it
- * will be dumped after data is loaded anyway, so don't do it.
- * (There's an automatic dependency in the opposite direction
- * anyway, so don't need to add one manually here.)
- */
- if (!constrs[j].separate)
- addObjectDependency(&tbinfo->dobj,
- constrs[j].dobj.dumpId);
- /*
- * If the constraint is inherited, this will be detected later
- * (in pre-8.4 databases). We also detect later if the
- * constraint must be split out from the table definition.
- */
- }
- PQclear(res);
- }
- }
- destroyPQExpBuffer(q);
- }
- /*
- * Test whether a column should be printed as part of table's CREATE TABLE.
- * Column number is zero-based.
- *
- * Normally this is always true, but it's false for dropped columns, as well
- * as those that were inherited without any local definition. (If we print
- * such a column it will mistakenly get pg_attribute.attislocal set to true.)
- * However, in binary_upgrade mode, we must print all such columns anyway and
- * fix the attislocal/attisdropped state later, so as to keep control of the
- * physical column order.
- *
- * This function exists because there are scattered nonobvious places that
- * must be kept in sync with this decision.
- */
- bool
- shouldPrintColumn(TableInfo *tbinfo, int colno)
- {
- if (binary_upgrade)
- return true;
- return (tbinfo->attislocal[colno] && !tbinfo->attisdropped[colno]);
- }
- /*
- * getTSParsers:
- * read all text search parsers in the system catalogs and return them
- * in the TSParserInfo* structure
- *
- * numTSParsers is set to the number of parsers read in
- */
- TSParserInfo *
- getTSParsers(Archive *fout, int *numTSParsers)
- {
- PGresult *res;
- int ntups;
- int i;
- PQExpBuffer query;
- TSParserInfo *prsinfo;
- int i_tableoid;
- int i_oid;
- int i_prsname;
- int i_prsnamespace;
- int i_prsstart;
- int i_prstoken;
- int i_prsend;
- int i_prsheadline;
- int i_prslextype;
- /* Before 8.3, there is no built-in text search support */
- if (fout->remoteVersion < 80300)
- {
- *numTSParsers = 0;
- return NULL;
- }
- query = createPQExpBuffer();
- /*
- * find all text search objects, including builtin ones; we filter out
- * system-defined objects at dump-out time.
- */
- /* Make sure we are in proper schema */
- selectSourceSchema(fout, "pg_catalog");
- appendPQExpBufferStr(query, "SELECT tableoid, oid, prsname, prsnamespace, "
- "prsstart::oid, prstoken::oid, "
- "prsend::oid, prsheadline::oid, prslextype::oid "
- "FROM pg_ts_parser");
- res = ExecuteSqlQuery(fout, query->data, PGRES_TUPLES_OK);
- ntups = PQntuples(res);
- *numTSParsers = ntups;
- prsinfo = (TSParserInfo *) pg_malloc(ntups * sizeof(TSParserInfo));
- i_tableoid = PQfnumber(res, "tableoid");
- i_oid = PQfnumber(res, "oid");
- i_prsname = PQfnumber(res, "prsname");
- i_prsnamespace = PQfnumber(res, "prsnamespace");
- i_prsstart = PQfnumber(res, "prsstart");
- i_prstoken = PQfnumber(res, "prstoken");
- i_prsend = PQfnumber(res, "prsend");
- i_prsheadline = PQfnumber(res, "prsheadline");
- i_prslextype = PQfnumber(res, "prslextype");
- for (i = 0; i < ntups; i++)
- {
- prsinfo[i].dobj.objType = DO_TSPARSER;
- prsinfo[i].dobj.catId.tableoid = atooid(PQgetvalue(res, i, i_tableoid));
- prsinfo[i].dobj.catId.oid = atooid(PQgetvalue(res, i, i_oid));
- AssignDumpId(&prsinfo[i].dobj);
- prsinfo[i].dobj.name = pg_strdup(PQgetvalue(res, i, i_prsname));
- prsinfo[i].dobj.namespace =
- findNamespace(fout,
- atooid(PQgetvalue(res, i, i_prsnamespace)),
- prsinfo[i].dobj.catId.oid);
- prsinfo[i].prsstart = atooid(PQgetvalue(res, i, i_prsstart));
- prsinfo[i].prstoken = atooid(PQgetvalue(res, i, i_prstoken));
- prsinfo[i].prsend = atooid(PQgetvalue(res, i, i_prsend));
- prsinfo[i].prsheadline = atooid(PQgetvalue(res, i, i_prsheadline));
- prsinfo[i].prslextype = atooid(PQgetvalue(res, i, i_prslextype));
- /* Decide whether we want to dump it */
- selectDumpableObject(&(prsinfo[i].dobj));
- }
- PQclear(res);
- destroyPQExpBuffer(query);
- return prsinfo;
- }
- /*
- * getTSDictionaries:
- * read all text search dictionaries in the system catalogs and return them
- * in the TSDictInfo* structure
- *
- * numTSDicts is set to the number of dictionaries read in
- */
- TSDictInfo *
- getTSDictionaries(Archive *fout, int *numTSDicts)
- {
- PGresult *res;
- int ntups;
- int i;
- PQExpBuffer query;
- TSDictInfo *dictinfo;
- int i_tableoid;
- int i_oid;
- int i_dictname;
- int i_dictnamespace;
- int i_rolname;
- int i_dicttemplate;
- int i_dictinitoption;
- /* Before 8.3, there is no built-in text search support */
- if (fout->remoteVersion < 80300)
- {
- *numTSDicts = 0;
- return NULL;
- }
- query = createPQExpBuffer();
- /* Make sure we are in proper schema */
- selectSourceSchema(fout, "pg_catalog");
- appendPQExpBuffer(query, "SELECT tableoid, oid, dictname, "
- "dictnamespace, (%s dictowner) AS rolname, "
- "dicttemplate, dictinitoption "
- "FROM pg_ts_dict",
- username_subquery);
- res = ExecuteSqlQuery(fout, query->data, PGRES_TUPLES_OK);
- ntups = PQntuples(res);
- *numTSDicts = ntups;
- dictinfo = (TSDictInfo *) pg_malloc(ntups * sizeof(TSDictInfo));
- i_tableoid = PQfnumber(res, "tableoid");
- i_oid = PQfnumber(res, "oid");
- i_dictname = PQfnumber(res, "dictname");
- i_dictnamespace = PQfnumber(res, "dictnamespace");
- i_rolname = PQfnumber(res, "rolname");
- i_dictinitoption = PQfnumber(res, "dictinitoption");
- i_dicttemplate = PQfnumber(res, "dicttemplate");
- for (i = 0; i < ntups; i++)
- {
- dictinfo[i].dobj.objType = DO_TSDICT;
- dictinfo[i].dobj.catId.tableoid = atooid(PQgetvalue(res, i, i_tableoid));
- dictinfo[i].dobj.catId.oid = atooid(PQgetvalue(res, i, i_oid));
- AssignDumpId(&dictinfo[i].dobj);
- dictinfo[i].dobj.name = pg_strdup(PQgetvalue(res, i, i_dictname));
- dictinfo[i].dobj.namespace =
- findNamespace(fout,
- atooid(PQgetvalue(res, i, i_dictnamespace)),
- dictinfo[i].dobj.catId.oid);
- dictinfo[i].rolname = pg_strdup(PQgetvalue(res, i, i_rolname));
- dictinfo[i].dicttemplate = atooid(PQgetvalue(res, i, i_dicttemplate));
- if (PQgetisnull(res, i, i_dictinitoption))
- dictinfo[i].dictinitoption = NULL;
- else
- dictinfo[i].dictinitoption = pg_strdup(PQgetvalue(res, i, i_dictinitoption));
- /* Decide whether we want to dump it */
- selectDumpableObject(&(dictinfo[i].dobj));
- }
- PQclear(res);
- destroyPQExpBuffer(query);
- return dictinfo;
- }
- /*
- * getTSTemplates:
- * read all text search templates in the system catalogs and return them
- * in the TSTemplateInfo* structure
- *
- * numTSTemplates is set to the number of templates read in
- */
- TSTemplateInfo *
- getTSTemplates(Archive *fout, int *numTSTemplates)
- {
- PGresult *res;
- int ntups;
- int i;
- PQExpBuffer query;
- TSTemplateInfo *tmplinfo;
- int i_tableoid;
- int i_oid;
- int i_tmplname;
- int i_tmplnamespace;
- int i_tmplinit;
- int i_tmpllexize;
- /* Before 8.3, there is no built-in text search support */
- if (fout->remoteVersion < 80300)
- {
- *numTSTemplates = 0;
- return NULL;
- }
- query = createPQExpBuffer();
- /* Make sure we are in proper schema */
- selectSourceSchema(fout, "pg_catalog");
- appendPQExpBufferStr(query, "SELECT tableoid, oid, tmplname, "
- "tmplnamespace, tmplinit::oid, tmpllexize::oid "
- "FROM pg_ts_template");
- res = ExecuteSqlQuery(fout, query->data, PGRES_TUPLES_OK);
- ntups = PQntuples(res);
- *numTSTemplates = ntups;
- tmplinfo = (TSTemplateInfo *) pg_malloc(ntups * sizeof(TSTemplateInfo));
- i_tableoid = PQfnumber(res, "tableoid");
- i_oid = PQfnumber(res, "oid");
- i_tmplname = PQfnumber(res, "tmplname");
- i_tmplnamespace = PQfnumber(res, "tmplnamespace");
- i_tmplinit = PQfnumber(res, "tmplinit");
- i_tmpllexize = PQfnumber(res, "tmpllexize");
- for (i = 0; i < ntups; i++)
- {
- tmplinfo[i].dobj.objType = DO_TSTEMPLATE;
- tmplinfo[i].dobj.catId.tableoid = atooid(PQgetvalue(res, i, i_tableoid));
- tmplinfo[i].dobj.catId.oid = atooid(PQgetvalue(res, i, i_oid));
- AssignDumpId(&tmplinfo[i].dobj);
- tmplinfo[i].dobj.name = pg_strdup(PQgetvalue(res, i, i_tmplname));
- tmplinfo[i].dobj.namespace =
- findNamespace(fout,
- atooid(PQgetvalue(res, i, i_tmplnamespace)),
- tmplinfo[i].dobj.catId.oid);
- tmplinfo[i].tmplinit = atooid(PQgetvalue(res, i, i_tmplinit));
- tmplinfo[i].tmpllexize = atooid(PQgetvalue(res, i, i_tmpllexize));
- /* Decide whether we want to dump it */
- selectDumpableObject(&(tmplinfo[i].dobj));
- }
- PQclear(res);
- destroyPQExpBuffer(query);
- return tmplinfo;
- }
- /*
- * getTSConfigurations:
- * read all text search configurations in the system catalogs and return
- * them in the TSConfigInfo* structure
- *
- * numTSConfigs is set to the number of configurations read in
- */
- TSConfigInfo *
- getTSConfigurations(Archive *fout, int *numTSConfigs)
- {
- PGresult *res;
- int ntups;
- int i;
- PQExpBuffer query;
- TSConfigInfo *cfginfo;
- int i_tableoid;
- int i_oid;
- int i_cfgname;
- int i_cfgnamespace;
- int i_rolname;
- int i_cfgparser;
- /* Before 8.3, there is no built-in text search support */
- if (fout->remoteVersion < 80300)
- {
- *numTSConfigs = 0;
- return NULL;
- }
- query = createPQExpBuffer();
- /* Make sure we are in proper schema */
- selectSourceSchema(fout, "pg_catalog");
- appendPQExpBuffer(query, "SELECT tableoid, oid, cfgname, "
- "cfgnamespace, (%s cfgowner) AS rolname, cfgparser "
- "FROM pg_ts_config",
- username_subquery);
- res = ExecuteSqlQuery(fout, query->data, PGRES_TUPLES_OK);
- ntups = PQntuples(res);
- *numTSConfigs = ntups;
- cfginfo = (TSConfigInfo *) pg_malloc(ntups * sizeof(TSConfigInfo));
- i_tableoid = PQfnumber(res, "tableoid");
- i_oid = PQfnumber(res, "oid");
- i_cfgname = PQfnumber(res, "cfgname");
- i_cfgnamespace = PQfnumber(res, "cfgnamespace");
- i_rolname = PQfnumber(res, "rolname");
- i_cfgparser = PQfnumber(res, "cfgparser");
- for (i = 0; i < ntups; i++)
- {
- cfginfo[i].dobj.objType = DO_TSCONFIG;
- cfginfo[i].dobj.catId.tableoid = atooid(PQgetvalue(res, i, i_tableoid));
- cfginfo[i].dobj.catId.oid = atooid(PQgetvalue(res, i, i_oid));
- AssignDumpId(&cfginfo[i].dobj);
- cfginfo[i].dobj.name = pg_strdup(PQgetvalue(res, i, i_cfgname));
- cfginfo[i].dobj.namespace =
- findNamespace(fout,
- atooid(PQgetvalue(res, i, i_cfgnamespace)),
- cfginfo[i].dobj.catId.oid);
- cfginfo[i].rolname = pg_strdup(PQgetvalue(res, i, i_rolname));
- cfginfo[i].cfgparser = atooid(PQgetvalue(res, i, i_cfgparser));
- /* Decide whether we want to dump it */
- selectDumpableObject(&(cfginfo[i].dobj));
- }
- PQclear(res);
- destroyPQExpBuffer(query);
- return cfginfo;
- }
- /*
- * getForeignDataWrappers:
- * read all foreign-data wrappers in the system catalogs and return
- * them in the FdwInfo* structure
- *
- * numForeignDataWrappers is set to the number of fdws read in
- */
- FdwInfo *
- getForeignDataWrappers(Archive *fout, int *numForeignDataWrappers)
- {
- PGresult *res;
- int ntups;
- int i;
- PQExpBuffer query;
- FdwInfo *fdwinfo;
- int i_tableoid;
- int i_oid;
- int i_fdwname;
- int i_rolname;
- int i_fdwhandler;
- int i_fdwvalidator;
- int i_fdwacl;
- int i_fdwoptions;
- /* Before 8.4, there are no foreign-data wrappers */
- if (fout->remoteVersion < 80400)
- {
- *numForeignDataWrappers = 0;
- return NULL;
- }
- query = createPQExpBuffer();
- /* Make sure we are in proper schema */
- selectSourceSchema(fout, "pg_catalog");
- if (fout->remoteVersion >= 90100)
- {
- appendPQExpBuffer(query, "SELECT tableoid, oid, fdwname, "
- "(%s fdwowner) AS rolname, "
- "fdwhandler::pg_catalog.regproc, "
- "fdwvalidator::pg_catalog.regproc, fdwacl, "
- "array_to_string(ARRAY("
- "SELECT quote_ident(option_name) || ' ' || "
- "quote_literal(option_value) "
- "FROM pg_options_to_table(fdwoptions) "
- "ORDER BY option_name"
- "), E',\n ') AS fdwoptions "
- "FROM pg_foreign_data_wrapper",
- username_subquery);
- }
- else
- {
- appendPQExpBuffer(query, "SELECT tableoid, oid, fdwname, "
- "(%s fdwowner) AS rolname, "
- "'-' AS fdwhandler, "
- "fdwvalidator::pg_catalog.regproc, fdwacl, "
- "array_to_string(ARRAY("
- "SELECT quote_ident(option_name) || ' ' || "
- "quote_literal(option_value) "
- "FROM pg_options_to_table(fdwoptions) "
- "ORDER BY option_name"
- "), E',\n ') AS fdwoptions "
- "FROM pg_foreign_data_wrapper",
- username_subquery);
- }
- res = ExecuteSqlQuery(fout, query->data, PGRES_TUPLES_OK);
- ntups = PQntuples(res);
- *numForeignDataWrappers = ntups;
- fdwinfo = (FdwInfo *) pg_malloc(ntups * sizeof(FdwInfo));
- i_tableoid = PQfnumber(res, "tableoid");
- i_oid = PQfnumber(res, "oid");
- i_fdwname = PQfnumber(res, "fdwname");
- i_rolname = PQfnumber(res, "rolname");
- i_fdwhandler = PQfnumber(res, "fdwhandler");
- i_fdwvalidator = PQfnumber(res, "fdwvalidator");
- i_fdwacl = PQfnumber(res, "fdwacl");
- i_fdwoptions = PQfnumber(res, "fdwoptions");
- for (i = 0; i < ntups; i++)
- {
- fdwinfo[i].dobj.objType = DO_FDW;
- fdwinfo[i].dobj.catId.tableoid = atooid(PQgetvalue(res, i, i_tableoid));
- fdwinfo[i].dobj.catId.oid = atooid(PQgetvalue(res, i, i_oid));
- AssignDumpId(&fdwinfo[i].dobj);
- fdwinfo[i].dobj.name = pg_strdup(PQgetvalue(res, i, i_fdwname));
- fdwinfo[i].dobj.namespace = NULL;
- fdwinfo[i].rolname = pg_strdup(PQgetvalue(res, i, i_rolname));
- fdwinfo[i].fdwhandler = pg_strdup(PQgetvalue(res, i, i_fdwhandler));
- fdwinfo[i].fdwvalidator = pg_strdup(PQgetvalue(res, i, i_fdwvalidator));
- fdwinfo[i].fdwoptions = pg_strdup(PQgetvalue(res, i, i_fdwoptions));
- fdwinfo[i].fdwacl = pg_strdup(PQgetvalue(res, i, i_fdwacl));
- /* Decide whether we want to dump it */
- selectDumpableObject(&(fdwinfo[i].dobj));
- }
- PQclear(res);
- destroyPQExpBuffer(query);
- return fdwinfo;
- }
- /*
- * getForeignServers:
- * read all foreign servers in the system catalogs and return
- * them in the ForeignServerInfo * structure
- *
- * numForeignServers is set to the number of servers read in
- */
- ForeignServerInfo *
- getForeignServers(Archive *fout, int *numForeignServers)
- {
- PGresult *res;
- int ntups;
- int i;
- PQExpBuffer query;
- ForeignServerInfo *srvinfo;
- int i_tableoid;
- int i_oid;
- int i_srvname;
- int i_rolname;
- int i_srvfdw;
- int i_srvtype;
- int i_srvversion;
- int i_srvacl;
- int i_srvoptions;
- /* Before 8.4, there are no foreign servers */
- if (fout->remoteVersion < 80400)
- {
- *numForeignServers = 0;
- return NULL;
- }
- query = createPQExpBuffer();
- /* Make sure we are in proper schema */
- selectSourceSchema(fout, "pg_catalog");
- appendPQExpBuffer(query, "SELECT tableoid, oid, srvname, "
- "(%s srvowner) AS rolname, "
- "srvfdw, srvtype, srvversion, srvacl,"
- "array_to_string(ARRAY("
- "SELECT quote_ident(option_name) || ' ' || "
- "quote_literal(option_value) "
- "FROM pg_options_to_table(srvoptions) "
- "ORDER BY option_name"
- "), E',\n ') AS srvoptions "
- "FROM pg_foreign_server",
- username_subquery);
- res = ExecuteSqlQuery(fout, query->data, PGRES_TUPLES_OK);
- ntups = PQntuples(res);
- *numForeignServers = ntups;
- srvinfo = (ForeignServerInfo *) pg_malloc(ntups * sizeof(ForeignServerInfo));
- i_tableoid = PQfnumber(res, "tableoid");
- i_oid = PQfnumber(res, "oid");
- i_srvname = PQfnumber(res, "srvname");
- i_rolname = PQfnumber(res, "rolname");
- i_srvfdw = PQfnumber(res, "srvfdw");
- i_srvtype = PQfnumber(res, "srvtype");
- i_srvversion = PQfnumber(res, "srvversion");
- i_srvacl = PQfnumber(res, "srvacl");
- i_srvoptions = PQfnumber(res, "srvoptions");
- for (i = 0; i < ntups; i++)
- {
- srvinfo[i].dobj.objType = DO_FOREIGN_SERVER;
- srvinfo[i].dobj.catId.tableoid = atooid(PQgetvalue(res, i, i_tableoid));
- srvinfo[i].dobj.catId.oid = atooid(PQgetvalue(res, i, i_oid));
- AssignDumpId(&srvinfo[i].dobj);
- srvinfo[i].dobj.name = pg_strdup(PQgetvalue(res, i, i_srvname));
- srvinfo[i].dobj.namespace = NULL;
- srvinfo[i].rolname = pg_strdup(PQgetvalue(res, i, i_rolname));
- srvinfo[i].srvfdw = atooid(PQgetvalue(res, i, i_srvfdw));
- srvinfo[i].srvtype = pg_strdup(PQgetvalue(res, i, i_srvtype));
- srvinfo[i].srvversion = pg_strdup(PQgetvalue(res, i, i_srvversion));
- srvinfo[i].srvoptions = pg_strdup(PQgetvalue(res, i, i_srvoptions));
- srvinfo[i].srvacl = pg_strdup(PQgetvalue(res, i, i_srvacl));
- /* Decide whether we want to dump it */
- selectDumpableObject(&(srvinfo[i].dobj));
- }
- PQclear(res);
- destroyPQExpBuffer(query);
- return srvinfo;
- }
- /*
- * getDefaultACLs:
- * read all default ACL information in the system catalogs and return
- * them in the DefaultACLInfo structure
- *
- * numDefaultACLs is set to the number of ACLs read in
- */
- DefaultACLInfo *
- getDefaultACLs(Archive *fout, int *numDefaultACLs)
- {
- DefaultACLInfo *daclinfo;
- PQExpBuffer query;
- PGresult *res;
- int i_oid;
- int i_tableoid;
- int i_defaclrole;
- int i_defaclnamespace;
- int i_defaclobjtype;
- int i_defaclacl;
- int i,
- ntups;
- if (fout->remoteVersion < 90000)
- {
- *numDefaultACLs = 0;
- return NULL;
- }
- query = createPQExpBuffer();
- /* Make sure we are in proper schema */
- selectSourceSchema(fout, "pg_catalog");
- appendPQExpBuffer(query, "SELECT oid, tableoid, "
- "(%s defaclrole) AS defaclrole, "
- "defaclnamespace, "
- "defaclobjtype, "
- "defaclacl "
- "FROM pg_default_acl",
- username_subquery);
- res = ExecuteSqlQuery(fout, query->data, PGRES_TUPLES_OK);
- ntups = PQntuples(res);
- *numDefaultACLs = ntups;
- daclinfo = (DefaultACLInfo *) pg_malloc(ntups * sizeof(DefaultACLInfo));
- i_oid = PQfnumber(res, "oid");
- i_tableoid = PQfnumber(res, "tableoid");
- i_defaclrole = PQfnumber(res, "defaclrole");
- i_defaclnamespace = PQfnumber(res, "defaclnamespace");
- i_defaclobjtype = PQfnumber(res, "defaclobjtype");
- i_defaclacl = PQfnumber(res, "defaclacl");
- for (i = 0; i < ntups; i++)
- {
- Oid nspid = atooid(PQgetvalue(res, i, i_defaclnamespace));
- daclinfo[i].dobj.objType = DO_DEFAULT_ACL;
- daclinfo[i].dobj.catId.tableoid = atooid(PQgetvalue(res, i, i_tableoid));
- daclinfo[i].dobj.catId.oid = atooid(PQgetvalue(res, i, i_oid));
- AssignDumpId(&daclinfo[i].dobj);
- /* cheesy ... is it worth coming up with a better object name? */
- daclinfo[i].dobj.name = pg_strdup(PQgetvalue(res, i, i_defaclobjtype));
- if (nspid != InvalidOid)
- daclinfo[i].dobj.namespace = findNamespace(fout, nspid,
- daclinfo[i].dobj.catId.oid);
- else
- daclinfo[i].dobj.namespace = NULL;
- daclinfo[i].defaclrole = pg_strdup(PQgetvalue(res, i, i_defaclrole));
- daclinfo[i].defaclobjtype = *(PQgetvalue(res, i, i_defaclobjtype));
- daclinfo[i].defaclacl = pg_strdup(PQgetvalue(res, i, i_defaclacl));
- /* Decide whether we want to dump it */
- selectDumpableDefaultACL(&(daclinfo[i]));
- }
- PQclear(res);
- destroyPQExpBuffer(query);
- return daclinfo;
- }
- /*
- * dumpComment --
- *
- * This routine is used to dump any comments associated with the
- * object handed to this routine. The routine takes a constant character
- * string for the target part of the comment-creation command, plus
- * the namespace and owner of the object (for labeling the ArchiveEntry),
- * plus catalog ID and subid which are the lookup key for pg_description,
- * plus the dump ID for the object (for setting a dependency).
- * If a matching pg_description entry is found, it is dumped.
- *
- * Note: although this routine takes a dumpId for dependency purposes,
- * that purpose is just to mark the dependency in the emitted dump file
- * for possible future use by pg_restore. We do NOT use it for determining
- * ordering of the comment in the dump file, because this routine is called
- * after dependency sorting occurs. This routine should be called just after
- * calling ArchiveEntry() for the specified object.
- */
- static void
- dumpComment(Archive *fout, const char *target,
- const char *namespace, const char *owner,
- CatalogId catalogId, int subid, DumpId dumpId)
- {
- CommentItem *comments;
- int ncomments;
- /* Comments are schema not data ... except blob comments are data */
- if (strncmp(target, "LARGE OBJECT ", 13) != 0)
- {
- if (dataOnly)
- return;
- }
- else
- {
- if (schemaOnly)
- return;
- }
- /* Search for comments associated with catalogId, using table */
- ncomments = findComments(fout, catalogId.tableoid, catalogId.oid,
- &comments);
- /* Is there one matching the subid? */
- while (ncomments > 0)
- {
- if (comments->objsubid == subid)
- break;
- comments++;
- ncomments--;
- }
- /* If a comment exists, build COMMENT ON statement */
- if (ncomments > 0)
- {
- PQExpBuffer query = createPQExpBuffer();
- appendPQExpBuffer(query, "COMMENT ON %s IS ", target);
- appendStringLiteralAH(query, comments->descr, fout);
- appendPQExpBufferStr(query, ";\n");
- /*
- * We mark comments as SECTION_NONE because they really belong in the
- * same section as their parent, whether that is pre-data or
- * post-data.
- */
- ArchiveEntry(fout, nilCatalogId, createDumpId(),
- target, namespace, NULL, owner,
- false, "COMMENT", SECTION_NONE,
- query->data, "", NULL,
- &(dumpId), 1,
- NULL, NULL);
- destroyPQExpBuffer(query);
- }
- }
- /*
- * dumpTableComment --
- *
- * As above, but dump comments for both the specified table (or view)
- * and its columns.
- */
- static void
- dumpTableComment(Archive *fout, TableInfo *tbinfo,
- const char *reltypename)
- {
- CommentItem *comments;
- int ncomments;
- PQExpBuffer query;
- PQExpBuffer target;
- /* Comments are SCHEMA not data */
- if (dataOnly)
- return;
- /* Search for comments associated with relation, using table */
- ncomments = findComments(fout,
- tbinfo->dobj.catId.tableoid,
- tbinfo->dobj.catId.oid,
- &comments);
- /* If comments exist, build COMMENT ON statements */
- if (ncomments <= 0)
- return;
- query = createPQExpBuffer();
- target = createPQExpBuffer();
- while (ncomments > 0)
- {
- const char *descr = comments->descr;
- int objsubid = comments->objsubid;
- if (objsubid == 0)
- {
- resetPQExpBuffer(target);
- appendPQExpBuffer(target, "%s %s", reltypename,
- fmtId(tbinfo->dobj.name));
- resetPQExpBuffer(query);
- appendPQExpBuffer(query, "COMMENT ON %s IS ", target->data);
- appendStringLiteralAH(query, descr, fout);
- appendPQExpBufferStr(query, ";\n");
- ArchiveEntry(fout, nilCatalogId, createDumpId(),
- target->data,
- tbinfo->dobj.namespace->dobj.name,
- NULL, tbinfo->rolname,
- false, "COMMENT", SECTION_NONE,
- query->data, "", NULL,
- &(tbinfo->dobj.dumpId), 1,
- NULL, NULL);
- }
- else if (objsubid > 0 && objsubid <= tbinfo->numatts)
- {
- resetPQExpBuffer(target);
- appendPQExpBuffer(target, "COLUMN %s.",
- fmtId(tbinfo->dobj.name));
- appendPQExpBufferStr(target, fmtId(tbinfo->attnames[objsubid - 1]));
- resetPQExpBuffer(query);
- appendPQExpBuffer(query, "COMMENT ON %s IS ", target->data);
- appendStringLiteralAH(query, descr, fout);
- appendPQExpBufferStr(query, ";\n");
- ArchiveEntry(fout, nilCatalogId, createDumpId(),
- target->data,
- tbinfo->dobj.namespace->dobj.name,
- NULL, tbinfo->rolname,
- false, "COMMENT", SECTION_NONE,
- query->data, "", NULL,
- &(tbinfo->dobj.dumpId), 1,
- NULL, NULL);
- }
- comments++;
- ncomments--;
- }
- destroyPQExpBuffer(query);
- destroyPQExpBuffer(target);
- }
- /*
- * findComments --
- *
- * Find the comment(s), if any, associated with the given object. All the
- * objsubid values associated with the given classoid/objoid are found with
- * one search.
- */
- static int
- findComments(Archive *fout, Oid classoid, Oid objoid,
- CommentItem **items)
- {
- /* static storage for table of comments */
- static CommentItem *comments = NULL;
- static int ncomments = -1;
- CommentItem *middle = NULL;
- CommentItem *low;
- CommentItem *high;
- int nmatch;
- /* Get comments if we didn't already */
- if (ncomments < 0)
- ncomments = collectComments(fout, &comments);
- /*
- * Pre-7.2, pg_description does not contain classoid, so collectComments
- * just stores a zero. If there's a collision on object OID, well, you
- * get duplicate comments.
- */
- if (fout->remoteVersion < 70200)
- classoid = 0;
- /*
- * Do binary search to find some item matching the object.
- */
- low = &comments[0];
- high = &comments[ncomments - 1];
- while (low <= high)
- {
- middle = low + (high - low) / 2;
- if (classoid < middle->classoid)
- high = middle - 1;
- else if (classoid > middle->classoid)
- low = middle + 1;
- else if (objoid < middle->objoid)
- high = middle - 1;
- else if (objoid > middle->objoid)
- low = middle + 1;
- else
- break; /* found a match */
- }
- if (low > high) /* no matches */
- {
- *items = NULL;
- return 0;
- }
- /*
- * Now determine how many items match the object. The search loop
- * invariant still holds: only items between low and high inclusive could
- * match.
- */
- nmatch = 1;
- while (middle > low)
- {
- if (classoid != middle[-1].classoid ||
- objoid != middle[-1].objoid)
- break;
- middle--;
- nmatch++;
- }
- *items = middle;
- middle += nmatch;
- while (middle <= high)
- {
- if (classoid != middle->classoid ||
- objoid != middle->objoid)
- break;
- middle++;
- nmatch++;
- }
- return nmatch;
- }
- /*
- * collectComments --
- *
- * Construct a table of all comments available for database objects.
- * We used to do per-object queries for the comments, but it's much faster
- * to pull them all over at once, and on most databases the memory cost
- * isn't high.
- *
- * The table is sorted by classoid/objid/objsubid for speed in lookup.
- */
- static int
- collectComments(Archive *fout, CommentItem **items)
- {
- PGresult *res;
- PQExpBuffer query;
- int i_description;
- int i_classoid;
- int i_objoid;
- int i_objsubid;
- int ntups;
- int i;
- CommentItem *comments;
- /*
- * Note we do NOT change source schema here; preserve the caller's
- * setting, instead.
- */
- query = createPQExpBuffer();
- if (fout->remoteVersion >= 70300)
- {
- appendPQExpBufferStr(query, "SELECT description, classoid, objoid, objsubid "
- "FROM pg_catalog.pg_description "
- "ORDER BY classoid, objoid, objsubid");
- }
- else if (fout->remoteVersion >= 70200)
- {
- appendPQExpBufferStr(query, "SELECT description, classoid, objoid, objsubid "
- "FROM pg_description "
- "ORDER BY classoid, objoid, objsubid");
- }
- else
- {
- /* Note: this will fail to find attribute comments in pre-7.2... */
- appendPQExpBufferStr(query, "SELECT description, 0 AS classoid, objoid, 0 AS objsubid "
- "FROM pg_description "
- "ORDER BY objoid");
- }
- res = ExecuteSqlQuery(fout, query->data, PGRES_TUPLES_OK);
- /* Construct lookup table containing OIDs in numeric form */
- i_description = PQfnumber(res, "description");
- i_classoid = PQfnumber(res, "classoid");
- i_objoid = PQfnumber(res, "objoid");
- i_objsubid = PQfnumber(res, "objsubid");
- ntups = PQntuples(res);
- comments = (CommentItem *) pg_malloc(ntups * sizeof(CommentItem));
- for (i = 0; i < ntups; i++)
- {
- comments[i].descr = PQgetvalue(res, i, i_description);
- comments[i].classoid = atooid(PQgetvalue(res, i, i_classoid));
- comments[i].objoid = atooid(PQgetvalue(res, i, i_objoid));
- comments[i].objsubid = atoi(PQgetvalue(res, i, i_objsubid));
- }
- /* Do NOT free the PGresult since we are keeping pointers into it */
- destroyPQExpBuffer(query);
- *items = comments;
- return ntups;
- }
- /*
- * dumpDumpableObject
- *
- * This routine and its subsidiaries are responsible for creating
- * ArchiveEntries (TOC objects) for each object to be dumped.
- */
- static void
- dumpDumpableObject(Archive *fout, DumpableObject *dobj)
- {
- switch (dobj->objType)
- {
- case DO_NAMESPACE:
- dumpNamespace(fout, (NamespaceInfo *) dobj);
- break;
- case DO_EXTENSION:
- dumpExtension(fout, (ExtensionInfo *) dobj);
- break;
- case DO_TYPE:
- dumpType(fout, (TypeInfo *) dobj);
- break;
- case DO_SHELL_TYPE:
- dumpShellType(fout, (ShellTypeInfo *) dobj);
- break;
- case DO_FUNC:
- dumpFunc(fout, (FuncInfo *) dobj);
- break;
- case DO_AGG:
- dumpAgg(fout, (AggInfo *) dobj);
- break;
- case DO_OPERATOR:
- dumpOpr(fout, (OprInfo *) dobj);
- break;
- case DO_OPCLASS:
- dumpOpclass(fout, (OpclassInfo *) dobj);
- break;
- case DO_OPFAMILY:
- dumpOpfamily(fout, (OpfamilyInfo *) dobj);
- break;
- case DO_COLLATION:
- dumpCollation(fout, (CollInfo *) dobj);
- break;
- case DO_CONVERSION:
- dumpConversion(fout, (ConvInfo *) dobj);
- break;
- case DO_TABLE:
- dumpTable(fout, (TableInfo *) dobj);
- break;
- case DO_ATTRDEF:
- dumpAttrDef(fout, (AttrDefInfo *) dobj);
- break;
- case DO_INDEX:
- dumpIndex(fout, (IndxInfo *) dobj);
- break;
- case DO_REFRESH_MATVIEW:
- refreshMatViewData(fout, (TableDataInfo *) dobj);
- break;
- case DO_RULE:
- dumpRule(fout, (RuleInfo *) dobj);
- break;
- case DO_TRIGGER:
- dumpTrigger(fout, (TriggerInfo *) dobj);
- break;
- case DO_EVENT_TRIGGER:
- dumpEventTrigger(fout, (EventTriggerInfo *) dobj);
- break;
- case DO_CONSTRAINT:
- dumpConstraint(fout, (ConstraintInfo *) dobj);
- break;
- case DO_FK_CONSTRAINT:
- dumpConstraint(fout, (ConstraintInfo *) dobj);
- break;
- case DO_PROCLANG:
- dumpProcLang(fout, (ProcLangInfo *) dobj);
- break;
- case DO_CAST:
- dumpCast(fout, (CastInfo *) dobj);
- break;
- case DO_TABLE_DATA:
- if (((TableDataInfo *) dobj)->tdtable->relkind == RELKIND_SEQUENCE)
- dumpSequenceData(fout, (TableDataInfo *) dobj);
- else
- dumpTableData(fout, (TableDataInfo *) dobj);
- break;
- case DO_DUMMY_TYPE:
- /* table rowtypes and array types are never dumped separately */
- break;
- case DO_TSPARSER:
- dumpTSParser(fout, (TSParserInfo *) dobj);
- break;
- case DO_TSDICT:
- dumpTSDictionary(fout, (TSDictInfo *) dobj);
- break;
- case DO_TSTEMPLATE:
- dumpTSTemplate(fout, (TSTemplateInfo *) dobj);
- break;
- case DO_TSCONFIG:
- dumpTSConfig(fout, (TSConfigInfo *) dobj);
- break;
- case DO_FDW:
- dumpForeignDataWrapper(fout, (FdwInfo *) dobj);
- break;
- case DO_FOREIGN_SERVER:
- dumpForeignServer(fout, (ForeignServerInfo *) dobj);
- break;
- case DO_DEFAULT_ACL:
- dumpDefaultACL(fout, (DefaultACLInfo *) dobj);
- break;
- case DO_BLOB:
- dumpBlob(fout, (BlobInfo *) dobj);
- break;
- case DO_BLOB_DATA:
- ArchiveEntry(fout, dobj->catId, dobj->dumpId,
- dobj->name, NULL, NULL, "",
- false, "BLOBS", SECTION_DATA,
- "", "", NULL,
- NULL, 0,
- dumpBlobs, NULL);
- break;
- case DO_PRE_DATA_BOUNDARY:
- case DO_POST_DATA_BOUNDARY:
- /* never dumped, nothing to do */
- break;
- }
- }
- /*
- * dumpNamespace
- * writes out to fout the queries to recreate a user-defined namespace
- */
- static void
- dumpNamespace(Archive *fout, NamespaceInfo *nspinfo)
- {
- PQExpBuffer q;
- PQExpBuffer delq;
- PQExpBuffer labelq;
- char *qnspname;
- /* Skip if not to be dumped */
- if (!nspinfo->dobj.dump || dataOnly)
- return;
- /* don't dump dummy namespace from pre-7.3 source */
- if (strlen(nspinfo->dobj.name) == 0)
- return;
- q = createPQExpBuffer();
- delq = createPQExpBuffer();
- labelq = createPQExpBuffer();
- qnspname = pg_strdup(fmtId(nspinfo->dobj.name));
- appendPQExpBuffer(delq, "DROP SCHEMA %s;\n", qnspname);
- appendPQExpBuffer(q, "CREATE SCHEMA %s;\n", qnspname);
- appendPQExpBuffer(labelq, "SCHEMA %s", qnspname);
- if (binary_upgrade)
- binary_upgrade_extension_member(q, &nspinfo->dobj, labelq->data);
- ArchiveEntry(fout, nspinfo->dobj.catId, nspinfo->dobj.dumpId,
- nspinfo->dobj.name,
- NULL, NULL,
- nspinfo->rolname,
- false, "SCHEMA", SECTION_PRE_DATA,
- q->data, delq->data, NULL,
- NULL, 0,
- NULL, NULL);
- /* Dump Schema Comments and Security Labels */
- dumpComment(fout, labelq->data,
- NULL, nspinfo->rolname,
- nspinfo->dobj.catId, 0, nspinfo->dobj.dumpId);
- dumpSecLabel(fout, labelq->data,
- NULL, nspinfo->rolname,
- nspinfo->dobj.catId, 0, nspinfo->dobj.dumpId);
- dumpACL(fout, nspinfo->dobj.catId, nspinfo->dobj.dumpId, "SCHEMA",
- qnspname, NULL, nspinfo->dobj.name, NULL,
- nspinfo->rolname, nspinfo->nspacl);
- free(qnspname);
- destroyPQExpBuffer(q);
- destroyPQExpBuffer(delq);
- destroyPQExpBuffer(labelq);
- }
- /*
- * dumpExtension
- * writes out to fout the queries to recreate an extension
- */
- static void
- dumpExtension(Archive *fout, ExtensionInfo *extinfo)
- {
- PQExpBuffer q;
- PQExpBuffer delq;
- PQExpBuffer labelq;
- char *qextname;
- /* Skip if not to be dumped */
- if (!extinfo->dobj.dump || dataOnly)
- return;
- q = createPQExpBuffer();
- delq = createPQExpBuffer();
- labelq = createPQExpBuffer();
- qextname = pg_strdup(fmtId(extinfo->dobj.name));
- appendPQExpBuffer(delq, "DROP EXTENSION %s;\n", qextname);
- if (!binary_upgrade)
- {
- /*
- * In a regular dump, we use IF NOT EXISTS so that there isn't a
- * problem if the extension already exists in the target database;
- * this is essential for installed-by-default extensions such as
- * plpgsql.
- *
- * In binary-upgrade mode, that doesn't work well, so instead we skip
- * built-in extensions based on their OIDs; see
- * selectDumpableExtension.
- */
- appendPQExpBuffer(q, "CREATE EXTENSION IF NOT EXISTS %s WITH SCHEMA %s;\n",
- qextname, fmtId(extinfo->namespace));
- }
- else
- {
- int i;
- int n;
- appendPQExpBufferStr(q, "-- For binary upgrade, create an empty extension and insert objects into it\n");
- /*
- * We unconditionally create the extension, so we must drop it if it
- * exists. This could happen if the user deleted 'plpgsql' and then
- * readded it, causing its oid to be greater than FirstNormalObjectId.
- * The FirstNormalObjectId test was kept to avoid repeatedly dropping
- * and recreating extensions like 'plpgsql'.
- */
- appendPQExpBuffer(q, "DROP EXTENSION IF EXISTS %s;\n", qextname);
- appendPQExpBufferStr(q,
- "SELECT binary_upgrade.create_empty_extension(");
- appendStringLiteralAH(q, extinfo->dobj.name, fout);
- appendPQExpBufferStr(q, ", ");
- appendStringLiteralAH(q, extinfo->namespace, fout);
- appendPQExpBufferStr(q, ", ");
- appendPQExpBuffer(q, "%s, ", extinfo->relocatable ? "true" : "false");
- appendStringLiteralAH(q, extinfo->extversion, fout);
- appendPQExpBufferStr(q, ", ");
- /*
- * Note that we're pushing extconfig (an OID array) back into
- * pg_extension exactly as-is. This is OK because pg_class OIDs are
- * preserved in binary upgrade.
- */
- if (strlen(extinfo->extconfig) > 2)
- appendStringLiteralAH(q, extinfo->extconfig, fout);
- else
- appendPQExpBufferStr(q, "NULL");
- appendPQExpBufferStr(q, ", ");
- if (strlen(extinfo->extcondition) > 2)
- appendStringLiteralAH(q, extinfo->extcondition, fout);
- else
- appendPQExpBufferStr(q, "NULL");
- appendPQExpBufferStr(q, ", ");
- appendPQExpBufferStr(q, "ARRAY[");
- n = 0;
- for (i = 0; i < extinfo->dobj.nDeps; i++)
- {
- DumpableObject *extobj;
- extobj = findObjectByDumpId(extinfo->dobj.dependencies[i]);
- if (extobj && extobj->objType == DO_EXTENSION)
- {
- if (n++ > 0)
- appendPQExpBufferChar(q, ',');
- appendStringLiteralAH(q, extobj->name, fout);
- }
- }
- appendPQExpBufferStr(q, "]::pg_catalog.text[]");
- appendPQExpBufferStr(q, ");\n");
- }
- appendPQExpBuffer(labelq, "EXTENSION %s", qextname);
- ArchiveEntry(fout, extinfo->dobj.catId, extinfo->dobj.dumpId,
- extinfo->dobj.name,
- NULL, NULL,
- "",
- false, "EXTENSION", SECTION_PRE_DATA,
- q->data, delq->data, NULL,
- NULL, 0,
- NULL, NULL);
- /* Dump Extension Comments and Security Labels */
- dumpComment(fout, labelq->data,
- NULL, "",
- extinfo->dobj.catId, 0, extinfo->dobj.dumpId);
- dumpSecLabel(fout, labelq->data,
- NULL, "",
- extinfo->dobj.catId, 0, extinfo->dobj.dumpId);
- free(qextname);
- destroyPQExpBuffer(q);
- destroyPQExpBuffer(delq);
- destroyPQExpBuffer(labelq);
- }
- /*
- * dumpType
- * writes out to fout the queries to recreate a user-defined type
- */
- static void
- dumpType(Archive *fout, TypeInfo *tyinfo)
- {
- /* Skip if not to be dumped */
- if (!tyinfo->dobj.dump || dataOnly)
- return;
- /* Dump out in proper style */
- if (tyinfo->typtype == TYPTYPE_BASE)
- dumpBaseType(fout, tyinfo);
- else if (tyinfo->typtype == TYPTYPE_DOMAIN)
- dumpDomain(fout, tyinfo);
- else if (tyinfo->typtype == TYPTYPE_COMPOSITE)
- dumpCompositeType(fout, tyinfo);
- else if (tyinfo->typtype == TYPTYPE_ENUM)
- dumpEnumType(fout, tyinfo);
- else if (tyinfo->typtype == TYPTYPE_RANGE)
- dumpRangeType(fout, tyinfo);
- else
- write_msg(NULL, "WARNING: typtype of data type \"%s\" appears to be invalid\n",
- tyinfo->dobj.name);
- }
- /*
- * dumpEnumType
- * writes out to fout the queries to recreate a user-defined enum type
- */
- static void
- dumpEnumType(Archive *fout, TypeInfo *tyinfo)
- {
- PQExpBuffer q = createPQExpBuffer();
- PQExpBuffer delq = createPQExpBuffer();
- PQExpBuffer labelq = createPQExpBuffer();
- PQExpBuffer query = createPQExpBuffer();
- PGresult *res;
- int num,
- i;
- Oid enum_oid;
- char *qtypname;
- char *label;
- /* Set proper schema search path */
- selectSourceSchema(fout, "pg_catalog");
- if (fout->remoteVersion >= 90100)
- appendPQExpBuffer(query, "SELECT oid, enumlabel "
- "FROM pg_catalog.pg_enum "
- "WHERE enumtypid = '%u'"
- "ORDER BY enumsortorder",
- tyinfo->dobj.catId.oid);
- else
- appendPQExpBuffer(query, "SELECT oid, enumlabel "
- "FROM pg_catalog.pg_enum "
- "WHERE enumtypid = '%u'"
- "ORDER BY oid",
- tyinfo->dobj.catId.oid);
- res = ExecuteSqlQuery(fout, query->data, PGRES_TUPLES_OK);
- num = PQntuples(res);
- qtypname = pg_strdup(fmtId(tyinfo->dobj.name));
- /*
- * DROP must be fully qualified in case same name appears in pg_catalog.
- * CASCADE shouldn't be required here as for normal types since the I/O
- * functions are generic and do not get dropped.
- */
- appendPQExpBuffer(delq, "DROP TYPE %s.",
- fmtId(tyinfo->dobj.namespace->dobj.name));
- appendPQExpBuffer(delq, "%s;\n",
- qtypname);
- if (binary_upgrade)
- binary_upgrade_set_type_oids_by_type_oid(fout, q,
- tyinfo->dobj.catId.oid);
- appendPQExpBuffer(q, "CREATE TYPE %s AS ENUM (",
- qtypname);
- if (!binary_upgrade)
- {
- /* Labels with server-assigned oids */
- for (i = 0; i < num; i++)
- {
- label = PQgetvalue(res, i, PQfnumber(res, "enumlabel"));
- if (i > 0)
- appendPQExpBufferChar(q, ',');
- appendPQExpBufferStr(q, "\n ");
- appendStringLiteralAH(q, label, fout);
- }
- }
- appendPQExpBufferStr(q, "\n);\n");
- if (binary_upgrade)
- {
- /* Labels with dump-assigned (preserved) oids */
- for (i = 0; i < num; i++)
- {
- enum_oid = atooid(PQgetvalue(res, i, PQfnumber(res, "oid")));
- label = PQgetvalue(res, i, PQfnumber(res, "enumlabel"));
- if (i == 0)
- appendPQExpBufferStr(q, "\n-- For binary upgrade, must preserve pg_enum oids\n");
- appendPQExpBuffer(q,
- "SELECT binary_upgrade.set_next_pg_enum_oid('%u'::pg_catalog.oid);\n",
- enum_oid);
- appendPQExpBuffer(q, "ALTER TYPE %s.",
- fmtId(tyinfo->dobj.namespace->dobj.name));
- appendPQExpBuffer(q, "%s ADD VALUE ",
- qtypname);
- appendStringLiteralAH(q, label, fout);
- appendPQExpBufferStr(q, ";\n\n");
- }
- }
- appendPQExpBuffer(labelq, "TYPE %s", qtypname);
- if (binary_upgrade)
- binary_upgrade_extension_member(q, &tyinfo->dobj, labelq->data);
- ArchiveEntry(fout, tyinfo->dobj.catId, tyinfo->dobj.dumpId,
- tyinfo->dobj.name,
- tyinfo->dobj.namespace->dobj.name,
- NULL,
- tyinfo->rolname, false,
- "TYPE", SECTION_PRE_DATA,
- q->data, delq->data, NULL,
- NULL, 0,
- NULL, NULL);
- /* Dump Type Comments and Security Labels */
- dumpComment(fout, labelq->data,
- tyinfo->dobj.namespace->dobj.name, tyinfo->rolname,
- tyinfo->dobj.catId, 0, tyinfo->dobj.dumpId);
- dumpSecLabel(fout, labelq->data,
- tyinfo->dobj.namespace->dobj.name, tyinfo->rolname,
- tyinfo->dobj.catId, 0, tyinfo->dobj.dumpId);
- dumpACL(fout, tyinfo->dobj.catId, tyinfo->dobj.dumpId, "TYPE",
- qtypname, NULL, tyinfo->dobj.name,
- tyinfo->dobj.namespace->dobj.name,
- tyinfo->rolname, tyinfo->typacl);
- PQclear(res);
- destroyPQExpBuffer(q);
- destroyPQExpBuffer(delq);
- destroyPQExpBuffer(labelq);
- destroyPQExpBuffer(query);
- }
- /*
- * dumpRangeType
- * writes out to fout the queries to recreate a user-defined range type
- */
- static void
- dumpRangeType(Archive *fout, TypeInfo *tyinfo)
- {
- PQExpBuffer q = createPQExpBuffer();
- PQExpBuffer delq = createPQExpBuffer();
- PQExpBuffer labelq = createPQExpBuffer();
- PQExpBuffer query = createPQExpBuffer();
- PGresult *res;
- Oid collationOid;
- char *qtypname;
- char *procname;
- /*
- * select appropriate schema to ensure names in CREATE are properly
- * qualified
- */
- selectSourceSchema(fout, tyinfo->dobj.namespace->dobj.name);
- appendPQExpBuffer(query,
- "SELECT pg_catalog.format_type(rngsubtype, NULL) AS rngsubtype, "
- "opc.opcname AS opcname, "
- "(SELECT nspname FROM pg_catalog.pg_namespace nsp "
- " WHERE nsp.oid = opc.opcnamespace) AS opcnsp, "
- "opc.opcdefault, "
- "CASE WHEN rngcollation = st.typcollation THEN 0 "
- " ELSE rngcollation END AS collation, "
- "rngcanonical, rngsubdiff "
- "FROM pg_catalog.pg_range r, pg_catalog.pg_type st, "
- " pg_catalog.pg_opclass opc "
- "WHERE st.oid = rngsubtype AND opc.oid = rngsubopc AND "
- "rngtypid = '%u'",
- tyinfo->dobj.catId.oid);
- res = ExecuteSqlQueryForSingleRow(fout, query->data);
- qtypname = pg_strdup(fmtId(tyinfo->dobj.name));
- /*
- * DROP must be fully qualified in case same name appears in pg_catalog.
- * CASCADE shouldn't be required here as for normal types since the I/O
- * functions are generic and do not get dropped.
- */
- appendPQExpBuffer(delq, "DROP TYPE %s.",
- fmtId(tyinfo->dobj.namespace->dobj.name));
- appendPQExpBuffer(delq, "%s;\n",
- qtypname);
- if (binary_upgrade)
- binary_upgrade_set_type_oids_by_type_oid(fout,
- q, tyinfo->dobj.catId.oid);
- appendPQExpBuffer(q, "CREATE TYPE %s AS RANGE (",
- qtypname);
- appendPQExpBuffer(q, "\n subtype = %s",
- PQgetvalue(res, 0, PQfnumber(res, "rngsubtype")));
- /* print subtype_opclass only if not default for subtype */
- if (PQgetvalue(res, 0, PQfnumber(res, "opcdefault"))[0] != 't')
- {
- char *opcname = PQgetvalue(res, 0, PQfnumber(res, "opcname"));
- char *nspname = PQgetvalue(res, 0, PQfnumber(res, "opcnsp"));
- /* always schema-qualify, don't try to be smart */
- appendPQExpBuffer(q, ",\n subtype_opclass = %s.",
- fmtId(nspname));
- appendPQExpBufferStr(q, fmtId(opcname));
- }
- collationOid = atooid(PQgetvalue(res, 0, PQfnumber(res, "collation")));
- if (OidIsValid(collationOid))
- {
- CollInfo *coll = findCollationByOid(collationOid);
- if (coll)
- {
- /* always schema-qualify, don't try to be smart */
- appendPQExpBuffer(q, ",\n collation = %s.",
- fmtId(coll->dobj.namespace->dobj.name));
- appendPQExpBufferStr(q, fmtId(coll->dobj.name));
- }
- }
- procname = PQgetvalue(res, 0, PQfnumber(res, "rngcanonical"));
- if (strcmp(procname, "-") != 0)
- appendPQExpBuffer(q, ",\n canonical = %s", procname);
- procname = PQgetvalue(res, 0, PQfnumber(res, "rngsubdiff"));
- if (strcmp(procname, "-") != 0)
- appendPQExpBuffer(q, ",\n subtype_diff = %s", procname);
- appendPQExpBufferStr(q, "\n);\n");
- appendPQExpBuffer(labelq, "TYPE %s", qtypname);
- if (binary_upgrade)
- binary_upgrade_extension_member(q, &tyinfo->dobj, labelq->data);
- ArchiveEntry(fout, tyinfo->dobj.catId, tyinfo->dobj.dumpId,
- tyinfo->dobj.name,
- tyinfo->dobj.namespace->dobj.name,
- NULL,
- tyinfo->rolname, false,
- "TYPE", SECTION_PRE_DATA,
- q->data, delq->data, NULL,
- NULL, 0,
- NULL, NULL);
- /* Dump Type Comments and Security Labels */
- dumpComment(fout, labelq->data,
- tyinfo->dobj.namespace->dobj.name, tyinfo->rolname,
- tyinfo->dobj.catId, 0, tyinfo->dobj.dumpId);
- dumpSecLabel(fout, labelq->data,
- tyinfo->dobj.namespace->dobj.name, tyinfo->rolname,
- tyinfo->dobj.catId, 0, tyinfo->dobj.dumpId);
- dumpACL(fout, tyinfo->dobj.catId, tyinfo->dobj.dumpId, "TYPE",
- qtypname, NULL, tyinfo->dobj.name,
- tyinfo->dobj.namespace->dobj.name,
- tyinfo->rolname, tyinfo->typacl);
- PQclear(res);
- destroyPQExpBuffer(q);
- destroyPQExpBuffer(delq);
- destroyPQExpBuffer(labelq);
- destroyPQExpBuffer(query);
- }
- /*
- * dumpBaseType
- * writes out to fout the queries to recreate a user-defined base type
- */
- static void
- dumpBaseType(Archive *fout, TypeInfo *tyinfo)
- {
- PQExpBuffer q = createPQExpBuffer();
- PQExpBuffer delq = createPQExpBuffer();
- PQExpBuffer labelq = createPQExpBuffer();
- PQExpBuffer query = createPQExpBuffer();
- PGresult *res;
- char *qtypname;
- char *typlen;
- char *typinput;
- char *typoutput;
- char *typreceive;
- char *typsend;
- char *typmodin;
- char *typmodout;
- char *typanalyze;
- Oid typreceiveoid;
- Oid typsendoid;
- Oid typmodinoid;
- Oid typmodoutoid;
- Oid typanalyzeoid;
- char *typcategory;
- char *typispreferred;
- char *typdelim;
- char *typbyval;
- char *typalign;
- char *typstorage;
- char *typcollatable;
- char *typdefault;
- bool typdefault_is_literal = false;
- /* Set proper schema search path so regproc references list correctly */
- selectSourceSchema(fout, tyinfo->dobj.namespace->dobj.name);
- /* Fetch type-specific details */
- if (fout->remoteVersion >= 90100)
- {
- appendPQExpBuffer(query, "SELECT typlen, "
- "typinput, typoutput, typreceive, typsend, "
- "typmodin, typmodout, typanalyze, "
- "typreceive::pg_catalog.oid AS typreceiveoid, "
- "typsend::pg_catalog.oid AS typsendoid, "
- "typmodin::pg_catalog.oid AS typmodinoid, "
- "typmodout::pg_catalog.oid AS typmodoutoid, "
- "typanalyze::pg_catalog.oid AS typanalyzeoid, "
- "typcategory, typispreferred, "
- "typdelim, typbyval, typalign, typstorage, "
- "(typcollation <> 0) AS typcollatable, "
- "pg_catalog.pg_get_expr(typdefaultbin, 0) AS typdefaultbin, typdefault "
- "FROM pg_catalog.pg_type "
- "WHERE oid = '%u'::pg_catalog.oid",
- tyinfo->dobj.catId.oid);
- }
- else if (fout->remoteVersion >= 80400)
- {
- appendPQExpBuffer(query, "SELECT typlen, "
- "typinput, typoutput, typreceive, typsend, "
- "typmodin, typmodout, typanalyze, "
- "typreceive::pg_catalog.oid AS typreceiveoid, "
- "typsend::pg_catalog.oid AS typsendoid, "
- "typmodin::pg_catalog.oid AS typmodinoid, "
- "typmodout::pg_catalog.oid AS typmodoutoid, "
- "typanalyze::pg_catalog.oid AS typanalyzeoid, "
- "typcategory, typispreferred, "
- "typdelim, typbyval, typalign, typstorage, "
- "false AS typcollatable, "
- "pg_catalog.pg_get_expr(typdefaultbin, 0) AS typdefaultbin, typdefault "
- "FROM pg_catalog.pg_type "
- "WHERE oid = '%u'::pg_catalog.oid",
- tyinfo->dobj.catId.oid);
- }
- else if (fout->remoteVersion >= 80300)
- {
- /* Before 8.4, pg_get_expr does not allow 0 for its second arg */
- appendPQExpBuffer(query, "SELECT typlen, "
- "typinput, typoutput, typreceive, typsend, "
- "typmodin, typmodout, typanalyze, "
- "typreceive::pg_catalog.oid AS typreceiveoid, "
- "typsend::pg_catalog.oid AS typsendoid, "
- "typmodin::pg_catalog.oid AS typmodinoid, "
- "typmodout::pg_catalog.oid AS typmodoutoid, "
- "typanalyze::pg_catalog.oid AS typanalyzeoid, "
- "'U' AS typcategory, false AS typispreferred, "
- "typdelim, typbyval, typalign, typstorage, "
- "false AS typcollatable, "
- "pg_catalog.pg_get_expr(typdefaultbin, 'pg_catalog.pg_type'::pg_catalog.regclass) AS typdefaultbin, typdefault "
- "FROM pg_catalog.pg_type "
- "WHERE oid = '%u'::pg_catalog.oid",
- tyinfo->dobj.catId.oid);
- }
- else if (fout->remoteVersion >= 80000)
- {
- appendPQExpBuffer(query, "SELECT typlen, "
- "typinput, typoutput, typreceive, typsend, "
- "'-' AS typmodin, '-' AS typmodout, "
- "typanalyze, "
- "typreceive::pg_catalog.oid AS typreceiveoid, "
- "typsend::pg_catalog.oid AS typsendoid, "
- "0 AS typmodinoid, 0 AS typmodoutoid, "
- "typanalyze::pg_catalog.oid AS typanalyzeoid, "
- "'U' AS typcategory, false AS typispreferred, "
- "typdelim, typbyval, typalign, typstorage, "
- "false AS typcollatable, "
- "pg_catalog.pg_get_expr(typdefaultbin, 'pg_catalog.pg_type'::pg_catalog.regclass) AS typdefaultbin, typdefault "
- "FROM pg_catalog.pg_type "
- "WHERE oid = '%u'::pg_catalog.oid",
- tyinfo->dobj.catId.oid);
- }
- else if (fout->remoteVersion >= 70400)
- {
- appendPQExpBuffer(query, "SELECT typlen, "
- "typinput, typoutput, typreceive, typsend, "
- "'-' AS typmodin, '-' AS typmodout, "
- "'-' AS typanalyze, "
- "typreceive::pg_catalog.oid AS typreceiveoid, "
- "typsend::pg_catalog.oid AS typsendoid, "
- "0 AS typmodinoid, 0 AS typmodoutoid, "
- "0 AS typanalyzeoid, "
- "'U' AS typcategory, false AS typispreferred, "
- "typdelim, typbyval, typalign, typstorage, "
- "false AS typcollatable, "
- "pg_catalog.pg_get_expr(typdefaultbin, 'pg_catalog.pg_type'::pg_catalog.regclass) AS typdefaultbin, typdefault "
- "FROM pg_catalog.pg_type "
- "WHERE oid = '%u'::pg_catalog.oid",
- tyinfo->dobj.catId.oid);
- }
- else if (fout->remoteVersion >= 70300)
- {
- appendPQExpBuffer(query, "SELECT typlen, "
- "typinput, typoutput, "
- "'-' AS typreceive, '-' AS typsend, "
- "'-' AS typmodin, '-' AS typmodout, "
- "'-' AS typanalyze, "
- "0 AS typreceiveoid, 0 AS typsendoid, "
- "0 AS typmodinoid, 0 AS typmodoutoid, "
- "0 AS typanalyzeoid, "
- "'U' AS typcategory, false AS typispreferred, "
- "typdelim, typbyval, typalign, typstorage, "
- "false AS typcollatable, "
- "pg_catalog.pg_get_expr(typdefaultbin, 'pg_catalog.pg_type'::pg_catalog.regclass) AS typdefaultbin, typdefault "
- "FROM pg_catalog.pg_type "
- "WHERE oid = '%u'::pg_catalog.oid",
- tyinfo->dobj.catId.oid);
- }
- else if (fout->remoteVersion >= 70200)
- {
- /*
- * Note: although pre-7.3 catalogs contain typreceive and typsend,
- * ignore them because they are not right.
- */
- appendPQExpBuffer(query, "SELECT typlen, "
- "typinput, typoutput, "
- "'-' AS typreceive, '-' AS typsend, "
- "'-' AS typmodin, '-' AS typmodout, "
- "'-' AS typanalyze, "
- "0 AS typreceiveoid, 0 AS typsendoid, "
- "0 AS typmodinoid, 0 AS typmodoutoid, "
- "0 AS typanalyzeoid, "
- "'U' AS typcategory, false AS typispreferred, "
- "typdelim, typbyval, typalign, typstorage, "
- "false AS typcollatable, "
- "NULL AS typdefaultbin, typdefault "
- "FROM pg_type "
- "WHERE oid = '%u'::oid",
- tyinfo->dobj.catId.oid);
- }
- else if (fout->remoteVersion >= 70100)
- {
- /*
- * Ignore pre-7.2 typdefault; the field exists but has an unusable
- * representation.
- */
- appendPQExpBuffer(query, "SELECT typlen, "
- "typinput, typoutput, "
- "'-' AS typreceive, '-' AS typsend, "
- "'-' AS typmodin, '-' AS typmodout, "
- "'-' AS typanalyze, "
- "0 AS typreceiveoid, 0 AS typsendoid, "
- "0 AS typmodinoid, 0 AS typmodoutoid, "
- "0 AS typanalyzeoid, "
- "'U' AS typcategory, false AS typispreferred, "
- "typdelim, typbyval, typalign, typstorage, "
- "false AS typcollatable, "
- "NULL AS typdefaultbin, NULL AS typdefault "
- "FROM pg_type "
- "WHERE oid = '%u'::oid",
- tyinfo->dobj.catId.oid);
- }
- else
- {
- appendPQExpBuffer(query, "SELECT typlen, "
- "typinput, typoutput, "
- "'-' AS typreceive, '-' AS typsend, "
- "'-' AS typmodin, '-' AS typmodout, "
- "'-' AS typanalyze, "
- "0 AS typreceiveoid, 0 AS typsendoid, "
- "0 AS typmodinoid, 0 AS typmodoutoid, "
- "0 AS typanalyzeoid, "
- "'U' AS typcategory, false AS typispreferred, "
- "typdelim, typbyval, typalign, "
- "'p'::char AS typstorage, "
- "false AS typcollatable, "
- "NULL AS typdefaultbin, NULL AS typdefault "
- "FROM pg_type "
- "WHERE oid = '%u'::oid",
- tyinfo->dobj.catId.oid);
- }
- res = ExecuteSqlQueryForSingleRow(fout, query->data);
- typlen = PQgetvalue(res, 0, PQfnumber(res, "typlen"));
- typinput = PQgetvalue(res, 0, PQfnumber(res, "typinput"));
- typoutput = PQgetvalue(res, 0, PQfnumber(res, "typoutput"));
- typreceive = PQgetvalue(res, 0, PQfnumber(res, "typreceive"));
- typsend = PQgetvalue(res, 0, PQfnumber(res, "typsend"));
- typmodin = PQgetvalue(res, 0, PQfnumber(res, "typmodin"));
- typmodout = PQgetvalue(res, 0, PQfnumber(res, "typmodout"));
- typanalyze = PQgetvalue(res, 0, PQfnumber(res, "typanalyze"));
- typreceiveoid = atooid(PQgetvalue(res, 0, PQfnumber(res, "typreceiveoid")));
- typsendoid = atooid(PQgetvalue(res, 0, PQfnumber(res, "typsendoid")));
- typmodinoid = atooid(PQgetvalue(res, 0, PQfnumber(res, "typmodinoid")));
- typmodoutoid = atooid(PQgetvalue(res, 0, PQfnumber(res, "typmodoutoid")));
- typanalyzeoid = atooid(PQgetvalue(res, 0, PQfnumber(res, "typanalyzeoid")));
- typcategory = PQgetvalue(res, 0, PQfnumber(res, "typcategory"));
- typispreferred = PQgetvalue(res, 0, PQfnumber(res, "typispreferred"));
- typdelim = PQgetvalue(res, 0, PQfnumber(res, "typdelim"));
- typbyval = PQgetvalue(res, 0, PQfnumber(res, "typbyval"));
- typalign = PQgetvalue(res, 0, PQfnumber(res, "typalign"));
- typstorage = PQgetvalue(res, 0, PQfnumber(res, "typstorage"));
- typcollatable = PQgetvalue(res, 0, PQfnumber(res, "typcollatable"));
- if (!PQgetisnull(res, 0, PQfnumber(res, "typdefaultbin")))
- typdefault = PQgetvalue(res, 0, PQfnumber(res, "typdefaultbin"));
- else if (!PQgetisnull(res, 0, PQfnumber(res, "typdefault")))
- {
- typdefault = PQgetvalue(res, 0, PQfnumber(res, "typdefault"));
- typdefault_is_literal = true; /* it needs quotes */
- }
- else
- typdefault = NULL;
- qtypname = pg_strdup(fmtId(tyinfo->dobj.name));
- /*
- * DROP must be fully qualified in case same name appears in pg_catalog.
- * The reason we include CASCADE is that the circular dependency between
- * the type and its I/O functions makes it impossible to drop the type any
- * other way.
- */
- appendPQExpBuffer(delq, "DROP TYPE %s.",
- fmtId(tyinfo->dobj.namespace->dobj.name));
- appendPQExpBuffer(delq, "%s CASCADE;\n",
- qtypname);
- /* We might already have a shell type, but setting pg_type_oid is harmless */
- if (binary_upgrade)
- binary_upgrade_set_type_oids_by_type_oid(fout, q,
- tyinfo->dobj.catId.oid);
- appendPQExpBuffer(q,
- "CREATE TYPE %s (\n"
- " INTERNALLENGTH = %s",
- qtypname,
- (strcmp(typlen, "-1") == 0) ? "variable" : typlen);
- if (fout->remoteVersion >= 70300)
- {
- /* regproc result is correctly quoted as of 7.3 */
- appendPQExpBuffer(q, ",\n INPUT = %s", typinput);
- appendPQExpBuffer(q, ",\n OUTPUT = %s", typoutput);
- if (OidIsValid(typreceiveoid))
- appendPQExpBuffer(q, ",\n RECEIVE = %s", typreceive);
- if (OidIsValid(typsendoid))
- appendPQExpBuffer(q, ",\n SEND = %s", typsend);
- if (OidIsValid(typmodinoid))
- appendPQExpBuffer(q, ",\n TYPMOD_IN = %s", typmodin);
- if (OidIsValid(typmodoutoid))
- appendPQExpBuffer(q, ",\n TYPMOD_OUT = %s", typmodout);
- if (OidIsValid(typanalyzeoid))
- appendPQExpBuffer(q, ",\n ANALYZE = %s", typanalyze);
- }
- else
- {
- /* regproc delivers an unquoted name before 7.3 */
- /* cannot combine these because fmtId uses static result area */
- appendPQExpBuffer(q, ",\n INPUT = %s", fmtId(typinput));
- appendPQExpBuffer(q, ",\n OUTPUT = %s", fmtId(typoutput));
- /* receive/send/typmodin/typmodout/analyze need not be printed */
- }
- if (strcmp(typcollatable, "t") == 0)
- appendPQExpBufferStr(q, ",\n COLLATABLE = true");
- if (typdefault != NULL)
- {
- appendPQExpBufferStr(q, ",\n DEFAULT = ");
- if (typdefault_is_literal)
- appendStringLiteralAH(q, typdefault, fout);
- else
- appendPQExpBufferStr(q, typdefault);
- }
- if (OidIsValid(tyinfo->typelem))
- {
- char *elemType;
- /* reselect schema in case changed by function dump */
- selectSourceSchema(fout, tyinfo->dobj.namespace->dobj.name);
- elemType = getFormattedTypeName(fout, tyinfo->typelem, zeroAsOpaque);
- appendPQExpBuffer(q, ",\n ELEMENT = %s", elemType);
- free(elemType);
- }
- if (strcmp(typcategory, "U") != 0)
- {
- appendPQExpBufferStr(q, ",\n CATEGORY = ");
- appendStringLiteralAH(q, typcategory, fout);
- }
- if (strcmp(typispreferred, "t") == 0)
- appendPQExpBufferStr(q, ",\n PREFERRED = true");
- if (typdelim && strcmp(typdelim, ",") != 0)
- {
- appendPQExpBufferStr(q, ",\n DELIMITER = ");
- appendStringLiteralAH(q, typdelim, fout);
- }
- if (strcmp(typalign, "c") == 0)
- appendPQExpBufferStr(q, ",\n ALIGNMENT = char");
- else if (strcmp(typalign, "s") == 0)
- appendPQExpBufferStr(q, ",\n ALIGNMENT = int2");
- else if (strcmp(typalign, "i") == 0)
- appendPQExpBufferStr(q, ",\n ALIGNMENT = int4");
- else if (strcmp(typalign, "d") == 0)
- appendPQExpBufferStr(q, ",\n ALIGNMENT = double");
- if (strcmp(typstorage, "p") == 0)
- appendPQExpBufferStr(q, ",\n STORAGE = plain");
- else if (strcmp(typstorage, "e") == 0)
- appendPQExpBufferStr(q, ",\n STORAGE = external");
- else if (strcmp(typstorage, "x") == 0)
- appendPQExpBufferStr(q, ",\n STORAGE = extended");
- else if (strcmp(typstorage, "m") == 0)
- appendPQExpBufferStr(q, ",\n STORAGE = main");
- if (strcmp(typbyval, "t") == 0)
- appendPQExpBufferStr(q, ",\n PASSEDBYVALUE");
- appendPQExpBufferStr(q, "\n);\n");
- appendPQExpBuffer(labelq, "TYPE %s", qtypname);
- if (binary_upgrade)
- binary_upgrade_extension_member(q, &tyinfo->dobj, labelq->data);
- ArchiveEntry(fout, tyinfo->dobj.catId, tyinfo->dobj.dumpId,
- tyinfo->dobj.name,
- tyinfo->dobj.namespace->dobj.name,
- NULL,
- tyinfo->rolname, false,
- "TYPE", SECTION_PRE_DATA,
- q->data, delq->data, NULL,
- NULL, 0,
- NULL, NULL);
- /* Dump Type Comments and Security Labels */
- dumpComment(fout, labelq->data,
- tyinfo->dobj.namespace->dobj.name, tyinfo->rolname,
- tyinfo->dobj.catId, 0, tyinfo->dobj.dumpId);
- dumpSecLabel(fout, labelq->data,
- tyinfo->dobj.namespace->dobj.name, tyinfo->rolname,
- tyinfo->dobj.catId, 0, tyinfo->dobj.dumpId);
- dumpACL(fout, tyinfo->dobj.catId, tyinfo->dobj.dumpId, "TYPE",
- qtypname, NULL, tyinfo->dobj.name,
- tyinfo->dobj.namespace->dobj.name,
- tyinfo->rolname, tyinfo->typacl);
- PQclear(res);
- destroyPQExpBuffer(q);
- destroyPQExpBuffer(delq);
- destroyPQExpBuffer(labelq);
- destroyPQExpBuffer(query);
- }
- /*
- * dumpDomain
- * writes out to fout the queries to recreate a user-defined domain
- */
- static void
- dumpDomain(Archive *fout, TypeInfo *tyinfo)
- {
- PQExpBuffer q = createPQExpBuffer();
- PQExpBuffer delq = createPQExpBuffer();
- PQExpBuffer labelq = createPQExpBuffer();
- PQExpBuffer query = createPQExpBuffer();
- PGresult *res;
- int i;
- char *qtypname;
- char *typnotnull;
- char *typdefn;
- char *typdefault;
- Oid typcollation;
- bool typdefault_is_literal = false;
- /* Set proper schema search path so type references list correctly */
- selectSourceSchema(fout, tyinfo->dobj.namespace->dobj.name);
- /* Fetch domain specific details */
- if (fout->remoteVersion >= 90100)
- {
- /* typcollation is new in 9.1 */
- appendPQExpBuffer(query, "SELECT t.typnotnull, "
- "pg_catalog.format_type(t.typbasetype, t.typtypmod) AS typdefn, "
- "pg_catalog.pg_get_expr(t.typdefaultbin, 'pg_catalog.pg_type'::pg_catalog.regclass) AS typdefaultbin, "
- "t.typdefault, "
- "CASE WHEN t.typcollation <> u.typcollation "
- "THEN t.typcollation ELSE 0 END AS typcollation "
- "FROM pg_catalog.pg_type t "
- "LEFT JOIN pg_catalog.pg_type u ON (t.typbasetype = u.oid) "
- "WHERE t.oid = '%u'::pg_catalog.oid",
- tyinfo->dobj.catId.oid);
- }
- else
- {
- /* We assume here that remoteVersion must be at least 70300 */
- appendPQExpBuffer(query, "SELECT typnotnull, "
- "pg_catalog.format_type(typbasetype, typtypmod) AS typdefn, "
- "pg_catalog.pg_get_expr(typdefaultbin, 'pg_catalog.pg_type'::pg_catalog.regclass) AS typdefaultbin, "
- "typdefault, 0 AS typcollation "
- "FROM pg_catalog.pg_type "
- "WHERE oid = '%u'::pg_catalog.oid",
- tyinfo->dobj.catId.oid);
- }
- res = ExecuteSqlQueryForSingleRow(fout, query->data);
- typnotnull = PQgetvalue(res, 0, PQfnumber(res, "typnotnull"));
- typdefn = PQgetvalue(res, 0, PQfnumber(res, "typdefn"));
- if (!PQgetisnull(res, 0, PQfnumber(res, "typdefaultbin")))
- typdefault = PQgetvalue(res, 0, PQfnumber(res, "typdefaultbin"));
- else if (!PQgetisnull(res, 0, PQfnumber(res, "typdefault")))
- {
- typdefault = PQgetvalue(res, 0, PQfnumber(res, "typdefault"));
- typdefault_is_literal = true; /* it needs quotes */
- }
- else
- typdefault = NULL;
- typcollation = atooid(PQgetvalue(res, 0, PQfnumber(res, "typcollation")));
- if (binary_upgrade)
- binary_upgrade_set_type_oids_by_type_oid(fout, q,
- tyinfo->dobj.catId.oid);
- qtypname = pg_strdup(fmtId(tyinfo->dobj.name));
- appendPQExpBuffer(q,
- "CREATE DOMAIN %s AS %s",
- qtypname,
- typdefn);
- /* Print collation only if different from base type's collation */
- if (OidIsValid(typcollation))
- {
- CollInfo *coll;
- coll = findCollationByOid(typcollation);
- if (coll)
- {
- /* always schema-qualify, don't try to be smart */
- appendPQExpBuffer(q, " COLLATE %s.",
- fmtId(coll->dobj.namespace->dobj.name));
- appendPQExpBufferStr(q, fmtId(coll->dobj.name));
- }
- }
- if (typnotnull[0] == 't')
- appendPQExpBufferStr(q, " NOT NULL");
- if (typdefault != NULL)
- {
- appendPQExpBufferStr(q, " DEFAULT ");
- if (typdefault_is_literal)
- appendStringLiteralAH(q, typdefault, fout);
- else
- appendPQExpBufferStr(q, typdefault);
- }
- PQclear(res);
- /*
- * Add any CHECK constraints for the domain
- */
- for (i = 0; i < tyinfo->nDomChecks; i++)
- {
- ConstraintInfo *domcheck = &(tyinfo->domChecks[i]);
- if (!domcheck->separate)
- appendPQExpBuffer(q, "\n\tCONSTRAINT %s %s",
- fmtId(domcheck->dobj.name), domcheck->condef);
- }
- appendPQExpBufferStr(q, ";\n");
- /*
- * DROP must be fully qualified in case same name appears in pg_catalog
- */
- appendPQExpBuffer(delq, "DROP DOMAIN %s.",
- fmtId(tyinfo->dobj.namespace->dobj.name));
- appendPQExpBuffer(delq, "%s;\n",
- qtypname);
- appendPQExpBuffer(labelq, "DOMAIN %s", qtypname);
- if (binary_upgrade)
- binary_upgrade_extension_member(q, &tyinfo->dobj, labelq->data);
- ArchiveEntry(fout, tyinfo->dobj.catId, tyinfo->dobj.dumpId,
- tyinfo->dobj.name,
- tyinfo->dobj.namespace->dobj.name,
- NULL,
- tyinfo->rolname, false,
- "DOMAIN", SECTION_PRE_DATA,
- q->data, delq->data, NULL,
- NULL, 0,
- NULL, NULL);
- /* Dump Domain Comments and Security Labels */
- dumpComment(fout, labelq->data,
- tyinfo->dobj.namespace->dobj.name, tyinfo->rolname,
- tyinfo->dobj.catId, 0, tyinfo->dobj.dumpId);
- dumpSecLabel(fout, labelq->data,
- tyinfo->dobj.namespace->dobj.name, tyinfo->rolname,
- tyinfo->dobj.catId, 0, tyinfo->dobj.dumpId);
- dumpACL(fout, tyinfo->dobj.catId, tyinfo->dobj.dumpId, "TYPE",
- qtypname, NULL, tyinfo->dobj.name,
- tyinfo->dobj.namespace->dobj.name,
- tyinfo->rolname, tyinfo->typacl);
- destroyPQExpBuffer(q);
- destroyPQExpBuffer(delq);
- destroyPQExpBuffer(labelq);
- destroyPQExpBuffer(query);
- }
- /*
- * dumpCompositeType
- * writes out to fout the queries to recreate a user-defined stand-alone
- * composite type
- */
- static void
- dumpCompositeType(Archive *fout, TypeInfo *tyinfo)
- {
- PQExpBuffer q = createPQExpBuffer();
- PQExpBuffer dropped = createPQExpBuffer();
- PQExpBuffer delq = createPQExpBuffer();
- PQExpBuffer labelq = createPQExpBuffer();
- PQExpBuffer query = createPQExpBuffer();
- PGresult *res;
- char *qtypname;
- int ntups;
- int i_attname;
- int i_atttypdefn;
- int i_attlen;
- int i_attalign;
- int i_attisdropped;
- int i_attcollation;
- int i_typrelid;
- int i;
- int actual_atts;
- /* Set proper schema search path so type references list correctly */
- selectSourceSchema(fout, tyinfo->dobj.namespace->dobj.name);
- /* Fetch type specific details */
- if (fout->remoteVersion >= 90100)
- {
- /*
- * attcollation is new in 9.1. Since we only want to dump COLLATE
- * clauses for attributes whose collation is different from their
- * type's default, we use a CASE here to suppress uninteresting
- * attcollations cheaply. atttypid will be 0 for dropped columns;
- * collation does not matter for those.
- */
- appendPQExpBuffer(query, "SELECT a.attname, "
- "pg_catalog.format_type(a.atttypid, a.atttypmod) AS atttypdefn, "
- "a.attlen, a.attalign, a.attisdropped, "
- "CASE WHEN a.attcollation <> at.typcollation "
- "THEN a.attcollation ELSE 0 END AS attcollation, "
- "ct.typrelid "
- "FROM pg_catalog.pg_type ct "
- "JOIN pg_catalog.pg_attribute a ON a.attrelid = ct.typrelid "
- "LEFT JOIN pg_catalog.pg_type at ON at.oid = a.atttypid "
- "WHERE ct.oid = '%u'::pg_catalog.oid "
- "ORDER BY a.attnum ",
- tyinfo->dobj.catId.oid);
- }
- else
- {
- /*
- * We assume here that remoteVersion must be at least 70300. Since
- * ALTER TYPE could not drop columns until 9.1, attisdropped should
- * always be false.
- */
- appendPQExpBuffer(query, "SELECT a.attname, "
- "pg_catalog.format_type(a.atttypid, a.atttypmod) AS atttypdefn, "
- "a.attlen, a.attalign, a.attisdropped, "
- "0 AS attcollation, "
- "ct.typrelid "
- "FROM pg_catalog.pg_type ct, pg_catalog.pg_attribute a "
- "WHERE ct.oid = '%u'::pg_catalog.oid "
- "AND a.attrelid = ct.typrelid "
- "ORDER BY a.attnum ",
- tyinfo->dobj.catId.oid);
- }
- res = ExecuteSqlQuery(fout, query->data, PGRES_TUPLES_OK);
- ntups = PQntuples(res);
- i_attname = PQfnumber(res, "attname");
- i_atttypdefn = PQfnumber(res, "atttypdefn");
- i_attlen = PQfnumber(res, "attlen");
- i_attalign = PQfnumber(res, "attalign");
- i_attisdropped = PQfnumber(res, "attisdropped");
- i_attcollation = PQfnumber(res, "attcollation");
- i_typrelid = PQfnumber(res, "typrelid");
- if (binary_upgrade)
- {
- Oid typrelid = atooid(PQgetvalue(res, 0, i_typrelid));
- binary_upgrade_set_type_oids_by_type_oid(fout, q,
- tyinfo->dobj.catId.oid);
- binary_upgrade_set_pg_class_oids(fout, q, typrelid, false);
- }
- qtypname = pg_strdup(fmtId(tyinfo->dobj.name));
- appendPQExpBuffer(q, "CREATE TYPE %s AS (",
- qtypname);
- actual_atts = 0;
- for (i = 0; i < ntups; i++)
- {
- char *attname;
- char *atttypdefn;
- char *attlen;
- char *attalign;
- bool attisdropped;
- Oid attcollation;
- attname = PQgetvalue(res, i, i_attname);
- atttypdefn = PQgetvalue(res, i, i_atttypdefn);
- attlen = PQgetvalue(res, i, i_attlen);
- attalign = PQgetvalue(res, i, i_attalign);
- attisdropped = (PQgetvalue(res, i, i_attisdropped)[0] == 't');
- attcollation = atooid(PQgetvalue(res, i, i_attcollation));
- if (attisdropped && !binary_upgrade)
- continue;
- /* Format properly if not first attr */
- if (actual_atts++ > 0)
- appendPQExpBufferChar(q, ',');
- appendPQExpBufferStr(q, "\n\t");
- if (!attisdropped)
- {
- appendPQExpBuffer(q, "%s %s", fmtId(attname), atttypdefn);
- /* Add collation if not default for the column type */
- if (OidIsValid(attcollation))
- {
- CollInfo *coll;
- coll = findCollationByOid(attcollation);
- if (coll)
- {
- /* always schema-qualify, don't try to be smart */
- appendPQExpBuffer(q, " COLLATE %s.",
- fmtId(coll->dobj.namespace->dobj.name));
- appendPQExpBufferStr(q, fmtId(coll->dobj.name));
- }
- }
- }
- else
- {
- /*
- * This is a dropped attribute and we're in binary_upgrade mode.
- * Insert a placeholder for it in the CREATE TYPE command, and set
- * length and alignment with direct UPDATE to the catalogs
- * afterwards. See similar code in dumpTableSchema().
- */
- appendPQExpBuffer(q, "%s INTEGER /* dummy */", fmtId(attname));
- /* stash separately for insertion after the CREATE TYPE */
- appendPQExpBufferStr(dropped,
- "\n-- For binary upgrade, recreate dropped column.\n");
- appendPQExpBuffer(dropped, "UPDATE pg_catalog.pg_attribute\n"
- "SET attlen = %s, "
- "attalign = '%s', attbyval = false\n"
- "WHERE attname = ", attlen, attalign);
- appendStringLiteralAH(dropped, attname, fout);
- appendPQExpBufferStr(dropped, "\n AND attrelid = ");
- appendStringLiteralAH(dropped, qtypname, fout);
- appendPQExpBufferStr(dropped, "::pg_catalog.regclass;\n");
- appendPQExpBuffer(dropped, "ALTER TYPE %s ",
- qtypname);
- appendPQExpBuffer(dropped, "DROP ATTRIBUTE %s;\n",
- fmtId(attname));
- }
- }
- appendPQExpBufferStr(q, "\n);\n");
- appendPQExpBufferStr(q, dropped->data);
- /*
- * DROP must be fully qualified in case same name appears in pg_catalog
- */
- appendPQExpBuffer(delq, "DROP TYPE %s.",
- fmtId(tyinfo->dobj.namespace->dobj.name));
- appendPQExpBuffer(delq, "%s;\n",
- qtypname);
- appendPQExpBuffer(labelq, "TYPE %s", qtypname);
- if (binary_upgrade)
- binary_upgrade_extension_member(q, &tyinfo->dobj, labelq->data);
- ArchiveEntry(fout, tyinfo->dobj.catId, tyinfo->dobj.dumpId,
- tyinfo->dobj.name,
- tyinfo->dobj.namespace->dobj.name,
- NULL,
- tyinfo->rolname, false,
- "TYPE", SECTION_PRE_DATA,
- q->data, delq->data, NULL,
- NULL, 0,
- NULL, NULL);
- /* Dump Type Comments and Security Labels */
- dumpComment(fout, labelq->data,
- tyinfo->dobj.namespace->dobj.name, tyinfo->rolname,
- tyinfo->dobj.catId, 0, tyinfo->dobj.dumpId);
- dumpSecLabel(fout, labelq->data,
- tyinfo->dobj.namespace->dobj.name, tyinfo->rolname,
- tyinfo->dobj.catId, 0, tyinfo->dobj.dumpId);
- dumpACL(fout, tyinfo->dobj.catId, tyinfo->dobj.dumpId, "TYPE",
- qtypname, NULL, tyinfo->dobj.name,
- tyinfo->dobj.namespace->dobj.name,
- tyinfo->rolname, tyinfo->typacl);
- PQclear(res);
- destroyPQExpBuffer(q);
- destroyPQExpBuffer(dropped);
- destroyPQExpBuffer(delq);
- destroyPQExpBuffer(labelq);
- destroyPQExpBuffer(query);
- /* Dump any per-column comments */
- dumpCompositeTypeColComments(fout, tyinfo);
- }
- /*
- * dumpCompositeTypeColComments
- * writes out to fout the queries to recreate comments on the columns of
- * a user-defined stand-alone composite type
- */
- static void
- dumpCompositeTypeColComments(Archive *fout, TypeInfo *tyinfo)
- {
- CommentItem *comments;
- int ncomments;
- PGresult *res;
- PQExpBuffer query;
- PQExpBuffer target;
- Oid pgClassOid;
- int i;
- int ntups;
- int i_attname;
- int i_attnum;
- query = createPQExpBuffer();
- /* We assume here that remoteVersion must be at least 70300 */
- appendPQExpBuffer(query,
- "SELECT c.tableoid, a.attname, a.attnum "
- "FROM pg_catalog.pg_class c, pg_catalog.pg_attribute a "
- "WHERE c.oid = '%u' AND c.oid = a.attrelid "
- " AND NOT a.attisdropped "
- "ORDER BY a.attnum ",
- tyinfo->typrelid);
- /* Fetch column attnames */
- res = ExecuteSqlQuery(fout, query->data, PGRES_TUPLES_OK);
- ntups = PQntuples(res);
- if (ntups < 1)
- {
- PQclear(res);
- destroyPQExpBuffer(query);
- return;
- }
- pgClassOid = atooid(PQgetvalue(res, 0, PQfnumber(res, "tableoid")));
- /* Search for comments associated with type's pg_class OID */
- ncomments = findComments(fout,
- pgClassOid,
- tyinfo->typrelid,
- &comments);
- /* If no comments exist, we're done */
- if (ncomments <= 0)
- {
- PQclear(res);
- destroyPQExpBuffer(query);
- return;
- }
- /* Build COMMENT ON statements */
- target = createPQExpBuffer();
- i_attnum = PQfnumber(res, "attnum");
- i_attname = PQfnumber(res, "attname");
- while (ncomments > 0)
- {
- const char *attname;
- attname = NULL;
- for (i = 0; i < ntups; i++)
- {
- if (atoi(PQgetvalue(res, i, i_attnum)) == comments->objsubid)
- {
- attname = PQgetvalue(res, i, i_attname);
- break;
- }
- }
- if (attname) /* just in case we don't find it */
- {
- const char *descr = comments->descr;
- resetPQExpBuffer(target);
- appendPQExpBuffer(target, "COLUMN %s.",
- fmtId(tyinfo->dobj.name));
- appendPQExpBufferStr(target, fmtId(attname));
- resetPQExpBuffer(query);
- appendPQExpBuffer(query, "COMMENT ON %s IS ", target->data);
- appendStringLiteralAH(query, descr, fout);
- appendPQExpBufferStr(query, ";\n");
- ArchiveEntry(fout, nilCatalogId, createDumpId(),
- target->data,
- tyinfo->dobj.namespace->dobj.name,
- NULL, tyinfo->rolname,
- false, "COMMENT", SECTION_NONE,
- query->data, "", NULL,
- &(tyinfo->dobj.dumpId), 1,
- NULL, NULL);
- }
- comments++;
- ncomments--;
- }
- PQclear(res);
- destroyPQExpBuffer(query);
- destroyPQExpBuffer(target);
- }
- /*
- * dumpShellType
- * writes out to fout the queries to create a shell type
- *
- * We dump a shell definition in advance of the I/O functions for the type.
- */
- static void
- dumpShellType(Archive *fout, ShellTypeInfo *stinfo)
- {
- PQExpBuffer q;
- /* Skip if not to be dumped */
- if (!stinfo->dobj.dump || dataOnly)
- return;
- q = createPQExpBuffer();
- /*
- * Note the lack of a DROP command for the shell type; any required DROP
- * is driven off the base type entry, instead. This interacts with
- * _printTocEntry()'s use of the presence of a DROP command to decide
- * whether an entry needs an ALTER OWNER command. We don't want to alter
- * the shell type's owner immediately on creation; that should happen only
- * after it's filled in, otherwise the backend complains.
- */
- if (binary_upgrade)
- binary_upgrade_set_type_oids_by_type_oid(fout, q,
- stinfo->baseType->dobj.catId.oid);
- appendPQExpBuffer(q, "CREATE TYPE %s;\n",
- fmtId(stinfo->dobj.name));
- ArchiveEntry(fout, stinfo->dobj.catId, stinfo->dobj.dumpId,
- stinfo->dobj.name,
- stinfo->dobj.namespace->dobj.name,
- NULL,
- stinfo->baseType->rolname, false,
- "SHELL TYPE", SECTION_PRE_DATA,
- q->data, "", NULL,
- NULL, 0,
- NULL, NULL);
- destroyPQExpBuffer(q);
- }
- /*
- * Determine whether we want to dump definitions for procedural languages.
- * Since the languages themselves don't have schemas, we can't rely on
- * the normal schema-based selection mechanism. We choose to dump them
- * whenever neither --schema nor --table was given. (Before 8.1, we used
- * the dump flag of the PL's call handler function, but in 8.1 this will
- * probably always be false since call handlers are created in pg_catalog.)
- *
- * For some backwards compatibility with the older behavior, we forcibly
- * dump a PL if its handler function (and validator if any) are in a
- * dumpable namespace. That case is not checked here.
- *
- * Also, if the PL belongs to an extension, we do not use this heuristic.
- * That case isn't checked here either.
- */
- static bool
- shouldDumpProcLangs(void)
- {
- if (!include_everything)
- return false;
- /* And they're schema not data */
- if (dataOnly)
- return false;
- return true;
- }
- /*
- * dumpProcLang
- * writes out to fout the queries to recreate a user-defined
- * procedural language
- */
- static void
- dumpProcLang(Archive *fout, ProcLangInfo *plang)
- {
- PQExpBuffer defqry;
- PQExpBuffer delqry;
- PQExpBuffer labelq;
- bool useParams;
- char *qlanname;
- char *lanschema;
- FuncInfo *funcInfo;
- FuncInfo *inlineInfo = NULL;
- FuncInfo *validatorInfo = NULL;
- /* Skip if not to be dumped */
- if (!plang->dobj.dump || dataOnly)
- return;
- /*
- * Try to find the support function(s). It is not an error if we don't
- * find them --- if the functions are in the pg_catalog schema, as is
- * standard in 8.1 and up, then we won't have loaded them. (In this case
- * we will emit a parameterless CREATE LANGUAGE command, which will
- * require PL template knowledge in the backend to reload.)
- */
- funcInfo = findFuncByOid(plang->lanplcallfoid);
- if (funcInfo != NULL && !funcInfo->dobj.dump)
- funcInfo = NULL; /* treat not-dumped same as not-found */
- if (OidIsValid(plang->laninline))
- {
- inlineInfo = findFuncByOid(plang->laninline);
- if (inlineInfo != NULL && !inlineInfo->dobj.dump)
- inlineInfo = NULL;
- }
- if (OidIsValid(plang->lanvalidator))
- {
- validatorInfo = findFuncByOid(plang->lanvalidator);
- if (validatorInfo != NULL && !validatorInfo->dobj.dump)
- validatorInfo = NULL;
- }
- /*
- * If the functions are dumpable then emit a traditional CREATE LANGUAGE
- * with parameters. Otherwise, dump only if shouldDumpProcLangs() says to
- * dump it.
- *
- * However, for a language that belongs to an extension, we must not use
- * the shouldDumpProcLangs heuristic, but just dump the language iff we're
- * told to (via dobj.dump). Generally the support functions will belong
- * to the same extension and so have the same dump flags ... if they
- * don't, this might not work terribly nicely.
- */
- useParams = (funcInfo != NULL &&
- (inlineInfo != NULL || !OidIsValid(plang->laninline)) &&
- (validatorInfo != NULL || !OidIsValid(plang->lanvalidator)));
- if (!plang->dobj.ext_member)
- {
- if (!useParams && !shouldDumpProcLangs())
- return;
- }
- defqry = createPQExpBuffer();
- delqry = createPQExpBuffer();
- labelq = createPQExpBuffer();
- qlanname = pg_strdup(fmtId(plang->dobj.name));
- /*
- * If dumping a HANDLER clause, treat the language as being in the handler
- * function's schema; this avoids cluttering the HANDLER clause. Otherwise
- * it doesn't really have a schema.
- */
- if (useParams)
- lanschema = funcInfo->dobj.namespace->dobj.name;
- else
- lanschema = NULL;
- appendPQExpBuffer(delqry, "DROP PROCEDURAL LANGUAGE %s;\n",
- qlanname);
- if (useParams)
- {
- appendPQExpBuffer(defqry, "CREATE %sPROCEDURAL LANGUAGE %s",
- plang->lanpltrusted ? "TRUSTED " : "",
- qlanname);
- appendPQExpBuffer(defqry, " HANDLER %s",
- fmtId(funcInfo->dobj.name));
- if (OidIsValid(plang->laninline))
- {
- appendPQExpBufferStr(defqry, " INLINE ");
- /* Cope with possibility that inline is in different schema */
- if (inlineInfo->dobj.namespace != funcInfo->dobj.namespace)
- appendPQExpBuffer(defqry, "%s.",
- fmtId(inlineInfo->dobj.namespace->dobj.name));
- appendPQExpBufferStr(defqry, fmtId(inlineInfo->dobj.name));
- }
- if (OidIsValid(plang->lanvalidator))
- {
- appendPQExpBufferStr(defqry, " VALIDATOR ");
- /* Cope with possibility that validator is in different schema */
- if (validatorInfo->dobj.namespace != funcInfo->dobj.namespace)
- appendPQExpBuffer(defqry, "%s.",
- fmtId(validatorInfo->dobj.namespace->dobj.name));
- appendPQExpBufferStr(defqry, fmtId(validatorInfo->dobj.name));
- }
- }
- else
- {
- /*
- * If not dumping parameters, then use CREATE OR REPLACE so that the
- * command will not fail if the language is preinstalled in the target
- * database. We restrict the use of REPLACE to this case so as to
- * eliminate the risk of replacing a language with incompatible
- * parameter settings: this command will only succeed at all if there
- * is a pg_pltemplate entry, and if there is one, the existing entry
- * must match it too.
- */
- appendPQExpBuffer(defqry, "CREATE OR REPLACE PROCEDURAL LANGUAGE %s",
- qlanname);
- }
- appendPQExpBufferStr(defqry, ";\n");
- appendPQExpBuffer(labelq, "LANGUAGE %s", qlanname);
- if (binary_upgrade)
- binary_upgrade_extension_member(defqry, &plang->dobj, labelq->data);
- ArchiveEntry(fout, plang->dobj.catId, plang->dobj.dumpId,
- plang->dobj.name,
- lanschema, NULL, plang->lanowner,
- false, "PROCEDURAL LANGUAGE", SECTION_PRE_DATA,
- defqry->data, delqry->data, NULL,
- NULL, 0,
- NULL, NULL);
- /* Dump Proc Lang Comments and Security Labels */
- dumpComment(fout, labelq->data,
- NULL, "",
- plang->dobj.catId, 0, plang->dobj.dumpId);
- dumpSecLabel(fout, labelq->data,
- NULL, "",
- plang->dobj.catId, 0, plang->dobj.dumpId);
- if (plang->lanpltrusted)
- dumpACL(fout, plang->dobj.catId, plang->dobj.dumpId, "LANGUAGE",
- qlanname, NULL, plang->dobj.name,
- lanschema,
- plang->lanowner, plang->lanacl);
- free(qlanname);
- destroyPQExpBuffer(defqry);
- destroyPQExpBuffer(delqry);
- destroyPQExpBuffer(labelq);
- }
- /*
- * format_function_arguments: generate function name and argument list
- *
- * This is used when we can rely on pg_get_function_arguments to format
- * the argument list. Note, however, that pg_get_function_arguments
- * does not special-case zero-argument aggregates.
- */
- static char *
- format_function_arguments(FuncInfo *finfo, char *funcargs, bool is_agg)
- {
- PQExpBufferData fn;
- initPQExpBuffer(&fn);
- appendPQExpBufferStr(&fn, fmtId(finfo->dobj.name));
- if (is_agg && finfo->nargs == 0)
- appendPQExpBufferStr(&fn, "(*)");
- else
- appendPQExpBuffer(&fn, "(%s)", funcargs);
- return fn.data;
- }
- /*
- * format_function_arguments_old: generate function name and argument list
- *
- * The argument type names are qualified if needed. The function name
- * is never qualified.
- *
- * This is used only with pre-8.4 servers, so we aren't expecting to see
- * VARIADIC or TABLE arguments, nor are there any defaults for arguments.
- *
- * Any or all of allargtypes, argmodes, argnames may be NULL.
- */
- static char *
- format_function_arguments_old(Archive *fout,
- FuncInfo *finfo, int nallargs,
- char **allargtypes,
- char **argmodes,
- char **argnames)
- {
- PQExpBufferData fn;
- int j;
- initPQExpBuffer(&fn);
- appendPQExpBuffer(&fn, "%s(", fmtId(finfo->dobj.name));
- for (j = 0; j < nallargs; j++)
- {
- Oid typid;
- char *typname;
- const char *argmode;
- const char *argname;
- typid = allargtypes ? atooid(allargtypes[j]) : finfo->argtypes[j];
- typname = getFormattedTypeName(fout, typid, zeroAsOpaque);
- if (argmodes)
- {
- switch (argmodes[j][0])
- {
- case PROARGMODE_IN:
- argmode = "";
- break;
- case PROARGMODE_OUT:
- argmode = "OUT ";
- break;
- case PROARGMODE_INOUT:
- argmode = "INOUT ";
- break;
- default:
- write_msg(NULL, "WARNING: bogus value in proargmodes array\n");
- argmode = "";
- break;
- }
- }
- else
- argmode = "";
- argname = argnames ? argnames[j] : (char *) NULL;
- if (argname && argname[0] == '\0')
- argname = NULL;
- appendPQExpBuffer(&fn, "%s%s%s%s%s",
- (j > 0) ? ", " : "",
- argmode,
- argname ? fmtId(argname) : "",
- argname ? " " : "",
- typname);
- free(typname);
- }
- appendPQExpBufferChar(&fn, ')');
- return fn.data;
- }
- /*
- * format_function_signature: generate function name and argument list
- *
- * This is like format_function_arguments_old except that only a minimal
- * list of input argument types is generated; this is sufficient to
- * reference the function, but not to define it.
- *
- * If honor_quotes is false then the function name is never quoted.
- * This is appropriate for use in TOC tags, but not in SQL commands.
- */
- static char *
- format_function_signature(Archive *fout, FuncInfo *finfo, bool honor_quotes)
- {
- PQExpBufferData fn;
- int j;
- initPQExpBuffer(&fn);
- if (honor_quotes)
- appendPQExpBuffer(&fn, "%s(", fmtId(finfo->dobj.name));
- else
- appendPQExpBuffer(&fn, "%s(", finfo->dobj.name);
- for (j = 0; j < finfo->nargs; j++)
- {
- char *typname;
- if (j > 0)
- appendPQExpBufferStr(&fn, ", ");
- typname = getFormattedTypeName(fout, finfo->argtypes[j],
- zeroAsOpaque);
- appendPQExpBufferStr(&fn, typname);
- free(typname);
- }
- appendPQExpBufferChar(&fn, ')');
- return fn.data;
- }
- /*
- * dumpFunc:
- * dump out one function
- */
- static void
- dumpFunc(Archive *fout, FuncInfo *finfo)
- {
- PQExpBuffer query;
- PQExpBuffer q;
- PQExpBuffer delqry;
- PQExpBuffer labelq;
- PQExpBuffer asPart;
- PGresult *res;
- char *funcsig; /* identity signature */
- char *funcfullsig = NULL; /* full signature */
- char *funcsig_tag;
- char *proretset;
- char *prosrc;
- char *probin;
- char *funcargs;
- char *funciargs;
- char *funcresult;
- char *proallargtypes;
- char *proargmodes;
- char *proargnames;
- char *proiswindow;
- char *provolatile;
- char *proisstrict;
- char *prosecdef;
- char *proleakproof;
- char *proconfig;
- char *procost;
- char *prorows;
- char *lanname;
- char *rettypename;
- int nallargs;
- char **allargtypes = NULL;
- char **argmodes = NULL;
- char **argnames = NULL;
- char **configitems = NULL;
- int nconfigitems = 0;
- int i;
- /* Skip if not to be dumped */
- if (!finfo->dobj.dump || dataOnly)
- return;
- query = createPQExpBuffer();
- q = createPQExpBuffer();
- delqry = createPQExpBuffer();
- labelq = createPQExpBuffer();
- asPart = createPQExpBuffer();
- /* Set proper schema search path so type references list correctly */
- selectSourceSchema(fout, finfo->dobj.namespace->dobj.name);
- /* Fetch function-specific details */
- if (fout->remoteVersion >= 90200)
- {
- /*
- * proleakproof was added at v9.2
- */
- appendPQExpBuffer(query,
- "SELECT proretset, prosrc, probin, "
- "pg_catalog.pg_get_function_arguments(oid) AS funcargs, "
- "pg_catalog.pg_get_function_identity_arguments(oid) AS funciargs, "
- "pg_catalog.pg_get_function_result(oid) AS funcresult, "
- "proiswindow, provolatile, proisstrict, prosecdef, "
- "proleakproof, proconfig, procost, prorows, "
- "(SELECT lanname FROM pg_catalog.pg_language WHERE oid = prolang) AS lanname "
- "FROM pg_catalog.pg_proc "
- "WHERE oid = '%u'::pg_catalog.oid",
- finfo->dobj.catId.oid);
- }
- else if (fout->remoteVersion >= 80400)
- {
- /*
- * In 8.4 and up we rely on pg_get_function_arguments and
- * pg_get_function_result instead of examining proallargtypes etc.
- */
- appendPQExpBuffer(query,
- "SELECT proretset, prosrc, probin, "
- "pg_catalog.pg_get_function_arguments(oid) AS funcargs, "
- "pg_catalog.pg_get_function_identity_arguments(oid) AS funciargs, "
- "pg_catalog.pg_get_function_result(oid) AS funcresult, "
- "proiswindow, provolatile, proisstrict, prosecdef, "
- "false AS proleakproof, "
- " proconfig, procost, prorows, "
- "(SELECT lanname FROM pg_catalog.pg_language WHERE oid = prolang) AS lanname "
- "FROM pg_catalog.pg_proc "
- "WHERE oid = '%u'::pg_catalog.oid",
- finfo->dobj.catId.oid);
- }
- else if (fout->remoteVersion >= 80300)
- {
- appendPQExpBuffer(query,
- "SELECT proretset, prosrc, probin, "
- "proallargtypes, proargmodes, proargnames, "
- "false AS proiswindow, "
- "provolatile, proisstrict, prosecdef, "
- "false AS proleakproof, "
- "proconfig, procost, prorows, "
- "(SELECT lanname FROM pg_catalog.pg_language WHERE oid = prolang) AS lanname "
- "FROM pg_catalog.pg_proc "
- "WHERE oid = '%u'::pg_catalog.oid",
- finfo->dobj.catId.oid);
- }
- else if (fout->remoteVersion >= 80100)
- {
- appendPQExpBuffer(query,
- "SELECT proretset, prosrc, probin, "
- "proallargtypes, proargmodes, proargnames, "
- "false AS proiswindow, "
- "provolatile, proisstrict, prosecdef, "
- "false AS proleakproof, "
- "null AS proconfig, 0 AS procost, 0 AS prorows, "
- "(SELECT lanname FROM pg_catalog.pg_language WHERE oid = prolang) AS lanname "
- "FROM pg_catalog.pg_proc "
- "WHERE oid = '%u'::pg_catalog.oid",
- finfo->dobj.catId.oid);
- }
- else if (fout->remoteVersion >= 80000)
- {
- appendPQExpBuffer(query,
- "SELECT proretset, prosrc, probin, "
- "null AS proallargtypes, "
- "null AS proargmodes, "
- "proargnames, "
- "false AS proiswindow, "
- "provolatile, proisstrict, prosecdef, "
- "false AS proleakproof, "
- "null AS proconfig, 0 AS procost, 0 AS prorows, "
- "(SELECT lanname FROM pg_catalog.pg_language WHERE oid = prolang) AS lanname "
- "FROM pg_catalog.pg_proc "
- "WHERE oid = '%u'::pg_catalog.oid",
- finfo->dobj.catId.oid);
- }
- else if (fout->remoteVersion >= 70300)
- {
- appendPQExpBuffer(query,
- "SELECT proretset, prosrc, probin, "
- "null AS proallargtypes, "
- "null AS proargmodes, "
- "null AS proargnames, "
- "false AS proiswindow, "
- "provolatile, proisstrict, prosecdef, "
- "false AS proleakproof, "
- "null AS proconfig, 0 AS procost, 0 AS prorows, "
- "(SELECT lanname FROM pg_catalog.pg_language WHERE oid = prolang) AS lanname "
- "FROM pg_catalog.pg_proc "
- "WHERE oid = '%u'::pg_catalog.oid",
- finfo->dobj.catId.oid);
- }
- else if (fout->remoteVersion >= 70100)
- {
- appendPQExpBuffer(query,
- "SELECT proretset, prosrc, probin, "
- "null AS proallargtypes, "
- "null AS proargmodes, "
- "null AS proargnames, "
- "false AS proiswindow, "
- "case when proiscachable then 'i' else 'v' end AS provolatile, "
- "proisstrict, "
- "false AS prosecdef, "
- "false AS proleakproof, "
- "null AS proconfig, 0 AS procost, 0 AS prorows, "
- "(SELECT lanname FROM pg_language WHERE oid = prolang) AS lanname "
- "FROM pg_proc "
- "WHERE oid = '%u'::oid",
- finfo->dobj.catId.oid);
- }
- else
- {
- appendPQExpBuffer(query,
- "SELECT proretset, prosrc, probin, "
- "null AS proallargtypes, "
- "null AS proargmodes, "
- "null AS proargnames, "
- "false AS proiswindow, "
- "CASE WHEN proiscachable THEN 'i' ELSE 'v' END AS provolatile, "
- "false AS proisstrict, "
- "false AS prosecdef, "
- "false AS proleakproof, "
- "NULL AS proconfig, 0 AS procost, 0 AS prorows, "
- "(SELECT lanname FROM pg_language WHERE oid = prolang) AS lanname "
- "FROM pg_proc "
- "WHERE oid = '%u'::oid",
- finfo->dobj.catId.oid);
- }
- res = ExecuteSqlQueryForSingleRow(fout, query->data);
- proretset = PQgetvalue(res, 0, PQfnumber(res, "proretset"));
- prosrc = PQgetvalue(res, 0, PQfnumber(res, "prosrc"));
- probin = PQgetvalue(res, 0, PQfnumber(res, "probin"));
- if (fout->remoteVersion >= 80400)
- {
- funcargs = PQgetvalue(res, 0, PQfnumber(res, "funcargs"));
- funciargs = PQgetvalue(res, 0, PQfnumber(res, "funciargs"));
- funcresult = PQgetvalue(res, 0, PQfnumber(res, "funcresult"));
- proallargtypes = proargmodes = proargnames = NULL;
- }
- else
- {
- proallargtypes = PQgetvalue(res, 0, PQfnumber(res, "proallargtypes"));
- proargmodes = PQgetvalue(res, 0, PQfnumber(res, "proargmodes"));
- proargnames = PQgetvalue(res, 0, PQfnumber(res, "proargnames"));
- funcargs = funciargs = funcresult = NULL;
- }
- proiswindow = PQgetvalue(res, 0, PQfnumber(res, "proiswindow"));
- provolatile = PQgetvalue(res, 0, PQfnumber(res, "provolatile"));
- proisstrict = PQgetvalue(res, 0, PQfnumber(res, "proisstrict"));
- prosecdef = PQgetvalue(res, 0, PQfnumber(res, "prosecdef"));
- proleakproof = PQgetvalue(res, 0, PQfnumber(res, "proleakproof"));
- proconfig = PQgetvalue(res, 0, PQfnumber(res, "proconfig"));
- procost = PQgetvalue(res, 0, PQfnumber(res, "procost"));
- prorows = PQgetvalue(res, 0, PQfnumber(res, "prorows"));
- lanname = PQgetvalue(res, 0, PQfnumber(res, "lanname"));
- /*
- * See backend/commands/functioncmds.c for details of how the 'AS' clause
- * is used. In 8.4 and up, an unused probin is NULL (here ""); previous
- * versions would set it to "-". There are no known cases in which prosrc
- * is unused, so the tests below for "-" are probably useless.
- */
- if (probin[0] != '\0' && strcmp(probin, "-") != 0)
- {
- appendPQExpBufferStr(asPart, "AS ");
- appendStringLiteralAH(asPart, probin, fout);
- if (strcmp(prosrc, "-") != 0)
- {
- appendPQExpBufferStr(asPart, ", ");
- /*
- * where we have bin, use dollar quoting if allowed and src
- * contains quote or backslash; else use regular quoting.
- */
- if (disable_dollar_quoting ||
- (strchr(prosrc, '\'') == NULL && strchr(prosrc, '\\') == NULL))
- appendStringLiteralAH(asPart, prosrc, fout);
- else
- appendStringLiteralDQ(asPart, prosrc, NULL);
- }
- }
- else
- {
- if (strcmp(prosrc, "-") != 0)
- {
- appendPQExpBufferStr(asPart, "AS ");
- /* with no bin, dollar quote src unconditionally if allowed */
- if (disable_dollar_quoting)
- appendStringLiteralAH(asPart, prosrc, fout);
- else
- appendStringLiteralDQ(asPart, prosrc, NULL);
- }
- }
- nallargs = finfo->nargs; /* unless we learn different from allargs */
- if (proallargtypes && *proallargtypes)
- {
- int nitems = 0;
- if (!parsePGArray(proallargtypes, &allargtypes, &nitems) ||
- nitems < finfo->nargs)
- {
- write_msg(NULL, "WARNING: could not parse proallargtypes array\n");
- if (allargtypes)
- free(allargtypes);
- allargtypes = NULL;
- }
- else
- nallargs = nitems;
- }
- if (proargmodes && *proargmodes)
- {
- int nitems = 0;
- if (!parsePGArray(proargmodes, &argmodes, &nitems) ||
- nitems != nallargs)
- {
- write_msg(NULL, "WARNING: could not parse proargmodes array\n");
- if (argmodes)
- free(argmodes);
- argmodes = NULL;
- }
- }
- if (proargnames && *proargnames)
- {
- int nitems = 0;
- if (!parsePGArray(proargnames, &argnames, &nitems) ||
- nitems != nallargs)
- {
- write_msg(NULL, "WARNING: could not parse proargnames array\n");
- if (argnames)
- free(argnames);
- argnames = NULL;
- }
- }
- if (proconfig && *proconfig)
- {
- if (!parsePGArray(proconfig, &configitems, &nconfigitems))
- {
- write_msg(NULL, "WARNING: could not parse proconfig array\n");
- if (configitems)
- free(configitems);
- configitems = NULL;
- nconfigitems = 0;
- }
- }
- if (funcargs)
- {
- /* 8.4 or later; we rely on server-side code for most of the work */
- funcfullsig = format_function_arguments(finfo, funcargs, false);
- funcsig = format_function_arguments(finfo, funciargs, false);
- }
- else
- /* pre-8.4, do it ourselves */
- funcsig = format_function_arguments_old(fout,
- finfo, nallargs, allargtypes,
- argmodes, argnames);
- funcsig_tag = format_function_signature(fout, finfo, false);
- /*
- * DROP must be fully qualified in case same name appears in pg_catalog
- */
- appendPQExpBuffer(delqry, "DROP FUNCTION %s.%s;\n",
- fmtId(finfo->dobj.namespace->dobj.name),
- funcsig);
- appendPQExpBuffer(q, "CREATE FUNCTION %s ", funcfullsig ? funcfullsig :
- funcsig);
- if (funcresult)
- appendPQExpBuffer(q, "RETURNS %s", funcresult);
- else
- {
- rettypename = getFormattedTypeName(fout, finfo->prorettype,
- zeroAsOpaque);
- appendPQExpBuffer(q, "RETURNS %s%s",
- (proretset[0] == 't') ? "SETOF " : "",
- rettypename);
- free(rettypename);
- }
- appendPQExpBuffer(q, "\n LANGUAGE %s", fmtId(lanname));
- if (proiswindow[0] == 't')
- appendPQExpBufferStr(q, " WINDOW");
- if (provolatile[0] != PROVOLATILE_VOLATILE)
- {
- if (provolatile[0] == PROVOLATILE_IMMUTABLE)
- appendPQExpBufferStr(q, " IMMUTABLE");
- else if (provolatile[0] == PROVOLATILE_STABLE)
- appendPQExpBufferStr(q, " STABLE");
- else if (provolatile[0] != PROVOLATILE_VOLATILE)
- exit_horribly(NULL, "unrecognized provolatile value for function \"%s\"\n",
- finfo->dobj.name);
- }
- if (proisstrict[0] == 't')
- appendPQExpBufferStr(q, " STRICT");
- if (prosecdef[0] == 't')
- appendPQExpBufferStr(q, " SECURITY DEFINER");
- if (proleakproof[0] == 't')
- appendPQExpBufferStr(q, " LEAKPROOF");
- /*
- * COST and ROWS are emitted only if present and not default, so as not to
- * break backwards-compatibility of the dump without need. Keep this code
- * in sync with the defaults in functioncmds.c.
- */
- if (strcmp(procost, "0") != 0)
- {
- if (strcmp(lanname, "internal") == 0 || strcmp(lanname, "c") == 0)
- {
- /* default cost is 1 */
- if (strcmp(procost, "1") != 0)
- appendPQExpBuffer(q, " COST %s", procost);
- }
- else
- {
- /* default cost is 100 */
- if (strcmp(procost, "100") != 0)
- appendPQExpBuffer(q, " COST %s", procost);
- }
- }
- if (proretset[0] == 't' &&
- strcmp(prorows, "0") != 0 && strcmp(prorows, "1000") != 0)
- appendPQExpBuffer(q, " ROWS %s", prorows);
- for (i = 0; i < nconfigitems; i++)
- {
- /* we feel free to scribble on configitems[] here */
- char *configitem = configitems[i];
- char *pos;
- pos = strchr(configitem, '=');
- if (pos == NULL)
- continue;
- *pos++ = '\0';
- appendPQExpBuffer(q, "\n SET %s TO ", fmtId(configitem));
- /*
- * Some GUC variable names are 'LIST' type and hence must not be
- * quoted.
- */
- if (pg_strcasecmp(configitem, "DateStyle") == 0
- || pg_strcasecmp(configitem, "search_path") == 0)
- appendPQExpBufferStr(q, pos);
- else
- appendStringLiteralAH(q, pos, fout);
- }
- appendPQExpBuffer(q, "\n %s;\n", asPart->data);
- appendPQExpBuffer(labelq, "FUNCTION %s", funcsig);
- if (binary_upgrade)
- binary_upgrade_extension_member(q, &finfo->dobj, labelq->data);
- ArchiveEntry(fout, finfo->dobj.catId, finfo->dobj.dumpId,
- funcsig_tag,
- finfo->dobj.namespace->dobj.name,
- NULL,
- finfo->rolname, false,
- "FUNCTION", SECTION_PRE_DATA,
- q->data, delqry->data, NULL,
- NULL, 0,
- NULL, NULL);
- /* Dump Function Comments and Security Labels */
- dumpComment(fout, labelq->data,
- finfo->dobj.namespace->dobj.name, finfo->rolname,
- finfo->dobj.catId, 0, finfo->dobj.dumpId);
- dumpSecLabel(fout, labelq->data,
- finfo->dobj.namespace->dobj.name, finfo->rolname,
- finfo->dobj.catId, 0, finfo->dobj.dumpId);
- dumpACL(fout, finfo->dobj.catId, finfo->dobj.dumpId, "FUNCTION",
- funcsig, NULL, funcsig_tag,
- finfo->dobj.namespace->dobj.name,
- finfo->rolname, finfo->proacl);
- PQclear(res);
- destroyPQExpBuffer(query);
- destroyPQExpBuffer(q);
- destroyPQExpBuffer(delqry);
- destroyPQExpBuffer(labelq);
- destroyPQExpBuffer(asPart);
- free(funcsig);
- if (funcfullsig)
- free(funcfullsig);
- free(funcsig_tag);
- if (allargtypes)
- free(allargtypes);
- if (argmodes)
- free(argmodes);
- if (argnames)
- free(argnames);
- if (configitems)
- free(configitems);
- }
- /*
- * Dump a user-defined cast
- */
- static void
- dumpCast(Archive *fout, CastInfo *cast)
- {
- PQExpBuffer defqry;
- PQExpBuffer delqry;
- PQExpBuffer labelq;
- FuncInfo *funcInfo = NULL;
- /* Skip if not to be dumped */
- if (!cast->dobj.dump || dataOnly)
- return;
- /* Cannot dump if we don't have the cast function's info */
- if (OidIsValid(cast->castfunc))
- {
- funcInfo = findFuncByOid(cast->castfunc);
- if (funcInfo == NULL)
- return;
- }
- /*
- * As per discussion we dump casts if one or more of the underlying
- * objects (the conversion function and the two data types) are not
- * builtin AND if all of the non-builtin objects are included in the dump.
- * Builtin meaning, the namespace name does not start with "pg_".
- *
- * However, for a cast that belongs to an extension, we must not use this
- * heuristic, but just dump the cast iff we're told to (via dobj.dump).
- */
- if (!cast->dobj.ext_member)
- {
- TypeInfo *sourceInfo = findTypeByOid(cast->castsource);
- TypeInfo *targetInfo = findTypeByOid(cast->casttarget);
- if (sourceInfo == NULL || targetInfo == NULL)
- return;
- /*
- * Skip this cast if all objects are from pg_
- */
- if ((funcInfo == NULL ||
- strncmp(funcInfo->dobj.namespace->dobj.name, "pg_", 3) == 0) &&
- strncmp(sourceInfo->dobj.namespace->dobj.name, "pg_", 3) == 0 &&
- strncmp(targetInfo->dobj.namespace->dobj.name, "pg_", 3) == 0)
- return;
- /*
- * Skip cast if function isn't from pg_ and is not to be dumped.
- */
- if (funcInfo &&
- strncmp(funcInfo->dobj.namespace->dobj.name, "pg_", 3) != 0 &&
- !funcInfo->dobj.dump)
- return;
- /*
- * Same for the source type
- */
- if (strncmp(sourceInfo->dobj.namespace->dobj.name, "pg_", 3) != 0 &&
- !sourceInfo->dobj.dump)
- return;
- /*
- * and the target type.
- */
- if (strncmp(targetInfo->dobj.namespace->dobj.name, "pg_", 3) != 0 &&
- !targetInfo->dobj.dump)
- return;
- }
- /* Make sure we are in proper schema (needed for getFormattedTypeName) */
- selectSourceSchema(fout, "pg_catalog");
- defqry = createPQExpBuffer();
- delqry = createPQExpBuffer();
- labelq = createPQExpBuffer();
- appendPQExpBuffer(delqry, "DROP CAST (%s AS %s);\n",
- getFormattedTypeName(fout, cast->castsource, zeroAsNone),
- getFormattedTypeName(fout, cast->casttarget, zeroAsNone));
- appendPQExpBuffer(defqry, "CREATE CAST (%s AS %s) ",
- getFormattedTypeName(fout, cast->castsource, zeroAsNone),
- getFormattedTypeName(fout, cast->casttarget, zeroAsNone));
- switch (cast->castmethod)
- {
- case COERCION_METHOD_BINARY:
- appendPQExpBufferStr(defqry, "WITHOUT FUNCTION");
- break;
- case COERCION_METHOD_INOUT:
- appendPQExpBufferStr(defqry, "WITH INOUT");
- break;
- case COERCION_METHOD_FUNCTION:
- if (funcInfo)
- {
- char *fsig = format_function_signature(fout, funcInfo, true);
- /*
- * Always qualify the function name, in case it is not in
- * pg_catalog schema (format_function_signature won't qualify
- * it).
- */
- appendPQExpBuffer(defqry, "WITH FUNCTION %s.%s",
- fmtId(funcInfo->dobj.namespace->dobj.name), fsig);
- free(fsig);
- }
- else
- write_msg(NULL, "WARNING: bogus value in pg_cast.castfunc or pg_cast.castmethod field\n");
- break;
- default:
- write_msg(NULL, "WARNING: bogus value in pg_cast.castmethod field\n");
- }
- if (cast->castcontext == 'a')
- appendPQExpBufferStr(defqry, " AS ASSIGNMENT");
- else if (cast->castcontext == 'i')
- appendPQExpBufferStr(defqry, " AS IMPLICIT");
- appendPQExpBufferStr(defqry, ";\n");
- appendPQExpBuffer(labelq, "CAST (%s AS %s)",
- getFormattedTypeName(fout, cast->castsource, zeroAsNone),
- getFormattedTypeName(fout, cast->casttarget, zeroAsNone));
- if (binary_upgrade)
- binary_upgrade_extension_member(defqry, &cast->dobj, labelq->data);
- ArchiveEntry(fout, cast->dobj.catId, cast->dobj.dumpId,
- labelq->data,
- "pg_catalog", NULL, "",
- false, "CAST", SECTION_PRE_DATA,
- defqry->data, delqry->data, NULL,
- NULL, 0,
- NULL, NULL);
- /* Dump Cast Comments */
- dumpComment(fout, labelq->data,
- NULL, "",
- cast->dobj.catId, 0, cast->dobj.dumpId);
- destroyPQExpBuffer(defqry);
- destroyPQExpBuffer(delqry);
- destroyPQExpBuffer(labelq);
- }
- /*
- * dumpOpr
- * write out a single operator definition
- */
- static void
- dumpOpr(Archive *fout, OprInfo *oprinfo)
- {
- PQExpBuffer query;
- PQExpBuffer q;
- PQExpBuffer delq;
- PQExpBuffer labelq;
- PQExpBuffer oprid;
- PQExpBuffer details;
- const char *name;
- PGresult *res;
- int i_oprkind;
- int i_oprcode;
- int i_oprleft;
- int i_oprright;
- int i_oprcom;
- int i_oprnegate;
- int i_oprrest;
- int i_oprjoin;
- int i_oprcanmerge;
- int i_oprcanhash;
- char *oprkind;
- char *oprcode;
- char *oprleft;
- char *oprright;
- char *oprcom;
- char *oprnegate;
- char *oprrest;
- char *oprjoin;
- char *oprcanmerge;
- char *oprcanhash;
- char *oprregproc;
- char *oprref;
- /* Skip if not to be dumped */
- if (!oprinfo->dobj.dump || dataOnly)
- return;
- /*
- * some operators are invalid because they were the result of user
- * defining operators before commutators exist
- */
- if (!OidIsValid(oprinfo->oprcode))
- return;
- query = createPQExpBuffer();
- q = createPQExpBuffer();
- delq = createPQExpBuffer();
- labelq = createPQExpBuffer();
- oprid = createPQExpBuffer();
- details = createPQExpBuffer();
- /* Make sure we are in proper schema so regoperator works correctly */
- selectSourceSchema(fout, oprinfo->dobj.namespace->dobj.name);
- if (fout->remoteVersion >= 80300)
- {
- appendPQExpBuffer(query, "SELECT oprkind, "
- "oprcode::pg_catalog.regprocedure, "
- "oprleft::pg_catalog.regtype, "
- "oprright::pg_catalog.regtype, "
- "oprcom::pg_catalog.regoperator, "
- "oprnegate::pg_catalog.regoperator, "
- "oprrest::pg_catalog.regprocedure, "
- "oprjoin::pg_catalog.regprocedure, "
- "oprcanmerge, oprcanhash "
- "FROM pg_catalog.pg_operator "
- "WHERE oid = '%u'::pg_catalog.oid",
- oprinfo->dobj.catId.oid);
- }
- else if (fout->remoteVersion >= 70300)
- {
- appendPQExpBuffer(query, "SELECT oprkind, "
- "oprcode::pg_catalog.regprocedure, "
- "oprleft::pg_catalog.regtype, "
- "oprright::pg_catalog.regtype, "
- "oprcom::pg_catalog.regoperator, "
- "oprnegate::pg_catalog.regoperator, "
- "oprrest::pg_catalog.regprocedure, "
- "oprjoin::pg_catalog.regprocedure, "
- "(oprlsortop != 0) AS oprcanmerge, "
- "oprcanhash "
- "FROM pg_catalog.pg_operator "
- "WHERE oid = '%u'::pg_catalog.oid",
- oprinfo->dobj.catId.oid);
- }
- else if (fout->remoteVersion >= 70100)
- {
- appendPQExpBuffer(query, "SELECT oprkind, oprcode, "
- "CASE WHEN oprleft = 0 THEN '-' "
- "ELSE format_type(oprleft, NULL) END AS oprleft, "
- "CASE WHEN oprright = 0 THEN '-' "
- "ELSE format_type(oprright, NULL) END AS oprright, "
- "oprcom, oprnegate, oprrest, oprjoin, "
- "(oprlsortop != 0) AS oprcanmerge, "
- "oprcanhash "
- "FROM pg_operator "
- "WHERE oid = '%u'::oid",
- oprinfo->dobj.catId.oid);
- }
- else
- {
- appendPQExpBuffer(query, "SELECT oprkind, oprcode, "
- "CASE WHEN oprleft = 0 THEN '-'::name "
- "ELSE (SELECT typname FROM pg_type WHERE oid = oprleft) END AS oprleft, "
- "CASE WHEN oprright = 0 THEN '-'::name "
- "ELSE (SELECT typname FROM pg_type WHERE oid = oprright) END AS oprright, "
- "oprcom, oprnegate, oprrest, oprjoin, "
- "(oprlsortop != 0) AS oprcanmerge, "
- "oprcanhash "
- "FROM pg_operator "
- "WHERE oid = '%u'::oid",
- oprinfo->dobj.catId.oid);
- }
- res = ExecuteSqlQueryForSingleRow(fout, query->data);
- i_oprkind = PQfnumber(res, "oprkind");
- i_oprcode = PQfnumber(res, "oprcode");
- i_oprleft = PQfnumber(res, "oprleft");
- i_oprright = PQfnumber(res, "oprright");
- i_oprcom = PQfnumber(res, "oprcom");
- i_oprnegate = PQfnumber(res, "oprnegate");
- i_oprrest = PQfnumber(res, "oprrest");
- i_oprjoin = PQfnumber(res, "oprjoin");
- i_oprcanmerge = PQfnumber(res, "oprcanmerge");
- i_oprcanhash = PQfnumber(res, "oprcanhash");
- oprkind = PQgetvalue(res, 0, i_oprkind);
- oprcode = PQgetvalue(res, 0, i_oprcode);
- oprleft = PQgetvalue(res, 0, i_oprleft);
- oprright = PQgetvalue(res, 0, i_oprright);
- oprcom = PQgetvalue(res, 0, i_oprcom);
- oprnegate = PQgetvalue(res, 0, i_oprnegate);
- oprrest = PQgetvalue(res, 0, i_oprrest);
- oprjoin = PQgetvalue(res, 0, i_oprjoin);
- oprcanmerge = PQgetvalue(res, 0, i_oprcanmerge);
- oprcanhash = PQgetvalue(res, 0, i_oprcanhash);
- oprregproc = convertRegProcReference(fout, oprcode);
- if (oprregproc)
- {
- appendPQExpBuffer(details, " PROCEDURE = %s", oprregproc);
- free(oprregproc);
- }
- appendPQExpBuffer(oprid, "%s (",
- oprinfo->dobj.name);
- /*
- * right unary means there's a left arg and left unary means there's a
- * right arg
- */
- if (strcmp(oprkind, "r") == 0 ||
- strcmp(oprkind, "b") == 0)
- {
- if (fout->remoteVersion >= 70100)
- name = oprleft;
- else
- name = fmtId(oprleft);
- appendPQExpBuffer(details, ",\n LEFTARG = %s", name);
- appendPQExpBufferStr(oprid, name);
- }
- else
- appendPQExpBufferStr(oprid, "NONE");
- if (strcmp(oprkind, "l") == 0 ||
- strcmp(oprkind, "b") == 0)
- {
- if (fout->remoteVersion >= 70100)
- name = oprright;
- else
- name = fmtId(oprright);
- appendPQExpBuffer(details, ",\n RIGHTARG = %s", name);
- appendPQExpBuffer(oprid, ", %s)", name);
- }
- else
- appendPQExpBufferStr(oprid, ", NONE)");
- oprref = convertOperatorReference(fout, oprcom);
- if (oprref)
- {
- appendPQExpBuffer(details, ",\n COMMUTATOR = %s", oprref);
- free(oprref);
- }
- oprref = convertOperatorReference(fout, oprnegate);
- if (oprref)
- {
- appendPQExpBuffer(details, ",\n NEGATOR = %s", oprref);
- free(oprref);
- }
- if (strcmp(oprcanmerge, "t") == 0)
- appendPQExpBufferStr(details, ",\n MERGES");
- if (strcmp(oprcanhash, "t") == 0)
- appendPQExpBufferStr(details, ",\n HASHES");
- oprregproc = convertRegProcReference(fout, oprrest);
- if (oprregproc)
- {
- appendPQExpBuffer(details, ",\n RESTRICT = %s", oprregproc);
- free(oprregproc);
- }
- oprregproc = convertRegProcReference(fout, oprjoin);
- if (oprregproc)
- {
- appendPQExpBuffer(details, ",\n JOIN = %s", oprregproc);
- free(oprregproc);
- }
- /*
- * DROP must be fully qualified in case same name appears in pg_catalog
- */
- appendPQExpBuffer(delq, "DROP OPERATOR %s.%s;\n",
- fmtId(oprinfo->dobj.namespace->dobj.name),
- oprid->data);
- appendPQExpBuffer(q, "CREATE OPERATOR %s (\n%s\n);\n",
- oprinfo->dobj.name, details->data);
- appendPQExpBuffer(labelq, "OPERATOR %s", oprid->data);
- if (binary_upgrade)
- binary_upgrade_extension_member(q, &oprinfo->dobj, labelq->data);
- ArchiveEntry(fout, oprinfo->dobj.catId, oprinfo->dobj.dumpId,
- oprinfo->dobj.name,
- oprinfo->dobj.namespace->dobj.name,
- NULL,
- oprinfo->rolname,
- false, "OPERATOR", SECTION_PRE_DATA,
- q->data, delq->data, NULL,
- NULL, 0,
- NULL, NULL);
- /* Dump Operator Comments */
- dumpComment(fout, labelq->data,
- oprinfo->dobj.namespace->dobj.name, oprinfo->rolname,
- oprinfo->dobj.catId, 0, oprinfo->dobj.dumpId);
- PQclear(res);
- destroyPQExpBuffer(query);
- destroyPQExpBuffer(q);
- destroyPQExpBuffer(delq);
- destroyPQExpBuffer(labelq);
- destroyPQExpBuffer(oprid);
- destroyPQExpBuffer(details);
- }
- /*
- * Convert a function reference obtained from pg_operator
- *
- * Returns allocated string of what to print, or NULL if function references
- * is InvalidOid. Returned string is expected to be free'd by the caller.
- *
- * In 7.3 the input is a REGPROCEDURE display; we have to strip the
- * argument-types part. In prior versions, the input is a REGPROC display.
- */
- static char *
- convertRegProcReference(Archive *fout, const char *proc)
- {
- /* In all cases "-" means a null reference */
- if (strcmp(proc, "-") == 0)
- return NULL;
- if (fout->remoteVersion >= 70300)
- {
- char *name;
- char *paren;
- bool inquote;
- name = pg_strdup(proc);
- /* find non-double-quoted left paren */
- inquote = false;
- for (paren = name; *paren; paren++)
- {
- if (*paren == '(' && !inquote)
- {
- *paren = '\0';
- break;
- }
- if (*paren == '"')
- inquote = !inquote;
- }
- return name;
- }
- /* REGPROC before 7.3 does not quote its result */
- return pg_strdup(fmtId(proc));
- }
- /*
- * Convert an operator cross-reference obtained from pg_operator
- *
- * Returns an allocated string of what to print, or NULL to print nothing.
- * Caller is responsible for free'ing result string.
- *
- * In 7.3 and up the input is a REGOPERATOR display; we have to strip the
- * argument-types part, and add OPERATOR() decoration if the name is
- * schema-qualified. In older versions, the input is just a numeric OID,
- * which we search our operator list for.
- */
- static char *
- convertOperatorReference(Archive *fout, const char *opr)
- {
- OprInfo *oprInfo;
- /* In all cases "0" means a null reference */
- if (strcmp(opr, "0") == 0)
- return NULL;
- if (fout->remoteVersion >= 70300)
- {
- char *name;
- char *oname;
- char *ptr;
- bool inquote;
- bool sawdot;
- name = pg_strdup(opr);
- /* find non-double-quoted left paren, and check for non-quoted dot */
- inquote = false;
- sawdot = false;
- for (ptr = name; *ptr; ptr++)
- {
- if (*ptr == '"')
- inquote = !inquote;
- else if (*ptr == '.' && !inquote)
- sawdot = true;
- else if (*ptr == '(' && !inquote)
- {
- *ptr = '\0';
- break;
- }
- }
- /* If not schema-qualified, don't need to add OPERATOR() */
- if (!sawdot)
- return name;
- oname = psprintf("OPERATOR(%s)", name);
- free(name);
- return oname;
- }
- oprInfo = findOprByOid(atooid(opr));
- if (oprInfo == NULL)
- {
- write_msg(NULL, "WARNING: could not find operator with OID %s\n",
- opr);
- return NULL;
- }
- return pg_strdup(oprInfo->dobj.name);
- }
- /*
- * Convert a function OID obtained from pg_ts_parser or pg_ts_template
- *
- * It is sufficient to use REGPROC rather than REGPROCEDURE, since the
- * argument lists of these functions are predetermined. Note that the
- * caller should ensure we are in the proper schema, because the results
- * are search path dependent!
- */
- static const char *
- convertTSFunction(Archive *fout, Oid funcOid)
- {
- char *result;
- char query[128];
- PGresult *res;
- snprintf(query, sizeof(query),
- "SELECT '%u'::pg_catalog.regproc", funcOid);
- res = ExecuteSqlQueryForSingleRow(fout, query);
- result = pg_strdup(PQgetvalue(res, 0, 0));
- PQclear(res);
- return result;
- }
- /*
- * dumpOpclass
- * write out a single operator class definition
- */
- static void
- dumpOpclass(Archive *fout, OpclassInfo *opcinfo)
- {
- PQExpBuffer query;
- PQExpBuffer q;
- PQExpBuffer delq;
- PQExpBuffer labelq;
- PGresult *res;
- int ntups;
- int i_opcintype;
- int i_opckeytype;
- int i_opcdefault;
- int i_opcfamily;
- int i_opcfamilyname;
- int i_opcfamilynsp;
- int i_amname;
- int i_amopstrategy;
- int i_amopreqcheck;
- int i_amopopr;
- int i_sortfamily;
- int i_sortfamilynsp;
- int i_amprocnum;
- int i_amproc;
- int i_amproclefttype;
- int i_amprocrighttype;
- char *opcintype;
- char *opckeytype;
- char *opcdefault;
- char *opcfamily;
- char *opcfamilyname;
- char *opcfamilynsp;
- char *amname;
- char *amopstrategy;
- char *amopreqcheck;
- char *amopopr;
- char *sortfamily;
- char *sortfamilynsp;
- char *amprocnum;
- char *amproc;
- char *amproclefttype;
- char *amprocrighttype;
- bool needComma;
- int i;
- /* Skip if not to be dumped */
- if (!opcinfo->dobj.dump || dataOnly)
- return;
- /*
- * XXX currently we do not implement dumping of operator classes from
- * pre-7.3 databases. This could be done but it seems not worth the
- * trouble.
- */
- if (fout->remoteVersion < 70300)
- return;
- query = createPQExpBuffer();
- q = createPQExpBuffer();
- delq = createPQExpBuffer();
- labelq = createPQExpBuffer();
- /* Make sure we are in proper schema so regoperator works correctly */
- selectSourceSchema(fout, opcinfo->dobj.namespace->dobj.name);
- /* Get additional fields from the pg_opclass row */
- if (fout->remoteVersion >= 80300)
- {
- appendPQExpBuffer(query, "SELECT opcintype::pg_catalog.regtype, "
- "opckeytype::pg_catalog.regtype, "
- "opcdefault, opcfamily, "
- "opfname AS opcfamilyname, "
- "nspname AS opcfamilynsp, "
- "(SELECT amname FROM pg_catalog.pg_am WHERE oid = opcmethod) AS amname "
- "FROM pg_catalog.pg_opclass c "
- "LEFT JOIN pg_catalog.pg_opfamily f ON f.oid = opcfamily "
- "LEFT JOIN pg_catalog.pg_namespace n ON n.oid = opfnamespace "
- "WHERE c.oid = '%u'::pg_catalog.oid",
- opcinfo->dobj.catId.oid);
- }
- else
- {
- appendPQExpBuffer(query, "SELECT opcintype::pg_catalog.regtype, "
- "opckeytype::pg_catalog.regtype, "
- "opcdefault, NULL AS opcfamily, "
- "NULL AS opcfamilyname, "
- "NULL AS opcfamilynsp, "
- "(SELECT amname FROM pg_catalog.pg_am WHERE oid = opcamid) AS amname "
- "FROM pg_catalog.pg_opclass "
- "WHERE oid = '%u'::pg_catalog.oid",
- opcinfo->dobj.catId.oid);
- }
- res = ExecuteSqlQueryForSingleRow(fout, query->data);
- i_opcintype = PQfnumber(res, "opcintype");
- i_opckeytype = PQfnumber(res, "opckeytype");
- i_opcdefault = PQfnumber(res, "opcdefault");
- i_opcfamily = PQfnumber(res, "opcfamily");
- i_opcfamilyname = PQfnumber(res, "opcfamilyname");
- i_opcfamilynsp = PQfnumber(res, "opcfamilynsp");
- i_amname = PQfnumber(res, "amname");
- opcintype = PQgetvalue(res, 0, i_opcintype);
- opckeytype = PQgetvalue(res, 0, i_opckeytype);
- opcdefault = PQgetvalue(res, 0, i_opcdefault);
- /* opcfamily will still be needed after we PQclear res */
- opcfamily = pg_strdup(PQgetvalue(res, 0, i_opcfamily));
- opcfamilyname = PQgetvalue(res, 0, i_opcfamilyname);
- opcfamilynsp = PQgetvalue(res, 0, i_opcfamilynsp);
- /* amname will still be needed after we PQclear res */
- amname = pg_strdup(PQgetvalue(res, 0, i_amname));
- /*
- * DROP must be fully qualified in case same name appears in pg_catalog
- */
- appendPQExpBuffer(delq, "DROP OPERATOR CLASS %s",
- fmtId(opcinfo->dobj.namespace->dobj.name));
- appendPQExpBuffer(delq, ".%s",
- fmtId(opcinfo->dobj.name));
- appendPQExpBuffer(delq, " USING %s;\n",
- fmtId(amname));
- /* Build the fixed portion of the CREATE command */
- appendPQExpBuffer(q, "CREATE OPERATOR CLASS %s\n ",
- fmtId(opcinfo->dobj.name));
- if (strcmp(opcdefault, "t") == 0)
- appendPQExpBufferStr(q, "DEFAULT ");
- appendPQExpBuffer(q, "FOR TYPE %s USING %s",
- opcintype,
- fmtId(amname));
- if (strlen(opcfamilyname) > 0 &&
- (strcmp(opcfamilyname, opcinfo->dobj.name) != 0 ||
- strcmp(opcfamilynsp, opcinfo->dobj.namespace->dobj.name) != 0))
- {
- appendPQExpBufferStr(q, " FAMILY ");
- if (strcmp(opcfamilynsp, opcinfo->dobj.namespace->dobj.name) != 0)
- appendPQExpBuffer(q, "%s.", fmtId(opcfamilynsp));
- appendPQExpBuffer(q, "%s", fmtId(opcfamilyname));
- }
- appendPQExpBufferStr(q, " AS\n ");
- needComma = false;
- if (strcmp(opckeytype, "-") != 0)
- {
- appendPQExpBuffer(q, "STORAGE %s",
- opckeytype);
- needComma = true;
- }
- PQclear(res);
- /*
- * Now fetch and print the OPERATOR entries (pg_amop rows).
- *
- * Print only those opfamily members that are tied to the opclass by
- * pg_depend entries.
- *
- * XXX RECHECK is gone as of 8.4, but we'll still print it if dumping an
- * older server's opclass in which it is used. This is to avoid
- * hard-to-detect breakage if a newer pg_dump is used to dump from an
- * older server and then reload into that old version. This can go away
- * once 8.3 is so old as to not be of interest to anyone.
- */
- resetPQExpBuffer(query);
- if (fout->remoteVersion >= 90100)
- {
- appendPQExpBuffer(query, "SELECT amopstrategy, false AS amopreqcheck, "
- "amopopr::pg_catalog.regoperator, "
- "opfname AS sortfamily, "
- "nspname AS sortfamilynsp "
- "FROM pg_catalog.pg_amop ao JOIN pg_catalog.pg_depend ON "
- "(classid = 'pg_catalog.pg_amop'::pg_catalog.regclass AND objid = ao.oid) "
- "LEFT JOIN pg_catalog.pg_opfamily f ON f.oid = amopsortfamily "
- "LEFT JOIN pg_catalog.pg_namespace n ON n.oid = opfnamespace "
- "WHERE refclassid = 'pg_catalog.pg_opclass'::pg_catalog.regclass "
- "AND refobjid = '%u'::pg_catalog.oid "
- "AND amopfamily = '%s'::pg_catalog.oid "
- "ORDER BY amopstrategy",
- opcinfo->dobj.catId.oid,
- opcfamily);
- }
- else if (fout->remoteVersion >= 80400)
- {
- appendPQExpBuffer(query, "SELECT amopstrategy, false AS amopreqcheck, "
- "amopopr::pg_catalog.regoperator, "
- "NULL AS sortfamily, "
- "NULL AS sortfamilynsp "
- "FROM pg_catalog.pg_amop ao, pg_catalog.pg_depend "
- "WHERE refclassid = 'pg_catalog.pg_opclass'::pg_catalog.regclass "
- "AND refobjid = '%u'::pg_catalog.oid "
- "AND classid = 'pg_catalog.pg_amop'::pg_catalog.regclass "
- "AND objid = ao.oid "
- "ORDER BY amopstrategy",
- opcinfo->dobj.catId.oid);
- }
- else if (fout->remoteVersion >= 80300)
- {
- appendPQExpBuffer(query, "SELECT amopstrategy, amopreqcheck, "
- "amopopr::pg_catalog.regoperator, "
- "NULL AS sortfamily, "
- "NULL AS sortfamilynsp "
- "FROM pg_catalog.pg_amop ao, pg_catalog.pg_depend "
- "WHERE refclassid = 'pg_catalog.pg_opclass'::pg_catalog.regclass "
- "AND refobjid = '%u'::pg_catalog.oid "
- "AND classid = 'pg_catalog.pg_amop'::pg_catalog.regclass "
- "AND objid = ao.oid "
- "ORDER BY amopstrategy",
- opcinfo->dobj.catId.oid);
- }
- else
- {
- /*
- * Here, we print all entries since there are no opfamilies and hence
- * no loose operators to worry about.
- */
- appendPQExpBuffer(query, "SELECT amopstrategy, amopreqcheck, "
- "amopopr::pg_catalog.regoperator, "
- "NULL AS sortfamily, "
- "NULL AS sortfamilynsp "
- "FROM pg_catalog.pg_amop "
- "WHERE amopclaid = '%u'::pg_catalog.oid "
- "ORDER BY amopstrategy",
- opcinfo->dobj.catId.oid);
- }
- res = ExecuteSqlQuery(fout, query->data, PGRES_TUPLES_OK);
- ntups = PQntuples(res);
- i_amopstrategy = PQfnumber(res, "amopstrategy");
- i_amopreqcheck = PQfnumber(res, "amopreqcheck");
- i_amopopr = PQfnumber(res, "amopopr");
- i_sortfamily = PQfnumber(res, "sortfamily");
- i_sortfamilynsp = PQfnumber(res, "sortfamilynsp");
- for (i = 0; i < ntups; i++)
- {
- amopstrategy = PQgetvalue(res, i, i_amopstrategy);
- amopreqcheck = PQgetvalue(res, i, i_amopreqcheck);
- amopopr = PQgetvalue(res, i, i_amopopr);
- sortfamily = PQgetvalue(res, i, i_sortfamily);
- sortfamilynsp = PQgetvalue(res, i, i_sortfamilynsp);
- if (needComma)
- appendPQExpBufferStr(q, " ,\n ");
- appendPQExpBuffer(q, "OPERATOR %s %s",
- amopstrategy, amopopr);
- if (strlen(sortfamily) > 0)
- {
- appendPQExpBufferStr(q, " FOR ORDER BY ");
- if (strcmp(sortfamilynsp, opcinfo->dobj.namespace->dobj.name) != 0)
- appendPQExpBuffer(q, "%s.", fmtId(sortfamilynsp));
- appendPQExpBufferStr(q, fmtId(sortfamily));
- }
- if (strcmp(amopreqcheck, "t") == 0)
- appendPQExpBufferStr(q, " RECHECK");
- needComma = true;
- }
- PQclear(res);
- /*
- * Now fetch and print the FUNCTION entries (pg_amproc rows).
- *
- * Print only those opfamily members that are tied to the opclass by
- * pg_depend entries.
- *
- * We print the amproclefttype/amprocrighttype even though in most cases
- * the backend could deduce the right values, because of the corner case
- * of a btree sort support function for a cross-type comparison. That's
- * only allowed in 9.2 and later, but for simplicity print them in all
- * versions that have the columns.
- */
- resetPQExpBuffer(query);
- if (fout->remoteVersion >= 80300)
- {
- appendPQExpBuffer(query, "SELECT amprocnum, "
- "amproc::pg_catalog.regprocedure, "
- "amproclefttype::pg_catalog.regtype, "
- "amprocrighttype::pg_catalog.regtype "
- "FROM pg_catalog.pg_amproc ap, pg_catalog.pg_depend "
- "WHERE refclassid = 'pg_catalog.pg_opclass'::pg_catalog.regclass "
- "AND refobjid = '%u'::pg_catalog.oid "
- "AND classid = 'pg_catalog.pg_amproc'::pg_catalog.regclass "
- "AND objid = ap.oid "
- "ORDER BY amprocnum",
- opcinfo->dobj.catId.oid);
- }
- else
- {
- appendPQExpBuffer(query, "SELECT amprocnum, "
- "amproc::pg_catalog.regprocedure, "
- "'' AS amproclefttype, "
- "'' AS amprocrighttype "
- "FROM pg_catalog.pg_amproc "
- "WHERE amopclaid = '%u'::pg_catalog.oid "
- "ORDER BY amprocnum",
- opcinfo->dobj.catId.oid);
- }
- res = ExecuteSqlQuery(fout, query->data, PGRES_TUPLES_OK);
- ntups = PQntuples(res);
- i_amprocnum = PQfnumber(res, "amprocnum");
- i_amproc = PQfnumber(res, "amproc");
- i_amproclefttype = PQfnumber(res, "amproclefttype");
- i_amprocrighttype = PQfnumber(res, "amprocrighttype");
- for (i = 0; i < ntups; i++)
- {
- amprocnum = PQgetvalue(res, i, i_amprocnum);
- amproc = PQgetvalue(res, i, i_amproc);
- amproclefttype = PQgetvalue(res, i, i_amproclefttype);
- amprocrighttype = PQgetvalue(res, i, i_amprocrighttype);
- if (needComma)
- appendPQExpBufferStr(q, " ,\n ");
- appendPQExpBuffer(q, "FUNCTION %s", amprocnum);
- if (*amproclefttype && *amprocrighttype)
- appendPQExpBuffer(q, " (%s, %s)", amproclefttype, amprocrighttype);
- appendPQExpBuffer(q, " %s", amproc);
- needComma = true;
- }
- PQclear(res);
- appendPQExpBufferStr(q, ";\n");
- appendPQExpBuffer(labelq, "OPERATOR CLASS %s",
- fmtId(opcinfo->dobj.name));
- appendPQExpBuffer(labelq, " USING %s",
- fmtId(amname));
- if (binary_upgrade)
- binary_upgrade_extension_member(q, &opcinfo->dobj, labelq->data);
- ArchiveEntry(fout, opcinfo->dobj.catId, opcinfo->dobj.dumpId,
- opcinfo->dobj.name,
- opcinfo->dobj.namespace->dobj.name,
- NULL,
- opcinfo->rolname,
- false, "OPERATOR CLASS", SECTION_PRE_DATA,
- q->data, delq->data, NULL,
- NULL, 0,
- NULL, NULL);
- /* Dump Operator Class Comments */
- dumpComment(fout, labelq->data,
- NULL, opcinfo->rolname,
- opcinfo->dobj.catId, 0, opcinfo->dobj.dumpId);
- free(amname);
- destroyPQExpBuffer(query);
- destroyPQExpBuffer(q);
- destroyPQExpBuffer(delq);
- destroyPQExpBuffer(labelq);
- }
- /*
- * dumpOpfamily
- * write out a single operator family definition
- *
- * Note: this also dumps any "loose" operator members that aren't bound to a
- * specific opclass within the opfamily.
- */
- static void
- dumpOpfamily(Archive *fout, OpfamilyInfo *opfinfo)
- {
- PQExpBuffer query;
- PQExpBuffer q;
- PQExpBuffer delq;
- PQExpBuffer labelq;
- PGresult *res;
- PGresult *res_ops;
- PGresult *res_procs;
- int ntups;
- int i_amname;
- int i_amopstrategy;
- int i_amopreqcheck;
- int i_amopopr;
- int i_sortfamily;
- int i_sortfamilynsp;
- int i_amprocnum;
- int i_amproc;
- int i_amproclefttype;
- int i_amprocrighttype;
- char *amname;
- char *amopstrategy;
- char *amopreqcheck;
- char *amopopr;
- char *sortfamily;
- char *sortfamilynsp;
- char *amprocnum;
- char *amproc;
- char *amproclefttype;
- char *amprocrighttype;
- bool needComma;
- int i;
- /* Skip if not to be dumped */
- if (!opfinfo->dobj.dump || dataOnly)
- return;
- /*
- * We want to dump the opfamily only if (1) it contains "loose" operators
- * or functions, or (2) it contains an opclass with a different name or
- * owner. Otherwise it's sufficient to let it be created during creation
- * of the contained opclass, and not dumping it improves portability of
- * the dump. Since we have to fetch the loose operators/funcs anyway, do
- * that first.
- */
- query = createPQExpBuffer();
- q = createPQExpBuffer();
- delq = createPQExpBuffer();
- labelq = createPQExpBuffer();
- /* Make sure we are in proper schema so regoperator works correctly */
- selectSourceSchema(fout, opfinfo->dobj.namespace->dobj.name);
- /*
- * Fetch only those opfamily members that are tied directly to the
- * opfamily by pg_depend entries.
- *
- * XXX RECHECK is gone as of 8.4, but we'll still print it if dumping an
- * older server's opclass in which it is used. This is to avoid
- * hard-to-detect breakage if a newer pg_dump is used to dump from an
- * older server and then reload into that old version. This can go away
- * once 8.3 is so old as to not be of interest to anyone.
- */
- if (fout->remoteVersion >= 90100)
- {
- appendPQExpBuffer(query, "SELECT amopstrategy, false AS amopreqcheck, "
- "amopopr::pg_catalog.regoperator, "
- "opfname AS sortfamily, "
- "nspname AS sortfamilynsp "
- "FROM pg_catalog.pg_amop ao JOIN pg_catalog.pg_depend ON "
- "(classid = 'pg_catalog.pg_amop'::pg_catalog.regclass AND objid = ao.oid) "
- "LEFT JOIN pg_catalog.pg_opfamily f ON f.oid = amopsortfamily "
- "LEFT JOIN pg_catalog.pg_namespace n ON n.oid = opfnamespace "
- "WHERE refclassid = 'pg_catalog.pg_opfamily'::pg_catalog.regclass "
- "AND refobjid = '%u'::pg_catalog.oid "
- "AND amopfamily = '%u'::pg_catalog.oid "
- "ORDER BY amopstrategy",
- opfinfo->dobj.catId.oid,
- opfinfo->dobj.catId.oid);
- }
- else if (fout->remoteVersion >= 80400)
- {
- appendPQExpBuffer(query, "SELECT amopstrategy, false AS amopreqcheck, "
- "amopopr::pg_catalog.regoperator, "
- "NULL AS sortfamily, "
- "NULL AS sortfamilynsp "
- "FROM pg_catalog.pg_amop ao, pg_catalog.pg_depend "
- "WHERE refclassid = 'pg_catalog.pg_opfamily'::pg_catalog.regclass "
- "AND refobjid = '%u'::pg_catalog.oid "
- "AND classid = 'pg_catalog.pg_amop'::pg_catalog.regclass "
- "AND objid = ao.oid "
- "ORDER BY amopstrategy",
- opfinfo->dobj.catId.oid);
- }
- else
- {
- appendPQExpBuffer(query, "SELECT amopstrategy, amopreqcheck, "
- "amopopr::pg_catalog.regoperator, "
- "NULL AS sortfamily, "
- "NULL AS sortfamilynsp "
- "FROM pg_catalog.pg_amop ao, pg_catalog.pg_depend "
- "WHERE refclassid = 'pg_catalog.pg_opfamily'::pg_catalog.regclass "
- "AND refobjid = '%u'::pg_catalog.oid "
- "AND classid = 'pg_catalog.pg_amop'::pg_catalog.regclass "
- "AND objid = ao.oid "
- "ORDER BY amopstrategy",
- opfinfo->dobj.catId.oid);
- }
- res_ops = ExecuteSqlQuery(fout, query->data, PGRES_TUPLES_OK);
- resetPQExpBuffer(query);
- appendPQExpBuffer(query, "SELECT amprocnum, "
- "amproc::pg_catalog.regprocedure, "
- "amproclefttype::pg_catalog.regtype, "
- "amprocrighttype::pg_catalog.regtype "
- "FROM pg_catalog.pg_amproc ap, pg_catalog.pg_depend "
- "WHERE refclassid = 'pg_catalog.pg_opfamily'::pg_catalog.regclass "
- "AND refobjid = '%u'::pg_catalog.oid "
- "AND classid = 'pg_catalog.pg_amproc'::pg_catalog.regclass "
- "AND objid = ap.oid "
- "ORDER BY amprocnum",
- opfinfo->dobj.catId.oid);
- res_procs = ExecuteSqlQuery(fout, query->data, PGRES_TUPLES_OK);
- if (PQntuples(res_ops) == 0 && PQntuples(res_procs) == 0)
- {
- /* No loose members, so check contained opclasses */
- resetPQExpBuffer(query);
- appendPQExpBuffer(query, "SELECT 1 "
- "FROM pg_catalog.pg_opclass c, pg_catalog.pg_opfamily f, pg_catalog.pg_depend "
- "WHERE f.oid = '%u'::pg_catalog.oid "
- "AND refclassid = 'pg_catalog.pg_opfamily'::pg_catalog.regclass "
- "AND refobjid = f.oid "
- "AND classid = 'pg_catalog.pg_opclass'::pg_catalog.regclass "
- "AND objid = c.oid "
- "AND (opcname != opfname OR opcnamespace != opfnamespace OR opcowner != opfowner) "
- "LIMIT 1",
- opfinfo->dobj.catId.oid);
- res = ExecuteSqlQuery(fout, query->data, PGRES_TUPLES_OK);
- if (PQntuples(res) == 0)
- {
- /* no need to dump it, so bail out */
- PQclear(res);
- PQclear(res_ops);
- PQclear(res_procs);
- destroyPQExpBuffer(query);
- destroyPQExpBuffer(q);
- destroyPQExpBuffer(delq);
- destroyPQExpBuffer(labelq);
- return;
- }
- PQclear(res);
- }
- /* Get additional fields from the pg_opfamily row */
- resetPQExpBuffer(query);
- appendPQExpBuffer(query, "SELECT "
- "(SELECT amname FROM pg_catalog.pg_am WHERE oid = opfmethod) AS amname "
- "FROM pg_catalog.pg_opfamily "
- "WHERE oid = '%u'::pg_catalog.oid",
- opfinfo->dobj.catId.oid);
- res = ExecuteSqlQueryForSingleRow(fout, query->data);
- i_amname = PQfnumber(res, "amname");
- /* amname will still be needed after we PQclear res */
- amname = pg_strdup(PQgetvalue(res, 0, i_amname));
- /*
- * DROP must be fully qualified in case same name appears in pg_catalog
- */
- appendPQExpBuffer(delq, "DROP OPERATOR FAMILY %s",
- fmtId(opfinfo->dobj.namespace->dobj.name));
- appendPQExpBuffer(delq, ".%s",
- fmtId(opfinfo->dobj.name));
- appendPQExpBuffer(delq, " USING %s;\n",
- fmtId(amname));
- /* Build the fixed portion of the CREATE command */
- appendPQExpBuffer(q, "CREATE OPERATOR FAMILY %s",
- fmtId(opfinfo->dobj.name));
- appendPQExpBuffer(q, " USING %s;\n",
- fmtId(amname));
- PQclear(res);
- /* Do we need an ALTER to add loose members? */
- if (PQntuples(res_ops) > 0 || PQntuples(res_procs) > 0)
- {
- appendPQExpBuffer(q, "ALTER OPERATOR FAMILY %s",
- fmtId(opfinfo->dobj.name));
- appendPQExpBuffer(q, " USING %s ADD\n ",
- fmtId(amname));
- needComma = false;
- /*
- * Now fetch and print the OPERATOR entries (pg_amop rows).
- */
- ntups = PQntuples(res_ops);
- i_amopstrategy = PQfnumber(res_ops, "amopstrategy");
- i_amopreqcheck = PQfnumber(res_ops, "amopreqcheck");
- i_amopopr = PQfnumber(res_ops, "amopopr");
- i_sortfamily = PQfnumber(res_ops, "sortfamily");
- i_sortfamilynsp = PQfnumber(res_ops, "sortfamilynsp");
- for (i = 0; i < ntups; i++)
- {
- amopstrategy = PQgetvalue(res_ops, i, i_amopstrategy);
- amopreqcheck = PQgetvalue(res_ops, i, i_amopreqcheck);
- amopopr = PQgetvalue(res_ops, i, i_amopopr);
- sortfamily = PQgetvalue(res_ops, i, i_sortfamily);
- sortfamilynsp = PQgetvalue(res_ops, i, i_sortfamilynsp);
- if (needComma)
- appendPQExpBufferStr(q, " ,\n ");
- appendPQExpBuffer(q, "OPERATOR %s %s",
- amopstrategy, amopopr);
- if (strlen(sortfamily) > 0)
- {
- appendPQExpBufferStr(q, " FOR ORDER BY ");
- if (strcmp(sortfamilynsp, opfinfo->dobj.namespace->dobj.name) != 0)
- appendPQExpBuffer(q, "%s.", fmtId(sortfamilynsp));
- appendPQExpBufferStr(q, fmtId(sortfamily));
- }
- if (strcmp(amopreqcheck, "t") == 0)
- appendPQExpBufferStr(q, " RECHECK");
- needComma = true;
- }
- /*
- * Now fetch and print the FUNCTION entries (pg_amproc rows).
- */
- ntups = PQntuples(res_procs);
- i_amprocnum = PQfnumber(res_procs, "amprocnum");
- i_amproc = PQfnumber(res_procs, "amproc");
- i_amproclefttype = PQfnumber(res_procs, "amproclefttype");
- i_amprocrighttype = PQfnumber(res_procs, "amprocrighttype");
- for (i = 0; i < ntups; i++)
- {
- amprocnum = PQgetvalue(res_procs, i, i_amprocnum);
- amproc = PQgetvalue(res_procs, i, i_amproc);
- amproclefttype = PQgetvalue(res_procs, i, i_amproclefttype);
- amprocrighttype = PQgetvalue(res_procs, i, i_amprocrighttype);
- if (needComma)
- appendPQExpBufferStr(q, " ,\n ");
- appendPQExpBuffer(q, "FUNCTION %s (%s, %s) %s",
- amprocnum, amproclefttype, amprocrighttype,
- amproc);
- needComma = true;
- }
- appendPQExpBufferStr(q, ";\n");
- }
- appendPQExpBuffer(labelq, "OPERATOR FAMILY %s",
- fmtId(opfinfo->dobj.name));
- appendPQExpBuffer(labelq, " USING %s",
- fmtId(amname));
- if (binary_upgrade)
- binary_upgrade_extension_member(q, &opfinfo->dobj, labelq->data);
- ArchiveEntry(fout, opfinfo->dobj.catId, opfinfo->dobj.dumpId,
- opfinfo->dobj.name,
- opfinfo->dobj.namespace->dobj.name,
- NULL,
- opfinfo->rolname,
- false, "OPERATOR FAMILY", SECTION_PRE_DATA,
- q->data, delq->data, NULL,
- NULL, 0,
- NULL, NULL);
- /* Dump Operator Family Comments */
- dumpComment(fout, labelq->data,
- NULL, opfinfo->rolname,
- opfinfo->dobj.catId, 0, opfinfo->dobj.dumpId);
- free(amname);
- PQclear(res_ops);
- PQclear(res_procs);
- destroyPQExpBuffer(query);
- destroyPQExpBuffer(q);
- destroyPQExpBuffer(delq);
- destroyPQExpBuffer(labelq);
- }
- /*
- * dumpCollation
- * write out a single collation definition
- */
- static void
- dumpCollation(Archive *fout, CollInfo *collinfo)
- {
- PQExpBuffer query;
- PQExpBuffer q;
- PQExpBuffer delq;
- PQExpBuffer labelq;
- PGresult *res;
- int i_collcollate;
- int i_collctype;
- const char *collcollate;
- const char *collctype;
- /* Skip if not to be dumped */
- if (!collinfo->dobj.dump || dataOnly)
- return;
- query = createPQExpBuffer();
- q = createPQExpBuffer();
- delq = createPQExpBuffer();
- labelq = createPQExpBuffer();
- /* Make sure we are in proper schema */
- selectSourceSchema(fout, collinfo->dobj.namespace->dobj.name);
- /* Get collation-specific details */
- appendPQExpBuffer(query, "SELECT "
- "collcollate, "
- "collctype "
- "FROM pg_catalog.pg_collation c "
- "WHERE c.oid = '%u'::pg_catalog.oid",
- collinfo->dobj.catId.oid);
- res = ExecuteSqlQueryForSingleRow(fout, query->data);
- i_collcollate = PQfnumber(res, "collcollate");
- i_collctype = PQfnumber(res, "collctype");
- collcollate = PQgetvalue(res, 0, i_collcollate);
- collctype = PQgetvalue(res, 0, i_collctype);
- /*
- * DROP must be fully qualified in case same name appears in pg_catalog
- */
- appendPQExpBuffer(delq, "DROP COLLATION %s",
- fmtId(collinfo->dobj.namespace->dobj.name));
- appendPQExpBuffer(delq, ".%s;\n",
- fmtId(collinfo->dobj.name));
- appendPQExpBuffer(q, "CREATE COLLATION %s (lc_collate = ",
- fmtId(collinfo->dobj.name));
- appendStringLiteralAH(q, collcollate, fout);
- appendPQExpBufferStr(q, ", lc_ctype = ");
- appendStringLiteralAH(q, collctype, fout);
- appendPQExpBufferStr(q, ");\n");
- appendPQExpBuffer(labelq, "COLLATION %s", fmtId(collinfo->dobj.name));
- if (binary_upgrade)
- binary_upgrade_extension_member(q, &collinfo->dobj, labelq->data);
- ArchiveEntry(fout, collinfo->dobj.catId, collinfo->dobj.dumpId,
- collinfo->dobj.name,
- collinfo->dobj.namespace->dobj.name,
- NULL,
- collinfo->rolname,
- false, "COLLATION", SECTION_PRE_DATA,
- q->data, delq->data, NULL,
- NULL, 0,
- NULL, NULL);
- /* Dump Collation Comments */
- dumpComment(fout, labelq->data,
- collinfo->dobj.namespace->dobj.name, collinfo->rolname,
- collinfo->dobj.catId, 0, collinfo->dobj.dumpId);
- PQclear(res);
- destroyPQExpBuffer(query);
- destroyPQExpBuffer(q);
- destroyPQExpBuffer(delq);
- destroyPQExpBuffer(labelq);
- }
- /*
- * dumpConversion
- * write out a single conversion definition
- */
- static void
- dumpConversion(Archive *fout, ConvInfo *convinfo)
- {
- PQExpBuffer query;
- PQExpBuffer q;
- PQExpBuffer delq;
- PQExpBuffer labelq;
- PGresult *res;
- int i_conforencoding;
- int i_contoencoding;
- int i_conproc;
- int i_condefault;
- const char *conforencoding;
- const char *contoencoding;
- const char *conproc;
- bool condefault;
- /* Skip if not to be dumped */
- if (!convinfo->dobj.dump || dataOnly)
- return;
- query = createPQExpBuffer();
- q = createPQExpBuffer();
- delq = createPQExpBuffer();
- labelq = createPQExpBuffer();
- /* Make sure we are in proper schema */
- selectSourceSchema(fout, convinfo->dobj.namespace->dobj.name);
- /* Get conversion-specific details */
- appendPQExpBuffer(query, "SELECT "
- "pg_catalog.pg_encoding_to_char(conforencoding) AS conforencoding, "
- "pg_catalog.pg_encoding_to_char(contoencoding) AS contoencoding, "
- "conproc, condefault "
- "FROM pg_catalog.pg_conversion c "
- "WHERE c.oid = '%u'::pg_catalog.oid",
- convinfo->dobj.catId.oid);
- res = ExecuteSqlQueryForSingleRow(fout, query->data);
- i_conforencoding = PQfnumber(res, "conforencoding");
- i_contoencoding = PQfnumber(res, "contoencoding");
- i_conproc = PQfnumber(res, "conproc");
- i_condefault = PQfnumber(res, "condefault");
- conforencoding = PQgetvalue(res, 0, i_conforencoding);
- contoencoding = PQgetvalue(res, 0, i_contoencoding);
- conproc = PQgetvalue(res, 0, i_conproc);
- condefault = (PQgetvalue(res, 0, i_condefault)[0] == 't');
- /*
- * DROP must be fully qualified in case same name appears in pg_catalog
- */
- appendPQExpBuffer(delq, "DROP CONVERSION %s",
- fmtId(convinfo->dobj.namespace->dobj.name));
- appendPQExpBuffer(delq, ".%s;\n",
- fmtId(convinfo->dobj.name));
- appendPQExpBuffer(q, "CREATE %sCONVERSION %s FOR ",
- (condefault) ? "DEFAULT " : "",
- fmtId(convinfo->dobj.name));
- appendStringLiteralAH(q, conforencoding, fout);
- appendPQExpBufferStr(q, " TO ");
- appendStringLiteralAH(q, contoencoding, fout);
- /* regproc is automatically quoted in 7.3 and above */
- appendPQExpBuffer(q, " FROM %s;\n", conproc);
- appendPQExpBuffer(labelq, "CONVERSION %s", fmtId(convinfo->dobj.name));
- if (binary_upgrade)
- binary_upgrade_extension_member(q, &convinfo->dobj, labelq->data);
- ArchiveEntry(fout, convinfo->dobj.catId, convinfo->dobj.dumpId,
- convinfo->dobj.name,
- convinfo->dobj.namespace->dobj.name,
- NULL,
- convinfo->rolname,
- false, "CONVERSION", SECTION_PRE_DATA,
- q->data, delq->data, NULL,
- NULL, 0,
- NULL, NULL);
- /* Dump Conversion Comments */
- dumpComment(fout, labelq->data,
- convinfo->dobj.namespace->dobj.name, convinfo->rolname,
- convinfo->dobj.catId, 0, convinfo->dobj.dumpId);
- PQclear(res);
- destroyPQExpBuffer(query);
- destroyPQExpBuffer(q);
- destroyPQExpBuffer(delq);
- destroyPQExpBuffer(labelq);
- }
- /*
- * format_aggregate_signature: generate aggregate name and argument list
- *
- * The argument type names are qualified if needed. The aggregate name
- * is never qualified.
- */
- static char *
- format_aggregate_signature(AggInfo *agginfo, Archive *fout, bool honor_quotes)
- {
- PQExpBufferData buf;
- int j;
- initPQExpBuffer(&buf);
- if (honor_quotes)
- appendPQExpBufferStr(&buf, fmtId(agginfo->aggfn.dobj.name));
- else
- appendPQExpBufferStr(&buf, agginfo->aggfn.dobj.name);
- if (agginfo->aggfn.nargs == 0)
- appendPQExpBuffer(&buf, "(*)");
- else
- {
- appendPQExpBufferChar(&buf, '(');
- for (j = 0; j < agginfo->aggfn.nargs; j++)
- {
- char *typname;
- typname = getFormattedTypeName(fout, agginfo->aggfn.argtypes[j],
- zeroAsOpaque);
- appendPQExpBuffer(&buf, "%s%s",
- (j > 0) ? ", " : "",
- typname);
- free(typname);
- }
- appendPQExpBufferChar(&buf, ')');
- }
- return buf.data;
- }
- /*
- * dumpAgg
- * write out a single aggregate definition
- */
- static void
- dumpAgg(Archive *fout, AggInfo *agginfo)
- {
- PQExpBuffer query;
- PQExpBuffer q;
- PQExpBuffer delq;
- PQExpBuffer labelq;
- PQExpBuffer details;
- char *aggsig; /* identity signature */
- char *aggfullsig = NULL; /* full signature */
- char *aggsig_tag;
- PGresult *res;
- int i_aggtransfn;
- int i_aggfinalfn;
- int i_aggmtransfn;
- int i_aggminvtransfn;
- int i_aggmfinalfn;
- int i_aggfinalextra;
- int i_aggmfinalextra;
- int i_aggsortop;
- int i_hypothetical;
- int i_aggtranstype;
- int i_aggtransspace;
- int i_aggmtranstype;
- int i_aggmtransspace;
- int i_agginitval;
- int i_aggminitval;
- int i_convertok;
- const char *aggtransfn;
- const char *aggfinalfn;
- const char *aggmtransfn;
- const char *aggminvtransfn;
- const char *aggmfinalfn;
- bool aggfinalextra;
- bool aggmfinalextra;
- const char *aggsortop;
- char *aggsortconvop;
- bool hypothetical;
- const char *aggtranstype;
- const char *aggtransspace;
- const char *aggmtranstype;
- const char *aggmtransspace;
- const char *agginitval;
- const char *aggminitval;
- bool convertok;
- /* Skip if not to be dumped */
- if (!agginfo->aggfn.dobj.dump || dataOnly)
- return;
- query = createPQExpBuffer();
- q = createPQExpBuffer();
- delq = createPQExpBuffer();
- labelq = createPQExpBuffer();
- details = createPQExpBuffer();
- /* Make sure we are in proper schema */
- selectSourceSchema(fout, agginfo->aggfn.dobj.namespace->dobj.name);
- /* Get aggregate-specific details */
- if (fout->remoteVersion >= 90400)
- {
- appendPQExpBuffer(query, "SELECT aggtransfn, "
- "aggfinalfn, aggtranstype::pg_catalog.regtype, "
- "aggmtransfn, aggminvtransfn, aggmfinalfn, "
- "aggmtranstype::pg_catalog.regtype, "
- "aggfinalextra, aggmfinalextra, "
- "aggsortop::pg_catalog.regoperator, "
- "(aggkind = 'h') AS hypothetical, "
- "aggtransspace, agginitval, "
- "aggmtransspace, aggminitval, "
- "true AS convertok, "
- "pg_catalog.pg_get_function_arguments(p.oid) AS funcargs, "
- "pg_catalog.pg_get_function_identity_arguments(p.oid) AS funciargs "
- "FROM pg_catalog.pg_aggregate a, pg_catalog.pg_proc p "
- "WHERE a.aggfnoid = p.oid "
- "AND p.oid = '%u'::pg_catalog.oid",
- agginfo->aggfn.dobj.catId.oid);
- }
- else if (fout->remoteVersion >= 80400)
- {
- appendPQExpBuffer(query, "SELECT aggtransfn, "
- "aggfinalfn, aggtranstype::pg_catalog.regtype, "
- "'-' AS aggmtransfn, '-' AS aggminvtransfn, "
- "'-' AS aggmfinalfn, 0 AS aggmtranstype, "
- "false AS aggfinalextra, false AS aggmfinalextra, "
- "aggsortop::pg_catalog.regoperator, "
- "false AS hypothetical, "
- "0 AS aggtransspace, agginitval, "
- "0 AS aggmtransspace, NULL AS aggminitval, "
- "true AS convertok, "
- "pg_catalog.pg_get_function_arguments(p.oid) AS funcargs, "
- "pg_catalog.pg_get_function_identity_arguments(p.oid) AS funciargs "
- "FROM pg_catalog.pg_aggregate a, pg_catalog.pg_proc p "
- "WHERE a.aggfnoid = p.oid "
- "AND p.oid = '%u'::pg_catalog.oid",
- agginfo->aggfn.dobj.catId.oid);
- }
- else if (fout->remoteVersion >= 80100)
- {
- appendPQExpBuffer(query, "SELECT aggtransfn, "
- "aggfinalfn, aggtranstype::pg_catalog.regtype, "
- "'-' AS aggmtransfn, '-' AS aggminvtransfn, "
- "'-' AS aggmfinalfn, 0 AS aggmtranstype, "
- "false AS aggfinalextra, false AS aggmfinalextra, "
- "aggsortop::pg_catalog.regoperator, "
- "false AS hypothetical, "
- "0 AS aggtransspace, agginitval, "
- "0 AS aggmtransspace, NULL AS aggminitval, "
- "true AS convertok "
- "FROM pg_catalog.pg_aggregate a, pg_catalog.pg_proc p "
- "WHERE a.aggfnoid = p.oid "
- "AND p.oid = '%u'::pg_catalog.oid",
- agginfo->aggfn.dobj.catId.oid);
- }
- else if (fout->remoteVersion >= 70300)
- {
- appendPQExpBuffer(query, "SELECT aggtransfn, "
- "aggfinalfn, aggtranstype::pg_catalog.regtype, "
- "'-' AS aggmtransfn, '-' AS aggminvtransfn, "
- "'-' AS aggmfinalfn, 0 AS aggmtranstype, "
- "false AS aggfinalextra, false AS aggmfinalextra, "
- "0 AS aggsortop, "
- "false AS hypothetical, "
- "0 AS aggtransspace, agginitval, "
- "0 AS aggmtransspace, NULL AS aggminitval, "
- "true AS convertok "
- "FROM pg_catalog.pg_aggregate a, pg_catalog.pg_proc p "
- "WHERE a.aggfnoid = p.oid "
- "AND p.oid = '%u'::pg_catalog.oid",
- agginfo->aggfn.dobj.catId.oid);
- }
- else if (fout->remoteVersion >= 70100)
- {
- appendPQExpBuffer(query, "SELECT aggtransfn, aggfinalfn, "
- "format_type(aggtranstype, NULL) AS aggtranstype, "
- "'-' AS aggmtransfn, '-' AS aggminvtransfn, "
- "'-' AS aggmfinalfn, 0 AS aggmtranstype, "
- "false AS aggfinalextra, false AS aggmfinalextra, "
- "0 AS aggsortop, "
- "false AS hypothetical, "
- "0 AS aggtransspace, agginitval, "
- "0 AS aggmtransspace, NULL AS aggminitval, "
- "true AS convertok "
- "FROM pg_aggregate "
- "WHERE oid = '%u'::oid",
- agginfo->aggfn.dobj.catId.oid);
- }
- else
- {
- appendPQExpBuffer(query, "SELECT aggtransfn1 AS aggtransfn, "
- "aggfinalfn, "
- "(SELECT typname FROM pg_type WHERE oid = aggtranstype1) AS aggtranstype, "
- "'-' AS aggmtransfn, '-' AS aggminvtransfn, "
- "'-' AS aggmfinalfn, 0 AS aggmtranstype, "
- "false AS aggfinalextra, false AS aggmfinalextra, "
- "0 AS aggsortop, "
- "false AS hypothetical, "
- "0 AS aggtransspace, agginitval1 AS agginitval, "
- "0 AS aggmtransspace, NULL AS aggminitval, "
- "(aggtransfn2 = 0 and aggtranstype2 = 0 and agginitval2 is null) AS convertok "
- "FROM pg_aggregate "
- "WHERE oid = '%u'::oid",
- agginfo->aggfn.dobj.catId.oid);
- }
- res = ExecuteSqlQueryForSingleRow(fout, query->data);
- i_aggtransfn = PQfnumber(res, "aggtransfn");
- i_aggfinalfn = PQfnumber(res, "aggfinalfn");
- i_aggmtransfn = PQfnumber(res, "aggmtransfn");
- i_aggminvtransfn = PQfnumber(res, "aggminvtransfn");
- i_aggmfinalfn = PQfnumber(res, "aggmfinalfn");
- i_aggfinalextra = PQfnumber(res, "aggfinalextra");
- i_aggmfinalextra = PQfnumber(res, "aggmfinalextra");
- i_aggsortop = PQfnumber(res, "aggsortop");
- i_hypothetical = PQfnumber(res, "hypothetical");
- i_aggtranstype = PQfnumber(res, "aggtranstype");
- i_aggtransspace = PQfnumber(res, "aggtransspace");
- i_aggmtranstype = PQfnumber(res, "aggmtranstype");
- i_aggmtransspace = PQfnumber(res, "aggmtransspace");
- i_agginitval = PQfnumber(res, "agginitval");
- i_aggminitval = PQfnumber(res, "aggminitval");
- i_convertok = PQfnumber(res, "convertok");
- aggtransfn = PQgetvalue(res, 0, i_aggtransfn);
- aggfinalfn = PQgetvalue(res, 0, i_aggfinalfn);
- aggmtransfn = PQgetvalue(res, 0, i_aggmtransfn);
- aggminvtransfn = PQgetvalue(res, 0, i_aggminvtransfn);
- aggmfinalfn = PQgetvalue(res, 0, i_aggmfinalfn);
- aggfinalextra = (PQgetvalue(res, 0, i_aggfinalextra)[0] == 't');
- aggmfinalextra = (PQgetvalue(res, 0, i_aggmfinalextra)[0] == 't');
- aggsortop = PQgetvalue(res, 0, i_aggsortop);
- hypothetical = (PQgetvalue(res, 0, i_hypothetical)[0] == 't');
- aggtranstype = PQgetvalue(res, 0, i_aggtranstype);
- aggtransspace = PQgetvalue(res, 0, i_aggtransspace);
- aggmtranstype = PQgetvalue(res, 0, i_aggmtranstype);
- aggmtransspace = PQgetvalue(res, 0, i_aggmtransspace);
- agginitval = PQgetvalue(res, 0, i_agginitval);
- aggminitval = PQgetvalue(res, 0, i_aggminitval);
- convertok = (PQgetvalue(res, 0, i_convertok)[0] == 't');
- if (fout->remoteVersion >= 80400)
- {
- /* 8.4 or later; we rely on server-side code for most of the work */
- char *funcargs;
- char *funciargs;
- funcargs = PQgetvalue(res, 0, PQfnumber(res, "funcargs"));
- funciargs = PQgetvalue(res, 0, PQfnumber(res, "funciargs"));
- aggfullsig = format_function_arguments(&agginfo->aggfn, funcargs, true);
- aggsig = format_function_arguments(&agginfo->aggfn, funciargs, true);
- }
- else
- /* pre-8.4, do it ourselves */
- aggsig = format_aggregate_signature(agginfo, fout, true);
- aggsig_tag = format_aggregate_signature(agginfo, fout, false);
- if (!convertok)
- {
- write_msg(NULL, "WARNING: aggregate function %s could not be dumped correctly for this database version; ignored\n",
- aggsig);
- if (aggfullsig)
- free(aggfullsig);
- free(aggsig);
- return;
- }
- if (fout->remoteVersion >= 70300)
- {
- /* If using 7.3's regproc or regtype, data is already quoted */
- appendPQExpBuffer(details, " SFUNC = %s,\n STYPE = %s",
- aggtransfn,
- aggtranstype);
- }
- else if (fout->remoteVersion >= 70100)
- {
- /* format_type quotes, regproc does not */
- appendPQExpBuffer(details, " SFUNC = %s,\n STYPE = %s",
- fmtId(aggtransfn),
- aggtranstype);
- }
- else
- {
- /* need quotes all around */
- appendPQExpBuffer(details, " SFUNC = %s,\n",
- fmtId(aggtransfn));
- appendPQExpBuffer(details, " STYPE = %s",
- fmtId(aggtranstype));
- }
- if (strcmp(aggtransspace, "0") != 0)
- {
- appendPQExpBuffer(details, ",\n SSPACE = %s",
- aggtransspace);
- }
- if (!PQgetisnull(res, 0, i_agginitval))
- {
- appendPQExpBufferStr(details, ",\n INITCOND = ");
- appendStringLiteralAH(details, agginitval, fout);
- }
- if (strcmp(aggfinalfn, "-") != 0)
- {
- appendPQExpBuffer(details, ",\n FINALFUNC = %s",
- aggfinalfn);
- if (aggfinalextra)
- appendPQExpBufferStr(details, ",\n FINALFUNC_EXTRA");
- }
- if (strcmp(aggmtransfn, "-") != 0)
- {
- appendPQExpBuffer(details, ",\n MSFUNC = %s,\n MINVFUNC = %s,\n MSTYPE = %s",
- aggmtransfn,
- aggminvtransfn,
- aggmtranstype);
- }
- if (strcmp(aggmtransspace, "0") != 0)
- {
- appendPQExpBuffer(details, ",\n MSSPACE = %s",
- aggmtransspace);
- }
- if (!PQgetisnull(res, 0, i_aggminitval))
- {
- appendPQExpBufferStr(details, ",\n MINITCOND = ");
- appendStringLiteralAH(details, aggminitval, fout);
- }
- if (strcmp(aggmfinalfn, "-") != 0)
- {
- appendPQExpBuffer(details, ",\n MFINALFUNC = %s",
- aggmfinalfn);
- if (aggmfinalextra)
- appendPQExpBufferStr(details, ",\n MFINALFUNC_EXTRA");
- }
- aggsortconvop = convertOperatorReference(fout, aggsortop);
- if (aggsortconvop)
- {
- appendPQExpBuffer(details, ",\n SORTOP = %s",
- aggsortconvop);
- free(aggsortconvop);
- }
- if (hypothetical)
- appendPQExpBufferStr(details, ",\n HYPOTHETICAL");
- /*
- * DROP must be fully qualified in case same name appears in pg_catalog
- */
- appendPQExpBuffer(delq, "DROP AGGREGATE %s.%s;\n",
- fmtId(agginfo->aggfn.dobj.namespace->dobj.name),
- aggsig);
- appendPQExpBuffer(q, "CREATE AGGREGATE %s (\n%s\n);\n",
- aggfullsig ? aggfullsig : aggsig, details->data);
- appendPQExpBuffer(labelq, "AGGREGATE %s", aggsig);
- if (binary_upgrade)
- binary_upgrade_extension_member(q, &agginfo->aggfn.dobj, labelq->data);
- ArchiveEntry(fout, agginfo->aggfn.dobj.catId, agginfo->aggfn.dobj.dumpId,
- aggsig_tag,
- agginfo->aggfn.dobj.namespace->dobj.name,
- NULL,
- agginfo->aggfn.rolname,
- false, "AGGREGATE", SECTION_PRE_DATA,
- q->data, delq->data, NULL,
- NULL, 0,
- NULL, NULL);
- /* Dump Aggregate Comments */
- dumpComment(fout, labelq->data,
- agginfo->aggfn.dobj.namespace->dobj.name, agginfo->aggfn.rolname,
- agginfo->aggfn.dobj.catId, 0, agginfo->aggfn.dobj.dumpId);
- dumpSecLabel(fout, labelq->data,
- agginfo->aggfn.dobj.namespace->dobj.name, agginfo->aggfn.rolname,
- agginfo->aggfn.dobj.catId, 0, agginfo->aggfn.dobj.dumpId);
- /*
- * Since there is no GRANT ON AGGREGATE syntax, we have to make the ACL
- * command look like a function's GRANT; in particular this affects the
- * syntax for zero-argument aggregates and ordered-set aggregates.
- */
- free(aggsig);
- free(aggsig_tag);
- aggsig = format_function_signature(fout, &agginfo->aggfn, true);
- aggsig_tag = format_function_signature(fout, &agginfo->aggfn, false);
- dumpACL(fout, agginfo->aggfn.dobj.catId, agginfo->aggfn.dobj.dumpId,
- "FUNCTION",
- aggsig, NULL, aggsig_tag,
- agginfo->aggfn.dobj.namespace->dobj.name,
- agginfo->aggfn.rolname, agginfo->aggfn.proacl);
- free(aggsig);
- if (aggfullsig)
- free(aggfullsig);
- free(aggsig_tag);
- PQclear(res);
- destroyPQExpBuffer(query);
- destroyPQExpBuffer(q);
- destroyPQExpBuffer(delq);
- destroyPQExpBuffer(labelq);
- destroyPQExpBuffer(details);
- }
- /*
- * dumpTSParser
- * write out a single text search parser
- */
- static void
- dumpTSParser(Archive *fout, TSParserInfo *prsinfo)
- {
- PQExpBuffer q;
- PQExpBuffer delq;
- PQExpBuffer labelq;
- /* Skip if not to be dumped */
- if (!prsinfo->dobj.dump || dataOnly)
- return;
- q = createPQExpBuffer();
- delq = createPQExpBuffer();
- labelq = createPQExpBuffer();
- /* Make sure we are in proper schema */
- selectSourceSchema(fout, prsinfo->dobj.namespace->dobj.name);
- appendPQExpBuffer(q, "CREATE TEXT SEARCH PARSER %s (\n",
- fmtId(prsinfo->dobj.name));
- appendPQExpBuffer(q, " START = %s,\n",
- convertTSFunction(fout, prsinfo->prsstart));
- appendPQExpBuffer(q, " GETTOKEN = %s,\n",
- convertTSFunction(fout, prsinfo->prstoken));
- appendPQExpBuffer(q, " END = %s,\n",
- convertTSFunction(fout, prsinfo->prsend));
- if (prsinfo->prsheadline != InvalidOid)
- appendPQExpBuffer(q, " HEADLINE = %s,\n",
- convertTSFunction(fout, prsinfo->prsheadline));
- appendPQExpBuffer(q, " LEXTYPES = %s );\n",
- convertTSFunction(fout, prsinfo->prslextype));
- /*
- * DROP must be fully qualified in case same name appears in pg_catalog
- */
- appendPQExpBuffer(delq, "DROP TEXT SEARCH PARSER %s",
- fmtId(prsinfo->dobj.namespace->dobj.name));
- appendPQExpBuffer(delq, ".%s;\n",
- fmtId(prsinfo->dobj.name));
- appendPQExpBuffer(labelq, "TEXT SEARCH PARSER %s",
- fmtId(prsinfo->dobj.name));
- if (binary_upgrade)
- binary_upgrade_extension_member(q, &prsinfo->dobj, labelq->data);
- ArchiveEntry(fout, prsinfo->dobj.catId, prsinfo->dobj.dumpId,
- prsinfo->dobj.name,
- prsinfo->dobj.namespace->dobj.name,
- NULL,
- "",
- false, "TEXT SEARCH PARSER", SECTION_PRE_DATA,
- q->data, delq->data, NULL,
- NULL, 0,
- NULL, NULL);
- /* Dump Parser Comments */
- dumpComment(fout, labelq->data,
- NULL, "",
- prsinfo->dobj.catId, 0, prsinfo->dobj.dumpId);
- destroyPQExpBuffer(q);
- destroyPQExpBuffer(delq);
- destroyPQExpBuffer(labelq);
- }
- /*
- * dumpTSDictionary
- * write out a single text search dictionary
- */
- static void
- dumpTSDictionary(Archive *fout, TSDictInfo *dictinfo)
- {
- PQExpBuffer q;
- PQExpBuffer delq;
- PQExpBuffer labelq;
- PQExpBuffer query;
- PGresult *res;
- char *nspname;
- char *tmplname;
- /* Skip if not to be dumped */
- if (!dictinfo->dobj.dump || dataOnly)
- return;
- q = createPQExpBuffer();
- delq = createPQExpBuffer();
- labelq = createPQExpBuffer();
- query = createPQExpBuffer();
- /* Fetch name and namespace of the dictionary's template */
- selectSourceSchema(fout, "pg_catalog");
- appendPQExpBuffer(query, "SELECT nspname, tmplname "
- "FROM pg_ts_template p, pg_namespace n "
- "WHERE p.oid = '%u' AND n.oid = tmplnamespace",
- dictinfo->dicttemplate);
- res = ExecuteSqlQueryForSingleRow(fout, query->data);
- nspname = PQgetvalue(res, 0, 0);
- tmplname = PQgetvalue(res, 0, 1);
- /* Make sure we are in proper schema */
- selectSourceSchema(fout, dictinfo->dobj.namespace->dobj.name);
- appendPQExpBuffer(q, "CREATE TEXT SEARCH DICTIONARY %s (\n",
- fmtId(dictinfo->dobj.name));
- appendPQExpBufferStr(q, " TEMPLATE = ");
- if (strcmp(nspname, dictinfo->dobj.namespace->dobj.name) != 0)
- appendPQExpBuffer(q, "%s.", fmtId(nspname));
- appendPQExpBufferStr(q, fmtId(tmplname));
- PQclear(res);
- /* the dictinitoption can be dumped straight into the command */
- if (dictinfo->dictinitoption)
- appendPQExpBuffer(q, ",\n %s", dictinfo->dictinitoption);
- appendPQExpBufferStr(q, " );\n");
- /*
- * DROP must be fully qualified in case same name appears in pg_catalog
- */
- appendPQExpBuffer(delq, "DROP TEXT SEARCH DICTIONARY %s",
- fmtId(dictinfo->dobj.namespace->dobj.name));
- appendPQExpBuffer(delq, ".%s;\n",
- fmtId(dictinfo->dobj.name));
- appendPQExpBuffer(labelq, "TEXT SEARCH DICTIONARY %s",
- fmtId(dictinfo->dobj.name));
- if (binary_upgrade)
- binary_upgrade_extension_member(q, &dictinfo->dobj, labelq->data);
- ArchiveEntry(fout, dictinfo->dobj.catId, dictinfo->dobj.dumpId,
- dictinfo->dobj.name,
- dictinfo->dobj.namespace->dobj.name,
- NULL,
- dictinfo->rolname,
- false, "TEXT SEARCH DICTIONARY", SECTION_PRE_DATA,
- q->data, delq->data, NULL,
- NULL, 0,
- NULL, NULL);
- /* Dump Dictionary Comments */
- dumpComment(fout, labelq->data,
- NULL, dictinfo->rolname,
- dictinfo->dobj.catId, 0, dictinfo->dobj.dumpId);
- destroyPQExpBuffer(q);
- destroyPQExpBuffer(delq);
- destroyPQExpBuffer(labelq);
- destroyPQExpBuffer(query);
- }
- /*
- * dumpTSTemplate
- * write out a single text search template
- */
- static void
- dumpTSTemplate(Archive *fout, TSTemplateInfo *tmplinfo)
- {
- PQExpBuffer q;
- PQExpBuffer delq;
- PQExpBuffer labelq;
- /* Skip if not to be dumped */
- if (!tmplinfo->dobj.dump || dataOnly)
- return;
- q = createPQExpBuffer();
- delq = createPQExpBuffer();
- labelq = createPQExpBuffer();
- /* Make sure we are in proper schema */
- selectSourceSchema(fout, tmplinfo->dobj.namespace->dobj.name);
- appendPQExpBuffer(q, "CREATE TEXT SEARCH TEMPLATE %s (\n",
- fmtId(tmplinfo->dobj.name));
- if (tmplinfo->tmplinit != InvalidOid)
- appendPQExpBuffer(q, " INIT = %s,\n",
- convertTSFunction(fout, tmplinfo->tmplinit));
- appendPQExpBuffer(q, " LEXIZE = %s );\n",
- convertTSFunction(fout, tmplinfo->tmpllexize));
- /*
- * DROP must be fully qualified in case same name appears in pg_catalog
- */
- appendPQExpBuffer(delq, "DROP TEXT SEARCH TEMPLATE %s",
- fmtId(tmplinfo->dobj.namespace->dobj.name));
- appendPQExpBuffer(delq, ".%s;\n",
- fmtId(tmplinfo->dobj.name));
- appendPQExpBuffer(labelq, "TEXT SEARCH TEMPLATE %s",
- fmtId(tmplinfo->dobj.name));
- if (binary_upgrade)
- binary_upgrade_extension_member(q, &tmplinfo->dobj, labelq->data);
- ArchiveEntry(fout, tmplinfo->dobj.catId, tmplinfo->dobj.dumpId,
- tmplinfo->dobj.name,
- tmplinfo->dobj.namespace->dobj.name,
- NULL,
- "",
- false, "TEXT SEARCH TEMPLATE", SECTION_PRE_DATA,
- q->data, delq->data, NULL,
- NULL, 0,
- NULL, NULL);
- /* Dump Template Comments */
- dumpComment(fout, labelq->data,
- NULL, "",
- tmplinfo->dobj.catId, 0, tmplinfo->dobj.dumpId);
- destroyPQExpBuffer(q);
- destroyPQExpBuffer(delq);
- destroyPQExpBuffer(labelq);
- }
- /*
- * dumpTSConfig
- * write out a single text search configuration
- */
- static void
- dumpTSConfig(Archive *fout, TSConfigInfo *cfginfo)
- {
- PQExpBuffer q;
- PQExpBuffer delq;
- PQExpBuffer labelq;
- PQExpBuffer query;
- PGresult *res;
- char *nspname;
- char *prsname;
- int ntups,
- i;
- int i_tokenname;
- int i_dictname;
- /* Skip if not to be dumped */
- if (!cfginfo->dobj.dump || dataOnly)
- return;
- q = createPQExpBuffer();
- delq = createPQExpBuffer();
- labelq = createPQExpBuffer();
- query = createPQExpBuffer();
- /* Fetch name and namespace of the config's parser */
- selectSourceSchema(fout, "pg_catalog");
- appendPQExpBuffer(query, "SELECT nspname, prsname "
- "FROM pg_ts_parser p, pg_namespace n "
- "WHERE p.oid = '%u' AND n.oid = prsnamespace",
- cfginfo->cfgparser);
- res = ExecuteSqlQueryForSingleRow(fout, query->data);
- nspname = PQgetvalue(res, 0, 0);
- prsname = PQgetvalue(res, 0, 1);
- /* Make sure we are in proper schema */
- selectSourceSchema(fout, cfginfo->dobj.namespace->dobj.name);
- appendPQExpBuffer(q, "CREATE TEXT SEARCH CONFIGURATION %s (\n",
- fmtId(cfginfo->dobj.name));
- appendPQExpBufferStr(q, " PARSER = ");
- if (strcmp(nspname, cfginfo->dobj.namespace->dobj.name) != 0)
- appendPQExpBuffer(q, "%s.", fmtId(nspname));
- appendPQExpBuffer(q, "%s );\n", fmtId(prsname));
- PQclear(res);
- resetPQExpBuffer(query);
- appendPQExpBuffer(query,
- "SELECT \n"
- " ( SELECT alias FROM pg_catalog.ts_token_type('%u'::pg_catalog.oid) AS t \n"
- " WHERE t.tokid = m.maptokentype ) AS tokenname, \n"
- " m.mapdict::pg_catalog.regdictionary AS dictname \n"
- "FROM pg_catalog.pg_ts_config_map AS m \n"
- "WHERE m.mapcfg = '%u' \n"
- "ORDER BY m.mapcfg, m.maptokentype, m.mapseqno",
- cfginfo->cfgparser, cfginfo->dobj.catId.oid);
- res = ExecuteSqlQuery(fout, query->data, PGRES_TUPLES_OK);
- ntups = PQntuples(res);
- i_tokenname = PQfnumber(res, "tokenname");
- i_dictname = PQfnumber(res, "dictname");
- for (i = 0; i < ntups; i++)
- {
- char *tokenname = PQgetvalue(res, i, i_tokenname);
- char *dictname = PQgetvalue(res, i, i_dictname);
- if (i == 0 ||
- strcmp(tokenname, PQgetvalue(res, i - 1, i_tokenname)) != 0)
- {
- /* starting a new token type, so start a new command */
- if (i > 0)
- appendPQExpBufferStr(q, ";\n");
- appendPQExpBuffer(q, "\nALTER TEXT SEARCH CONFIGURATION %s\n",
- fmtId(cfginfo->dobj.name));
- /* tokenname needs quoting, dictname does NOT */
- appendPQExpBuffer(q, " ADD MAPPING FOR %s WITH %s",
- fmtId(tokenname), dictname);
- }
- else
- appendPQExpBuffer(q, ", %s", dictname);
- }
- if (ntups > 0)
- appendPQExpBufferStr(q, ";\n");
- PQclear(res);
- /*
- * DROP must be fully qualified in case same name appears in pg_catalog
- */
- appendPQExpBuffer(delq, "DROP TEXT SEARCH CONFIGURATION %s",
- fmtId(cfginfo->dobj.namespace->dobj.name));
- appendPQExpBuffer(delq, ".%s;\n",
- fmtId(cfginfo->dobj.name));
- appendPQExpBuffer(labelq, "TEXT SEARCH CONFIGURATION %s",
- fmtId(cfginfo->dobj.name));
- if (binary_upgrade)
- binary_upgrade_extension_member(q, &cfginfo->dobj, labelq->data);
- ArchiveEntry(fout, cfginfo->dobj.catId, cfginfo->dobj.dumpId,
- cfginfo->dobj.name,
- cfginfo->dobj.namespace->dobj.name,
- NULL,
- cfginfo->rolname,
- false, "TEXT SEARCH CONFIGURATION", SECTION_PRE_DATA,
- q->data, delq->data, NULL,
- NULL, 0,
- NULL, NULL);
- /* Dump Configuration Comments */
- dumpComment(fout, labelq->data,
- NULL, cfginfo->rolname,
- cfginfo->dobj.catId, 0, cfginfo->dobj.dumpId);
- destroyPQExpBuffer(q);
- destroyPQExpBuffer(delq);
- destroyPQExpBuffer(labelq);
- destroyPQExpBuffer(query);
- }
- /*
- * dumpForeignDataWrapper
- * write out a single foreign-data wrapper definition
- */
- static void
- dumpForeignDataWrapper(Archive *fout, FdwInfo *fdwinfo)
- {
- PQExpBuffer q;
- PQExpBuffer delq;
- PQExpBuffer labelq;
- char *qfdwname;
- /* Skip if not to be dumped */
- if (!fdwinfo->dobj.dump || dataOnly)
- return;
- /*
- * FDWs that belong to an extension are dumped based on their "dump"
- * field. Otherwise omit them if we are only dumping some specific object.
- */
- if (!fdwinfo->dobj.ext_member)
- if (!include_everything)
- return;
- q = createPQExpBuffer();
- delq = createPQExpBuffer();
- labelq = createPQExpBuffer();
- qfdwname = pg_strdup(fmtId(fdwinfo->dobj.name));
- appendPQExpBuffer(q, "CREATE FOREIGN DATA WRAPPER %s",
- qfdwname);
- if (strcmp(fdwinfo->fdwhandler, "-") != 0)
- appendPQExpBuffer(q, " HANDLER %s", fdwinfo->fdwhandler);
- if (strcmp(fdwinfo->fdwvalidator, "-") != 0)
- appendPQExpBuffer(q, " VALIDATOR %s", fdwinfo->fdwvalidator);
- if (strlen(fdwinfo->fdwoptions) > 0)
- appendPQExpBuffer(q, " OPTIONS (\n %s\n)", fdwinfo->fdwoptions);
- appendPQExpBufferStr(q, ";\n");
- appendPQExpBuffer(delq, "DROP FOREIGN DATA WRAPPER %s;\n",
- qfdwname);
- appendPQExpBuffer(labelq, "FOREIGN DATA WRAPPER %s",
- qfdwname);
- if (binary_upgrade)
- binary_upgrade_extension_member(q, &fdwinfo->dobj, labelq->data);
- ArchiveEntry(fout, fdwinfo->dobj.catId, fdwinfo->dobj.dumpId,
- fdwinfo->dobj.name,
- NULL,
- NULL,
- fdwinfo->rolname,
- false, "FOREIGN DATA WRAPPER", SECTION_PRE_DATA,
- q->data, delq->data, NULL,
- NULL, 0,
- NULL, NULL);
- /* Handle the ACL */
- dumpACL(fout, fdwinfo->dobj.catId, fdwinfo->dobj.dumpId,
- "FOREIGN DATA WRAPPER",
- qfdwname, NULL, fdwinfo->dobj.name,
- NULL, fdwinfo->rolname,
- fdwinfo->fdwacl);
- /* Dump Foreign Data Wrapper Comments */
- dumpComment(fout, labelq->data,
- NULL, fdwinfo->rolname,
- fdwinfo->dobj.catId, 0, fdwinfo->dobj.dumpId);
- free(qfdwname);
- destroyPQExpBuffer(q);
- destroyPQExpBuffer(delq);
- destroyPQExpBuffer(labelq);
- }
- /*
- * dumpForeignServer
- * write out a foreign server definition
- */
- static void
- dumpForeignServer(Archive *fout, ForeignServerInfo *srvinfo)
- {
- PQExpBuffer q;
- PQExpBuffer delq;
- PQExpBuffer labelq;
- PQExpBuffer query;
- PGresult *res;
- char *qsrvname;
- char *fdwname;
- /* Skip if not to be dumped */
- if (!srvinfo->dobj.dump || dataOnly || !include_everything)
- return;
- q = createPQExpBuffer();
- delq = createPQExpBuffer();
- labelq = createPQExpBuffer();
- query = createPQExpBuffer();
- qsrvname = pg_strdup(fmtId(srvinfo->dobj.name));
- /* look up the foreign-data wrapper */
- selectSourceSchema(fout, "pg_catalog");
- appendPQExpBuffer(query, "SELECT fdwname "
- "FROM pg_foreign_data_wrapper w "
- "WHERE w.oid = '%u'",
- srvinfo->srvfdw);
- res = ExecuteSqlQueryForSingleRow(fout, query->data);
- fdwname = PQgetvalue(res, 0, 0);
- appendPQExpBuffer(q, "CREATE SERVER %s", qsrvname);
- if (srvinfo->srvtype && strlen(srvinfo->srvtype) > 0)
- {
- appendPQExpBufferStr(q, " TYPE ");
- appendStringLiteralAH(q, srvinfo->srvtype, fout);
- }
- if (srvinfo->srvversion && strlen(srvinfo->srvversion) > 0)
- {
- appendPQExpBufferStr(q, " VERSION ");
- appendStringLiteralAH(q, srvinfo->srvversion, fout);
- }
- appendPQExpBufferStr(q, " FOREIGN DATA WRAPPER ");
- appendPQExpBufferStr(q, fmtId(fdwname));
- if (srvinfo->srvoptions && strlen(srvinfo->srvoptions) > 0)
- appendPQExpBuffer(q, " OPTIONS (\n %s\n)", srvinfo->srvoptions);
- appendPQExpBufferStr(q, ";\n");
- appendPQExpBuffer(delq, "DROP SERVER %s;\n",
- qsrvname);
- appendPQExpBuffer(labelq, "SERVER %s", qsrvname);
- if (binary_upgrade)
- binary_upgrade_extension_member(q, &srvinfo->dobj, labelq->data);
- ArchiveEntry(fout, srvinfo->dobj.catId, srvinfo->dobj.dumpId,
- srvinfo->dobj.name,
- NULL,
- NULL,
- srvinfo->rolname,
- false, "SERVER", SECTION_PRE_DATA,
- q->data, delq->data, NULL,
- NULL, 0,
- NULL, NULL);
- /* Handle the ACL */
- dumpACL(fout, srvinfo->dobj.catId, srvinfo->dobj.dumpId,
- "FOREIGN SERVER",
- qsrvname, NULL, srvinfo->dobj.name,
- NULL, srvinfo->rolname,
- srvinfo->srvacl);
- /* Dump user mappings */
- dumpUserMappings(fout,
- srvinfo->dobj.name, NULL,
- srvinfo->rolname,
- srvinfo->dobj.catId, srvinfo->dobj.dumpId);
- /* Dump Foreign Server Comments */
- dumpComment(fout, labelq->data,
- NULL, srvinfo->rolname,
- srvinfo->dobj.catId, 0, srvinfo->dobj.dumpId);
- free(qsrvname);
- destroyPQExpBuffer(q);
- destroyPQExpBuffer(delq);
- destroyPQExpBuffer(labelq);
- }
- /*
- * dumpUserMappings
- *
- * This routine is used to dump any user mappings associated with the
- * server handed to this routine. Should be called after ArchiveEntry()
- * for the server.
- */
- static void
- dumpUserMappings(Archive *fout,
- const char *servername, const char *namespace,
- const char *owner,
- CatalogId catalogId, DumpId dumpId)
- {
- PQExpBuffer q;
- PQExpBuffer delq;
- PQExpBuffer query;
- PQExpBuffer tag;
- PGresult *res;
- int ntups;
- int i_usename;
- int i_umoptions;
- int i;
- q = createPQExpBuffer();
- tag = createPQExpBuffer();
- delq = createPQExpBuffer();
- query = createPQExpBuffer();
- /*
- * We read from the publicly accessible view pg_user_mappings, so as not
- * to fail if run by a non-superuser. Note that the view will show
- * umoptions as null if the user hasn't got privileges for the associated
- * server; this means that pg_dump will dump such a mapping, but with no
- * OPTIONS clause. A possible alternative is to skip such mappings
- * altogether, but it's not clear that that's an improvement.
- */
- selectSourceSchema(fout, "pg_catalog");
- appendPQExpBuffer(query,
- "SELECT usename, "
- "array_to_string(ARRAY("
- "SELECT quote_ident(option_name) || ' ' || "
- "quote_literal(option_value) "
- "FROM pg_options_to_table(umoptions) "
- "ORDER BY option_name"
- "), E',\n ') AS umoptions "
- "FROM pg_user_mappings "
- "WHERE srvid = '%u' "
- "ORDER BY usename",
- catalogId.oid);
- res = ExecuteSqlQuery(fout, query->data, PGRES_TUPLES_OK);
- ntups = PQntuples(res);
- i_usename = PQfnumber(res, "usename");
- i_umoptions = PQfnumber(res, "umoptions");
- for (i = 0; i < ntups; i++)
- {
- char *usename;
- char *umoptions;
- usename = PQgetvalue(res, i, i_usename);
- umoptions = PQgetvalue(res, i, i_umoptions);
- resetPQExpBuffer(q);
- appendPQExpBuffer(q, "CREATE USER MAPPING FOR %s", fmtId(usename));
- appendPQExpBuffer(q, " SERVER %s", fmtId(servername));
- if (umoptions && strlen(umoptions) > 0)
- appendPQExpBuffer(q, " OPTIONS (\n %s\n)", umoptions);
- appendPQExpBufferStr(q, ";\n");
- resetPQExpBuffer(delq);
- appendPQExpBuffer(delq, "DROP USER MAPPING FOR %s", fmtId(usename));
- appendPQExpBuffer(delq, " SERVER %s;\n", fmtId(servername));
- resetPQExpBuffer(tag);
- appendPQExpBuffer(tag, "USER MAPPING %s SERVER %s",
- usename, servername);
- ArchiveEntry(fout, nilCatalogId, createDumpId(),
- tag->data,
- namespace,
- NULL,
- owner, false,
- "USER MAPPING", SECTION_PRE_DATA,
- q->data, delq->data, NULL,
- &dumpId, 1,
- NULL, NULL);
- }
- PQclear(res);
- destroyPQExpBuffer(query);
- destroyPQExpBuffer(delq);
- destroyPQExpBuffer(tag);
- destroyPQExpBuffer(q);
- }
- /*
- * Write out default privileges information
- */
- static void
- dumpDefaultACL(Archive *fout, DefaultACLInfo *daclinfo)
- {
- PQExpBuffer q;
- PQExpBuffer tag;
- const char *type;
- /* Skip if not to be dumped */
- if (!daclinfo->dobj.dump || dataOnly || aclsSkip)
- return;
- q = createPQExpBuffer();
- tag = createPQExpBuffer();
- switch (daclinfo->defaclobjtype)
- {
- case DEFACLOBJ_RELATION:
- type = "TABLES";
- break;
- case DEFACLOBJ_SEQUENCE:
- type = "SEQUENCES";
- break;
- case DEFACLOBJ_FUNCTION:
- type = "FUNCTIONS";
- break;
- case DEFACLOBJ_TYPE:
- type = "TYPES";
- break;
- default:
- /* shouldn't get here */
- exit_horribly(NULL,
- "unrecognized object type in default privileges: %d\n",
- (int) daclinfo->defaclobjtype);
- type = ""; /* keep compiler quiet */
- }
- appendPQExpBuffer(tag, "DEFAULT PRIVILEGES FOR %s", type);
- /* build the actual command(s) for this tuple */
- if (!buildDefaultACLCommands(type,
- daclinfo->dobj.namespace != NULL ?
- daclinfo->dobj.namespace->dobj.name : NULL,
- daclinfo->defaclacl,
- daclinfo->defaclrole,
- fout->remoteVersion,
- q))
- exit_horribly(NULL, "could not parse default ACL list (%s)\n",
- daclinfo->defaclacl);
- ArchiveEntry(fout, daclinfo->dobj.catId, daclinfo->dobj.dumpId,
- tag->data,
- daclinfo->dobj.namespace ? daclinfo->dobj.namespace->dobj.name : NULL,
- NULL,
- daclinfo->defaclrole,
- false, "DEFAULT ACL", SECTION_POST_DATA,
- q->data, "", NULL,
- NULL, 0,
- NULL, NULL);
- destroyPQExpBuffer(tag);
- destroyPQExpBuffer(q);
- }
- /*----------
- * Write out grant/revoke information
- *
- * 'objCatId' is the catalog ID of the underlying object.
- * 'objDumpId' is the dump ID of the underlying object.
- * 'type' must be one of
- * TABLE, SEQUENCE, FUNCTION, LANGUAGE, SCHEMA, DATABASE, TABLESPACE,
- * FOREIGN DATA WRAPPER, SERVER, or LARGE OBJECT.
- * 'name' is the formatted name of the object. Must be quoted etc. already.
- * 'subname' is the formatted name of the sub-object, if any. Must be quoted.
- * 'tag' is the tag for the archive entry (typ. unquoted name of object).
- * 'nspname' is the namespace the object is in (NULL if none).
- * 'owner' is the owner, NULL if there is no owner (for languages).
- * 'acls' is the string read out of the fooacl system catalog field;
- * it will be parsed here.
- *----------
- */
- static void
- dumpACL(Archive *fout, CatalogId objCatId, DumpId objDumpId,
- const char *type, const char *name, const char *subname,
- const char *tag, const char *nspname, const char *owner,
- const char *acls)
- {
- PQExpBuffer sql;
- /* Do nothing if ACL dump is not enabled */
- if (aclsSkip)
- return;
- /* --data-only skips ACLs *except* BLOB ACLs */
- if (dataOnly && strcmp(type, "LARGE OBJECT") != 0)
- return;
- sql = createPQExpBuffer();
- if (!buildACLCommands(name, subname, type, acls, owner,
- "", fout->remoteVersion, sql))
- exit_horribly(NULL,
- "could not parse ACL list (%s) for object \"%s\" (%s)\n",
- acls, name, type);
- if (sql->len > 0)
- ArchiveEntry(fout, nilCatalogId, createDumpId(),
- tag, nspname,
- NULL,
- owner ? owner : "",
- false, "ACL", SECTION_NONE,
- sql->data, "", NULL,
- &(objDumpId), 1,
- NULL, NULL);
- destroyPQExpBuffer(sql);
- }
- /*
- * dumpSecLabel
- *
- * This routine is used to dump any security labels associated with the
- * object handed to this routine. The routine takes a constant character
- * string for the target part of the security-label command, plus
- * the namespace and owner of the object (for labeling the ArchiveEntry),
- * plus catalog ID and subid which are the lookup key for pg_seclabel,
- * plus the dump ID for the object (for setting a dependency).
- * If a matching pg_seclabel entry is found, it is dumped.
- *
- * Note: although this routine takes a dumpId for dependency purposes,
- * that purpose is just to mark the dependency in the emitted dump file
- * for possible future use by pg_restore. We do NOT use it for determining
- * ordering of the label in the dump file, because this routine is called
- * after dependency sorting occurs. This routine should be called just after
- * calling ArchiveEntry() for the specified object.
- */
- static void
- dumpSecLabel(Archive *fout, const char *target,
- const char *namespace, const char *owner,
- CatalogId catalogId, int subid, DumpId dumpId)
- {
- SecLabelItem *labels;
- int nlabels;
- int i;
- PQExpBuffer query;
- /* do nothing, if --no-security-labels is supplied */
- if (no_security_labels)
- return;
- /* Comments are schema not data ... except blob comments are data */
- if (strncmp(target, "LARGE OBJECT ", 13) != 0)
- {
- if (dataOnly)
- return;
- }
- else
- {
- if (schemaOnly)
- return;
- }
- /* Search for security labels associated with catalogId, using table */
- nlabels = findSecLabels(fout, catalogId.tableoid, catalogId.oid, &labels);
- query = createPQExpBuffer();
- for (i = 0; i < nlabels; i++)
- {
- /*
- * Ignore label entries for which the subid doesn't match.
- */
- if (labels[i].objsubid != subid)
- continue;
- appendPQExpBuffer(query,
- "SECURITY LABEL FOR %s ON %s IS ",
- fmtId(labels[i].provider), target);
- appendStringLiteralAH(query, labels[i].label, fout);
- appendPQExpBufferStr(query, ";\n");
- }
- if (query->len > 0)
- {
- ArchiveEntry(fout, nilCatalogId, createDumpId(),
- target, namespace, NULL, owner,
- false, "SECURITY LABEL", SECTION_NONE,
- query->data, "", NULL,
- &(dumpId), 1,
- NULL, NULL);
- }
- destroyPQExpBuffer(query);
- }
- /*
- * dumpTableSecLabel
- *
- * As above, but dump security label for both the specified table (or view)
- * and its columns.
- */
- static void
- dumpTableSecLabel(Archive *fout, TableInfo *tbinfo, const char *reltypename)
- {
- SecLabelItem *labels;
- int nlabels;
- int i;
- PQExpBuffer query;
- PQExpBuffer target;
- /* do nothing, if --no-security-labels is supplied */
- if (no_security_labels)
- return;
- /* SecLabel are SCHEMA not data */
- if (dataOnly)
- return;
- /* Search for comments associated with relation, using table */
- nlabels = findSecLabels(fout,
- tbinfo->dobj.catId.tableoid,
- tbinfo->dobj.catId.oid,
- &labels);
- /* If security labels exist, build SECURITY LABEL statements */
- if (nlabels <= 0)
- return;
- query = createPQExpBuffer();
- target = createPQExpBuffer();
- for (i = 0; i < nlabels; i++)
- {
- const char *colname;
- const char *provider = labels[i].provider;
- const char *label = labels[i].label;
- int objsubid = labels[i].objsubid;
- resetPQExpBuffer(target);
- if (objsubid == 0)
- {
- appendPQExpBuffer(target, "%s %s", reltypename,
- fmtId(tbinfo->dobj.name));
- }
- else
- {
- colname = getAttrName(objsubid, tbinfo);
- /* first fmtId result must be consumed before calling it again */
- appendPQExpBuffer(target, "COLUMN %s", fmtId(tbinfo->dobj.name));
- appendPQExpBuffer(target, ".%s", fmtId(colname));
- }
- appendPQExpBuffer(query, "SECURITY LABEL FOR %s ON %s IS ",
- fmtId(provider), target->data);
- appendStringLiteralAH(query, label, fout);
- appendPQExpBufferStr(query, ";\n");
- }
- if (query->len > 0)
- {
- resetPQExpBuffer(target);
- appendPQExpBuffer(target, "%s %s", reltypename,
- fmtId(tbinfo->dobj.name));
- ArchiveEntry(fout, nilCatalogId, createDumpId(),
- target->data,
- tbinfo->dobj.namespace->dobj.name,
- NULL, tbinfo->rolname,
- false, "SECURITY LABEL", SECTION_NONE,
- query->data, "", NULL,
- &(tbinfo->dobj.dumpId), 1,
- NULL, NULL);
- }
- destroyPQExpBuffer(query);
- destroyPQExpBuffer(target);
- }
- /*
- * findSecLabels
- *
- * Find the security label(s), if any, associated with the given object.
- * All the objsubid values associated with the given classoid/objoid are
- * found with one search.
- */
- static int
- findSecLabels(Archive *fout, Oid classoid, Oid objoid, SecLabelItem **items)
- {
- /* static storage for table of security labels */
- static SecLabelItem *labels = NULL;
- static int nlabels = -1;
- SecLabelItem *middle = NULL;
- SecLabelItem *low;
- SecLabelItem *high;
- int nmatch;
- /* Get security labels if we didn't already */
- if (nlabels < 0)
- nlabels = collectSecLabels(fout, &labels);
- if (nlabels <= 0) /* no labels, so no match is possible */
- {
- *items = NULL;
- return 0;
- }
- /*
- * Do binary search to find some item matching the object.
- */
- low = &labels[0];
- high = &labels[nlabels - 1];
- while (low <= high)
- {
- middle = low + (high - low) / 2;
- if (classoid < middle->classoid)
- high = middle - 1;
- else if (classoid > middle->classoid)
- low = middle + 1;
- else if (objoid < middle->objoid)
- high = middle - 1;
- else if (objoid > middle->objoid)
- low = middle + 1;
- else
- break; /* found a match */
- }
- if (low > high) /* no matches */
- {
- *items = NULL;
- return 0;
- }
- /*
- * Now determine how many items match the object. The search loop
- * invariant still holds: only items between low and high inclusive could
- * match.
- */
- nmatch = 1;
- while (middle > low)
- {
- if (classoid != middle[-1].classoid ||
- objoid != middle[-1].objoid)
- break;
- middle--;
- nmatch++;
- }
- *items = middle;
- middle += nmatch;
- while (middle <= high)
- {
- if (classoid != middle->classoid ||
- objoid != middle->objoid)
- break;
- middle++;
- nmatch++;
- }
- return nmatch;
- }
- /*
- * collectSecLabels
- *
- * Construct a table of all security labels available for database objects.
- * It's much faster to pull them all at once.
- *
- * The table is sorted by classoid/objid/objsubid for speed in lookup.
- */
- static int
- collectSecLabels(Archive *fout, SecLabelItem **items)
- {
- PGresult *res;
- PQExpBuffer query;
- int i_label;
- int i_provider;
- int i_classoid;
- int i_objoid;
- int i_objsubid;
- int ntups;
- int i;
- SecLabelItem *labels;
- query = createPQExpBuffer();
- appendPQExpBufferStr(query,
- "SELECT label, provider, classoid, objoid, objsubid "
- "FROM pg_catalog.pg_seclabel "
- "ORDER BY classoid, objoid, objsubid");
- res = ExecuteSqlQuery(fout, query->data, PGRES_TUPLES_OK);
- /* Construct lookup table containing OIDs in numeric form */
- i_label = PQfnumber(res, "label");
- i_provider = PQfnumber(res, "provider");
- i_classoid = PQfnumber(res, "classoid");
- i_objoid = PQfnumber(res, "objoid");
- i_objsubid = PQfnumber(res, "objsubid");
- ntups = PQntuples(res);
- labels = (SecLabelItem *) pg_malloc(ntups * sizeof(SecLabelItem));
- for (i = 0; i < ntups; i++)
- {
- labels[i].label = PQgetvalue(res, i, i_label);
- labels[i].provider = PQgetvalue(res, i, i_provider);
- labels[i].classoid = atooid(PQgetvalue(res, i, i_classoid));
- labels[i].objoid = atooid(PQgetvalue(res, i, i_objoid));
- labels[i].objsubid = atoi(PQgetvalue(res, i, i_objsubid));
- }
- /* Do NOT free the PGresult since we are keeping pointers into it */
- destroyPQExpBuffer(query);
- *items = labels;
- return ntups;
- }
- /*
- * dumpTable
- * write out to fout the declarations (not data) of a user-defined table
- */
- static void
- dumpTable(Archive *fout, TableInfo *tbinfo)
- {
- if (tbinfo->dobj.dump && !dataOnly)
- {
- char *namecopy;
- if (tbinfo->relkind == RELKIND_SEQUENCE)
- dumpSequence(fout, tbinfo);
- else
- dumpTableSchema(fout, tbinfo);
- /* Handle the ACL here */
- namecopy = pg_strdup(fmtId(tbinfo->dobj.name));
- dumpACL(fout, tbinfo->dobj.catId, tbinfo->dobj.dumpId,
- (tbinfo->relkind == RELKIND_SEQUENCE) ? "SEQUENCE" :
- "TABLE",
- namecopy, NULL, tbinfo->dobj.name,
- tbinfo->dobj.namespace->dobj.name, tbinfo->rolname,
- tbinfo->relacl);
- /*
- * Handle column ACLs, if any. Note: we pull these with a separate
- * query rather than trying to fetch them during getTableAttrs, so
- * that we won't miss ACLs on system columns.
- */
- if (fout->remoteVersion >= 80400)
- {
- PQExpBuffer query = createPQExpBuffer();
- PGresult *res;
- int i;
- appendPQExpBuffer(query,
- "SELECT attname, attacl FROM pg_catalog.pg_attribute "
- "WHERE attrelid = '%u' AND NOT attisdropped AND attacl IS NOT NULL "
- "ORDER BY attnum",
- tbinfo->dobj.catId.oid);
- res = ExecuteSqlQuery(fout, query->data, PGRES_TUPLES_OK);
- for (i = 0; i < PQntuples(res); i++)
- {
- char *attname = PQgetvalue(res, i, 0);
- char *attacl = PQgetvalue(res, i, 1);
- char *attnamecopy;
- char *acltag;
- attnamecopy = pg_strdup(fmtId(attname));
- acltag = psprintf("%s.%s", tbinfo->dobj.name, attname);
- /* Column's GRANT type is always TABLE */
- dumpACL(fout, tbinfo->dobj.catId, tbinfo->dobj.dumpId, "TABLE",
- namecopy, attnamecopy, acltag,
- tbinfo->dobj.namespace->dobj.name, tbinfo->rolname,
- attacl);
- free(attnamecopy);
- free(acltag);
- }
- PQclear(res);
- destroyPQExpBuffer(query);
- }
- free(namecopy);
- }
- }
- /*
- * Create the AS clause for a view or materialized view. The semicolon is
- * stripped because a materialized view must add a WITH NO DATA clause.
- *
- * This returns a new buffer which must be freed by the caller.
- */
- static PQExpBuffer
- createViewAsClause(Archive *fout, TableInfo *tbinfo)
- {
- PQExpBuffer query = createPQExpBuffer();
- PQExpBuffer result = createPQExpBuffer();
- PGresult *res;
- int len;
- /* Fetch the view definition */
- if (fout->remoteVersion >= 70300)
- {
- /* Beginning in 7.3, viewname is not unique; rely on OID */
- appendPQExpBuffer(query,
- "SELECT pg_catalog.pg_get_viewdef('%u'::pg_catalog.oid) AS viewdef",
- tbinfo->dobj.catId.oid);
- }
- else
- {
- appendPQExpBufferStr(query, "SELECT definition AS viewdef "
- "FROM pg_views WHERE viewname = ");
- appendStringLiteralAH(query, tbinfo->dobj.name, fout);
- }
- res = ExecuteSqlQuery(fout, query->data, PGRES_TUPLES_OK);
- if (PQntuples(res) != 1)
- {
- if (PQntuples(res) < 1)
- exit_horribly(NULL, "query to obtain definition of view \"%s\" returned no data\n",
- tbinfo->dobj.name);
- else
- exit_horribly(NULL, "query to obtain definition of view \"%s\" returned more than one definition\n",
- tbinfo->dobj.name);
- }
- len = PQgetlength(res, 0, 0);
- if (len == 0)
- exit_horribly(NULL, "definition of view \"%s\" appears to be empty (length zero)\n",
- tbinfo->dobj.name);
- /* Strip off the trailing semicolon so that other things may follow. */
- Assert(PQgetvalue(res, 0, 0)[len - 1] == ';');
- appendBinaryPQExpBuffer(result, PQgetvalue(res, 0, 0), len - 1);
- PQclear(res);
- destroyPQExpBuffer(query);
- return result;
- }
- /*
- * dumpTableSchema
- * write the declaration (not data) of one user-defined table or view
- */
- static void
- dumpTableSchema(Archive *fout, TableInfo *tbinfo)
- {
- PQExpBuffer q = createPQExpBuffer();
- PQExpBuffer delq = createPQExpBuffer();
- PQExpBuffer labelq = createPQExpBuffer();
- int numParents;
- TableInfo **parents;
- int actual_atts; /* number of attrs in this CREATE statement */
- const char *reltypename;
- char *storage;
- char *srvname;
- char *ftoptions;
- int j,
- k;
- /* Make sure we are in proper schema */
- selectSourceSchema(fout, tbinfo->dobj.namespace->dobj.name);
- if (binary_upgrade)
- binary_upgrade_set_type_oids_by_rel_oid(fout, q,
- tbinfo->dobj.catId.oid);
- /* Is it a table or a view? */
- if (tbinfo->relkind == RELKIND_VIEW)
- {
- PQExpBuffer result;
- reltypename = "VIEW";
- /*
- * DROP must be fully qualified in case same name appears in
- * pg_catalog
- */
- appendPQExpBuffer(delq, "DROP VIEW %s.",
- fmtId(tbinfo->dobj.namespace->dobj.name));
- appendPQExpBuffer(delq, "%s;\n",
- fmtId(tbinfo->dobj.name));
- if (binary_upgrade)
- binary_upgrade_set_pg_class_oids(fout, q,
- tbinfo->dobj.catId.oid, false);
- appendPQExpBuffer(q, "CREATE VIEW %s", fmtId(tbinfo->dobj.name));
- if (tbinfo->reloptions && strlen(tbinfo->reloptions) > 0)
- appendPQExpBuffer(q, " WITH (%s)", tbinfo->reloptions);
- result = createViewAsClause(fout, tbinfo);
- appendPQExpBuffer(q, " AS\n%s", result->data);
- destroyPQExpBuffer(result);
- if (tbinfo->checkoption != NULL)
- appendPQExpBuffer(q, "\n WITH %s CHECK OPTION", tbinfo->checkoption);
- appendPQExpBufferStr(q, ";\n");
- appendPQExpBuffer(labelq, "VIEW %s",
- fmtId(tbinfo->dobj.name));
- }
- else
- {
- switch (tbinfo->relkind)
- {
- case (RELKIND_FOREIGN_TABLE):
- {
- PQExpBuffer query = createPQExpBuffer();
- PGresult *res;
- int i_srvname;
- int i_ftoptions;
- reltypename = "FOREIGN TABLE";
- /* retrieve name of foreign server and generic options */
- appendPQExpBuffer(query,
- "SELECT fs.srvname, "
- "pg_catalog.array_to_string(ARRAY("
- "SELECT pg_catalog.quote_ident(option_name) || "
- "' ' || pg_catalog.quote_literal(option_value) "
- "FROM pg_catalog.pg_options_to_table(ftoptions) "
- "ORDER BY option_name"
- "), E',\n ') AS ftoptions "
- "FROM pg_catalog.pg_foreign_table ft "
- "JOIN pg_catalog.pg_foreign_server fs "
- "ON (fs.oid = ft.ftserver) "
- "WHERE ft.ftrelid = '%u'",
- tbinfo->dobj.catId.oid);
- res = ExecuteSqlQueryForSingleRow(fout, query->data);
- i_srvname = PQfnumber(res, "srvname");
- i_ftoptions = PQfnumber(res, "ftoptions");
- srvname = pg_strdup(PQgetvalue(res, 0, i_srvname));
- ftoptions = pg_strdup(PQgetvalue(res, 0, i_ftoptions));
- PQclear(res);
- destroyPQExpBuffer(query);
- break;
- }
- case (RELKIND_MATVIEW):
- reltypename = "MATERIALIZED VIEW";
- srvname = NULL;
- ftoptions = NULL;
- break;
- default:
- reltypename = "TABLE";
- srvname = NULL;
- ftoptions = NULL;
- }
- numParents = tbinfo->numParents;
- parents = tbinfo->parents;
- /*
- * DROP must be fully qualified in case same name appears in
- * pg_catalog
- */
- appendPQExpBuffer(delq, "DROP %s %s.", reltypename,
- fmtId(tbinfo->dobj.namespace->dobj.name));
- appendPQExpBuffer(delq, "%s;\n",
- fmtId(tbinfo->dobj.name));
- appendPQExpBuffer(labelq, "%s %s", reltypename,
- fmtId(tbinfo->dobj.name));
- if (binary_upgrade)
- binary_upgrade_set_pg_class_oids(fout, q,
- tbinfo->dobj.catId.oid, false);
- appendPQExpBuffer(q, "CREATE %s%s %s",
- tbinfo->relpersistence == RELPERSISTENCE_UNLOGGED ?
- "UNLOGGED " : "",
- reltypename,
- fmtId(tbinfo->dobj.name));
- /*
- * Attach to type, if reloftype; except in case of a binary upgrade,
- * we dump the table normally and attach it to the type afterward.
- */
- if (tbinfo->reloftype && !binary_upgrade)
- appendPQExpBuffer(q, " OF %s", tbinfo->reloftype);
- if (tbinfo->relkind != RELKIND_MATVIEW)
- {
- /* Dump the attributes */
- actual_atts = 0;
- for (j = 0; j < tbinfo->numatts; j++)
- {
- /*
- * Normally, dump if it's locally defined in this table, and
- * not dropped. But for binary upgrade, we'll dump all the
- * columns, and then fix up the dropped and nonlocal cases
- * below.
- */
- if (shouldPrintColumn(tbinfo, j))
- {
- /*
- * Default value --- suppress if to be printed separately.
- */
- bool has_default = (tbinfo->attrdefs[j] != NULL &&
- !tbinfo->attrdefs[j]->separate);
- /*
- * Not Null constraint --- suppress if inherited, except
- * in binary-upgrade case where that won't work.
- */
- bool has_notnull = (tbinfo->notnull[j] &&
- (!tbinfo->inhNotNull[j] ||
- binary_upgrade));
- /* Skip column if fully defined by reloftype */
- if (tbinfo->reloftype &&
- !has_default && !has_notnull && !binary_upgrade)
- continue;
- /* Format properly if not first attr */
- if (actual_atts == 0)
- appendPQExpBufferStr(q, " (");
- else
- appendPQExpBufferStr(q, ",");
- appendPQExpBufferStr(q, "\n ");
- actual_atts++;
- /* Attribute name */
- appendPQExpBufferStr(q, fmtId(tbinfo->attnames[j]));
- if (tbinfo->attisdropped[j])
- {
- /*
- * ALTER TABLE DROP COLUMN clears
- * pg_attribute.atttypid, so we will not have gotten a
- * valid type name; insert INTEGER as a stopgap. We'll
- * clean things up later.
- */
- appendPQExpBufferStr(q, " INTEGER /* dummy */");
- /* Skip all the rest, too */
- continue;
- }
- /* Attribute type */
- if (tbinfo->reloftype && !binary_upgrade)
- {
- appendPQExpBufferStr(q, " WITH OPTIONS");
- }
- else if (fout->remoteVersion >= 70100)
- {
- appendPQExpBuffer(q, " %s",
- tbinfo->atttypnames[j]);
- }
- else
- {
- /* If no format_type, fake it */
- appendPQExpBuffer(q, " %s",
- myFormatType(tbinfo->atttypnames[j],
- tbinfo->atttypmod[j]));
- }
- /* Add collation if not default for the type */
- if (OidIsValid(tbinfo->attcollation[j]))
- {
- CollInfo *coll;
- coll = findCollationByOid(tbinfo->attcollation[j]);
- if (coll)
- {
- /* always schema-qualify, don't try to be smart */
- appendPQExpBuffer(q, " COLLATE %s.",
- fmtId(coll->dobj.namespace->dobj.name));
- appendPQExpBufferStr(q, fmtId(coll->dobj.name));
- }
- }
- if (has_default)
- appendPQExpBuffer(q, " DEFAULT %s",
- tbinfo->attrdefs[j]->adef_expr);
- if (has_notnull)
- appendPQExpBufferStr(q, " NOT NULL");
- }
- }
- /*
- * Add non-inherited CHECK constraints, if any.
- */
- for (j = 0; j < tbinfo->ncheck; j++)
- {
- ConstraintInfo *constr = &(tbinfo->checkexprs[j]);
- if (constr->separate || !constr->conislocal)
- continue;
- if (actual_atts == 0)
- appendPQExpBufferStr(q, " (\n ");
- else
- appendPQExpBufferStr(q, ",\n ");
- appendPQExpBuffer(q, "CONSTRAINT %s ",
- fmtId(constr->dobj.name));
- appendPQExpBufferStr(q, constr->condef);
- actual_atts++;
- }
- if (actual_atts)
- appendPQExpBufferStr(q, "\n)");
- else if (!(tbinfo->reloftype && !binary_upgrade))
- {
- /*
- * We must have a parenthesized attribute list, even though
- * empty, when not using the OF TYPE syntax.
- */
- appendPQExpBufferStr(q, " (\n)");
- }
- if (numParents > 0 && !binary_upgrade)
- {
- appendPQExpBufferStr(q, "\nINHERITS (");
- for (k = 0; k < numParents; k++)
- {
- TableInfo *parentRel = parents[k];
- if (k > 0)
- appendPQExpBufferStr(q, ", ");
- if (parentRel->dobj.namespace != tbinfo->dobj.namespace)
- appendPQExpBuffer(q, "%s.",
- fmtId(parentRel->dobj.namespace->dobj.name));
- appendPQExpBufferStr(q, fmtId(parentRel->dobj.name));
- }
- appendPQExpBufferChar(q, ')');
- }
- if (tbinfo->relkind == RELKIND_FOREIGN_TABLE)
- appendPQExpBuffer(q, "\nSERVER %s", fmtId(srvname));
- }
- if ((tbinfo->reloptions && strlen(tbinfo->reloptions) > 0) ||
- (tbinfo->toast_reloptions && strlen(tbinfo->toast_reloptions) > 0))
- {
- bool addcomma = false;
- appendPQExpBufferStr(q, "\nWITH (");
- if (tbinfo->reloptions && strlen(tbinfo->reloptions) > 0)
- {
- addcomma = true;
- appendPQExpBufferStr(q, tbinfo->reloptions);
- }
- if (tbinfo->toast_reloptions && strlen(tbinfo->toast_reloptions) > 0)
- {
- appendPQExpBuffer(q, "%s%s", addcomma ? ", " : "",
- tbinfo->toast_reloptions);
- }
- appendPQExpBufferChar(q, ')');
- }
- /* Dump generic options if any */
- if (ftoptions && ftoptions[0])
- appendPQExpBuffer(q, "\nOPTIONS (\n %s\n)", ftoptions);
- /*
- * For materialized views, create the AS clause just like a view. At
- * this point, we always mark the view as not populated.
- */
- if (tbinfo->relkind == RELKIND_MATVIEW)
- {
- PQExpBuffer result;
- result = createViewAsClause(fout, tbinfo);
- appendPQExpBuffer(q, " AS\n%s\n WITH NO DATA;\n",
- result->data);
- destroyPQExpBuffer(result);
- }
- else
- appendPQExpBufferStr(q, ";\n");
- /*
- * To create binary-compatible heap files, we have to ensure the same
- * physical column order, including dropped columns, as in the
- * original. Therefore, we create dropped columns above and drop them
- * here, also updating their attlen/attalign values so that the
- * dropped column can be skipped properly. (We do not bother with
- * restoring the original attbyval setting.) Also, inheritance
- * relationships are set up by doing ALTER INHERIT rather than using
- * an INHERITS clause --- the latter would possibly mess up the column
- * order. That also means we have to take care about setting
- * attislocal correctly, plus fix up any inherited CHECK constraints.
- * Analogously, we set up typed tables using ALTER TABLE / OF here.
- */
- if (binary_upgrade && (tbinfo->relkind == RELKIND_RELATION ||
- tbinfo->relkind == RELKIND_FOREIGN_TABLE))
- {
- for (j = 0; j < tbinfo->numatts; j++)
- {
- if (tbinfo->attisdropped[j])
- {
- appendPQExpBufferStr(q, "\n-- For binary upgrade, recreate dropped column.\n");
- appendPQExpBuffer(q, "UPDATE pg_catalog.pg_attribute\n"
- "SET attlen = %d, "
- "attalign = '%c', attbyval = false\n"
- "WHERE attname = ",
- tbinfo->attlen[j],
- tbinfo->attalign[j]);
- appendStringLiteralAH(q, tbinfo->attnames[j], fout);
- appendPQExpBufferStr(q, "\n AND attrelid = ");
- appendStringLiteralAH(q, fmtId(tbinfo->dobj.name), fout);
- appendPQExpBufferStr(q, "::pg_catalog.regclass;\n");
- if (tbinfo->relkind == RELKIND_RELATION)
- appendPQExpBuffer(q, "ALTER TABLE ONLY %s ",
- fmtId(tbinfo->dobj.name));
- else
- appendPQExpBuffer(q, "ALTER FOREIGN TABLE %s ",
- fmtId(tbinfo->dobj.name));
- appendPQExpBuffer(q, "DROP COLUMN %s;\n",
- fmtId(tbinfo->attnames[j]));
- }
- else if (!tbinfo->attislocal[j])
- {
- Assert(tbinfo->relkind != RELKIND_FOREIGN_TABLE);
- appendPQExpBufferStr(q, "\n-- For binary upgrade, recreate inherited column.\n");
- appendPQExpBufferStr(q, "UPDATE pg_catalog.pg_attribute\n"
- "SET attislocal = false\n"
- "WHERE attname = ");
- appendStringLiteralAH(q, tbinfo->attnames[j], fout);
- appendPQExpBufferStr(q, "\n AND attrelid = ");
- appendStringLiteralAH(q, fmtId(tbinfo->dobj.name), fout);
- appendPQExpBufferStr(q, "::pg_catalog.regclass;\n");
- }
- }
- for (k = 0; k < tbinfo->ncheck; k++)
- {
- ConstraintInfo *constr = &(tbinfo->checkexprs[k]);
- if (constr->separate || constr->conislocal)
- continue;
- appendPQExpBufferStr(q, "\n-- For binary upgrade, set up inherited constraint.\n");
- appendPQExpBuffer(q, "ALTER TABLE ONLY %s ",
- fmtId(tbinfo->dobj.name));
- appendPQExpBuffer(q, " ADD CONSTRAINT %s ",
- fmtId(constr->dobj.name));
- appendPQExpBuffer(q, "%s;\n", constr->condef);
- appendPQExpBufferStr(q, "UPDATE pg_catalog.pg_constraint\n"
- "SET conislocal = false\n"
- "WHERE contype = 'c' AND conname = ");
- appendStringLiteralAH(q, constr->dobj.name, fout);
- appendPQExpBufferStr(q, "\n AND conrelid = ");
- appendStringLiteralAH(q, fmtId(tbinfo->dobj.name), fout);
- appendPQExpBufferStr(q, "::pg_catalog.regclass;\n");
- }
- if (numParents > 0)
- {
- appendPQExpBufferStr(q, "\n-- For binary upgrade, set up inheritance this way.\n");
- for (k = 0; k < numParents; k++)
- {
- TableInfo *parentRel = parents[k];
- appendPQExpBuffer(q, "ALTER TABLE ONLY %s INHERIT ",
- fmtId(tbinfo->dobj.name));
- if (parentRel->dobj.namespace != tbinfo->dobj.namespace)
- appendPQExpBuffer(q, "%s.",
- fmtId(parentRel->dobj.namespace->dobj.name));
- appendPQExpBuffer(q, "%s;\n",
- fmtId(parentRel->dobj.name));
- }
- }
- if (tbinfo->reloftype)
- {
- appendPQExpBufferStr(q, "\n-- For binary upgrade, set up typed tables this way.\n");
- appendPQExpBuffer(q, "ALTER TABLE ONLY %s OF %s;\n",
- fmtId(tbinfo->dobj.name),
- tbinfo->reloftype);
- }
- appendPQExpBufferStr(q, "\n-- For binary upgrade, set heap's relfrozenxid and relminmxid\n");
- appendPQExpBuffer(q, "UPDATE pg_catalog.pg_class\n"
- "SET relfrozenxid = '%u', relminmxid = '%u'\n"
- "WHERE oid = ",
- tbinfo->frozenxid, tbinfo->minmxid);
- appendStringLiteralAH(q, fmtId(tbinfo->dobj.name), fout);
- appendPQExpBufferStr(q, "::pg_catalog.regclass;\n");
- if (tbinfo->toast_oid)
- {
- /* We preserve the toast oids, so we can use it during restore */
- appendPQExpBufferStr(q, "\n-- For binary upgrade, set toast's relfrozenxid and relminmxid\n");
- appendPQExpBuffer(q, "UPDATE pg_catalog.pg_class\n"
- "SET relfrozenxid = '%u', relminmxid = '%u'\n"
- "WHERE oid = '%u';\n",
- tbinfo->toast_frozenxid,
- tbinfo->toast_minmxid, tbinfo->toast_oid);
- }
- }
- /*
- * In binary_upgrade mode, restore matviews' populated status by
- * poking pg_class directly. This is pretty ugly, but we can't use
- * REFRESH MATERIALIZED VIEW since it's possible that some underlying
- * matview is not populated even though this matview is.
- */
- if (binary_upgrade && tbinfo->relkind == RELKIND_MATVIEW &&
- tbinfo->relispopulated)
- {
- appendPQExpBufferStr(q, "\n-- For binary upgrade, mark materialized view as populated\n");
- appendPQExpBufferStr(q, "UPDATE pg_catalog.pg_class\n"
- "SET relispopulated = 't'\n"
- "WHERE oid = ");
- appendStringLiteralAH(q, fmtId(tbinfo->dobj.name), fout);
- appendPQExpBufferStr(q, "::pg_catalog.regclass;\n");
- }
- /*
- * Dump additional per-column properties that we can't handle in the
- * main CREATE TABLE command.
- */
- for (j = 0; j < tbinfo->numatts; j++)
- {
- /* None of this applies to dropped columns */
- if (tbinfo->attisdropped[j])
- continue;
- /*
- * If we didn't dump the column definition explicitly above, and
- * it is NOT NULL and did not inherit that property from a parent,
- * we have to mark it separately.
- */
- if (!shouldPrintColumn(tbinfo, j) &&
- tbinfo->notnull[j] && !tbinfo->inhNotNull[j])
- {
- appendPQExpBuffer(q, "ALTER TABLE ONLY %s ",
- fmtId(tbinfo->dobj.name));
- appendPQExpBuffer(q, "ALTER COLUMN %s SET NOT NULL;\n",
- fmtId(tbinfo->attnames[j]));
- }
- /*
- * Dump per-column statistics information. We only issue an ALTER
- * TABLE statement if the attstattarget entry for this column is
- * non-negative (i.e. it's not the default value)
- */
- if (tbinfo->attstattarget[j] >= 0)
- {
- appendPQExpBuffer(q, "ALTER TABLE ONLY %s ",
- fmtId(tbinfo->dobj.name));
- appendPQExpBuffer(q, "ALTER COLUMN %s ",
- fmtId(tbinfo->attnames[j]));
- appendPQExpBuffer(q, "SET STATISTICS %d;\n",
- tbinfo->attstattarget[j]);
- }
- /*
- * Dump per-column storage information. The statement is only
- * dumped if the storage has been changed from the type's default.
- */
- if (tbinfo->attstorage[j] != tbinfo->typstorage[j])
- {
- switch (tbinfo->attstorage[j])
- {
- case 'p':
- storage = "PLAIN";
- break;
- case 'e':
- storage = "EXTERNAL";
- break;
- case 'm':
- storage = "MAIN";
- break;
- case 'x':
- storage = "EXTENDED";
- break;
- default:
- storage = NULL;
- }
- /*
- * Only dump the statement if it's a storage type we recognize
- */
- if (storage != NULL)
- {
- appendPQExpBuffer(q, "ALTER TABLE ONLY %s ",
- fmtId(tbinfo->dobj.name));
- appendPQExpBuffer(q, "ALTER COLUMN %s ",
- fmtId(tbinfo->attnames[j]));
- appendPQExpBuffer(q, "SET STORAGE %s;\n",
- storage);
- }
- }
- /*
- * Dump per-column attributes.
- */
- if (tbinfo->attoptions[j] && tbinfo->attoptions[j][0] != '\0')
- {
- appendPQExpBuffer(q, "ALTER TABLE ONLY %s ",
- fmtId(tbinfo->dobj.name));
- appendPQExpBuffer(q, "ALTER COLUMN %s ",
- fmtId(tbinfo->attnames[j]));
- appendPQExpBuffer(q, "SET (%s);\n",
- tbinfo->attoptions[j]);
- }
- /*
- * Dump per-column fdw options.
- */
- if (tbinfo->relkind == RELKIND_FOREIGN_TABLE &&
- tbinfo->attfdwoptions[j] &&
- tbinfo->attfdwoptions[j][0] != '\0')
- {
- appendPQExpBuffer(q, "ALTER FOREIGN TABLE %s ",
- fmtId(tbinfo->dobj.name));
- appendPQExpBuffer(q, "ALTER COLUMN %s ",
- fmtId(tbinfo->attnames[j]));
- appendPQExpBuffer(q, "OPTIONS (\n %s\n);\n",
- tbinfo->attfdwoptions[j]);
- }
- }
- }
- /*
- * dump properties we only have ALTER TABLE syntax for
- */
- if ((tbinfo->relkind == RELKIND_RELATION || tbinfo->relkind == RELKIND_MATVIEW) &&
- tbinfo->relreplident != REPLICA_IDENTITY_DEFAULT)
- {
- if (tbinfo->relreplident == REPLICA_IDENTITY_INDEX)
- {
- /* nothing to do, will be set when the index is dumped */
- }
- else if (tbinfo->relreplident == REPLICA_IDENTITY_NOTHING)
- {
- appendPQExpBuffer(q, "\nALTER TABLE ONLY %s REPLICA IDENTITY NOTHING;\n",
- fmtId(tbinfo->dobj.name));
- }
- else if (tbinfo->relreplident == REPLICA_IDENTITY_FULL)
- {
- appendPQExpBuffer(q, "\nALTER TABLE ONLY %s REPLICA IDENTITY FULL;\n",
- fmtId(tbinfo->dobj.name));
- }
- }
- if (binary_upgrade)
- binary_upgrade_extension_member(q, &tbinfo->dobj, labelq->data);
- ArchiveEntry(fout, tbinfo->dobj.catId, tbinfo->dobj.dumpId,
- tbinfo->dobj.name,
- tbinfo->dobj.namespace->dobj.name,
- (tbinfo->relkind == RELKIND_VIEW) ? NULL : tbinfo->reltablespace,
- tbinfo->rolname,
- (strcmp(reltypename, "TABLE") == 0) ? tbinfo->hasoids : false,
- reltypename,
- tbinfo->postponed_def ? SECTION_POST_DATA : SECTION_PRE_DATA,
- q->data, delq->data, NULL,
- NULL, 0,
- NULL, NULL);
- /* Dump Table Comments */
- dumpTableComment(fout, tbinfo, reltypename);
- /* Dump Table Security Labels */
- dumpTableSecLabel(fout, tbinfo, reltypename);
- /* Dump comments on inlined table constraints */
- for (j = 0; j < tbinfo->ncheck; j++)
- {
- ConstraintInfo *constr = &(tbinfo->checkexprs[j]);
- if (constr->separate || !constr->conislocal)
- continue;
- dumpTableConstraintComment(fout, constr);
- }
- destroyPQExpBuffer(q);
- destroyPQExpBuffer(delq);
- destroyPQExpBuffer(labelq);
- }
- /*
- * dumpAttrDef --- dump an attribute's default-value declaration
- */
- static void
- dumpAttrDef(Archive *fout, AttrDefInfo *adinfo)
- {
- TableInfo *tbinfo = adinfo->adtable;
- int adnum = adinfo->adnum;
- PQExpBuffer q;
- PQExpBuffer delq;
- /* Skip if table definition not to be dumped */
- if (!tbinfo->dobj.dump || dataOnly)
- return;
- /* Skip if not "separate"; it was dumped in the table's definition */
- if (!adinfo->separate)
- return;
- q = createPQExpBuffer();
- delq = createPQExpBuffer();
- appendPQExpBuffer(q, "ALTER TABLE ONLY %s ",
- fmtId(tbinfo->dobj.name));
- appendPQExpBuffer(q, "ALTER COLUMN %s SET DEFAULT %s;\n",
- fmtId(tbinfo->attnames[adnum - 1]),
- adinfo->adef_expr);
- /*
- * DROP must be fully qualified in case same name appears in pg_catalog
- */
- appendPQExpBuffer(delq, "ALTER TABLE %s.",
- fmtId(tbinfo->dobj.namespace->dobj.name));
- appendPQExpBuffer(delq, "%s ",
- fmtId(tbinfo->dobj.name));
- appendPQExpBuffer(delq, "ALTER COLUMN %s DROP DEFAULT;\n",
- fmtId(tbinfo->attnames[adnum - 1]));
- ArchiveEntry(fout, adinfo->dobj.catId, adinfo->dobj.dumpId,
- tbinfo->attnames[adnum - 1],
- tbinfo->dobj.namespace->dobj.name,
- NULL,
- tbinfo->rolname,
- false, "DEFAULT", SECTION_PRE_DATA,
- q->data, delq->data, NULL,
- NULL, 0,
- NULL, NULL);
- destroyPQExpBuffer(q);
- destroyPQExpBuffer(delq);
- }
- /*
- * getAttrName: extract the correct name for an attribute
- *
- * The array tblInfo->attnames[] only provides names of user attributes;
- * if a system attribute number is supplied, we have to fake it.
- * We also do a little bit of bounds checking for safety's sake.
- */
- static const char *
- getAttrName(int attrnum, TableInfo *tblInfo)
- {
- if (attrnum > 0 && attrnum <= tblInfo->numatts)
- return tblInfo->attnames[attrnum - 1];
- switch (attrnum)
- {
- case SelfItemPointerAttributeNumber:
- return "ctid";
- case ObjectIdAttributeNumber:
- return "oid";
- case MinTransactionIdAttributeNumber:
- return "xmin";
- case MinCommandIdAttributeNumber:
- return "cmin";
- case MaxTransactionIdAttributeNumber:
- return "xmax";
- case MaxCommandIdAttributeNumber:
- return "cmax";
- case TableOidAttributeNumber:
- return "tableoid";
- }
- exit_horribly(NULL, "invalid column number %d for table \"%s\"\n",
- attrnum, tblInfo->dobj.name);
- return NULL; /* keep compiler quiet */
- }
- /*
- * dumpIndex
- * write out to fout a user-defined index
- */
- static void
- dumpIndex(Archive *fout, IndxInfo *indxinfo)
- {
- TableInfo *tbinfo = indxinfo->indextable;
- bool is_constraint = (indxinfo->indexconstraint != 0);
- PQExpBuffer q;
- PQExpBuffer delq;
- PQExpBuffer labelq;
- if (dataOnly)
- return;
- q = createPQExpBuffer();
- delq = createPQExpBuffer();
- labelq = createPQExpBuffer();
- appendPQExpBuffer(labelq, "INDEX %s",
- fmtId(indxinfo->dobj.name));
- /*
- * If there's an associated constraint, don't dump the index per se, but
- * do dump any comment for it. (This is safe because dependency ordering
- * will have ensured the constraint is emitted first.) Note that the
- * emitted comment has to be shown as depending on the constraint, not the
- * index, in such cases.
- */
- if (!is_constraint)
- {
- if (binary_upgrade)
- binary_upgrade_set_pg_class_oids(fout, q,
- indxinfo->dobj.catId.oid, true);
- /* Plain secondary index */
- appendPQExpBuffer(q, "%s;\n", indxinfo->indexdef);
- /* If the index is clustered, we need to record that. */
- if (indxinfo->indisclustered)
- {
- appendPQExpBuffer(q, "\nALTER TABLE %s CLUSTER",
- fmtId(tbinfo->dobj.name));
- appendPQExpBuffer(q, " ON %s;\n",
- fmtId(indxinfo->dobj.name));
- }
- /* If the index defines identity, we need to record that. */
- if (indxinfo->indisreplident)
- {
- appendPQExpBuffer(q, "\nALTER TABLE ONLY %s REPLICA IDENTITY USING",
- fmtId(tbinfo->dobj.name));
- appendPQExpBuffer(q, " INDEX %s;\n",
- fmtId(indxinfo->dobj.name));
- }
- /*
- * DROP must be fully qualified in case same name appears in
- * pg_catalog
- */
- appendPQExpBuffer(delq, "DROP INDEX %s.",
- fmtId(tbinfo->dobj.namespace->dobj.name));
- appendPQExpBuffer(delq, "%s;\n",
- fmtId(indxinfo->dobj.name));
- ArchiveEntry(fout, indxinfo->dobj.catId, indxinfo->dobj.dumpId,
- indxinfo->dobj.name,
- tbinfo->dobj.namespace->dobj.name,
- indxinfo->tablespace,
- tbinfo->rolname, false,
- "INDEX", SECTION_POST_DATA,
- q->data, delq->data, NULL,
- NULL, 0,
- NULL, NULL);
- }
- /* Dump Index Comments */
- dumpComment(fout, labelq->data,
- tbinfo->dobj.namespace->dobj.name,
- tbinfo->rolname,
- indxinfo->dobj.catId, 0,
- is_constraint ? indxinfo->indexconstraint :
- indxinfo->dobj.dumpId);
- destroyPQExpBuffer(q);
- destroyPQExpBuffer(delq);
- destroyPQExpBuffer(labelq);
- }
- /*
- * dumpConstraint
- * write out to fout a user-defined constraint
- */
- static void
- dumpConstraint(Archive *fout, ConstraintInfo *coninfo)
- {
- TableInfo *tbinfo = coninfo->contable;
- PQExpBuffer q;
- PQExpBuffer delq;
- /* Skip if not to be dumped */
- if (!coninfo->dobj.dump || dataOnly)
- return;
- q = createPQExpBuffer();
- delq = createPQExpBuffer();
- if (coninfo->contype == 'p' ||
- coninfo->contype == 'u' ||
- coninfo->contype == 'x')
- {
- /* Index-related constraint */
- IndxInfo *indxinfo;
- int k;
- indxinfo = (IndxInfo *) findObjectByDumpId(coninfo->conindex);
- if (indxinfo == NULL)
- exit_horribly(NULL, "missing index for constraint \"%s\"\n",
- coninfo->dobj.name);
- if (binary_upgrade)
- binary_upgrade_set_pg_class_oids(fout, q,
- indxinfo->dobj.catId.oid, true);
- appendPQExpBuffer(q, "ALTER TABLE ONLY %s\n",
- fmtId(tbinfo->dobj.name));
- appendPQExpBuffer(q, " ADD CONSTRAINT %s ",
- fmtId(coninfo->dobj.name));
- if (coninfo->condef)
- {
- /* pg_get_constraintdef should have provided everything */
- appendPQExpBuffer(q, "%s;\n", coninfo->condef);
- }
- else
- {
- appendPQExpBuffer(q, "%s (",
- coninfo->contype == 'p' ? "PRIMARY KEY" : "UNIQUE");
- for (k = 0; k < indxinfo->indnkeys; k++)
- {
- int indkey = (int) indxinfo->indkeys[k];
- const char *attname;
- if (indkey == InvalidAttrNumber)
- break;
- attname = getAttrName(indkey, tbinfo);
- appendPQExpBuffer(q, "%s%s",
- (k == 0) ? "" : ", ",
- fmtId(attname));
- }
- appendPQExpBufferChar(q, ')');
- if (indxinfo->options && strlen(indxinfo->options) > 0)
- appendPQExpBuffer(q, " WITH (%s)", indxinfo->options);
- if (coninfo->condeferrable)
- {
- appendPQExpBufferStr(q, " DEFERRABLE");
- if (coninfo->condeferred)
- appendPQExpBufferStr(q, " INITIALLY DEFERRED");
- }
- appendPQExpBufferStr(q, ";\n");
- }
- /* If the index is clustered, we need to record that. */
- if (indxinfo->indisclustered)
- {
- appendPQExpBuffer(q, "\nALTER TABLE %s CLUSTER",
- fmtId(tbinfo->dobj.name));
- appendPQExpBuffer(q, " ON %s;\n",
- fmtId(indxinfo->dobj.name));
- }
- /*
- * DROP must be fully qualified in case same name appears in
- * pg_catalog
- */
- appendPQExpBuffer(delq, "ALTER TABLE ONLY %s.",
- fmtId(tbinfo->dobj.namespace->dobj.name));
- appendPQExpBuffer(delq, "%s ",
- fmtId(tbinfo->dobj.name));
- appendPQExpBuffer(delq, "DROP CONSTRAINT %s;\n",
- fmtId(coninfo->dobj.name));
- ArchiveEntry(fout, coninfo->dobj.catId, coninfo->dobj.dumpId,
- coninfo->dobj.name,
- tbinfo->dobj.namespace->dobj.name,
- indxinfo->tablespace,
- tbinfo->rolname, false,
- "CONSTRAINT", SECTION_POST_DATA,
- q->data, delq->data, NULL,
- NULL, 0,
- NULL, NULL);
- }
- else if (coninfo->contype == 'f')
- {
- /*
- * XXX Potentially wrap in a 'SET CONSTRAINTS OFF' block so that the
- * current table data is not processed
- */
- appendPQExpBuffer(q, "ALTER TABLE ONLY %s\n",
- fmtId(tbinfo->dobj.name));
- appendPQExpBuffer(q, " ADD CONSTRAINT %s %s;\n",
- fmtId(coninfo->dobj.name),
- coninfo->condef);
- /*
- * DROP must be fully qualified in case same name appears in
- * pg_catalog
- */
- appendPQExpBuffer(delq, "ALTER TABLE ONLY %s.",
- fmtId(tbinfo->dobj.namespace->dobj.name));
- appendPQExpBuffer(delq, "%s ",
- fmtId(tbinfo->dobj.name));
- appendPQExpBuffer(delq, "DROP CONSTRAINT %s;\n",
- fmtId(coninfo->dobj.name));
- ArchiveEntry(fout, coninfo->dobj.catId, coninfo->dobj.dumpId,
- coninfo->dobj.name,
- tbinfo->dobj.namespace->dobj.name,
- NULL,
- tbinfo->rolname, false,
- "FK CONSTRAINT", SECTION_POST_DATA,
- q->data, delq->data, NULL,
- NULL, 0,
- NULL, NULL);
- }
- else if (coninfo->contype == 'c' && tbinfo)
- {
- /* CHECK constraint on a table */
- /* Ignore if not to be dumped separately */
- if (coninfo->separate)
- {
- /* not ONLY since we want it to propagate to children */
- appendPQExpBuffer(q, "ALTER TABLE %s\n",
- fmtId(tbinfo->dobj.name));
- appendPQExpBuffer(q, " ADD CONSTRAINT %s %s;\n",
- fmtId(coninfo->dobj.name),
- coninfo->condef);
- /*
- * DROP must be fully qualified in case same name appears in
- * pg_catalog
- */
- appendPQExpBuffer(delq, "ALTER TABLE %s.",
- fmtId(tbinfo->dobj.namespace->dobj.name));
- appendPQExpBuffer(delq, "%s ",
- fmtId(tbinfo->dobj.name));
- appendPQExpBuffer(delq, "DROP CONSTRAINT %s;\n",
- fmtId(coninfo->dobj.name));
- ArchiveEntry(fout, coninfo->dobj.catId, coninfo->dobj.dumpId,
- coninfo->dobj.name,
- tbinfo->dobj.namespace->dobj.name,
- NULL,
- tbinfo->rolname, false,
- "CHECK CONSTRAINT", SECTION_POST_DATA,
- q->data, delq->data, NULL,
- NULL, 0,
- NULL, NULL);
- }
- }
- else if (coninfo->contype == 'c' && tbinfo == NULL)
- {
- /* CHECK constraint on a domain */
- TypeInfo *tyinfo = coninfo->condomain;
- /* Ignore if not to be dumped separately */
- if (coninfo->separate)
- {
- appendPQExpBuffer(q, "ALTER DOMAIN %s\n",
- fmtId(tyinfo->dobj.name));
- appendPQExpBuffer(q, " ADD CONSTRAINT %s %s;\n",
- fmtId(coninfo->dobj.name),
- coninfo->condef);
- /*
- * DROP must be fully qualified in case same name appears in
- * pg_catalog
- */
- appendPQExpBuffer(delq, "ALTER DOMAIN %s.",
- fmtId(tyinfo->dobj.namespace->dobj.name));
- appendPQExpBuffer(delq, "%s ",
- fmtId(tyinfo->dobj.name));
- appendPQExpBuffer(delq, "DROP CONSTRAINT %s;\n",
- fmtId(coninfo->dobj.name));
- ArchiveEntry(fout, coninfo->dobj.catId, coninfo->dobj.dumpId,
- coninfo->dobj.name,
- tyinfo->dobj.namespace->dobj.name,
- NULL,
- tyinfo->rolname, false,
- "CHECK CONSTRAINT", SECTION_POST_DATA,
- q->data, delq->data, NULL,
- NULL, 0,
- NULL, NULL);
- }
- }
- else
- {
- exit_horribly(NULL, "unrecognized constraint type: %c\n",
- coninfo->contype);
- }
- /* Dump Constraint Comments --- only works for table constraints */
- if (tbinfo && coninfo->separate)
- dumpTableConstraintComment(fout, coninfo);
- destroyPQExpBuffer(q);
- destroyPQExpBuffer(delq);
- }
- /*
- * dumpTableConstraintComment --- dump a constraint's comment if any
- *
- * This is split out because we need the function in two different places
- * depending on whether the constraint is dumped as part of CREATE TABLE
- * or as a separate ALTER command.
- */
- static void
- dumpTableConstraintComment(Archive *fout, ConstraintInfo *coninfo)
- {
- TableInfo *tbinfo = coninfo->contable;
- PQExpBuffer labelq = createPQExpBuffer();
- appendPQExpBuffer(labelq, "CONSTRAINT %s ",
- fmtId(coninfo->dobj.name));
- appendPQExpBuffer(labelq, "ON %s",
- fmtId(tbinfo->dobj.name));
- dumpComment(fout, labelq->data,
- tbinfo->dobj.namespace->dobj.name,
- tbinfo->rolname,
- coninfo->dobj.catId, 0,
- coninfo->separate ? coninfo->dobj.dumpId : tbinfo->dobj.dumpId);
- destroyPQExpBuffer(labelq);
- }
- /*
- * findLastBuiltInOid -
- * find the last built in oid
- *
- * For 7.1 and 7.2, we do this by retrieving datlastsysoid from the
- * pg_database entry for the current database
- */
- static Oid
- findLastBuiltinOid_V71(Archive *fout, const char *dbname)
- {
- PGresult *res;
- Oid last_oid;
- PQExpBuffer query = createPQExpBuffer();
- resetPQExpBuffer(query);
- appendPQExpBufferStr(query, "SELECT datlastsysoid from pg_database where datname = ");
- appendStringLiteralAH(query, dbname, fout);
- res = ExecuteSqlQueryForSingleRow(fout, query->data);
- last_oid = atooid(PQgetvalue(res, 0, PQfnumber(res, "datlastsysoid")));
- PQclear(res);
- destroyPQExpBuffer(query);
- return last_oid;
- }
- /*
- * findLastBuiltInOid -
- * find the last built in oid
- *
- * For 7.0, we do this by assuming that the last thing that initdb does is to
- * create the pg_indexes view. This sucks in general, but seeing that 7.0.x
- * initdb won't be changing anymore, it'll do.
- */
- static Oid
- findLastBuiltinOid_V70(Archive *fout)
- {
- PGresult *res;
- int last_oid;
- res = ExecuteSqlQueryForSingleRow(fout,
- "SELECT oid FROM pg_class WHERE relname = 'pg_indexes'");
- last_oid = atooid(PQgetvalue(res, 0, PQfnumber(res, "oid")));
- PQclear(res);
- return last_oid;
- }
- /*
- * dumpSequence
- * write the declaration (not data) of one user-defined sequence
- */
- static void
- dumpSequence(Archive *fout, TableInfo *tbinfo)
- {
- PGresult *res;
- char *startv,
- *incby,
- *maxv = NULL,
- *minv = NULL,
- *cache;
- char bufm[100],
- bufx[100];
- bool cycled;
- PQExpBuffer query = createPQExpBuffer();
- PQExpBuffer delqry = createPQExpBuffer();
- PQExpBuffer labelq = createPQExpBuffer();
- /* Make sure we are in proper schema */
- selectSourceSchema(fout, tbinfo->dobj.namespace->dobj.name);
- snprintf(bufm, sizeof(bufm), INT64_FORMAT, SEQ_MINVALUE);
- snprintf(bufx, sizeof(bufx), INT64_FORMAT, SEQ_MAXVALUE);
- if (fout->remoteVersion >= 80400)
- {
- appendPQExpBuffer(query,
- "SELECT sequence_name, "
- "start_value, increment_by, "
- "CASE WHEN increment_by > 0 AND max_value = %s THEN NULL "
- " WHEN increment_by < 0 AND max_value = -1 THEN NULL "
- " ELSE max_value "
- "END AS max_value, "
- "CASE WHEN increment_by > 0 AND min_value = 1 THEN NULL "
- " WHEN increment_by < 0 AND min_value = %s THEN NULL "
- " ELSE min_value "
- "END AS min_value, "
- "cache_value, is_cycled FROM %s",
- bufx, bufm,
- fmtId(tbinfo->dobj.name));
- }
- else
- {
- appendPQExpBuffer(query,
- "SELECT sequence_name, "
- "0 AS start_value, increment_by, "
- "CASE WHEN increment_by > 0 AND max_value = %s THEN NULL "
- " WHEN increment_by < 0 AND max_value = -1 THEN NULL "
- " ELSE max_value "
- "END AS max_value, "
- "CASE WHEN increment_by > 0 AND min_value = 1 THEN NULL "
- " WHEN increment_by < 0 AND min_value = %s THEN NULL "
- " ELSE min_value "
- "END AS min_value, "
- "cache_value, is_cycled FROM %s",
- bufx, bufm,
- fmtId(tbinfo->dobj.name));
- }
- res = ExecuteSqlQuery(fout, query->data, PGRES_TUPLES_OK);
- if (PQntuples(res) != 1)
- {
- write_msg(NULL, ngettext("query to get data of sequence \"%s\" returned %d row (expected 1)\n",
- "query to get data of sequence \"%s\" returned %d rows (expected 1)\n",
- PQntuples(res)),
- tbinfo->dobj.name, PQntuples(res));
- exit_nicely(1);
- }
- /* Disable this check: it fails if sequence has been renamed */
- #ifdef NOT_USED
- if (strcmp(PQgetvalue(res, 0, 0), tbinfo->dobj.name) != 0)
- {
- write_msg(NULL, "query to get data of sequence \"%s\" returned name \"%s\"\n",
- tbinfo->dobj.name, PQgetvalue(res, 0, 0));
- exit_nicely(1);
- }
- #endif
- startv = PQgetvalue(res, 0, 1);
- incby = PQgetvalue(res, 0, 2);
- if (!PQgetisnull(res, 0, 3))
- maxv = PQgetvalue(res, 0, 3);
- if (!PQgetisnull(res, 0, 4))
- minv = PQgetvalue(res, 0, 4);
- cache = PQgetvalue(res, 0, 5);
- cycled = (strcmp(PQgetvalue(res, 0, 6), "t") == 0);
- /*
- * DROP must be fully qualified in case same name appears in pg_catalog
- */
- appendPQExpBuffer(delqry, "DROP SEQUENCE %s.",
- fmtId(tbinfo->dobj.namespace->dobj.name));
- appendPQExpBuffer(delqry, "%s;\n",
- fmtId(tbinfo->dobj.name));
- resetPQExpBuffer(query);
- if (binary_upgrade)
- {
- binary_upgrade_set_pg_class_oids(fout, query,
- tbinfo->dobj.catId.oid, false);
- binary_upgrade_set_type_oids_by_rel_oid(fout, query,
- tbinfo->dobj.catId.oid);
- }
- appendPQExpBuffer(query,
- "CREATE SEQUENCE %s\n",
- fmtId(tbinfo->dobj.name));
- if (fout->remoteVersion >= 80400)
- appendPQExpBuffer(query, " START WITH %s\n", startv);
- appendPQExpBuffer(query, " INCREMENT BY %s\n", incby);
- if (minv)
- appendPQExpBuffer(query, " MINVALUE %s\n", minv);
- else
- appendPQExpBufferStr(query, " NO MINVALUE\n");
- if (maxv)
- appendPQExpBuffer(query, " MAXVALUE %s\n", maxv);
- else
- appendPQExpBufferStr(query, " NO MAXVALUE\n");
- appendPQExpBuffer(query,
- " CACHE %s%s",
- cache, (cycled ? "\n CYCLE" : ""));
- appendPQExpBufferStr(query, ";\n");
- appendPQExpBuffer(labelq, "SEQUENCE %s", fmtId(tbinfo->dobj.name));
- /* binary_upgrade: no need to clear TOAST table oid */
- if (binary_upgrade)
- binary_upgrade_extension_member(query, &tbinfo->dobj,
- labelq->data);
- ArchiveEntry(fout, tbinfo->dobj.catId, tbinfo->dobj.dumpId,
- tbinfo->dobj.name,
- tbinfo->dobj.namespace->dobj.name,
- NULL,
- tbinfo->rolname,
- false, "SEQUENCE", SECTION_PRE_DATA,
- query->data, delqry->data, NULL,
- NULL, 0,
- NULL, NULL);
- /*
- * If the sequence is owned by a table column, emit the ALTER for it as a
- * separate TOC entry immediately following the sequence's own entry. It's
- * OK to do this rather than using full sorting logic, because the
- * dependency that tells us it's owned will have forced the table to be
- * created first. We can't just include the ALTER in the TOC entry
- * because it will fail if we haven't reassigned the sequence owner to
- * match the table's owner.
- *
- * We need not schema-qualify the table reference because both sequence
- * and table must be in the same schema.
- */
- if (OidIsValid(tbinfo->owning_tab))
- {
- TableInfo *owning_tab = findTableByOid(tbinfo->owning_tab);
- if (owning_tab && owning_tab->dobj.dump)
- {
- resetPQExpBuffer(query);
- appendPQExpBuffer(query, "ALTER SEQUENCE %s",
- fmtId(tbinfo->dobj.name));
- appendPQExpBuffer(query, " OWNED BY %s",
- fmtId(owning_tab->dobj.name));
- appendPQExpBuffer(query, ".%s;\n",
- fmtId(owning_tab->attnames[tbinfo->owning_col - 1]));
- ArchiveEntry(fout, nilCatalogId, createDumpId(),
- tbinfo->dobj.name,
- tbinfo->dobj.namespace->dobj.name,
- NULL,
- tbinfo->rolname,
- false, "SEQUENCE OWNED BY", SECTION_PRE_DATA,
- query->data, "", NULL,
- &(tbinfo->dobj.dumpId), 1,
- NULL, NULL);
- }
- }
- /* Dump Sequence Comments and Security Labels */
- dumpComment(fout, labelq->data,
- tbinfo->dobj.namespace->dobj.name, tbinfo->rolname,
- tbinfo->dobj.catId, 0, tbinfo->dobj.dumpId);
- dumpSecLabel(fout, labelq->data,
- tbinfo->dobj.namespace->dobj.name, tbinfo->rolname,
- tbinfo->dobj.catId, 0, tbinfo->dobj.dumpId);
- PQclear(res);
- destroyPQExpBuffer(query);
- destroyPQExpBuffer(delqry);
- destroyPQExpBuffer(labelq);
- }
- /*
- * dumpSequenceData
- * write the data of one user-defined sequence
- */
- static void
- dumpSequenceData(Archive *fout, TableDataInfo *tdinfo)
- {
- TableInfo *tbinfo = tdinfo->tdtable;
- PGresult *res;
- char *last;
- bool called;
- PQExpBuffer query = createPQExpBuffer();
- /* Make sure we are in proper schema */
- selectSourceSchema(fout, tbinfo->dobj.namespace->dobj.name);
- appendPQExpBuffer(query,
- "SELECT last_value, is_called FROM %s",
- fmtId(tbinfo->dobj.name));
- res = ExecuteSqlQuery(fout, query->data, PGRES_TUPLES_OK);
- if (PQntuples(res) != 1)
- {
- write_msg(NULL, ngettext("query to get data of sequence \"%s\" returned %d row (expected 1)\n",
- "query to get data of sequence \"%s\" returned %d rows (expected 1)\n",
- PQntuples(res)),
- tbinfo->dobj.name, PQntuples(res));
- exit_nicely(1);
- }
- last = PQgetvalue(res, 0, 0);
- called = (strcmp(PQgetvalue(res, 0, 1), "t") == 0);
- resetPQExpBuffer(query);
- appendPQExpBufferStr(query, "SELECT pg_catalog.setval(");
- appendStringLiteralAH(query, fmtId(tbinfo->dobj.name), fout);
- appendPQExpBuffer(query, ", %s, %s);\n",
- last, (called ? "true" : "false"));
- ArchiveEntry(fout, nilCatalogId, createDumpId(),
- tbinfo->dobj.name,
- tbinfo->dobj.namespace->dobj.name,
- NULL,
- tbinfo->rolname,
- false, "SEQUENCE SET", SECTION_DATA,
- query->data, "", NULL,
- &(tbinfo->dobj.dumpId), 1,
- NULL, NULL);
- PQclear(res);
- destroyPQExpBuffer(query);
- }
- /*
- * dumpTrigger
- * write the declaration of one user-defined table trigger
- */
- static void
- dumpTrigger(Archive *fout, TriggerInfo *tginfo)
- {
- TableInfo *tbinfo = tginfo->tgtable;
- PQExpBuffer query;
- PQExpBuffer delqry;
- PQExpBuffer labelq;
- char *tgargs;
- size_t lentgargs;
- const char *p;
- int findx;
- /*
- * we needn't check dobj.dump because TriggerInfo wouldn't have been
- * created in the first place for non-dumpable triggers
- */
- if (dataOnly)
- return;
- query = createPQExpBuffer();
- delqry = createPQExpBuffer();
- labelq = createPQExpBuffer();
- /*
- * DROP must be fully qualified in case same name appears in pg_catalog
- */
- appendPQExpBuffer(delqry, "DROP TRIGGER %s ",
- fmtId(tginfo->dobj.name));
- appendPQExpBuffer(delqry, "ON %s.",
- fmtId(tbinfo->dobj.namespace->dobj.name));
- appendPQExpBuffer(delqry, "%s;\n",
- fmtId(tbinfo->dobj.name));
- if (tginfo->tgdef)
- {
- appendPQExpBuffer(query, "%s;\n", tginfo->tgdef);
- }
- else
- {
- if (tginfo->tgisconstraint)
- {
- appendPQExpBufferStr(query, "CREATE CONSTRAINT TRIGGER ");
- appendPQExpBufferStr(query, fmtId(tginfo->tgconstrname));
- }
- else
- {
- appendPQExpBufferStr(query, "CREATE TRIGGER ");
- appendPQExpBufferStr(query, fmtId(tginfo->dobj.name));
- }
- appendPQExpBufferStr(query, "\n ");
- /* Trigger type */
- if (TRIGGER_FOR_BEFORE(tginfo->tgtype))
- appendPQExpBufferStr(query, "BEFORE");
- else if (TRIGGER_FOR_AFTER(tginfo->tgtype))
- appendPQExpBufferStr(query, "AFTER");
- else if (TRIGGER_FOR_INSTEAD(tginfo->tgtype))
- appendPQExpBufferStr(query, "INSTEAD OF");
- else
- {
- write_msg(NULL, "unexpected tgtype value: %d\n", tginfo->tgtype);
- exit_nicely(1);
- }
- findx = 0;
- if (TRIGGER_FOR_INSERT(tginfo->tgtype))
- {
- appendPQExpBufferStr(query, " INSERT");
- findx++;
- }
- if (TRIGGER_FOR_DELETE(tginfo->tgtype))
- {
- if (findx > 0)
- appendPQExpBufferStr(query, " OR DELETE");
- else
- appendPQExpBufferStr(query, " DELETE");
- findx++;
- }
- if (TRIGGER_FOR_UPDATE(tginfo->tgtype))
- {
- if (findx > 0)
- appendPQExpBufferStr(query, " OR UPDATE");
- else
- appendPQExpBufferStr(query, " UPDATE");
- findx++;
- }
- if (TRIGGER_FOR_TRUNCATE(tginfo->tgtype))
- {
- if (findx > 0)
- appendPQExpBufferStr(query, " OR TRUNCATE");
- else
- appendPQExpBufferStr(query, " TRUNCATE");
- findx++;
- }
- appendPQExpBuffer(query, " ON %s\n",
- fmtId(tbinfo->dobj.name));
- if (tginfo->tgisconstraint)
- {
- if (OidIsValid(tginfo->tgconstrrelid))
- {
- /* If we are using regclass, name is already quoted */
- if (fout->remoteVersion >= 70300)
- appendPQExpBuffer(query, " FROM %s\n ",
- tginfo->tgconstrrelname);
- else
- appendPQExpBuffer(query, " FROM %s\n ",
- fmtId(tginfo->tgconstrrelname));
- }
- if (!tginfo->tgdeferrable)
- appendPQExpBufferStr(query, "NOT ");
- appendPQExpBufferStr(query, "DEFERRABLE INITIALLY ");
- if (tginfo->tginitdeferred)
- appendPQExpBufferStr(query, "DEFERRED\n");
- else
- appendPQExpBufferStr(query, "IMMEDIATE\n");
- }
- if (TRIGGER_FOR_ROW(tginfo->tgtype))
- appendPQExpBufferStr(query, " FOR EACH ROW\n ");
- else
- appendPQExpBufferStr(query, " FOR EACH STATEMENT\n ");
- /* In 7.3, result of regproc is already quoted */
- if (fout->remoteVersion >= 70300)
- appendPQExpBuffer(query, "EXECUTE PROCEDURE %s(",
- tginfo->tgfname);
- else
- appendPQExpBuffer(query, "EXECUTE PROCEDURE %s(",
- fmtId(tginfo->tgfname));
- tgargs = (char *) PQunescapeBytea((unsigned char *) tginfo->tgargs,
- &lentgargs);
- p = tgargs;
- for (findx = 0; findx < tginfo->tgnargs; findx++)
- {
- /* find the embedded null that terminates this trigger argument */
- size_t tlen = strlen(p);
- if (p + tlen >= tgargs + lentgargs)
- {
- /* hm, not found before end of bytea value... */
- write_msg(NULL, "invalid argument string (%s) for trigger \"%s\" on table \"%s\"\n",
- tginfo->tgargs,
- tginfo->dobj.name,
- tbinfo->dobj.name);
- exit_nicely(1);
- }
- if (findx > 0)
- appendPQExpBufferStr(query, ", ");
- appendStringLiteralAH(query, p, fout);
- p += tlen + 1;
- }
- free(tgargs);
- appendPQExpBufferStr(query, ");\n");
- }
- if (tginfo->tgenabled != 't' && tginfo->tgenabled != 'O')
- {
- appendPQExpBuffer(query, "\nALTER TABLE %s ",
- fmtId(tbinfo->dobj.name));
- switch (tginfo->tgenabled)
- {
- case 'D':
- case 'f':
- appendPQExpBufferStr(query, "DISABLE");
- break;
- case 'A':
- appendPQExpBufferStr(query, "ENABLE ALWAYS");
- break;
- case 'R':
- appendPQExpBufferStr(query, "ENABLE REPLICA");
- break;
- default:
- appendPQExpBufferStr(query, "ENABLE");
- break;
- }
- appendPQExpBuffer(query, " TRIGGER %s;\n",
- fmtId(tginfo->dobj.name));
- }
- appendPQExpBuffer(labelq, "TRIGGER %s ",
- fmtId(tginfo->dobj.name));
- appendPQExpBuffer(labelq, "ON %s",
- fmtId(tbinfo->dobj.name));
- ArchiveEntry(fout, tginfo->dobj.catId, tginfo->dobj.dumpId,
- tginfo->dobj.name,
- tbinfo->dobj.namespace->dobj.name,
- NULL,
- tbinfo->rolname, false,
- "TRIGGER", SECTION_POST_DATA,
- query->data, delqry->data, NULL,
- NULL, 0,
- NULL, NULL);
- dumpComment(fout, labelq->data,
- tbinfo->dobj.namespace->dobj.name, tbinfo->rolname,
- tginfo->dobj.catId, 0, tginfo->dobj.dumpId);
- destroyPQExpBuffer(query);
- destroyPQExpBuffer(delqry);
- destroyPQExpBuffer(labelq);
- }
- /*
- * dumpEventTrigger
- * write the declaration of one user-defined event trigger
- */
- static void
- dumpEventTrigger(Archive *fout, EventTriggerInfo *evtinfo)
- {
- PQExpBuffer query;
- PQExpBuffer labelq;
- /* Skip if not to be dumped */
- if (!evtinfo->dobj.dump || dataOnly)
- return;
- query = createPQExpBuffer();
- labelq = createPQExpBuffer();
- appendPQExpBufferStr(query, "CREATE EVENT TRIGGER ");
- appendPQExpBufferStr(query, fmtId(evtinfo->dobj.name));
- appendPQExpBufferStr(query, " ON ");
- appendPQExpBufferStr(query, fmtId(evtinfo->evtevent));
- appendPQExpBufferStr(query, " ");
- if (strcmp("", evtinfo->evttags) != 0)
- {
- appendPQExpBufferStr(query, "\n WHEN TAG IN (");
- appendPQExpBufferStr(query, evtinfo->evttags);
- appendPQExpBufferStr(query, ") ");
- }
- appendPQExpBufferStr(query, "\n EXECUTE PROCEDURE ");
- appendPQExpBufferStr(query, evtinfo->evtfname);
- appendPQExpBufferStr(query, "();\n");
- if (evtinfo->evtenabled != 'O')
- {
- appendPQExpBuffer(query, "\nALTER EVENT TRIGGER %s ",
- fmtId(evtinfo->dobj.name));
- switch (evtinfo->evtenabled)
- {
- case 'D':
- appendPQExpBufferStr(query, "DISABLE");
- break;
- case 'A':
- appendPQExpBufferStr(query, "ENABLE ALWAYS");
- break;
- case 'R':
- appendPQExpBufferStr(query, "ENABLE REPLICA");
- break;
- default:
- appendPQExpBufferStr(query, "ENABLE");
- break;
- }
- appendPQExpBufferStr(query, ";\n");
- }
- appendPQExpBuffer(labelq, "EVENT TRIGGER %s ",
- fmtId(evtinfo->dobj.name));
- ArchiveEntry(fout, evtinfo->dobj.catId, evtinfo->dobj.dumpId,
- evtinfo->dobj.name, NULL, NULL, evtinfo->evtowner, false,
- "EVENT TRIGGER", SECTION_POST_DATA,
- query->data, "", NULL, NULL, 0, NULL, NULL);
- dumpComment(fout, labelq->data,
- NULL, NULL,
- evtinfo->dobj.catId, 0, evtinfo->dobj.dumpId);
- destroyPQExpBuffer(query);
- destroyPQExpBuffer(labelq);
- }
- /*
- * dumpRule
- * Dump a rule
- */
- static void
- dumpRule(Archive *fout, RuleInfo *rinfo)
- {
- TableInfo *tbinfo = rinfo->ruletable;
- PQExpBuffer query;
- PQExpBuffer cmd;
- PQExpBuffer delcmd;
- PQExpBuffer labelq;
- PGresult *res;
- /* Skip if not to be dumped */
- if (!rinfo->dobj.dump || dataOnly)
- return;
- /*
- * If it is an ON SELECT rule that is created implicitly by CREATE VIEW,
- * we do not want to dump it as a separate object.
- */
- if (!rinfo->separate)
- return;
- /*
- * Make sure we are in proper schema.
- */
- selectSourceSchema(fout, tbinfo->dobj.namespace->dobj.name);
- query = createPQExpBuffer();
- cmd = createPQExpBuffer();
- delcmd = createPQExpBuffer();
- labelq = createPQExpBuffer();
- if (fout->remoteVersion >= 70300)
- {
- appendPQExpBuffer(query,
- "SELECT pg_catalog.pg_get_ruledef('%u'::pg_catalog.oid) AS definition",
- rinfo->dobj.catId.oid);
- }
- else
- {
- /* Rule name was unique before 7.3 ... */
- appendPQExpBuffer(query,
- "SELECT pg_get_ruledef('%s') AS definition",
- rinfo->dobj.name);
- }
- res = ExecuteSqlQuery(fout, query->data, PGRES_TUPLES_OK);
- if (PQntuples(res) != 1)
- {
- write_msg(NULL, "query to get rule \"%s\" for table \"%s\" failed: wrong number of rows returned\n",
- rinfo->dobj.name, tbinfo->dobj.name);
- exit_nicely(1);
- }
- printfPQExpBuffer(cmd, "%s\n", PQgetvalue(res, 0, 0));
- /*
- * Add the command to alter the rules replication firing semantics if it
- * differs from the default.
- */
- if (rinfo->ev_enabled != 'O')
- {
- appendPQExpBuffer(cmd, "ALTER TABLE %s ", fmtId(tbinfo->dobj.name));
- switch (rinfo->ev_enabled)
- {
- case 'A':
- appendPQExpBuffer(cmd, "ENABLE ALWAYS RULE %s;\n",
- fmtId(rinfo->dobj.name));
- break;
- case 'R':
- appendPQExpBuffer(cmd, "ENABLE REPLICA RULE %s;\n",
- fmtId(rinfo->dobj.name));
- break;
- case 'D':
- appendPQExpBuffer(cmd, "DISABLE RULE %s;\n",
- fmtId(rinfo->dobj.name));
- break;
- }
- }
- /*
- * Apply view's reloptions when its ON SELECT rule is separate.
- */
- if (rinfo->reloptions && strlen(rinfo->reloptions) > 0)
- {
- appendPQExpBuffer(cmd, "ALTER VIEW %s SET (%s);\n",
- fmtId(tbinfo->dobj.name),
- rinfo->reloptions);
- }
- /*
- * DROP must be fully qualified in case same name appears in pg_catalog
- */
- appendPQExpBuffer(delcmd, "DROP RULE %s ",
- fmtId(rinfo->dobj.name));
- appendPQExpBuffer(delcmd, "ON %s.",
- fmtId(tbinfo->dobj.namespace->dobj.name));
- appendPQExpBuffer(delcmd, "%s;\n",
- fmtId(tbinfo->dobj.name));
- appendPQExpBuffer(labelq, "RULE %s",
- fmtId(rinfo->dobj.name));
- appendPQExpBuffer(labelq, " ON %s",
- fmtId(tbinfo->dobj.name));
- ArchiveEntry(fout, rinfo->dobj.catId, rinfo->dobj.dumpId,
- rinfo->dobj.name,
- tbinfo->dobj.namespace->dobj.name,
- NULL,
- tbinfo->rolname, false,
- "RULE", SECTION_POST_DATA,
- cmd->data, delcmd->data, NULL,
- NULL, 0,
- NULL, NULL);
- /* Dump rule comments */
- dumpComment(fout, labelq->data,
- tbinfo->dobj.namespace->dobj.name,
- tbinfo->rolname,
- rinfo->dobj.catId, 0, rinfo->dobj.dumpId);
- PQclear(res);
- destroyPQExpBuffer(query);
- destroyPQExpBuffer(cmd);
- destroyPQExpBuffer(delcmd);
- destroyPQExpBuffer(labelq);
- }
- /*
- * getExtensionMembership --- obtain extension membership data
- */
- void
- getExtensionMembership(Archive *fout, ExtensionInfo extinfo[],
- int numExtensions)
- {
- PQExpBuffer query;
- PGresult *res;
- int ntups,
- i;
- int i_classid,
- i_objid,
- i_refclassid,
- i_refobjid;
- DumpableObject *dobj,
- *refdobj;
- /* Nothing to do if no extensions */
- if (numExtensions == 0)
- return;
- /* Make sure we are in proper schema */
- selectSourceSchema(fout, "pg_catalog");
- query = createPQExpBuffer();
- /* refclassid constraint is redundant but may speed the search */
- appendPQExpBufferStr(query, "SELECT "
- "classid, objid, refclassid, refobjid "
- "FROM pg_depend "
- "WHERE refclassid = 'pg_extension'::regclass "
- "AND deptype = 'e' "
- "ORDER BY 3,4");
- res = ExecuteSqlQuery(fout, query->data, PGRES_TUPLES_OK);
- ntups = PQntuples(res);
- i_classid = PQfnumber(res, "classid");
- i_objid = PQfnumber(res, "objid");
- i_refclassid = PQfnumber(res, "refclassid");
- i_refobjid = PQfnumber(res, "refobjid");
- /*
- * Since we ordered the SELECT by referenced ID, we can expect that
- * multiple entries for the same extension will appear together; this
- * saves on searches.
- */
- refdobj = NULL;
- for (i = 0; i < ntups; i++)
- {
- CatalogId objId;
- CatalogId refobjId;
- objId.tableoid = atooid(PQgetvalue(res, i, i_classid));
- objId.oid = atooid(PQgetvalue(res, i, i_objid));
- refobjId.tableoid = atooid(PQgetvalue(res, i, i_refclassid));
- refobjId.oid = atooid(PQgetvalue(res, i, i_refobjid));
- if (refdobj == NULL ||
- refdobj->catId.tableoid != refobjId.tableoid ||
- refdobj->catId.oid != refobjId.oid)
- refdobj = findObjectByCatalogId(refobjId);
- /*
- * Failure to find objects mentioned in pg_depend is not unexpected,
- * since for example we don't collect info about TOAST tables.
- */
- if (refdobj == NULL)
- {
- #ifdef NOT_USED
- fprintf(stderr, "no referenced object %u %u\n",
- refobjId.tableoid, refobjId.oid);
- #endif
- continue;
- }
- dobj = findObjectByCatalogId(objId);
- if (dobj == NULL)
- {
- #ifdef NOT_USED
- fprintf(stderr, "no referencing object %u %u\n",
- objId.tableoid, objId.oid);
- #endif
- continue;
- }
- /* Record dependency so that getDependencies needn't repeat this */
- addObjectDependency(dobj, refdobj->dumpId);
- dobj->ext_member = true;
- /*
- * Normally, mark the member object as not to be dumped. But in
- * binary upgrades, we still dump the members individually, since the
- * idea is to exactly reproduce the database contents rather than
- * replace the extension contents with something different.
- */
- if (!binary_upgrade)
- dobj->dump = false;
- else
- dobj->dump = refdobj->dump;
- }
- PQclear(res);
- /*
- * Now identify extension configuration tables and create TableDataInfo
- * objects for them, ensuring their data will be dumped even though the
- * tables themselves won't be.
- *
- * Note that we create TableDataInfo objects even in schemaOnly mode, ie,
- * user data in a configuration table is treated like schema data. This
- * seems appropriate since system data in a config table would get
- * reloaded by CREATE EXTENSION.
- */
- for (i = 0; i < numExtensions; i++)
- {
- ExtensionInfo *curext = &(extinfo[i]);
- char *extconfig = curext->extconfig;
- char *extcondition = curext->extcondition;
- char **extconfigarray = NULL;
- char **extconditionarray = NULL;
- int nconfigitems;
- int nconditionitems;
- if (parsePGArray(extconfig, &extconfigarray, &nconfigitems) &&
- parsePGArray(extcondition, &extconditionarray, &nconditionitems) &&
- nconfigitems == nconditionitems)
- {
- int j;
- for (j = 0; j < nconfigitems; j++)
- {
- TableInfo *configtbl;
- Oid configtbloid = atooid(extconfigarray[j]);
- bool dumpobj = curext->dobj.dump;
- configtbl = findTableByOid(configtbloid);
- if (configtbl == NULL)
- continue;
- /*
- * Tables of not-to-be-dumped extensions shouldn't be dumped
- * unless the table or its schema is explicitly included
- */
- if (!curext->dobj.dump)
- {
- /* check table explicitly requested */
- if (table_include_oids.head != NULL &&
- simple_oid_list_member(&table_include_oids,
- configtbloid))
- dumpobj = true;
- /* check table's schema explicitly requested */
- if (configtbl->dobj.namespace->dobj.dump)
- dumpobj = true;
- }
- /* check table excluded by an exclusion switch */
- if (table_exclude_oids.head != NULL &&
- simple_oid_list_member(&table_exclude_oids,
- configtbloid))
- dumpobj = false;
- /* check schema excluded by an exclusion switch */
- if (simple_oid_list_member(&schema_exclude_oids,
- configtbl->dobj.namespace->dobj.catId.oid))
- dumpobj = false;
- if (dumpobj)
- {
- /*
- * Note: config tables are dumped without OIDs regardless
- * of the --oids setting. This is because row filtering
- * conditions aren't compatible with dumping OIDs.
- */
- makeTableDataInfo(configtbl, false);
- if (configtbl->dataObj != NULL)
- {
- if (strlen(extconditionarray[j]) > 0)
- configtbl->dataObj->filtercond = pg_strdup(extconditionarray[j]);
- }
- }
- }
- }
- if (extconfigarray)
- free(extconfigarray);
- if (extconditionarray)
- free(extconditionarray);
- }
- destroyPQExpBuffer(query);
- }
- /*
- * getDependencies --- obtain available dependency data
- */
- static void
- getDependencies(Archive *fout)
- {
- PQExpBuffer query;
- PGresult *res;
- int ntups,
- i;
- int i_classid,
- i_objid,
- i_refclassid,
- i_refobjid,
- i_deptype;
- DumpableObject *dobj,
- *refdobj;
- /* No dependency info available before 7.3 */
- if (fout->remoteVersion < 70300)
- return;
- if (g_verbose)
- write_msg(NULL, "reading dependency data\n");
- /* Make sure we are in proper schema */
- selectSourceSchema(fout, "pg_catalog");
- query = createPQExpBuffer();
- /*
- * PIN dependencies aren't interesting, and EXTENSION dependencies were
- * already processed by getExtensionMembership.
- */
- appendPQExpBufferStr(query, "SELECT "
- "classid, objid, refclassid, refobjid, deptype "
- "FROM pg_depend "
- "WHERE deptype != 'p' AND deptype != 'e' "
- "ORDER BY 1,2");
- res = ExecuteSqlQuery(fout, query->data, PGRES_TUPLES_OK);
- ntups = PQntuples(res);
- i_classid = PQfnumber(res, "classid");
- i_objid = PQfnumber(res, "objid");
- i_refclassid = PQfnumber(res, "refclassid");
- i_refobjid = PQfnumber(res, "refobjid");
- i_deptype = PQfnumber(res, "deptype");
- /*
- * Since we ordered the SELECT by referencing ID, we can expect that
- * multiple entries for the same object will appear together; this saves
- * on searches.
- */
- dobj = NULL;
- for (i = 0; i < ntups; i++)
- {
- CatalogId objId;
- CatalogId refobjId;
- char deptype;
- objId.tableoid = atooid(PQgetvalue(res, i, i_classid));
- objId.oid = atooid(PQgetvalue(res, i, i_objid));
- refobjId.tableoid = atooid(PQgetvalue(res, i, i_refclassid));
- refobjId.oid = atooid(PQgetvalue(res, i, i_refobjid));
- deptype = *(PQgetvalue(res, i, i_deptype));
- if (dobj == NULL ||
- dobj->catId.tableoid != objId.tableoid ||
- dobj->catId.oid != objId.oid)
- dobj = findObjectByCatalogId(objId);
- /*
- * Failure to find objects mentioned in pg_depend is not unexpected,
- * since for example we don't collect info about TOAST tables.
- */
- if (dobj == NULL)
- {
- #ifdef NOT_USED
- fprintf(stderr, "no referencing object %u %u\n",
- objId.tableoid, objId.oid);
- #endif
- continue;
- }
- refdobj = findObjectByCatalogId(refobjId);
- if (refdobj == NULL)
- {
- #ifdef NOT_USED
- fprintf(stderr, "no referenced object %u %u\n",
- refobjId.tableoid, refobjId.oid);
- #endif
- continue;
- }
- /*
- * Ordinarily, table rowtypes have implicit dependencies on their
- * tables. However, for a composite type the implicit dependency goes
- * the other way in pg_depend; which is the right thing for DROP but
- * it doesn't produce the dependency ordering we need. So in that one
- * case, we reverse the direction of the dependency.
- */
- if (deptype == 'i' &&
- dobj->objType == DO_TABLE &&
- refdobj->objType == DO_TYPE)
- addObjectDependency(refdobj, dobj->dumpId);
- else
- /* normal case */
- addObjectDependency(dobj, refdobj->dumpId);
- }
- PQclear(res);
- destroyPQExpBuffer(query);
- }
- /*
- * createBoundaryObjects - create dummy DumpableObjects to represent
- * dump section boundaries.
- */
- static DumpableObject *
- createBoundaryObjects(void)
- {
- DumpableObject *dobjs;
- dobjs = (DumpableObject *) pg_malloc(2 * sizeof(DumpableObject));
- dobjs[0].objType = DO_PRE_DATA_BOUNDARY;
- dobjs[0].catId = nilCatalogId;
- AssignDumpId(dobjs + 0);
- dobjs[0].name = pg_strdup("PRE-DATA BOUNDARY");
- dobjs[1].objType = DO_POST_DATA_BOUNDARY;
- dobjs[1].catId = nilCatalogId;
- AssignDumpId(dobjs + 1);
- dobjs[1].name = pg_strdup("POST-DATA BOUNDARY");
- return dobjs;
- }
- /*
- * addBoundaryDependencies - add dependencies as needed to enforce the dump
- * section boundaries.
- */
- static void
- addBoundaryDependencies(DumpableObject **dobjs, int numObjs,
- DumpableObject *boundaryObjs)
- {
- DumpableObject *preDataBound = boundaryObjs + 0;
- DumpableObject *postDataBound = boundaryObjs + 1;
- int i;
- for (i = 0; i < numObjs; i++)
- {
- DumpableObject *dobj = dobjs[i];
- /*
- * The classification of object types here must match the SECTION_xxx
- * values assigned during subsequent ArchiveEntry calls!
- */
- switch (dobj->objType)
- {
- case DO_NAMESPACE:
- case DO_EXTENSION:
- case DO_TYPE:
- case DO_SHELL_TYPE:
- case DO_FUNC:
- case DO_AGG:
- case DO_OPERATOR:
- case DO_OPCLASS:
- case DO_OPFAMILY:
- case DO_COLLATION:
- case DO_CONVERSION:
- case DO_TABLE:
- case DO_ATTRDEF:
- case DO_PROCLANG:
- case DO_CAST:
- case DO_DUMMY_TYPE:
- case DO_TSPARSER:
- case DO_TSDICT:
- case DO_TSTEMPLATE:
- case DO_TSCONFIG:
- case DO_FDW:
- case DO_FOREIGN_SERVER:
- case DO_BLOB:
- /* Pre-data objects: must come before the pre-data boundary */
- addObjectDependency(preDataBound, dobj->dumpId);
- break;
- case DO_TABLE_DATA:
- case DO_BLOB_DATA:
- /* Data objects: must come between the boundaries */
- addObjectDependency(dobj, preDataBound->dumpId);
- addObjectDependency(postDataBound, dobj->dumpId);
- break;
- case DO_INDEX:
- case DO_REFRESH_MATVIEW:
- case DO_TRIGGER:
- case DO_EVENT_TRIGGER:
- case DO_DEFAULT_ACL:
- /* Post-data objects: must come after the post-data boundary */
- addObjectDependency(dobj, postDataBound->dumpId);
- break;
- case DO_RULE:
- /* Rules are post-data, but only if dumped separately */
- if (((RuleInfo *) dobj)->separate)
- addObjectDependency(dobj, postDataBound->dumpId);
- break;
- case DO_CONSTRAINT:
- case DO_FK_CONSTRAINT:
- /* Constraints are post-data, but only if dumped separately */
- if (((ConstraintInfo *) dobj)->separate)
- addObjectDependency(dobj, postDataBound->dumpId);
- break;
- case DO_PRE_DATA_BOUNDARY:
- /* nothing to do */
- break;
- case DO_POST_DATA_BOUNDARY:
- /* must come after the pre-data boundary */
- addObjectDependency(dobj, preDataBound->dumpId);
- break;
- }
- }
- }
- /*
- * BuildArchiveDependencies - create dependency data for archive TOC entries
- *
- * The raw dependency data obtained by getDependencies() is not terribly
- * useful in an archive dump, because in many cases there are dependency
- * chains linking through objects that don't appear explicitly in the dump.
- * For example, a view will depend on its _RETURN rule while the _RETURN rule
- * will depend on other objects --- but the rule will not appear as a separate
- * object in the dump. We need to adjust the view's dependencies to include
- * whatever the rule depends on that is included in the dump.
- *
- * Just to make things more complicated, there are also "special" dependencies
- * such as the dependency of a TABLE DATA item on its TABLE, which we must
- * not rearrange because pg_restore knows that TABLE DATA only depends on
- * its table. In these cases we must leave the dependencies strictly as-is
- * even if they refer to not-to-be-dumped objects.
- *
- * To handle this, the convention is that "special" dependencies are created
- * during ArchiveEntry calls, and an archive TOC item that has any such
- * entries will not be touched here. Otherwise, we recursively search the
- * DumpableObject data structures to build the correct dependencies for each
- * archive TOC item.
- */
- static void
- BuildArchiveDependencies(Archive *fout)
- {
- ArchiveHandle *AH = (ArchiveHandle *) fout;
- TocEntry *te;
- /* Scan all TOC entries in the archive */
- for (te = AH->toc->next; te != AH->toc; te = te->next)
- {
- DumpableObject *dobj;
- DumpId *dependencies;
- int nDeps;
- int allocDeps;
- /* No need to process entries that will not be dumped */
- if (te->reqs == 0)
- continue;
- /* Ignore entries that already have "special" dependencies */
- if (te->nDeps > 0)
- continue;
- /* Otherwise, look up the item's original DumpableObject, if any */
- dobj = findObjectByDumpId(te->dumpId);
- if (dobj == NULL)
- continue;
- /* No work if it has no dependencies */
- if (dobj->nDeps <= 0)
- continue;
- /* Set up work array */
- allocDeps = 64;
- dependencies = (DumpId *) pg_malloc(allocDeps * sizeof(DumpId));
- nDeps = 0;
- /* Recursively find all dumpable dependencies */
- findDumpableDependencies(AH, dobj,
- &dependencies, &nDeps, &allocDeps);
- /* And save 'em ... */
- if (nDeps > 0)
- {
- dependencies = (DumpId *) pg_realloc(dependencies,
- nDeps * sizeof(DumpId));
- te->dependencies = dependencies;
- te->nDeps = nDeps;
- }
- else
- free(dependencies);
- }
- }
- /* Recursive search subroutine for BuildArchiveDependencies */
- static void
- findDumpableDependencies(ArchiveHandle *AH, DumpableObject *dobj,
- DumpId **dependencies, int *nDeps, int *allocDeps)
- {
- int i;
- /*
- * Ignore section boundary objects: if we search through them, we'll
- * report lots of bogus dependencies.
- */
- if (dobj->objType == DO_PRE_DATA_BOUNDARY ||
- dobj->objType == DO_POST_DATA_BOUNDARY)
- return;
- for (i = 0; i < dobj->nDeps; i++)
- {
- DumpId depid = dobj->dependencies[i];
- if (TocIDRequired(AH, depid) != 0)
- {
- /* Object will be dumped, so just reference it as a dependency */
- if (*nDeps >= *allocDeps)
- {
- *allocDeps *= 2;
- *dependencies = (DumpId *) pg_realloc(*dependencies,
- *allocDeps * sizeof(DumpId));
- }
- (*dependencies)[*nDeps] = depid;
- (*nDeps)++;
- }
- else
- {
- /*
- * Object will not be dumped, so recursively consider its deps. We
- * rely on the assumption that sortDumpableObjects already broke
- * any dependency loops, else we might recurse infinitely.
- */
- DumpableObject *otherdobj = findObjectByDumpId(depid);
- if (otherdobj)
- findDumpableDependencies(AH, otherdobj,
- dependencies, nDeps, allocDeps);
- }
- }
- }
- /*
- * selectSourceSchema - make the specified schema the active search path
- * in the source database.
- *
- * NB: pg_catalog is explicitly searched after the specified schema;
- * so user names are only qualified if they are cross-schema references,
- * and system names are only qualified if they conflict with a user name
- * in the current schema.
- *
- * Whenever the selected schema is not pg_catalog, be careful to qualify
- * references to system catalogs and types in our emitted commands!
- *
- * This function is called only from selectSourceSchemaOnAH and
- * selectSourceSchema.
- */
- static void
- selectSourceSchema(Archive *fout, const char *schemaName)
- {
- PQExpBuffer query;
- /* This is checked by the callers already */
- Assert(schemaName != NULL && *schemaName != '\0');
- /* Not relevant if fetching from pre-7.3 DB */
- if (fout->remoteVersion < 70300)
- return;
- query = createPQExpBuffer();
- appendPQExpBuffer(query, "SET search_path = %s",
- fmtId(schemaName));
- if (strcmp(schemaName, "pg_catalog") != 0)
- appendPQExpBufferStr(query, ", pg_catalog");
- ExecuteSqlStatement(fout, query->data);
- destroyPQExpBuffer(query);
- }
- /*
- * getFormattedTypeName - retrieve a nicely-formatted type name for the
- * given type name.
- *
- * NB: in 7.3 and up the result may depend on the currently-selected
- * schema; this is why we don't try to cache the names.
- */
- static char *
- getFormattedTypeName(Archive *fout, Oid oid, OidOptions opts)
- {
- char *result;
- PQExpBuffer query;
- PGresult *res;
- if (oid == 0)
- {
- if ((opts & zeroAsOpaque) != 0)
- return pg_strdup(g_opaque_type);
- else if ((opts & zeroAsAny) != 0)
- return pg_strdup("'any'");
- else if ((opts & zeroAsStar) != 0)
- return pg_strdup("*");
- else if ((opts & zeroAsNone) != 0)
- return pg_strdup("NONE");
- }
- query = createPQExpBuffer();
- if (fout->remoteVersion >= 70300)
- {
- appendPQExpBuffer(query, "SELECT pg_catalog.format_type('%u'::pg_catalog.oid, NULL)",
- oid);
- }
- else if (fout->remoteVersion >= 70100)
- {
- appendPQExpBuffer(query, "SELECT format_type('%u'::oid, NULL)",
- oid);
- }
- else
- {
- appendPQExpBuffer(query, "SELECT typname "
- "FROM pg_type "
- "WHERE oid = '%u'::oid",
- oid);
- }
- res = ExecuteSqlQueryForSingleRow(fout, query->data);
- if (fout->remoteVersion >= 70100)
- {
- /* already quoted */
- result = pg_strdup(PQgetvalue(res, 0, 0));
- }
- else
- {
- /* may need to quote it */
- result = pg_strdup(fmtId(PQgetvalue(res, 0, 0)));
- }
- PQclear(res);
- destroyPQExpBuffer(query);
- return result;
- }
- /*
- * myFormatType --- local implementation of format_type for use with 7.0.
- */
- static char *
- myFormatType(const char *typname, int32 typmod)
- {
- char *result;
- bool isarray = false;
- PQExpBuffer buf = createPQExpBuffer();
- /* Handle array types */
- if (typname[0] == '_')
- {
- isarray = true;
- typname++;
- }
- /* Show lengths on bpchar and varchar */
- if (strcmp(typname, "bpchar") == 0)
- {
- int len = (typmod - VARHDRSZ);
- appendPQExpBufferStr(buf, "character");
- if (len > 1)
- appendPQExpBuffer(buf, "(%d)",
- typmod - VARHDRSZ);
- }
- else if (strcmp(typname, "varchar") == 0)
- {
- appendPQExpBufferStr(buf, "character varying");
- if (typmod != -1)
- appendPQExpBuffer(buf, "(%d)",
- typmod - VARHDRSZ);
- }
- else if (strcmp(typname, "numeric") == 0)
- {
- appendPQExpBufferStr(buf, "numeric");
- if (typmod != -1)
- {
- int32 tmp_typmod;
- int precision;
- int scale;
- tmp_typmod = typmod - VARHDRSZ;
- precision = (tmp_typmod >> 16) & 0xffff;
- scale = tmp_typmod & 0xffff;
- appendPQExpBuffer(buf, "(%d,%d)",
- precision, scale);
- }
- }
- /*
- * char is an internal single-byte data type; Let's make sure we force it
- * through with quotes. - thomas 1998-12-13
- */
- else if (strcmp(typname, "char") == 0)
- appendPQExpBufferStr(buf, "\"char\"");
- else
- appendPQExpBufferStr(buf, fmtId(typname));
- /* Append array qualifier for array types */
- if (isarray)
- appendPQExpBufferStr(buf, "[]");
- result = pg_strdup(buf->data);
- destroyPQExpBuffer(buf);
- return result;
- }
- /*
- * Return a column list clause for the given relation.
- *
- * Special case: if there are no undropped columns in the relation, return
- * "", not an invalid "()" column list.
- */
- static const char *
- fmtCopyColumnList(const TableInfo *ti, PQExpBuffer buffer)
- {
- int numatts = ti->numatts;
- char **attnames = ti->attnames;
- bool *attisdropped = ti->attisdropped;
- bool needComma;
- int i;
- appendPQExpBufferChar(buffer, '(');
- needComma = false;
- for (i = 0; i < numatts; i++)
- {
- if (attisdropped[i])
- continue;
- if (needComma)
- appendPQExpBufferStr(buffer, ", ");
- appendPQExpBufferStr(buffer, fmtId(attnames[i]));
- needComma = true;
- }
- if (!needComma)
- return ""; /* no undropped columns */
- appendPQExpBufferChar(buffer, ')');
- return buffer->data;
- }
- /*
- * Execute an SQL query and verify that we got exactly one row back.
- */
- static PGresult *
- ExecuteSqlQueryForSingleRow(Archive *fout, char *query)
- {
- PGresult *res;
- int ntups;
- res = ExecuteSqlQuery(fout, query, PGRES_TUPLES_OK);
- /* Expecting a single result only */
- ntups = PQntuples(res);
- if (ntups != 1)
- exit_horribly(NULL,
- ngettext("query returned %d row instead of one: %s\n",
- "query returned %d rows instead of one: %s\n",
- ntups),
- ntups, query);
- return res;
- }