* out of it's tuple
*
* IDENTIFICATION
- * $Header: /cvsroot/pgsql/src/backend/utils/adt/ruleutils.c,v 1.27 1999/10/03 23:55:31 tgl Exp $
+ * $Header: /cvsroot/pgsql/src/backend/utils/adt/ruleutils.c,v 1.28 1999/10/04 04:37:23 tgl Exp $
*
* This software is copyrighted by Jan Wieck - Hamburg.
*
#include "catalog/pg_index.h"
#include "catalog/pg_operator.h"
#include "catalog/pg_shadow.h"
+#include "catalog/pg_type.h"
#include "utils/builtins.h"
#include "utils/lsyscache.h"
static void get_tle_expr(TargetEntry *tle, deparse_context *context);
static void get_const_expr(Const *constval, deparse_context *context);
static void get_sublink_expr(Node *node, deparse_context *context);
+static char *quote_identifier(char *ident);
static char *get_relation_name(Oid relid);
static char *get_attribute_name(Oid relid, int2 attnum);
static bool check_if_rte_used(Node *node, Index rt_index, int levelsup);
spi_nulls[1] = '\0';
spirc = SPI_execp(plan_getam, spi_args, spi_nulls, 1);
if (spirc != SPI_OK_SELECT)
- elog(ERROR, "failed to get pg_am tuple for index %s", nameout(&(idxrelrec->relname)));
+ elog(ERROR, "failed to get pg_am tuple for index %s",
+ nameout(&(idxrelrec->relname)));
if (SPI_processed != 1)
- elog(ERROR, "failed to get pg_am tuple for index %s", nameout(&(idxrelrec->relname)));
+ elog(ERROR, "failed to get pg_am tuple for index %s",
+ nameout(&(idxrelrec->relname)));
spi_tup = SPI_tuptable->vals[0];
spi_ttc = SPI_tuptable->tupdesc;
spi_fno = SPI_fnumber(spi_ttc, "amname");
* ----------
*/
initStringInfo(&buf);
- appendStringInfo(&buf, "CREATE %sINDEX \"%s\" ON \"%s\" USING %s (",
+ appendStringInfo(&buf, "CREATE %sINDEX %s ON %s USING %s (",
idxrec->indisunique ? "UNIQUE " : "",
- nameout(&(idxrelrec->relname)),
- nameout(&(indrelrec->relname)),
- SPI_getvalue(spi_tup, spi_ttc, spi_fno));
+ quote_identifier(nameout(&(idxrelrec->relname))),
+ quote_identifier(nameout(&(indrelrec->relname))),
+ quote_identifier(SPI_getvalue(spi_tup, spi_ttc,
+ spi_fno)));
/* ----------
* Collect the indexed attributes
* Add the indexed field name
* ----------
*/
- if (idxrec->indkey[keyno] == ObjectIdAttributeNumber - 1)
- appendStringInfo(&keybuf, "\"oid\"");
- else
- appendStringInfo(&keybuf, "\"%s\"",
- get_attribute_name(idxrec->indrelid,
- idxrec->indkey[keyno]));
+ appendStringInfo(&keybuf, "%s",
+ quote_identifier(get_attribute_name(idxrec->indrelid,
+ idxrec->indkey[keyno])));
/* ----------
* If not a functional index, add the operator class name
spi_tup = SPI_tuptable->vals[0];
spi_ttc = SPI_tuptable->tupdesc;
spi_fno = SPI_fnumber(spi_ttc, "opcname");
- appendStringInfo(&keybuf, " \"%s\"",
- SPI_getvalue(spi_tup, spi_ttc, spi_fno));
+ appendStringInfo(&keybuf, " %s",
+ quote_identifier(SPI_getvalue(spi_tup, spi_ttc,
+ spi_fno)));
}
}
elog(ERROR, "cache lookup for proc %u failed", idxrec->indproc);
procStruct = (Form_pg_proc) GETSTRUCT(proctup);
- appendStringInfo(&buf, "\"%s\" (%s) ",
- nameout(&(procStruct->proname)),
+ appendStringInfo(&buf, "%s(%s) ",
+ quote_identifier(nameout(&(procStruct->proname))),
keybuf.data);
spi_args[0] = ObjectIdGetDatum(idxrec->indclass[0]);
spi_tup = SPI_tuptable->vals[0];
spi_ttc = SPI_tuptable->tupdesc;
spi_fno = SPI_fnumber(spi_ttc, "opcname");
- appendStringInfo(&buf, "\"%s\"",
- SPI_getvalue(spi_tup, spi_ttc, spi_fno));
+ appendStringInfo(&buf, "%s",
+ quote_identifier(SPI_getvalue(spi_tup, spi_ttc,
+ spi_fno)));
}
else
/* ----------
* Build the rules definition text
* ----------
*/
- appendStringInfo(buf, "CREATE RULE \"%s\" AS ON ", rulename);
+ appendStringInfo(buf, "CREATE RULE %s AS ON ",
+ quote_identifier(rulename));
/* The event the rule is fired for */
switch (ev_type)
}
/* The relation the rule is fired on */
- appendStringInfo(buf, " TO \"%s\"", get_relation_name(ev_class));
+ appendStringInfo(buf, " TO %s",
+ quote_identifier(get_relation_name(ev_class)));
if (ev_attr > 0)
- appendStringInfo(buf, ".\"%s\"",
- get_attribute_name(ev_class, ev_attr));
+ appendStringInfo(buf, ".%s",
+ quote_identifier(get_attribute_name(ev_class,
+ ev_attr)));
/* If the rule has an event qualification, add it */
if (ev_qual == NULL)
/* and do if so */
if (tell_as)
- appendStringInfo(buf, " AS \"%s\"", tle->resdom->resname);
+ appendStringInfo(buf, " AS %s",
+ quote_identifier(tle->resdom->resname));
}
/* If we need other tables that *NEW* or *CURRENT* add the FROM clause */
appendStringInfo(buf, sep);
sep = ", ";
- appendStringInfo(buf, "\"%s\"", rte->relname);
+ appendStringInfo(buf, "%s",
+ quote_identifier(rte->relname));
if (strcmp(rte->relname, rte->refname) != 0)
- appendStringInfo(buf, " \"%s\"", rte->refname);
+ appendStringInfo(buf, " %s",
+ quote_identifier(rte->refname));
}
}
}
* ----------
*/
rte = (RangeTblEntry *) nth(query->resultRelation - 1, query->rtable);
- appendStringInfo(buf, "INSERT INTO \"%s\"", rte->relname);
+ appendStringInfo(buf, "INSERT INTO %s",
+ quote_identifier(rte->relname));
/* Add the target list */
sep = " (";
appendStringInfo(buf, sep);
sep = ", ";
- appendStringInfo(buf, "\"%s\"", tle->resdom->resname);
+ appendStringInfo(buf, "%s", quote_identifier(tle->resdom->resname));
}
appendStringInfo(buf, ") ");
* ----------
*/
rte = (RangeTblEntry *) nth(query->resultRelation - 1, query->rtable);
- appendStringInfo(buf, "UPDATE \"%s\" SET ", rte->relname);
+ appendStringInfo(buf, "UPDATE %s SET ",
+ quote_identifier(rte->relname));
/* Add the comma separated list of 'attname = value' */
sep = "";
appendStringInfo(buf, sep);
sep = ", ";
- appendStringInfo(buf, "\"%s\" = ", tle->resdom->resname);
+ appendStringInfo(buf, "%s = ",
+ quote_identifier(tle->resdom->resname));
get_tle_expr(tle, context);
}
* ----------
*/
rte = (RangeTblEntry *) nth(query->resultRelation - 1, query->rtable);
- appendStringInfo(buf, "DELETE FROM \"%s\"", rte->relname);
+ appendStringInfo(buf, "DELETE FROM %s",
+ quote_identifier(rte->relname));
/* Add a WHERE clause if given */
if (query->qual != NULL)
else if (!strcmp(rte->refname, "*CURRENT*"))
appendStringInfo(buf, "old.");
else
- appendStringInfo(buf, "\"%s\".", rte->refname);
+ appendStringInfo(buf, "%s.",
+ quote_identifier(rte->refname));
}
- appendStringInfo(buf, "\"%s\"",
- get_attribute_name(rte->relid,
- var->varattno));
+ appendStringInfo(buf, "%s",
+ quote_identifier(get_attribute_name(rte->relid,
+ var->varattno)));
}
break;
{
Aggref *aggref = (Aggref *) node;
- appendStringInfo(buf, "\"%s\"(", aggref->aggname);
+ appendStringInfo(buf, "%s(",
+ quote_identifier(aggref->aggname));
get_rule_expr(aggref->target, context);
appendStringInfo(buf, ")");
}
* Build a string of proname(args)
* ----------
*/
- appendStringInfo(buf, "\"%s\"(", proname);
+ appendStringInfo(buf, "%s(", quote_identifier(proname));
sep = "";
foreach(l, expr->args)
{
/* ----------
* get_const_expr
*
- * Make a string representation with the type cast out of a Const
+ * Make a string representation of a Const
* ----------
*/
static void
extval = (char *) (*fmgr_faddr(&finfo_output)) (constval->constvalue,
&isnull, -1);
- /*
- * We must quote any funny characters in the constant's representation.
- * XXX Any MULTIBYTE considerations here?
- */
- appendStringInfoChar(buf, '\'');
- for (valptr = extval; *valptr; valptr++)
+ switch (constval->consttype)
{
- char ch = *valptr;
- if (ch == '\'' || ch == '\\')
- {
- appendStringInfoChar(buf, '\\');
- appendStringInfoChar(buf, ch);
- }
- else if (ch >= 0 && ch < ' ')
- {
- appendStringInfo(buf, "\\%03o", ch);
- }
- else
- appendStringInfoChar(buf, ch);
+ case INT2OID:
+ case INT4OID:
+ case OIDOID: /* int types */
+ case FLOAT4OID:
+ case FLOAT8OID: /* float types */
+ /* These types are printed without quotes */
+ appendStringInfo(buf, extval);
+ break;
+ default:
+ /*
+ * We must quote any funny characters in the constant's
+ * representation.
+ * XXX Any MULTIBYTE considerations here?
+ */
+ appendStringInfoChar(buf, '\'');
+ for (valptr = extval; *valptr; valptr++)
+ {
+ char ch = *valptr;
+ if (ch == '\'' || ch == '\\')
+ {
+ appendStringInfoChar(buf, '\\');
+ appendStringInfoChar(buf, ch);
+ }
+ else if (ch >= 0 && ch < ' ')
+ appendStringInfo(buf, "\\%03o", (int) ch);
+ else
+ appendStringInfoChar(buf, ch);
+ }
+ appendStringInfoChar(buf, '\'');
+ break;
}
- appendStringInfoChar(buf, '\'');
- pfree(extval);
- extval = (char *) nameout(&(typeStruct->typname));
- /* probably would be better to recognize UNKNOWN by OID... */
- if (strcmp(extval, "unknown") != 0)
- appendStringInfo(buf, "::\"%s\"", extval);
pfree(extval);
+
+ switch (constval->consttype)
+ {
+ case INT4OID:
+ case FLOAT8OID:
+ case UNKNOWNOID:
+ /* These types can be left unlabeled */
+ break;
+ default:
+ extval = (char *) nameout(&(typeStruct->typname));
+ appendStringInfo(buf, "::%s", quote_identifier(extval));
+ pfree(extval);
+ break;
+ }
}
appendStringInfo(buf, "))");
}
+/* ----------
+ * quote_identifier - Quote an identifier only if needed
+ *
+ * When quotes are needed, we palloc the required space; slightly
+ * space-wasteful but well worth it for notational simplicity.
+ * ----------
+ */
+static char *
+quote_identifier(char *ident)
+{
+ /*
+ * Can avoid quoting if ident starts with a lowercase letter and
+ * contains only lowercase letters, digits, and underscores.
+ * Otherwise, supply quotes.
+ */
+ bool safe;
+ char *result;
+
+ /*
+ * would like to use <ctype.h> macros here, but they might yield
+ * unwanted locale-specific results...
+ */
+ safe = (ident[0] >= 'a' && ident[0] <= 'z');
+ if (safe)
+ {
+ char *ptr;
+
+ for (ptr = ident+1; *ptr; ptr++)
+ {
+ char ch = *ptr;
+
+ safe = ((ch >= 'a' && ch <= 'z') ||
+ (ch >= '0' && ch <= '9') ||
+ (ch == '_'));
+ if (! safe)
+ break;
+ }
+ }
+
+ if (safe)
+ return ident; /* no change needed */
+
+ result = (char *) palloc(strlen(ident) + 2 + 1);
+ sprintf(result, "\"%s\"", ident);
+ return result;
+}
/* ----------
* get_relation_name - Get a relation name by Oid
Form_pg_attribute attStruct;
atttup = SearchSysCacheTuple(ATTNUM,
- ObjectIdGetDatum(relid), (Datum) attnum, 0, 0);
+ ObjectIdGetDatum(relid), (Datum) attnum,
+ 0, 0);
if (!HeapTupleIsValid(atttup))
elog(ERROR, "cache lookup of attribute %d in relation %u failed",
attnum, relid);
OSDN Git Service
