From fb3f3ed788daec80a36e2fc8081647f449d67b5c Mon Sep 17 00:00:00 2001
From: "Alexander A. Klimov" <alexander.klimov@icinga.com>
Date: Thu, 3 Nov 2016 15:06:37 +0100
Subject: [PATCH] IniRepository: add support for per-table datasources

refs #13034
---
 library/Icinga/Repository/IniRepository.php | 45 ++++++++++++---------
 1 file changed, 27 insertions(+), 18 deletions(-)

diff --git a/library/Icinga/Repository/IniRepository.php b/library/Icinga/Repository/IniRepository.php
index 7413971a1..1df35d7e1 100644
--- a/library/Icinga/Repository/IniRepository.php
+++ b/library/Icinga/Repository/IniRepository.php
@@ -25,11 +25,11 @@ use Icinga\Exception\StatementException;
 abstract class IniRepository extends Repository implements Extensible, Updatable, Reducible
 {
     /**
-     * The datasource being used
+     * Per-table datasources
      *
-     * @var Config
+     * @var Config[string]
      */
-    protected $ds;
+    protected $datasources = array();
 
     /**
      * The tables for which triggers are available when inserting, updating or deleting rows
@@ -179,16 +179,16 @@ abstract class IniRepository extends Repository implements Extensible, Updatable
     {
         $newData = $this->requireStatementColumns($target, $data);
         $config = $this->onInsert($target, new ConfigObject($newData));
-        $section = $this->extractSectionName($config);
+        $section = $this->extractSectionName($config, $target);
 
-        if ($this->ds->hasSection($section)) {
+        if ($this->getDataSource($target)->hasSection($section)) {
             throw new StatementException(t('Cannot insert. Section "%s" does already exist'), $section);
         }
 
-        $this->ds->setSection($section, $config);
+        $this->getDataSource($target)->setSection($section, $config);
 
         try {
-            $this->ds->saveIni();
+            $this->getDataSource($target)->saveIni();
         } catch (Exception $e) {
             throw new StatementException(t('Failed to insert. An error occurred: %s'), $e->getMessage());
         }
@@ -206,7 +206,7 @@ abstract class IniRepository extends Repository implements Extensible, Updatable
     public function update($target, array $data, Filter $filter = null)
     {
         $newData = $this->requireStatementColumns($target, $data);
-        $keyColumn = $this->ds->getConfigObject()->getKeyColumn();
+        $keyColumn = $this->getDataSource($target)->getConfigObject()->getKeyColumn();
         if ($filter === null && isset($newData[$keyColumn])) {
             throw new StatementException(
                 t('Cannot update. Column "%s" holds a section\'s name which must be unique'),
@@ -214,7 +214,7 @@ abstract class IniRepository extends Repository implements Extensible, Updatable
             );
         }
 
-        $query = $this->ds->select();
+        $query = $this->getDataSource($target)->select();
         if ($filter !== null) {
             $query->addFilter($this->requireFilter($target, $filter));
         }
@@ -242,16 +242,16 @@ abstract class IniRepository extends Repository implements Extensible, Updatable
             unset($newConfig->$keyColumn);
 
             if ($newSection) {
-                if ($this->ds->hasSection($newSection)) {
+                if ($this->getDataSource($target)->hasSection($newSection)) {
                     throw new StatementException(t('Cannot update. Section "%s" does already exist'), $newSection);
                 }
 
-                $this->ds->removeSection($section)->setSection(
+                $this->getDataSource($target)->removeSection($section)->setSection(
                     $newSection,
                     $this->onUpdate($target, $config, $newConfig)
                 );
             } else {
-                $this->ds->setSection(
+                $this->getDataSource($target)->setSection(
                     $section,
                     $this->onUpdate($target, $config, $newConfig)
                 );
@@ -259,7 +259,7 @@ abstract class IniRepository extends Repository implements Extensible, Updatable
         }
 
         try {
-            $this->ds->saveIni();
+            $this->getDataSource($target)->saveIni();
         } catch (Exception $e) {
             throw new StatementException(t('Failed to update. An error occurred: %s'), $e->getMessage());
         }
@@ -275,19 +275,19 @@ abstract class IniRepository extends Repository implements Extensible, Updatable
      */
     public function delete($target, Filter $filter = null)
     {
-        $query = $this->ds->select();
+        $query = $this->getDataSource($target)->select();
         if ($filter !== null) {
             $query->addFilter($this->requireFilter($target, $filter));
         }
 
         /** @var ConfigObject $config */
         foreach ($query as $section => $config) {
-            $this->ds->removeSection($section);
+            $this->getDataSource($target)->removeSection($section);
             $this->onDelete($target, $config);
         }
 
         try {
-            $this->ds->saveIni();
+            $this->getDataSource($target)->saveIni();
         } catch (Exception $e) {
             throw new StatementException(t('Failed to delete. An error occurred: %s'), $e->getMessage());
         }
@@ -297,14 +297,15 @@ abstract class IniRepository extends Repository implements Extensible, Updatable
      * Extract and return the section name off of the given $config
      *
      * @param   array|ConfigObject  $config
+     * @param   string              $target The table whose datasource to get the key column from
      *
      * @return  string
      *
      * @throws  ProgrammingError    In case no valid section name is available
      */
-    protected function extractSectionName( & $config)
+    protected function extractSectionName( & $config, $target)
     {
-        $keyColumn = $this->ds->getConfigObject()->getKeyColumn();
+        $keyColumn = $this->getDataSource($target)->getConfigObject()->getKeyColumn();
         if (! is_array($config) && !$config instanceof ConfigObject) {
             throw new ProgrammingError('$config is neither an array nor a ConfigObject');
         } elseif (! isset($config[$keyColumn])) {
@@ -315,4 +316,12 @@ abstract class IniRepository extends Repository implements Extensible, Updatable
         unset($config[$keyColumn]);
         return $section;
     }
+
+    /**
+     * {@inheritDoc}
+     */
+    public function getDataSource($table = null)
+    {
+        return isset($this->datasources[$table]) ? $this->datasources[$table] : parent::getDataSource($table);
+    }
 }