You've already forked mariadb-columnstore-engine
mirror of
https://github.com/mariadb-corporation/mariadb-columnstore-engine.git
synced 2026-01-06 08:21:10 +03:00
89 lines
2.1 KiB
C++
89 lines
2.1 KiB
C++
/* Copyright (C) 2019 MariaDB Corporation
|
|
|
|
This program is free software; you can redistribute it and/or
|
|
modify it under the terms of the GNU General Public License
|
|
as published by the Free Software Foundation; version 2 of
|
|
the License.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program; if not, write to the Free Software
|
|
Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
|
|
MA 02110-1301, USA. */
|
|
|
|
#include "CopyTask.h"
|
|
#include <errno.h>
|
|
#include "SMLogging.h"
|
|
#include "messageFormat.h"
|
|
|
|
using namespace std;
|
|
|
|
namespace storagemanager
|
|
{
|
|
CopyTask::CopyTask(int sock, uint len) : PosixTask(sock, len)
|
|
{
|
|
}
|
|
|
|
CopyTask::~CopyTask()
|
|
{
|
|
}
|
|
|
|
#define check_error(msg, ret) \
|
|
if (success < 0) \
|
|
{ \
|
|
handleError(msg, errno); \
|
|
return ret; \
|
|
}
|
|
|
|
bool CopyTask::run()
|
|
{
|
|
int success;
|
|
SMLogging* logger = SMLogging::get();
|
|
|
|
uint8_t buf[2048] = {0};
|
|
|
|
if (getLength() > 2047)
|
|
{
|
|
handleError("CopyTask read", ENAMETOOLONG);
|
|
return true;
|
|
}
|
|
|
|
success = read(buf, getLength());
|
|
check_error("CopyTask read", false);
|
|
auto* cmd = (copy_cmd*)buf;
|
|
string filename1(cmd->file1.filename,
|
|
cmd->file1.flen); // need to copy this in case it's not null terminated
|
|
auto* filename2 = (f_name*)&buf[sizeof(copy_cmd) + cmd->file1.flen];
|
|
|
|
#ifdef SM_TRACE
|
|
logger->log(LOG_DEBUG, "copy %s to %s.", filename1.c_str(), filename2->filename);
|
|
#endif
|
|
int err;
|
|
try
|
|
{
|
|
err = ioc->copyFile(filename1.c_str(), filename2->filename);
|
|
}
|
|
catch (exception& e)
|
|
{
|
|
logger->log(LOG_ERR, "CopyTask: caught %s", e.what());
|
|
err = -1;
|
|
errno = EIO;
|
|
}
|
|
|
|
if (err)
|
|
{
|
|
handleError("CopyTask copy", errno);
|
|
return true;
|
|
}
|
|
|
|
auto* resp = (sm_response*)buf;
|
|
resp->returnCode = 0;
|
|
return write(*resp, 0);
|
|
}
|
|
|
|
} // namespace storagemanager
|