blob: d69fb97316578bbfca893b105c129391c7c47184 (
plain)
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
|
#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)
: auto_delete_(false),
start_info_(start_info),
data_lock_(data_mutex_, std::defer_lock),
process_lock_(process_mutex_, std::defer_lock) {}
PlatformSubProcessBase::~PlatformSubProcessBase() {}
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;
});
process_thread_.detach();
} catch (const std::exception& e) {
status_ = PlatformSubProcessStatus::FailedToStart;
}
}
void PlatformSubProcessBase::Wait(
std::optional<std::chrono::milliseconds> wait_time) {
auto status = GetStatus();
if (status == PlatformSubProcessStatus::Prepare) {
throw SubProcessException(
u"The process does not start. Can't wait for it.");
}
if (status == PlatformSubProcessStatus::FailedToStart) {
throw SubProcessException(
u"The process failed to start. Can't wait for it.");
}
if (status == PlatformSubProcessStatus::Exited) {
return;
}
if (wait_time) {
auto locked = process_lock_.try_lock_for(*wait_time);
if (locked) {
process_lock_.unlock();
}
} else {
process_lock_.lock();
process_lock_.unlock();
}
}
void PlatformSubProcessBase::Kill() {
auto status = GetStatus();
if (status == PlatformSubProcessStatus::Prepare) {
throw SubProcessException(u"The process does not start. Can't kill it.");
}
if (status == PlatformSubProcessStatus::FailedToStart) {
throw SubProcessException(u"The process failed to start. Can't kill it.");
}
if (status == PlatformSubProcessStatus::Exited) {
return;
}
PlatformKillProcess();
}
PlatformSubProcessStatus PlatformSubProcessBase::GetStatus() {
auto data_lock_guard = CreateDataLockGuard();
return status_;
}
} // namespace cru
|
