src/common/SubProcess.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

#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