#include "pg_backup.h"

Include dependency graph for pg_backup_db.h:

This graph shows which files directly or indirectly include this file:

Functions
int	ExecuteSqlCommandBuf (Archive AHX, const char buf, size_t bufLen)

void	ExecuteSqlStatement (Archive AHX, const char query)

PGresult *	ExecuteSqlQuery (Archive AHX, const char query, ExecStatusType status)

PGresult *	ExecuteSqlQueryForSingleRow (Archive fout, const char query)

void	EndDBCopyMode (Archive AHX, const char tocEntryTag)

void	StartTransaction (Archive *AHX)

void	CommitTransaction (Archive *AHX)

Function Documentation

◆ CommitTransaction()

void CommitTransaction ( Archive * AHX )

Definition at line 535 of file pg_backup_db.c.

 {
     ArchiveHandle *AH = (ArchiveHandle *) AHX;
  
     ExecuteSqlCommand(AH, "COMMIT", "could not commit database transaction");
 }

References ExecuteSqlCommand().

Referenced by IssueCommandPerBlob().

◆ EndDBCopyMode()

void EndDBCopyMode	(	Archive *	AHX,
		const char *	tocEntryTag
	)

Definition at line 498 of file pg_backup_db.c.

 {
     ArchiveHandle *AH = (ArchiveHandle *) AHX;
  
     if (AH->pgCopyIn)
     {
         PGresult   *res;
  
         if (PQputCopyEnd(AH->connection, NULL) <= 0)
             pg_fatal("error returned by PQputCopyEnd: %s",
                      PQerrorMessage(AH->connection));
  
         /* Check command status and return to normal libpq state */
         res = PQgetResult(AH->connection);
         if (PQresultStatus(res) != PGRES_COMMAND_OK)
             warn_or_exit_horribly(AH, "COPY failed for table \"%s\": %s",
                                   tocEntryTag, PQerrorMessage(AH->connection));
         PQclear(res);
  
         /* Do this to ensure we've pumped libpq back to idle state */
         if (PQgetResult(AH->connection) != NULL)
             pg_log_warning("unexpected extra results during COPY of table \"%s\"",
                            tocEntryTag);
  
         AH->pgCopyIn = false;
     }
 }

References _archiveHandle::connection, pg_fatal, pg_log_warning, _archiveHandle::pgCopyIn, PGRES_COMMAND_OK, PQclear(), PQerrorMessage(), PQgetResult(), PQputCopyEnd(), PQresultStatus(), res, and warn_or_exit_horribly().

Referenced by restore_toc_entry().

◆ ExecuteSqlCommandBuf()

int ExecuteSqlCommandBuf	(	Archive *	AHX,
		const char *	buf,
		size_t	bufLen
	)

Definition at line 443 of file pg_backup_db.c.

 {
     ArchiveHandle *AH = (ArchiveHandle *) AHX;
  
     if (AH->outputKind == OUTPUT_COPYDATA)
     {
         /*
          * COPY data.
          *
          * We drop the data on the floor if libpq has failed to enter COPY
          * mode; this allows us to behave reasonably when trying to continue
          * after an error in a COPY command.
          */
         if (AH->pgCopyIn &&
             PQputCopyData(AH->connection, buf, bufLen) <= 0)
             pg_fatal("error returned by PQputCopyData: %s",
                      PQerrorMessage(AH->connection));
     }
     else if (AH->outputKind == OUTPUT_OTHERDATA)
     {
         /*
          * Table data expressed as INSERT commands; or, in old dump files,
          * BLOB COMMENTS data (which is expressed as COMMENT ON commands).
          */
         ExecuteSimpleCommands(AH, buf, bufLen);
     }
     else
     {
         /*
          * General SQL commands; we assume that commands will not be split
          * across calls.
          *
          * In most cases the data passed to us will be a null-terminated
          * string, but if it's not, we have to add a trailing null.
          */
         if (buf[bufLen] == '\0')
             ExecuteSqlCommand(AH, buf, "could not execute query");
         else
         {
             char       *str = (char *) pg_malloc(bufLen + 1);
  
             memcpy(str, buf, bufLen);
             str[bufLen] = '\0';
             ExecuteSqlCommand(AH, str, "could not execute query");
             free(str);
         }
     }
  
     return bufLen;
 }

References buf, _archiveHandle::connection, ExecuteSimpleCommands(), ExecuteSqlCommand(), free, OUTPUT_COPYDATA, OUTPUT_OTHERDATA, _archiveHandle::outputKind, pg_fatal, pg_malloc(), _archiveHandle::pgCopyIn, PQerrorMessage(), PQputCopyData(), and str.

Referenced by ahwrite().

◆ ExecuteSqlQuery()

PGresult* ExecuteSqlQuery	(	Archive *	AHX,
		const char *	query,
		ExecStatusType	status
	)

Definition at line 288 of file pg_backup_db.c.

 {
     ArchiveHandle *AH = (ArchiveHandle *) AHX;
     PGresult   *res;
  
     res = PQexec(AH->connection, query);
     if (PQresultStatus(res) != status)
         die_on_query_failure(AH, query);
     return res;
 }

References _archiveHandle::connection, die_on_query_failure(), PQexec(), PQresultStatus(), and res.

Referenced by append_depends_on_extension(), buildMatViewRefreshDependencies(), collectBinaryUpgradeClassOids(), collectComments(), collectRoleNames(), collectSecLabels(), collectSequences(), createViewAsClause(), dumpCompositeType(), dumpDatabase(), dumpDatabaseConfig(), dumpEnumType(), dumpOpclass(), dumpOpfamily(), dumpRule(), dumpSequence(), dumpSequenceData(), dumpTable(), dumpTableData_copy(), dumpTableData_insert(), dumpTSConfig(), dumpUserMappings(), ExecuteSqlQueryForSingleRow(), expand_extension_name_patterns(), expand_foreign_server_name_patterns(), expand_schema_name_patterns(), expand_table_name_patterns(), getAccessMethods(), getAdditionalACLs(), getAggregates(), getCasts(), getCollations(), getConstraints(), getConversions(), getDefaultACLs(), getDependencies(), getDomainConstraints(), getEventTriggers(), getExtendedStatistics(), getExtensionMembership(), getExtensions(), getForeignDataWrappers(), getForeignServers(), getFuncs(), getIndexes(), getInherits(), getLOs(), getNamespaces(), getOpclasses(), getOperators(), getOpfamilies(), getPartitioningInfo(), getPolicies(), getProcLangs(), getPublicationNamespaces(), getPublications(), getPublicationTables(), getRules(), getSubscriptions(), getSubscriptionTables(), getTableAttrs(), getTables(), getTransforms(), getTriggers(), getTSConfigurations(), getTSDictionaries(), getTSParsers(), getTSTemplates(), getTypes(), processExtensionTables(), and set_restrict_relation_kind().

◆ ExecuteSqlQueryForSingleRow()

PGresult* ExecuteSqlQueryForSingleRow	(	Archive *	fout,
		const char *	query
	)

Definition at line 303 of file pg_backup_db.c.

 {
     PGresult   *res;
     int         ntups;
  
     res = ExecuteSqlQuery(fout, query, PGRES_TUPLES_OK);
  
     /* Expecting a single result only */
     ntups = PQntuples(res);
     if (ntups != 1)
         pg_fatal(ngettext("query returned %d row instead of one: %s",
                           "query returned %d rows instead of one: %s",
                           ntups),
                  ntups, query);
  
     return res;
 }

References ExecuteSqlQuery(), ngettext, pg_fatal, PGRES_TUPLES_OK, PQntuples(), and res.

Referenced by _check_database_version(), binary_upgrade_set_type_oids_by_type_oid(), ConnectDatabase(), convertTSFunction(), dumpAgg(), dumpBaseType(), dumpCollation(), dumpConversion(), dumpDatabase(), dumpDomain(), dumpForeignServer(), dumpFunc(), dumpOpclass(), dumpOpfamily(), dumpOpr(), dumpRangeType(), dumpSearchPath(), dumpStatisticsExt(), dumpTableAttach(), dumpTableSchema(), dumpTSConfig(), dumpTSDictionary(), expand_table_name_patterns(), get_language_name(), get_next_possible_free_pg_type_oid(), get_synchronized_snapshot(), getFormattedTypeName(), and setup_connection().

◆ ExecuteSqlStatement()

void ExecuteSqlStatement	(	Archive *	AHX,
		const char *	query
	)

Definition at line 276 of file pg_backup_db.c.

 {
     ArchiveHandle *AH = (ArchiveHandle *) AHX;
     PGresult   *res;
  
     res = PQexec(AH->connection, query);
     if (PQresultStatus(res) != PGRES_COMMAND_OK)
         die_on_query_failure(AH, query);
     PQclear(res);
 }

References _archiveHandle::connection, die_on_query_failure(), PGRES_COMMAND_OK, PQclear(), PQexec(), PQresultStatus(), and res.

Referenced by dumpAgg(), dumpBaseType(), dumpCompositeType(), dumpDomain(), dumpEnumType(), dumpFunc(), dumpOpr(), dumpRangeType(), dumpTable(), dumpTableAttach(), dumpTableData_insert(), expand_table_name_patterns(), getDomainConstraints(), getTables(), and setup_connection().

◆ StartTransaction()

void StartTransaction ( Archive * AHX )

Definition at line 527 of file pg_backup_db.c.

 {
     ArchiveHandle *AH = (ArchiveHandle *) AHX;
  
     ExecuteSqlCommand(AH, "BEGIN", "could not start database transaction");
 }