~george-edison55/twistload/trunk

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
/*
 * The MIT License (MIT)
 *
 * Copyright (c) 2017 Nathan Osman
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to
 * deal in the Software without restriction, including without limitation the
 * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
 * sell copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
 * IN THE SOFTWARE.
 */

#include <QMetaObject>

#include "allocationtask.h"
#include "file.h"

File::File()
    : mTask(nullptr)
{
}

File::File(const QString &filename)
    : mTask(nullptr),
      mFile(filename)
{
}

File::~File()
{
    if (mTask) {
        mTask->abort();
        mThread.wait();
    }
}

QString File::filename() const
{
    return mFile.fileName();
}

void File::setFilename(const QString &filename)
{
    mFile.setFileName(filename);
}

void File::open(qint64 size)
{
    if (size) {

        // Create a new thread and task and move the task to the thread
        mTask = new AllocationTask(mFile.fileName(), size);
        mTask->moveToThread(&mThread);

        // Ensure the task and thread are destroyed when complete
        connect(mTask, &AllocationTask::error, this, &File::error);
        connect(mTask, &AllocationTask::progress, this, &File::allocationProgress);
        connect(mTask, &AllocationTask::succeeded, this, &File::onSucceeded);
        connect(mTask, &AllocationTask::finished, this, [=]() {
            mTask->deleteLater();
            mTask = nullptr;
            mThread.quit();
        });

        // Start the thread and invoke the slot
        mThread.start();
        QMetaObject::invokeMethod(mTask, "run", Qt::QueuedConnection);

    } else {
        emit allocationSucceeded();
    }
}

void File::write(const QByteArray &data, qint64 offset)
{
    if (!mFile.seek(offset) || mFile.write(data) == -1) {
        emit error(mFile.errorString());
    }
}

void File::close()
{
    mFile.close();
}

void File::onSucceeded()
{
    // Attempt to open the file for writing
    if (!mFile.open(QIODevice::WriteOnly | QIODevice::Append)) {
        emit error(mFile.errorString());
        return;
    }

    emit allocationSucceeded();
}