-
Notifications
You must be signed in to change notification settings - Fork 1.2k
Expand file tree
/
Copy pathServerProcessSupervisor.cpp
More file actions
111 lines (93 loc) · 2.81 KB
/
Copy pathServerProcessSupervisor.cpp
File metadata and controls
111 lines (93 loc) · 2.81 KB
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
98
99
100
101
102
103
104
105
106
107
108
109
110
111
/*
* ServerProcessSupervisor.cpp
*
* Copyright (C) 2021 by RStudio, PBC
*
* Unless you have received this program directly from RStudio pursuant
* to the terms of a commercial license agreement with RStudio, then
* this program is licensed to you under the terms of version 3 of the
* GNU Affero General Public License. This program is distributed WITHOUT
* ANY EXPRESS OR IMPLIED WARRANTY, INCLUDING THOSE OF NON-INFRINGEMENT,
* MERCHANTABILITY OR FITNESS FOR A PARTICULAR PURPOSE. Please refer to the
* AGPL (http://www.gnu.org/licenses/agpl-3.0.txt) for more details.
*
*/
#include <server/ServerProcessSupervisor.hpp>
#include <shared_core/Error.hpp>
#include <core/DateTime.hpp>
#include <core/Thread.hpp>
#include <core/PeriodicCommand.hpp>
#include <core/system/Process.hpp>
#include <server/ServerScheduler.hpp>
using namespace rstudio::core;
namespace rstudio {
namespace server {
namespace process_supervisor {
namespace {
// mutex that protects access to the process supervisor's methods
boost::mutex s_mutex;
core::system::ProcessSupervisor& processSupervisor()
{
static core::system::ProcessSupervisor instance;
return instance;
}
bool pollProcessSupervisor()
{
LOCK_MUTEX(s_mutex)
{
processSupervisor().poll();
}
END_LOCK_MUTEX
return true;
}
} // anonymous namespace
Error runProgram(
const std::string& executable,
const std::vector<std::string>& args,
const std::string& input,
const core::system::ProcessOptions& options,
const boost::function<void(const core::system::ProcessResult&)>& onCompleted)
{
LOCK_MUTEX(s_mutex)
{
return processSupervisor().runProgram(executable,
args,
input,
options,
onCompleted);
}
END_LOCK_MUTEX
// fulfill closure and keep compiler happy
core::system::ProcessResult result;
result.exitStatus = EXIT_FAILURE;
result.stdErr = "Thread resource error occurred while running program " +
executable;
onCompleted(result);
return Success();
}
core::Error runProgram(
const std::string& executable,
const std::vector<std::string>& args,
const core::system::ProcessOptions& options,
const core::system::ProcessCallbacks& cb)
{
LOCK_MUTEX(s_mutex)
{
return processSupervisor().runProgram(
executable, args, options, cb);
}
END_LOCK_MUTEX
return Success();
}
Error initialize()
{
// periodically poll process supervisor
scheduler::addCommand(
boost::shared_ptr<ScheduledCommand>(new PeriodicCommand(
boost::posix_time::milliseconds(500), pollProcessSupervisor, false))
);
return Success();
}
} // namespace process_supervisor
} // namespace server
} // namespace rstudio