*
*
* IDENTIFICATION
- * $Header: /cvsroot/pgsql/src/bin/pg_dump/pg_dump.c,v 1.268 2002/07/02 05:49:51 momjian Exp $
+ * $Header: /cvsroot/pgsql/src/bin/pg_dump/pg_dump.c,v 1.269 2002/07/04 03:04:54 momjian Exp $
*
*-------------------------------------------------------------------------
*/
static void dumpFuncACL(Archive *fout, FuncInfo *finfo);
static void dumpAggACL(Archive *fout, AggInfo *finfo);
static void dumpACL(Archive *fout, const char *type, const char *name,
- const char *nspname, const char *usename,
- const char *acl, const char *objoid);
+ const char *name_noquotes, const char *nspname,
+ const char *usename, const char *acl, const char *objoid);
static void dumpTriggers(Archive *fout, TableInfo *tblinfo, int numTables);
static void dumpRules(Archive *fout, TableInfo *tblinfo, int numTables);
static void formatStringLiteral(PQExpBuffer buf, const char *str,
const formatLiteralOptions opts);
-static char *format_function_signature(FuncInfo *finfo);
+static char *format_function_signature(FuncInfo *finfo, bool honor_quotes);
static void dumpOneFunc(Archive *fout, FuncInfo *finfo);
static void dumpOneOpr(Archive *fout, OprInfo *oprinfo,
OprInfo *g_oprinfo, int numOperators);
if (strcmp(nspinfo->nspname, "public") == 0)
{
if (!aclsSkip && strcmp(nspinfo->nspacl, "{=UC}") != 0)
- dumpACL(fout, "SCHEMA", qnspname, NULL,
+ dumpACL(fout, "SCHEMA", qnspname, nspinfo->nspname, NULL,
nspinfo->usename, nspinfo->nspacl,
nspinfo->oid);
}
nspinfo->oid, "pg_namespace", 0, NULL);
if (!aclsSkip)
- dumpACL(fout, "SCHEMA", qnspname, NULL,
+ dumpACL(fout, "SCHEMA", qnspname, nspinfo->nspname, NULL,
nspinfo->usename, nspinfo->nspacl,
nspinfo->oid);
}
if (!aclsSkip)
{
- char * tmp = strdup(fmtId(lanname, force_quotes));
- dumpACL(fout, "LANGUAGE", tmp, finfo[fidx].pronamespace->nspname,
+ char *tmp = strdup(fmtId(lanname, force_quotes));
+ dumpACL(fout, "LANGUAGE", tmp, lanname,
+ finfo[fidx].pronamespace->nspname,
NULL, lanacl, lanoid);
free(tmp);
}
* is never qualified.
*/
static char *
-format_function_signature(FuncInfo *finfo)
+format_function_signature(FuncInfo *finfo, bool honor_quotes)
{
PQExpBufferData fn;
int j;
initPQExpBuffer(&fn);
- appendPQExpBuffer(&fn, "%s (", fmtId(finfo->proname, force_quotes));
+ if (honor_quotes)
+ appendPQExpBuffer(&fn, "%s(", fmtId(finfo->proname, force_quotes));
+ else
+ appendPQExpBuffer(&fn, "%s(", finfo->proname);
for (j = 0; j < finfo->nargs; j++)
{
char *typname;
static void
dumpFuncACL(Archive *fout, FuncInfo *finfo)
{
- char *funcsig;
+ char *funcsig, *funcsig_noquotes;
- funcsig = format_function_signature(finfo);
- dumpACL(fout, "FUNCTION", funcsig, finfo->pronamespace->nspname,
+ funcsig = format_function_signature(finfo, true);
+ funcsig_noquotes = format_function_signature(finfo, false);
+ dumpACL(fout, "FUNCTION", funcsig, funcsig_noquotes,
+ finfo->pronamespace->nspname,
finfo->usename, finfo->proacl, finfo->oid);
free(funcsig);
+ free(funcsig_noquotes);
}
PQExpBuffer asPart = createPQExpBuffer();
PGresult *res = NULL;
char *funcsig = NULL;
+ char *funcsig_noquotes = NULL;
int ntups;
char *proretset;
char *prosrc;
}
}
- funcsig = format_function_signature(finfo);
+ funcsig = format_function_signature(finfo, true);
+ funcsig_noquotes = format_function_signature(finfo, false);
/* DROP must be fully qualified in case same name appears in pg_catalog */
appendPQExpBuffer(delqry, "DROP FUNCTION %s.%s;\n",
finfo->proname);
exit_nicely();
}
- }
+ }
if (proimplicit[0] == 't')
appendPQExpBuffer(q, " IMPLICIT CAST");
appendPQExpBuffer(q, ";\n");
- ArchiveEntry(fout, finfo->oid, funcsig, finfo->pronamespace->nspname,
+ ArchiveEntry(fout, finfo->oid, funcsig_noquotes,
+ finfo->pronamespace->nspname,
finfo->usename, "FUNCTION", NULL,
q->data, delqry->data,
NULL, NULL, NULL);
destroyPQExpBuffer(delqry);
destroyPQExpBuffer(asPart);
free(funcsig);
+ free(funcsig_noquotes);
}
/*
* is never qualified.
*/
static char *
-format_aggregate_signature(AggInfo *agginfo, Archive *fout)
+format_aggregate_signature(AggInfo *agginfo, Archive *fout, bool honor_quotes)
{
PQExpBufferData buf;
bool anybasetype;
initPQExpBuffer(&buf);
- appendPQExpBuffer(&buf, "%s",
+ if (honor_quotes)
+ appendPQExpBuffer(&buf, "%s",
fmtId(agginfo->aggname, force_quotes));
+ else
+ appendPQExpBuffer(&buf, "%s", agginfo->aggname);
anybasetype = (strcmp(agginfo->aggbasetype, "0") == 0);
static void
dumpAggACL(Archive *fout, AggInfo *finfo)
{
- char *aggsig;
+ char *aggsig, *aggsig_noquotes;
- aggsig = format_aggregate_signature(finfo, fout);
- dumpACL(fout, "FUNCTION", aggsig, finfo->aggnamespace->nspname,
+ aggsig = format_aggregate_signature(finfo, fout, true);
+ aggsig_noquotes = format_aggregate_signature(finfo, fout, false);
+ dumpACL(fout, "FUNCTION", aggsig, aggsig_noquotes,
+ finfo->aggnamespace->nspname,
finfo->usename, finfo->aggacl, finfo->oid);
free(aggsig);
+ free(aggsig_noquotes);
}
PQExpBuffer q = createPQExpBuffer();
PQExpBuffer delq = createPQExpBuffer();
PQExpBuffer details = createPQExpBuffer();
- char *aggSig;
+ char *aggsig;
+ char *aggsig_noquotes;
PGresult *res;
int ntups;
int i_aggtransfn;
agginfo->fmtbasetype = strdup(PQgetvalue(res, 0, i_fmtbasetype));
convertok = (PQgetvalue(res, 0, i_convertok)[0] == 't');
- aggSig = format_aggregate_signature(agginfo, g_fout);
+ aggsig = format_aggregate_signature(agginfo, g_fout, true);
+ aggsig_noquotes = format_aggregate_signature(agginfo, g_fout, false);
if (!convertok)
{
write_msg(NULL, "WARNING: aggregate function %s could not be dumped correctly for this database version; ignored\n",
- aggSig);
+ aggsig);
appendPQExpBuffer(q, "-- WARNING: aggregate function %s could not be dumped correctly for this database version; ignored\n",
- aggSig);
- ArchiveEntry(fout, agginfo->oid, aggSig,
+ aggsig);
+ ArchiveEntry(fout, agginfo->oid, aggsig_noquotes,
agginfo->aggnamespace->nspname, agginfo->usename,
"WARNING", NULL,
q->data, "" /* Del */ ,
/* DROP must be fully qualified in case same name appears in pg_catalog */
appendPQExpBuffer(delq, "DROP AGGREGATE %s.%s;\n",
fmtId(agginfo->aggnamespace->nspname, force_quotes),
- aggSig);
+ aggsig);
appendPQExpBuffer(q, "CREATE AGGREGATE %s ( %s );\n",
fmtId(agginfo->aggname, force_quotes),
details->data);
- ArchiveEntry(fout, agginfo->oid, aggSig,
+ ArchiveEntry(fout, agginfo->oid, aggsig_noquotes,
agginfo->aggnamespace->nspname, agginfo->usename,
"AGGREGATE", NULL,
q->data, delq->data,
/*** Dump Aggregate Comments ***/
resetPQExpBuffer(q);
- appendPQExpBuffer(q, "AGGREGATE %s", aggSig);
+ appendPQExpBuffer(q, "AGGREGATE %s", aggsig);
if (g_fout->remoteVersion >= 70300)
dumpComment(fout, q->data,
agginfo->aggnamespace->nspname, agginfo->usename,
destroyPQExpBuffer(q);
destroyPQExpBuffer(delq);
destroyPQExpBuffer(details);
- free(aggSig);
+ free(aggsig);
+ free(aggsig_noquotes);
}
*/
static void
dumpACL(Archive *fout, const char *type, const char *name,
- const char *nspname, const char *usename,
+ const char *name_noquotes, const char *nspname, const char *usename,
const char *acls, const char *objoid)
{
char *aclbuf,
appendPQExpBuffer(sql, "%s;\n", fmtId(usename, force_quotes));
}
- ArchiveEntry(fout, objoid, name, nspname, usename ? usename : "",
+ ArchiveEntry(fout, objoid, name_noquotes, nspname, usename ? usename : "",
"ACL", NULL, sql->data, "", NULL, NULL, NULL);
free(aclbuf);
static void
dumpTableACL(Archive *fout, TableInfo *tbinfo)
{
- char * tmp = strdup( fmtId(tbinfo->relname, force_quotes) );
- dumpACL(fout, "TABLE", tmp, tbinfo->relnamespace->nspname,
- tbinfo->usename, tbinfo->relacl,
+ char *tmp = strdup(fmtId(tbinfo->relname, force_quotes));
+ dumpACL(fout, "TABLE", tmp, tbinfo->relname,
+ tbinfo->relnamespace->nspname, tbinfo->usename, tbinfo->relacl,
tbinfo->viewoid != NULL ? tbinfo->viewoid : tbinfo->oid);
free(tmp);
}
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"))
{
- appendPQExpBuffer(buf, "%s",
- fmtId(typname, true));
+ appendPQExpBuffer(buf, "%s", fmtId(typname, true));
}
else
{
- appendPQExpBuffer(buf, "%s",
- fmtId(typname, false));
+ appendPQExpBuffer(buf, "%s", fmtId(typname, false));
}
result = strdup(buf->data);
*
*
* IDENTIFICATION
- * $Header: /cvsroot/pgsql/src/bin/pg_dump/pg_restore.c,v 1.34 2002/05/10 22:36:27 tgl Exp $
+ * $Header: /cvsroot/pgsql/src/bin/pg_dump/pg_restore.c,v 1.35 2002/07/04 03:04:55 momjian Exp $
*
*-------------------------------------------------------------------------
*/
/* Forward decls */
static void usage(const char *progname);
static char *_cleanupName(char *name);
+static char *_cleanupFuncName(char *name);
typedef struct option optType;
case 'P': /* Function */
opts->selTypes = 1;
opts->selFunction = 1;
- opts->functionNames = optarg ? strdup(optarg) : NULL;
+ opts->functionNames = _cleanupFuncName(optarg);
break;
case 'I': /* Index */
opts->selTypes = 1;
" -O, --no-owner do not reconnect to database to match\n"
" object owner\n"
" -p, --port=PORT server port number\n"
- " -P, --function=NAME restore named function\n"
+ " -P, --function=NAME(args)restore named function\n"
" -r, --rearrange rearrange output to put indexes etc. at end\n"
" -R, --no-reconnect disallow ALL reconnections to the database\n"
" -s, --schema-only restore only the schema, no data\n"
" -O do not reconnect to database to match\n"
" object owner\n"
" -p PORT server port number\n"
- " -P NAME restore named function\n"
+ " -P NAME(args) restore named function\n"
" -r rearrange output to put indexes etc. at end\n"
" -R disallow ALL reconnections to the database\n"
" -s restore only the schema, no data\n"
}
return name;
}
+
+
+static char *
+_cleanupFuncName(char *name)
+{
+ int i;
+ char *ch;
+
+ if (!name || !name[0])
+ return NULL;
+
+ name = strdup(name);
+
+ if (name[0] == '"')
+ {
+ strcpy(name, &name[1]);
+ if (strchr(name, '"') != NULL)
+ strcpy(strchr(name, '"'), strchr(name, '"')+1);
+ }
+ /* otherwise, convert function name to lowercase... */
+ else
+ {
+ for (i = 0; name[i]; i++)
+ if (isupper((unsigned char) name[i]))
+ name[i] = tolower((unsigned char) name[i]);
+ }
+
+ /* strip out any space before paren */
+ ch = strchr(name,'(');
+ while (ch && ch > name && *(ch-1) == ' ')
+ {
+ strcpy(ch - 1, ch);
+ ch--;
+ }
+
+ /*
+ * Strip out spaces after commas in parameter list.
+ * We can't remove all spaces because some types, like
+ * 'double precision' have spaces.
+ */
+ if ((ch = strchr(name,'(')) != NULL)
+ {
+ while ((ch = strstr(ch,", ")) != NULL)
+ strcpy(ch + 1, ch + 2);
+ }
+
+ return name;
+}
projects / postgresql.git / commitdiff