external
/
citus
mirror of https://github.com/citusdata/citus.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix worker crash when coordinator disconnects

cmake_branch_point
Brian Cloutier 2018-04-10 23:36:02 +00:00 committed by Brian Cloutier
parent 73e1d81bb3
commit 667280650d
1 changed files with 27 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

43
src/backend/distributed/commands/transmit.c
View File

@ -48,31 +48,42 @@ RedirectCopyDataToRegularFile(const char *filename)
fileDesc = FileOpenForTransmit(filename, fileFlags, fileMode);
SendCopyInStart();
copyDone = ReceiveCopyData(copyData);
while (!copyDone)
PG_TRY();
{
/* if received data has contents, append to regular file */
if (copyData->len > 0)
SendCopyInStart();
copyDone = ReceiveCopyData(copyData);
while (!copyDone)
{
/* if received data has contents, append to regular file */
if (copyData->len > 0)
{
#if (PG_VERSION_NUM >= 100000)
int appended = FileWrite(fileDesc, copyData->data, copyData->len,
PG_WAIT_IO);
int appended = FileWrite(fileDesc, copyData->data, copyData->len,
PG_WAIT_IO);
#else
int appended = FileWrite(fileDesc, copyData->data, copyData->len);
int appended = FileWrite(fileDesc, copyData->data, copyData->len);
#endif
if (appended != copyData->len)
{
ereport(ERROR, (errcode_for_file_access(),
errmsg("could not append to received file: %m")));
if (appended != copyData->len)
{
ereport(ERROR, (errcode_for_file_access(),
errmsg("could not append to received file: %m")));
}
}
}
resetStringInfo(copyData);
copyDone = ReceiveCopyData(copyData);
resetStringInfo(copyData);
copyDone = ReceiveCopyData(copyData);
}
}
PG_CATCH();
{
FreeStringInfo(copyData);
FileClose(fileDesc);
PG_RE_THROW();
}
PG_END_TRY();
FreeStringInfo(copyData);
FileClose(fileDesc);