common.c File Reference

#include "postgres_fe.h"
#include <signal.h>
#include <unistd.h>
#include "common.h"
#include "common/connect.h"
#include "common/logging.h"
#include "common/string.h"
#include "fe_utils/cancel.h"
#include "fe_utils/query_utils.h"
#include "fe_utils/string_utils.h"

Include dependency graph for common.c:

Go to the source code of this file.

Macros
#define	PG_YESLETTER   gettext_noop("y")
#define	PG_NOLETTER   gettext_noop("n")

Functions
void	splitTableColumnsSpec (const char *spec, int encoding, char **table, const char **columns)
void	appendQualifiedRelation (PQExpBuffer buf, const char *spec, PGconn *conn, bool echo)
bool	yesno_prompt (const char *question)

Macro Definition Documentation

PG_NOLETTER

#define PG_NOLETTER   gettext_noop("n")

Definition at line 133 of file common.c.

Referenced by yesno_prompt().

PG_YESLETTER

#define PG_YESLETTER   gettext_noop("y")

Definition at line 131 of file common.c.

Referenced by yesno_prompt().

Function Documentation

appendQualifiedRelation()

void appendQualifiedRelation	(	PQExpBuffer	buf,
		const char *	spec,
		PGconn *	conn,
		bool	echo
	)

Definition at line 69 of file common.c.

References ALWAYS_SECURE_SEARCH_PATH_SQL, appendPQExpBufferStr(), appendStringLiteralConn(), PQExpBufferData::data, executeCommand(), executeQuery(), fmtQualifiedId(), initPQExpBuffer(), ngettext, pg_free(), pg_log_error, PQclear(), PQclientEncoding(), PQfinish(), PQgetvalue(), PQntuples(), splitTableColumnsSpec(), and termPQExpBuffer().

Referenced by cluster_one_database(), and run_reindex_command().

{
    char       *table;
    const char *columns;
    PQExpBufferData sql;
    PGresult   *res;
    int         ntups;

    splitTableColumnsSpec(spec, PQclientEncoding(conn), &table, &columns);

    /*
     * Query must remain ABSOLUTELY devoid of unqualified names.  This would
     * be unnecessary given a regclassin() variant taking a search_path
     * argument.
     */
    initPQExpBuffer(&sql);
    appendPQExpBufferStr(&sql,
                         "SELECT c.relname, ns.nspname\n"
                         " FROM pg_catalog.pg_class c,"
                         " pg_catalog.pg_namespace ns\n"
                         " WHERE c.relnamespace OPERATOR(pg_catalog.=) ns.oid\n"
                         "  AND c.oid OPERATOR(pg_catalog.=) ");
    appendStringLiteralConn(&sql, table, conn);
    appendPQExpBufferStr(&sql, "::pg_catalog.regclass;");

    executeCommand(conn, "RESET search_path;", echo);

    /*
     * One row is a typical result, as is a nonexistent relation ERROR.
     * regclassin() unconditionally accepts all-digits input as an OID; if no
     * relation has that OID; this query returns no rows.  Catalog corruption
     * might elicit other row counts.
     */
    res = executeQuery(conn, sql.data, echo);
    ntups = PQntuples(res);
    if (ntups != 1)
    {
        pg_log_error(ngettext("query returned %d row instead of one: %s",
                              "query returned %d rows instead of one: %s",
                              ntups),
                     ntups, sql.data);
        PQfinish(conn);
        exit(1);
    }
    appendPQExpBufferStr(buf,
                         fmtQualifiedId(PQgetvalue(res, 0, 1),
                                        PQgetvalue(res, 0, 0)));
    appendPQExpBufferStr(buf, columns);
    PQclear(res);
    termPQExpBuffer(&sql);
    pg_free(table);

    PQclear(executeQuery(conn, ALWAYS_SECURE_SEARCH_PATH_SQL, echo));
}

splitTableColumnsSpec()

void splitTableColumnsSpec	(	const char *	spec,
		int	encoding,
		char **	table,
		const char **	columns
	)

Definition at line 34 of file common.c.

References pnstrdup(), and PQmblenBounded().

Referenced by appendQualifiedRelation(), and vacuum_one_database().

{
    bool        inquotes = false;
    const char *cp = spec;

    /*
     * Find the first '(' not identifier-quoted.  Based on
     * dequote_downcase_identifier().
     */
    while (*cp && (*cp != '(' || inquotes))
    {
        if (*cp == '"')
        {
            if (inquotes && cp[1] == '"')
                cp++;           /* pair does not affect quoting */
            else
                inquotes = !inquotes;
            cp++;
        }
        else
            cp += PQmblenBounded(cp, encoding);
    }
    *table = pnstrdup(spec, cp - spec);
    *columns = cp;
}

yesno_prompt()

bool yesno_prompt

(

const char *

question

)

Definition at line 136 of file common.c.

References _, free, PG_NOLETTER, PG_YESLETTER, printf, simple_prompt(), and snprintf.

Referenced by main().

{
    char        prompt[256];

    /*------
       translator: This is a question followed by the translated options for
       "yes" and "no". */
    snprintf(prompt, sizeof(prompt), _("%s (%s/%s) "),
             _(question), _(PG_YESLETTER), _(PG_NOLETTER));

    for (;;)
    {
        char       *resp;

        resp = simple_prompt(prompt, true);

        if (strcmp(resp, _(PG_YESLETTER)) == 0)
        {
            free(resp);
            return true;
        }
        if (strcmp(resp, _(PG_NOLETTER)) == 0)
        {
            free(resp);
            return false;
        }
        free(resp);

        printf(_("Please answer \"%s\" or \"%s\".\n"),
               _(PG_YESLETTER), _(PG_NOLETTER));
    }
}