ksys: Add Task utilities

1 files changed, 28 insertions, 0 deletions

diff --git a/src/KingSystem/Utils/Thread/TaskQueueLock.h b/src/KingSystem/Utils/Thread/TaskQueueLock.h
new file mode 100644
index 00000000..2bc44035
--- /dev/null
+++ b/src/KingSystem/Utils/Thread/TaskQueueLock.h
@@ -0,0 +1,28 @@
+#pragma once
+
+#include <prim/seadRuntimeTypeInfo.h>
+#include "KingSystem/Utils/Thread/TaskThread.h"
+#include "KingSystem/Utils/Types.h"
+
+namespace ksys::util {
+
+class TaskQueueBase;
+
+class TaskQueueLock {
+    SEAD_RTTI_BASE(TaskQueueLock)
+public:
+    TaskQueueLock();
+    explicit TaskQueueLock(TaskThread* thread) : TaskQueueLock() { thread->lock(this); }
+    TaskQueueLock(const TaskQueueLock&) = delete;
+    virtual ~TaskQueueLock();
+    TaskQueueLock& operator=(const TaskQueueLock&) = delete;
+
+    void lock(TaskQueueBase* queue);
+    void unlock();
+
+private:
+    TaskQueueBase* mQueue = nullptr;
+};
+KSYS_CHECK_SIZE_NX150(TaskQueueLock, 0x10);
+
+}  // namespace ksys::util