Only DISCARD ALL should be in the command_no_begin list.

We allowes DISCARD PLANS and TEMP in a transaction.

diff --git a/src/bin/psql/common.c b/src/bin/psql/common.c
index 0f83dd3c45129a73b7c57e792ccee2bf84244493..64577084a2f6f0f6c0670d5578d918edbc03f71c 100644 (file)
--- a/src/bin/psql/common.c
+++ b/src/bin/psql/common.c
@@ -1352,8 +1352,20 @@ command_no_begin(const char *query)
                        return true;
        }
 
+       /* DISCARD ALL isn't allowed in xacts, but other variants are allowed. */
        if (wordlen == 7 && pg_strncasecmp(query, "discard", 7) == 0)
-               return true;
+       {
+               query += wordlen;
+
+               query = skip_white_space(query);
+
+               wordlen = 0;
+               while (isalpha((unsigned char) query[wordlen]))
+                       wordlen += PQmblen(&query[wordlen], pset.encoding);
+
+               if (wordlen == 3 && pg_strncasecmp(query, "all", 3) == 0)
+                       return true;
+       }
 
        return false;
 }