blob: 7268913da7ddfea296a0cff74c576ac479710939 (
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
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
|
<?php
namespace Icinga\Module\Director\Cli;
use gipfl\Json\JsonDecodeException;
use gipfl\Json\JsonString;
use Icinga\Cli\Command as CliCommand;
use Icinga\Module\Director\Application\MemoryLimit;
use Icinga\Module\Director\Core\CoreApi;
use Icinga\Module\Director\Db;
use Icinga\Module\Director\Objects\IcingaEndpoint;
use Icinga\Application\Config;
use RuntimeException;
class Command extends CliCommand
{
/** @var Db */
protected $db;
/** @var CoreApi */
private $api;
protected function renderJson($object, $pretty = true)
{
return JsonString::encode($object, $pretty ? JSON_PRETTY_PRINT : null) . "\n";
}
/**
* @param $json
* @return mixed
*/
protected function parseJson($json)
{
try {
return JsonString::decode($json);
} catch (JsonDecodeException $e) {
$this->fail('Invalid JSON: %s', $e->getMessage());
}
}
/**
* @param string $msg
* @return never-return
*/
public function fail($msg)
{
$args = func_get_args();
array_shift($args);
if (count($args)) {
$msg = vsprintf($msg, $args);
}
echo $this->screen->colorize("ERROR", 'red') . ": $msg\n";
exit(1);
}
/**
* @param null $endpointName
* @return CoreApi|\Icinga\Module\Director\Core\LegacyDeploymentApi
* @throws \Icinga\Exception\NotFoundError
*/
protected function api($endpointName = null)
{
if ($this->api === null) {
if ($endpointName === null) {
$endpoint = $this->db()->getDeploymentEndpoint();
} else {
$endpoint = IcingaEndpoint::load($endpointName, $this->db());
}
$this->api = $endpoint->api();
}
return $this->api;
}
/**
* Raise PHP resource limits
*
* @return self;
*/
protected function raiseLimits()
{
MemoryLimit::raiseTo('1024M');
ini_set('max_execution_time', '0');
if (version_compare(PHP_VERSION, '7.0.0') < 0) {
ini_set('zend.enable_gc', '0');
}
return $this;
}
/**
* @return Db
*/
protected function db()
{
if ($this->db === null) {
$resourceName = $this->params->get('dbResourceName');
if ($resourceName === null) {
// Hint: not using $this->Config() intentionally. This allows
// CLI commands in other modules to use this as a base class.
$resourceName = Config::module('director')->get('db', 'resource');
}
if ($resourceName) {
$this->db = Db::fromResourceName($resourceName);
} else {
throw new RuntimeException('Director is not configured correctly');
}
}
return $this->db;
}
}
|
