tyler.durden/icingaweb2-module-director
1
0
Fork 0
mirror of https://github.com/Icinga/icingaweb2-module-director.git synced 2025-07-31 01:34:12 +02:00
Code Issues Packages Projects Releases Wiki Activity

TemplateTree: add possibility to resolve objects

Browse Source
This commit is contained in:
Thomas Gelf 2017-07-31 20:57:41 +02:00
parent 118dddf93a
commit 1d683f972d
1 changed files with 58 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

58
library/Director/Resolver/TemplateTree.php
View File

@ -2,6 +2,7 @@
namespace Icinga\Module\Director\Resolver;
use Icinga\Application\Benchmark;
use Icinga\Exception\NotImplementedError;
use Icinga\Module\Director\Db;
use Icinga\Module\Director\Objects\IcingaObject;
@ -20,6 +21,12 @@ class TemplateTree
protected $tree;
protected $type;
protected $objectMaps;
protected $names;
public function __construct($type, Db $connection)
{
$this->type = $type;
@ -32,6 +39,51 @@ class TemplateTree
return $this->type;
}
public function listParentNamesForObject(IcingaObject $object)
{
if ($this->objectMaps === null) {
$this->loadObjectMaps();
}
$id = $object->getProperty('id');
$parents = [];
if (array_key_exists($id, $this->objectMaps)) {
foreach ($this->objectMaps[$id] as $pid) {
$parents[] = $this->names[$pid];
}
}
return $parents;
}
protected function loadObjectMaps()
{
$this->requireTree();
$map = [];
$db = $this->db;
$type = $this->type;
$table = "icinga_${type}_inheritance";
$query = $db->select()->from(
['i' => $table],
[
'object' => "i.${type}_id",
'parent' => "i.parent_${type}_id",
]
)->order('i.weight');
foreach ($db->fetchAll($query) as $row) {
$id = $row->object;
if (! array_key_exists($id, $map)) {
$map[$id] = [];
}
$map[$id][] = $row->parent;
}
$this->objectMaps = $map;
}
public function listParentIdsFor(IcingaObject $object)
{
return array_keys($this->getParentsFor($object));
@ -176,13 +228,16 @@ class TemplateTree
protected function prepareTree()
{
Benchmark::measure(sprintf('Prepare "%s" Template Tree', $this->type));
$templates = $this->fetchTemplates();
$parents = [];
$rootNodes = [];
$children = [];
$names = [];
foreach ($templates as $row) {
$id = (int) $row->id;
$pid = (int) $row->parent_id;
$names[$id] = $row->name;
if (! array_key_exists($id, $parents)) {
$parents[$id] = [];
}
@ -195,6 +250,7 @@ class TemplateTree
continue;
}
$names[$pid] = $row->parent_name;
$parents[$id][$pid] = $row->parent_name;
if (! array_key_exists($pid, $children)) {
@ -207,6 +263,8 @@ class TemplateTree
$this->parents = $parents;
$this->children = $children;
$this->rootNodes = $rootNodes;
$this->names = $names;
Benchmark::measure(sprintf('"%s" Template Tree ready', $this->type));
}
public function fetchObjects()