Fix psql's "\g target" meta-command to work with COPY TO STDOUT.

[postgresql] / src / bin / psql / copy.c

diff --git a/src/bin/psql/copy.c b/src/bin/psql/copy.c
index 02c85119fea662b6707d379d238d623a021af9d2..18a8c8eebb549599ef89968b423d685572425a2e 100644 (file)
--- a/src/bin/psql/copy.c
+++ b/src/bin/psql/copy.c
@@ -425,7 +425,10 @@ do_copy(const char *args)
  *
  * conn should be a database connection that you just issued COPY TO on
  * and got back a PGRES_COPY_OUT result.
+ *
  * copystream is the file stream for the data to go to.
+ * copystream can be NULL to eat the data without writing it anywhere.
+ *
  * The final status for the COPY is returned into *res (but note
  * we already reported the error, if it's not a success result).
  *
@@ -447,7 +450,7 @@ handleCopyOut(PGconn *conn, FILE *copystream, PGresult **res)
 
                if (buf)
                {
-                       if (OK && fwrite(buf, 1, ret, copystream) != ret)
+                       if (OK && copystream && fwrite(buf, 1, ret, copystream) != ret)
                        {
                                psql_error("could not write COPY data: %s\n",
                                                   strerror(errno));
@@ -458,7 +461,7 @@ handleCopyOut(PGconn *conn, FILE *copystream, PGresult **res)
                }
        }
 
-       if (OK && fflush(copystream))
+       if (OK && copystream && fflush(copystream))
        {
                psql_error("could not write COPY data: %s\n",
                                   strerror(errno));