tyler.durden
/
icingaweb2-module-director
mirror of https://github.com/Icinga/icingaweb2-module-director.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

TemplateTree: allow lookup for single objects too

This commit is contained in:
Thomas Gelf 2017-08-12 11:07:34 +02:00
parent 1356733332
commit 241bad7d81
1 changed files with 27 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -39,13 +39,17 @@ class TemplateTree
return $this->type;
}
public function listParentNamesForObject(IcingaObject $object)
public function listParentNamesFor(IcingaObject $object)
{
if ($this->objectMaps === null) {
$this->loadObjectMaps();
$id = (int) $object->getProperty('id');
$this->requireTree();
if (array_key_exists($id, $this->parents)) {
return array_values($this->parents[$id]);
}
$id = $object->getProperty('id');
$this->requireObjectMaps();
$parents = [];
if (array_key_exists($id, $this->objectMaps)) {
foreach ($this->objectMaps[$id] as $pid) {
@ -74,11 +78,11 @@ class TemplateTree
)->order('i.weight');
foreach ($db->fetchAll($query) as $row) {
$id = $row->object;
$id = (int) $row->object;
if (! array_key_exists($id, $map)) {
$map[$id] = [];
}
$map[$id][] = $row->parent;
$map[$id][] = (int) $row->parent;
}
$this->objectMaps = $map;
@ -124,12 +128,29 @@ class TemplateTree
}
}
protected function requireObjectMaps()
{
if ($this->objectMaps === null) {
$this->loadObjectMaps();
}
}
public function getParentsById($id)
{
$this->requireTree();
if (array_key_exists($id, $this->parents)) {
return $this->parents[$id];
}
$this->requireObjectMaps();
if (array_key_exists($id, $this->objectMaps)) {
$parents = [];
foreach ($this->objectMaps[$id] as $pid) {
$parents[$pid] = $this->names[$pid];
}
return $parents;
} else {
return [];
}
@ -144,7 +165,6 @@ class TemplateTree
if (false !== ($key = array_search($name, $ancestors))) {
unset($ancestors[$key]);
}
$ancestors[$pid] = $name;
}