ReFacTOR HostserviceproblemsummaryQuery

refs #9864

modules/monitoring/library/Monitoring/Backend/Ido/Query/HostserviceproblemsummaryQuery.php

@@ -3,9 +3,7 @@
 
 namespace Icinga\Module\Monitoring\Backend\Ido\Query;
 
-use Icinga\Data\Filter\Filter;
+class HostserviceproblemsummaryQuery extends IdoQuery
-
-class HostservicestatussummaryQuery extends IdoQuery
 {
     /**
      * {@inheritdoc}
@@ -13,18 +11,27 @@ class HostservicestatussummaryQuery extends IdoQuery
     protected $allowCustomVars = true;
 
     /**
-     * The HoststatusQuery
+     * The HoststatusQuery in use
      *
      * @var HoststatusQuery
      */
-    protected $hostQuery;
+    protected $hostStatusQuery;
 
     /**
      * {@inheritdoc}
      */
     protected $columnMap = array(
-        'statussummary' => array(
+        'services' => array(
             'host_name'                 => 'so.name1',
+            'service_description'       => 'so.name2'
+        ),
+        'hostgroups' => array(
+            'hostgroup_name'            => 'hgo.name1'
+        ),
+        'servicegroups' => array(
+            'servicegroup_name'         => 'sgo.name1'
+        ),
+        'problemsummary' => array(
             'unhandled_service_count'   => 'SUM(
                 CASE
                     WHEN (ss.problem_has_been_acknowledged + ss.scheduled_downtime_depth + COALESCE(hs.current_state, 0)) > 0
@@ -35,14 +42,26 @@ class HostservicestatussummaryQuery extends IdoQuery
         )
     );
 
+    /**
+     * Set the HoststatusQuery to use
+     *
+     * @param   HoststatusQuery     $query
+     *
+     * @return  $this
+     */
+    public function setHoststatusQuery(HoststatusQuery $query)
+    {
+        $this->hostStatusQuery = clone $query;
+        $this->hostStatusQuery->setIsSubQuery();
+        $this->hostStatusQuery->columns(array('object_id'));
+        return $this;
+    }
+
     /**
      * {@inheritdoc}
      */
     protected function joinBaseTables()
     {
-        $this->hostQuery = $this->createSubQuery('Hoststatus', array('object_id'));
-        $this->hostQuery->setIsSubQuery(); // TODO: Why is this necessary???
-
         $this->select->from(
             array('so' => $this->prefix . 'objects'),
             array()
@@ -50,7 +69,59 @@ class HostservicestatussummaryQuery extends IdoQuery
             array('s' => $this->prefix . 'services'),
             's.service_object_id = so.object_id',
             array()
-        )->join(
+        );
+
+        $this->select->group(array('so.name1'));
+        $this->select->where('so.is_active = 1');
+        $this->joinedVirtualTables['services'] = true;
+    }
+
+    /**
+     * Join host groups
+     */
+    protected function joinHostgroups()
+    {
+        $this->select->joinLeft(
+            array('hgm' => $this->prefix . 'hostgroup_members'),
+            'hgm.host_object_id = s.host_object_id',
+            array()
+        )->joinLeft(
+            array('hg' => $this->prefix . 'hostgroups'),
+            'hg.hostgroup_id = hgm.hostgroup_id',
+            array()
+        )->joinLeft(
+            array('hgo' => $this->prefix . 'objects'),
+            'hgo.object_id = hg.hostgroup_object_id AND hgo.is_active = 1 AND hgo.objecttype_id = 3',
+            array()
+        );
+    }
+
+    /**
+     * Join service groups
+     */
+    protected function joinServicegroups()
+    {
+        $this->select->joinLeft(
+            array('sgm' => $this->prefix . 'servicegroup_members'),
+            'sgm.service_object_id = so.object_id',
+            array()
+        )->joinLeft(
+            array('sg' => $this->prefix . 'servicegroups'),
+            'sgm.servicegroup_id = sg.servicegroup_id',
+            array()
+        )->joinLeft(
+            array('sgo' => $this->prefix . 'objects'),
+            'sgo.object_id = sg.servicegroup_object_id AND sgo.is_active = 1 AND sgo.objecttype_id = 4',
+            array()
+        );
+    }
+
+    /**
+     * Join the statussummary
+     */
+    protected function joinProblemsummary()
+    {
+        $this->select->join(
             array('ss' => $this->prefix . 'servicestatus'),
             'ss.service_object_id = so.object_id AND ss.current_state > 0',
             array()
@@ -59,50 +130,11 @@ class HostservicestatussummaryQuery extends IdoQuery
             'hs.host_object_id = s.host_object_id',
             array()
         )->join(
-            array('h' => $this->hostQuery),
+            array('h' => $this->hostStatusQuery),
             'h.object_id = s.host_object_id',
             array()
         );
 
-        $this->select->where('so.is_active = 1');
-        $this->select->group('so.name1');
         $this->select->having($this->getMappedField('unhandled_service_count') . ' > 0');
-        $this->joinedVirtualTables['statussummary'] = true;
-    }
-
-    /**
-     * {@inheritdoc}
-     */
-    public function addFilter(Filter $filter)
-    {
-        $this->hostQuery->addFilter($filter);
-        return $this;
-    }
-
-    /**
-     * {@inheritdoc}
-     */
-    public function order($columnOrAlias, $dir = null)
-    {
-        $this->hostQuery->order($columnOrAlias, $dir);
-        return $this;
-    }
-
-    /**
-     * {@inheritdoc}
-     */
-    public function where($condition, $value = null)
-    {
-        $this->hostQuery->where($condition, $value);
-        return $this;
-    }
-
-    /**
-     * {@inheritdoc}
-     */
-    public function limit($count = null, $offset = null)
-    {
-        $this->hostQuery->limit($count, $offset);
-        return $this;
     }
 }