Move and rename fmtReloptionsArray().

Move fmtReloptionsArray() from pg_dump.c to string_utils.c so that it
is available to other frontend code. In particular psql's \ev and \sv
commands need it to handle view reloptions. Also rename the function
to appendReloptionsArray(), which is a more accurate description of
what it does.

Author: Dean Rasheed
Reviewed-by: Peter Eisentraut
Discussion: http://www.postgresql.org/message-id/CAEZATCWZjCgKRyM-agE0p8ax15j9uyQoF=qew7D2xB6cF76T8A@mail.gmail.com

diff --git a/src/bin/pg_dump/pg_dump.c b/src/bin/pg_dump/pg_dump.c
index d3f515716d7e8c29453a4e167b12ab21087fafa0..582509396bb73127597abc10c4edba9e61c23283 100644 (file)
--- a/src/bin/pg_dump/pg_dump.c
+++ b/src/bin/pg_dump/pg_dump.c
@@ -261,8 +261,8 @@ static void binary_upgrade_extension_member(PQExpBuffer upgrade_buffer,
 static const char *getAttrName(int attrnum, TableInfo *tblInfo);
 static const char *fmtCopyColumnList(const TableInfo *ti, PQExpBuffer buffer);
 static bool nonemptyReloptions(const char *reloptions);
-static void fmtReloptionsArray(Archive *fout, PQExpBuffer buffer,
-                  const char *reloptions, const char *prefix);
+static void appendReloptionsArrayAH(PQExpBuffer buffer, const char *reloptions,
+                       const char *prefix, Archive *fout);
 static char *get_synchronized_snapshot(Archive *fout);
 static PGresult *ExecuteSqlQueryForSingleRow(Archive *fout, char *query);
 static void setupDumpWorker(Archive *AHX);
@@ -15046,7 +15046,7 @@ dumpTableSchema(Archive *fout, TableInfo *tbinfo)
        if (nonemptyReloptions(tbinfo->reloptions))
        {
            appendPQExpBufferStr(q, " WITH (");
-           fmtReloptionsArray(fout, q, tbinfo->reloptions, "");
+           appendReloptionsArrayAH(q, tbinfo->reloptions, "", fout);
            appendPQExpBufferChar(q, ')');
        }
        result = createViewAsClause(fout, tbinfo);
@@ -15301,13 +15301,14 @@ dumpTableSchema(Archive *fout, TableInfo *tbinfo)
            if (nonemptyReloptions(tbinfo->reloptions))
            {
                addcomma = true;
-               fmtReloptionsArray(fout, q, tbinfo->reloptions, "");
+               appendReloptionsArrayAH(q, tbinfo->reloptions, "", fout);
            }
            if (nonemptyReloptions(tbinfo->toast_reloptions))
            {
                if (addcomma)
                    appendPQExpBufferStr(q, ", ");
-               fmtReloptionsArray(fout, q, tbinfo->toast_reloptions, "toast.");
+               appendReloptionsArrayAH(q, tbinfo->toast_reloptions, "toast.",
+                                       fout);
            }
            appendPQExpBufferChar(q, ')');
        }
@@ -15908,7 +15909,7 @@ dumpConstraint(Archive *fout, ConstraintInfo *coninfo)
            if (nonemptyReloptions(indxinfo->indreloptions))
            {
                appendPQExpBufferStr(q, " WITH (");
-               fmtReloptionsArray(fout, q, indxinfo->indreloptions, "");
+               appendReloptionsArrayAH(q, indxinfo->indreloptions, "", fout);
                appendPQExpBufferChar(q, ')');
            }
 
@@ -16809,7 +16810,7 @@ dumpRule(Archive *fout, RuleInfo *rinfo)
    {
        appendPQExpBuffer(cmd, "ALTER VIEW %s SET (",
                          fmtId(tbinfo->dobj.name));
-       fmtReloptionsArray(fout, cmd, rinfo->reloptions, "");
+       appendReloptionsArrayAH(cmd, rinfo->reloptions, "", fout);
        appendPQExpBufferStr(cmd, ");\n");
    }
 
@@ -17707,67 +17708,17 @@ nonemptyReloptions(const char *reloptions)
  * Format a reloptions array and append it to the given buffer.
  *
  * "prefix" is prepended to the option names; typically it's "" or "toast.".
- *
- * Note: this logic should generally match the backend's flatten_reloptions()
- * (in adt/ruleutils.c).
  */
 static void
-fmtReloptionsArray(Archive *fout, PQExpBuffer buffer, const char *reloptions,
-                  const char *prefix)
+appendReloptionsArrayAH(PQExpBuffer buffer, const char *reloptions,
+                       const char *prefix, Archive *fout)
 {
-   char      **options;
-   int         noptions;
-   int         i;
+   bool        res;
 
-   if (!parsePGArray(reloptions, &options, &noptions))
-   {
+   res = appendReloptionsArray(buffer, reloptions, prefix, fout->encoding,
+                               fout->std_strings);
+   if (!res)
        write_msg(NULL, "WARNING: could not parse reloptions array\n");
-       if (options)
-           free(options);
-       return;
-   }
-
-   for (i = 0; i < noptions; i++)
-   {
-       char       *option = options[i];
-       char       *name;
-       char       *separator;
-       char       *value;
-
-       /*
-        * Each array element should have the form name=value.  If the "=" is
-        * missing for some reason, treat it like an empty value.
-        */
-       name = option;
-       separator = strchr(option, '=');
-       if (separator)
-       {
-           *separator = '\0';
-           value = separator + 1;
-       }
-       else
-           value = "";
-
-       if (i > 0)
-           appendPQExpBufferStr(buffer, ", ");
-       appendPQExpBuffer(buffer, "%s%s=", prefix, fmtId(name));
-
-       /*
-        * In general we need to quote the value; but to avoid unnecessary
-        * clutter, do not quote if it is an identifier that would not need
-        * quoting.  (We could also allow numbers, but that is a bit trickier
-        * than it looks --- for example, are leading zeroes significant?  We
-        * don't want to assume very much here about what custom reloptions
-        * might mean.)
-        */
-       if (strcmp(fmtId(value), value) == 0)
-           appendPQExpBufferStr(buffer, value);
-       else
-           appendStringLiteralAH(buffer, value, fout);
-   }
-
-   if (options)
-       free(options);
 }
 
 /*

diff --git a/src/fe_utils/string_utils.c b/src/fe_utils/string_utils.c
index c57d8366379b6165acf4a607288929c78f49f817..aeef12c3df66e565e39af194d167f0dc4448f53b 100644 (file)
--- a/src/fe_utils/string_utils.c
+++ b/src/fe_utils/string_utils.c
@@ -461,6 +461,78 @@ parsePGArray(const char *atext, char ***itemarray, int *nitems)
 }
 
 
+/*
+ * Format a reloptions array and append it to the given buffer.
+ *
+ * "prefix" is prepended to the option names; typically it's "" or "toast.".
+ *
+ * Returns false if the reloptions array could not be parsed (in which case
+ * nothing will have been appended to the buffer), or true on success.
+ *
+ * Note: this logic should generally match the backend's flatten_reloptions()
+ * (in adt/ruleutils.c).
+ */
+bool
+appendReloptionsArray(PQExpBuffer buffer, const char *reloptions,
+                     const char *prefix, int encoding, bool std_strings)
+{
+   char      **options;
+   int         noptions;
+   int         i;
+
+   if (!parsePGArray(reloptions, &options, &noptions))
+   {
+       if (options)
+           free(options);
+       return false;
+   }
+
+   for (i = 0; i < noptions; i++)
+   {
+       char       *option = options[i];
+       char       *name;
+       char       *separator;
+       char       *value;
+
+       /*
+        * Each array element should have the form name=value.  If the "=" is
+        * missing for some reason, treat it like an empty value.
+        */
+       name = option;
+       separator = strchr(option, '=');
+       if (separator)
+       {
+           *separator = '\0';
+           value = separator + 1;
+       }
+       else
+           value = "";
+
+       if (i > 0)
+           appendPQExpBufferStr(buffer, ", ");
+       appendPQExpBuffer(buffer, "%s%s=", prefix, fmtId(name));
+
+       /*
+        * In general we need to quote the value; but to avoid unnecessary
+        * clutter, do not quote if it is an identifier that would not need
+        * quoting.  (We could also allow numbers, but that is a bit trickier
+        * than it looks --- for example, are leading zeroes significant?  We
+        * don't want to assume very much here about what custom reloptions
+        * might mean.)
+        */
+       if (strcmp(fmtId(value), value) == 0)
+           appendPQExpBufferStr(buffer, value);
+       else
+           appendStringLiteral(buffer, value, encoding, std_strings);
+   }
+
+   if (options)
+       free(options);
+
+   return true;
+}
+
+
 /*
  * processSQLNamePattern
  *

diff --git a/src/include/fe_utils/string_utils.h b/src/include/fe_utils/string_utils.h
index 5d3fcc24271eb944338160b0324938d7c86c7403..733e337910b3853cf7abb830fe238cbd75059ed2 100644 (file)
--- a/src/include/fe_utils/string_utils.h
+++ b/src/include/fe_utils/string_utils.h
@@ -42,6 +42,9 @@ extern void appendByteaLiteral(PQExpBuffer buf,
 
 extern bool parsePGArray(const char *atext, char ***itemarray, int *nitems);
 
+extern bool appendReloptionsArray(PQExpBuffer buffer, const char *reloptions,
+                     const char *prefix, int encoding, bool std_strings);
+
 extern bool processSQLNamePattern(PGconn *conn, PQExpBuffer buf,
                      const char *pattern,
                      bool have_where, bool force_escape,