On Sat, 7 Aug 2010, Kris Jurka wrote:

On Fri, 6 Aug 2010, James William Pye wrote:

I think there's a snag in the patch:

postgres=# COPY data FROM '/Users/jwp/DATA.bcopy' WITH BINARY;
ERROR:  row field count is -1, expected 1
CONTEXT:  COPY data, line 4

Probably a quick/small fix away, I imagine.

Hmm, not quite sure why that is. That seems to imply that it's not using V3 protocol, but I thought binary copy could only be used with the V3 protocol. In any case, I think this new patch is more bulletproof.


Oh, duh. It's a server side copy not going through the client at all. Here's a hopefully final patch.

Kris Jurka
*** a/src/backend/commands/copy.c
--- b/src/backend/commands/copy.c
***************
*** 2058,2069 **** CopyFrom(CopyState cstate)
                        int16           fld_count;
                        ListCell   *cur;
  
!                       if (!CopyGetInt16(cstate, &fld_count) ||
!                               fld_count == -1)
                        {
                                done = true;
                                break;
                        }
  
                        if (fld_count != attr_count)
                                ereport(ERROR,
--- 2058,2088 ----
                        int16           fld_count;
                        ListCell   *cur;
  
!                       if (!CopyGetInt16(cstate, &fld_count))
                        {
                                done = true;
                                break;
                        }
+                       
+                       if (fld_count == -1)
+                       {
+                               /*
+                                * Received EOF marker.  Check to see if this 
is really
+                                * the EOF and complain if we find any more 
data.
+                                * In a V3 protocol copy, this also enforces 
that we wait
+                                * for the protocol end of copy (CopyDone/Fail).
+                                */
+                               int8 unused;
+                               if (CopyGetData(cstate, &unused, 
sizeof(unused), sizeof(unused)))
+                               {
+                                       ereport(ERROR,
+                                                       
(errcode(ERRCODE_BAD_COPY_FILE_FORMAT),
+                                                        errmsg("received copy 
data after EOF marker")));
+                               }
+ 
+                               done = true;
+                               break;
+                       }
  
                        if (fld_count != attr_count)
                                ereport(ERROR,
-- 
Sent via pgsql-hackers mailing list (pgsql-hackers@postgresql.org)
To make changes to your subscription:
http://www.postgresql.org/mailpref/pgsql-hackers

Reply via email to