src/KingSystem/Utils/Thread/ManagedTask.cpp


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

#include "KingSystem/Utils/Thread/ManagedTask.h"
#include "KingSystem/Utils/Thread/ManagedTaskHandle.h"
#include "KingSystem/Utils/Thread/TaskMgr.h"
#include "KingSystem/Utils/Thread/TaskQueueLock.h"

namespace ksys::util {

ManagedTask::ManagedTask(sead::Heap* heap) : Task(heap) {}

ManagedTask::ManagedTask(sead::IDisposer::HeapNullOption heap_null_option)
    : Task(nullptr, heap_null_option) {}

ManagedTask::~ManagedTask() {
    if (mHandle) {
        mHandle->finalize();
        mHandle = nullptr;
    }
    finalize_();
}

void ManagedTask::prepare_(TaskRequest* request) {
    mIsIdle = false;
    prepareImpl_(request);
}

void ManagedTask::run_() {
    Task::run_();
    onRun_();
}

void ManagedTask::onRunFinished_() {}

void ManagedTask::onFinish_() {
    if (auto* handle = mHandle) {
        handle->setIsSuccess(isSuccess());
        handle->setStatus(ManagedTaskHandle::Status::TaskFinished);
    }

    if (!mHandle && mMgr)
        mMgr->freeTask(this);
}

void ManagedTask::onPostFinish_() {
    mIsIdle = true;
}

void ManagedTask::preRemove_() {
    preRemoveImpl_();

    if (mHandle)
        mHandle->setStatus(ManagedTaskHandle::Status::TaskRemoved);

    if (!mHandle && mMgr)
        mMgr->freeTask(this);
}

void ManagedTask::postRemove_() {
    mIsIdle = true;
}

void ManagedTask::onRun_() {}

void ManagedTask::prepareImpl_(TaskRequest*) {}

void ManagedTask::preRemoveImpl_() {}

bool ManagedTask::isIdle() const {
    return mIsIdle;
}

void ManagedTask::setMgr(TaskMgr* mgr) {
    mMgr = mgr;
}

void ManagedTask::attachHandle(ManagedTaskHandle* handle, TaskQueueBase* queue) {
    if (mHandle)
        return;

    if (handle) {
        handle->attachTask({this, queue});
        handle->setStatus(ManagedTaskHandle::Status::TaskAttached);
    }
    mHandle = handle;
}

// NON_MATCHING: switch
void ManagedTask::detachHandle() {
    TaskQueueLock lock;
    lock.lock(mQueue);

    if (mHandle) {
        switch (mHandle->getStatus()) {
        case ManagedTaskHandle::Status::TaskRemoved:
        case ManagedTaskHandle::Status::TaskFinished:
            mHandle = nullptr;
            if (mMgr)
                mMgr->freeTask(this);
            break;
        default:
            mHandle = nullptr;
            break;
        }
    }
}

}  // namespace ksys::util