Relocating everything in the repo s.t. it can be merged into

the columnstore repo.
2025-07-30 19:23:07 +03:00 · 2019-08-14 11:07:24 -05:00
parent 87f9523028
commit fba0c1728c
78 changed files with 1 additions and 1 deletions
--- a/storage-manager/src/WriteTask.cpp
+++ b/storage-manager/src/WriteTask.cpp
@ -0,0 +1,119 @@
+/* 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 "WriteTask.h"
+#include "messageFormat.h"
+#include "IOCoordinator.h"
+#include "SMLogging.h"
+#include <errno.h>
+
+using namespace std;
+
+namespace storagemanager
+{
+
+WriteTask::WriteTask(int sock, uint len) : PosixTask(sock, len)
+{
+}
+
+WriteTask::~WriteTask()
+{
+}
+
+#define check_error(msg, ret) \
+    if (!success) \
+    { \
+        handleError(msg, errno); \
+        return ret; \
+    }
+
+#define min(x, y) (x < y ? x : y)
+
+bool WriteTask::run()
+{
+    SMLogging* logger = SMLogging::get();
+    bool success;
+    uint8_t cmdbuf[1024] = {0};
+    
+    success = read(cmdbuf, sizeof(write_cmd));
+    check_error("WriteTask read", false);
+    write_cmd *cmd = (write_cmd *) cmdbuf;
+    
+    if (cmd->flen > 1023 - sizeof(*cmd))
+    {
+        handleError("WriteTask", ENAMETOOLONG);
+        return true;
+    }
+    success = read(&cmdbuf[sizeof(*cmd)], cmd->flen);
+    check_error("WriteTask read", false);
+            
+    #ifdef SM_TRACE
+    logger->log(LOG_DEBUG,"write filename %s offset %i count %i.",cmd->filename,cmd->offset,cmd->count);
+    #endif
+            
+    ssize_t readCount = 0, writeCount = 0;
+    vector<uint8_t> databuf;
+    uint bufsize = min(1 << 20, cmd->count);   // 1 MB
+    databuf.resize(bufsize);
+
+    while (readCount < cmd->count)
+    {
+        uint toRead = min(cmd->count - readCount, bufsize);
+        success = read(&databuf[0], toRead);
+        check_error("WriteTask read data", false);
+        readCount += toRead;
+        uint writePos = 0;
+        ssize_t err;
+        while (writeCount < readCount)
+        {
+            try 
+            {
+                err = ioc->write(cmd->filename, &databuf[writePos], cmd->offset + writeCount, toRead - writePos);
+            }
+            catch (exception &e)
+            {
+                logger->log(LOG_DEBUG, "WriteTask: caught '%s'", e.what());
+                errno = EIO;
+                err = -1;
+            }
+            if (err <= 0)
+                break;
+            writeCount += err;
+            writePos += err;
+        }
+        if (writeCount != readCount)
+            break;
+    }
+    
+    uint8_t respbuf[sizeof(sm_response) + 4];
+    sm_response *resp = (sm_response *) respbuf;
+    uint payloadLen = 0;
+    if (cmd->count != 0 && writeCount == 0)
+    {
+        payloadLen = 4;
+        resp->returnCode = -1;
+        *((int *) &resp[1]) = errno;
+        success = write(*resp, payloadLen);
+    }
+    else
+        resp->returnCode = writeCount;
+    success = write(*resp, payloadLen);
+    return success;
+}
+
+}