ksys: Add Task utilities

1 files changed, 26 insertions, 0 deletions

diff --git a/src/KingSystem/Utils/Thread/TaskQueueLock.cpp b/src/KingSystem/Utils/Thread/TaskQueueLock.cpp
new file mode 100644
index 00000000..be7b1c5e
--- /dev/null
+++ b/src/KingSystem/Utils/Thread/TaskQueueLock.cpp
@@ -0,0 +1,26 @@
+#include "KingSystem/Utils/Thread/TaskQueueLock.h"
+#include "KingSystem/Utils/Thread/TaskQueueBase.h"
+
+namespace ksys::util {
+
+TaskQueueLock::TaskQueueLock() = default;
+
+TaskQueueLock::~TaskQueueLock() {
+    unlock();
+}
+
+void TaskQueueLock::lock(TaskQueueBase* queue) {
+    if (!mQueue) {
+        mQueue = queue;
+        mQueue->lock();
+    }
+}
+
+void TaskQueueLock::unlock() {
+    if (mQueue) {
+        mQueue->unlock();
+        mQueue = nullptr;
+    }
+}
+
+}  // namespace ksys::util