From 211b80abe252a46a98df8a9a3efb036041c6526d Mon Sep 17 00:00:00 2001
From: crupest <crupest@outlook.com>
Date: Sat, 16 Dec 2023 23:55:49 +0800
Subject: Add PlatformSubProcessBase.

---
 src/common/SubProcess.cpp | 51 +++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 51 insertions(+)

(limited to 'src/common/SubProcess.cpp')

diff --git a/src/common/SubProcess.cpp b/src/common/SubProcess.cpp
index 7b585e2f..f2bcf919 100644
--- a/src/common/SubProcess.cpp
+++ b/src/common/SubProcess.cpp
@@ -1,6 +1,57 @@
 #include "cru/common/SubProcess.h"
+#include "cru/common/Exception.h"
+
+#include <mutex>
 
 namespace cru {
+SubProcessException::SubProcessException(String message)
+    : Exception(std::move(message)) {}
+
+SubProcessException::~SubProcessException() {}
+
+PlatformSubProcessBase::PlatformSubProcessBase(
+    const PlatformSubProcessStartInfo& start_info)
+    : dispose_(DisposeKind::None),
+      start_info_(start_info),
+      data_lock_(data_mutex_, std::defer_lock),
+      process_lock_(process_mutex_, std::defer_lock) {}
+
+PlatformSubProcessBase::~PlatformSubProcessBase() {
+  auto data_lock_guard = CreateDataLockGuard();
+  if (status_ )
+    switch (dispose_) { DisposeKind::Join: }
+}
+
+void PlatformSubProcessBase::Start() {
+  auto data_lock_guard = CreateDataLockGuard();
+
+  if (status_ != PlatformSubProcessStatus::Prepare) {
+    throw SubProcessException(u"Sub-process has already run.");
+  }
+
+  status_ = PlatformSubProcessStatus::Running;
+
+  try {
+    PlatformCreateProcess();
+
+    process_thread_ = std::thread([this] {
+      auto process_lock_guard = CreateProcessLockGuard();
+      PlatformWaitForProcess();
+      auto data_lock_guard = CreateDataLockGuard();
+      status_ = PlatformSubProcessStatus::Exited;
+    });
+  } catch (const std::exception& e) {
+    status_ = PlatformSubProcessStatus::FailToStart;
+  }
+}
 
+void PlatformSubProcessBase::Wait(
+    std::optional<std::chrono::milliseconds> wait_time) {
+  if (wait_time) {
+    process_lock_.try_lock_for(*wait_time);
+  } else {
+    process_lock_.lock();
+  }
 }
 
+}  // namespace cru
-- 
cgit v1.2.3