diff options
Diffstat (limited to 'library/Director/Resolver/HostServiceBlacklist.php')
-rw-r--r-- | library/Director/Resolver/HostServiceBlacklist.php | 91 |
1 files changed, 91 insertions, 0 deletions
diff --git a/library/Director/Resolver/HostServiceBlacklist.php b/library/Director/Resolver/HostServiceBlacklist.php new file mode 100644 index 0000000..606855a --- /dev/null +++ b/library/Director/Resolver/HostServiceBlacklist.php @@ -0,0 +1,91 @@ +<?php + +namespace Icinga\Module\Director\Resolver; + +use Icinga\Module\Director\Db; +use Icinga\Module\Director\Objects\IcingaService; + +class HostServiceBlacklist +{ + /** @var Db */ + protected $db; + + protected $table = 'icinga_host_service_blacklist'; + + protected $mappings; + + public function __construct(Db $db) + { + $this->db = $db; + } + + protected function loadMappings() + { + $db = $this->db->getDbAdapter(); + $query = $db->select()->from(['hsb' => $this->table], [ + 'host_name' => 'h.object_name', + 'service_id' => 'hsb.service_id' + ])->join( + ['h' => 'icinga_host'], + 'hsb.host_id = h.id', + [] + ); + + $result = []; + foreach ($db->fetchAll($query) as $row) { + if (array_key_exists($row->service_id, $result)) { + $result[$row->service_id][] = $row->host_name; + } else { + $result[$row->service_id] = [$row->host_name]; + } + } + + return $result; + } + + public function preloadMappings() + { + $this->mappings = $this->loadMappings(); + + return $this; + } + + public function getBlacklistedHostnamesForService(IcingaService $service) + { + if ($this->mappings === null) { + return $this->fetchMappingsForService($service); + } else { + return $this->getPreLoadedMappingsForService($service); + } + } + + public function fetchMappingsForService(IcingaService $service) + { + if (! $service->hasBeenLoadedFromDb() || $service->get('id') === null) { + return []; + } + + $db = $this->db->getDbAdapter(); + $query = $db->select()->from(['hsb' => $this->table], [ + 'host_name' => 'h.object_name', + 'service_id' => 'hsb.service_id' + ])->join( + ['h' => 'icinga_host'], + 'hsb.host_id = h.id', + [] + )->where('hsb.service_id = ?', $service->get('id')); + + return $db->fetchCol($query); + } + + public function getPreLoadedMappingsForService(IcingaService $service) + { + if ($this->mappings !== null + && array_key_exists($service->get('id'), $this->mappings) + ) { + return $this->mappings[$service->get('id')]; + } + + return []; + } +} |