vendor/doctrine/dbal/lib/Doctrine/DBAL/Schema/DB2SchemaManager.php

   1 <?php
   2 /*
   3  *  $Id$
   4  *
   5  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
   6  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
   7  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
   8  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
   9  * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  10  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  11  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  12  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  13  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  14  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  15  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  16  *
  17  * This software consists of voluntary contributions made by many individuals
  18  * and is licensed under the MIT license. For more information, see
  19  * <http://www.doctrine-project.org>.
  20 */
  21
  22 namespace Doctrine\DBAL\Schema;
  23
  24 use Doctrine\DBAL\Event\SchemaIndexDefinitionEventArgs;
  25 use Doctrine\DBAL\Events;
  26
  27 /**
  28  * IBM Db2 Schema Manager
  29  *
  30  * @license     http://www.opensource.org/licenses/lgpl-license.php LGPL
  31  * @link        www.doctrine-project.com
  32  * @since       1.0
  33  * @version     $Revision$
  34  * @author      Benjamin Eberlei <kontakt@beberlei.de>
  35  */
  36 class DB2SchemaManager extends AbstractSchemaManager
  37 {
  38     /**
  39      * Return a list of all tables in the current database
  40      *
  41      * Apparently creator is the schema not the user who created it:
  42      * {@link http://publib.boulder.ibm.com/infocenter/dzichelp/v2r2/index.jsp?topic=/com.ibm.db29.doc.sqlref/db2z_sysibmsystablestable.htm}
  43      *
  44      * @return array
  45      */
  46     public function listTableNames()
  47     {
  48         $sql = $this->_platform->getListTablesSQL();
  49         $sql .= " AND CREATOR = UPPER('".$this->_conn->getUsername()."')";
  50
  51         $tables = $this->_conn->fetchAll($sql);
  52
  53         return $this->_getPortableTablesList($tables);
  54     }
  55
  56
  57     /**
  58      * Get Table Column Definition
  59      *
  60      * @param array $tableColumn
  61      * @return Column
  62      */
  63     protected function _getPortableTableColumnDefinition($tableColumn)
  64     {
  65         $tableColumn = array_change_key_case($tableColumn, \CASE_LOWER);
  66
  67         $length = null;
  68         $fixed = null;
  69         $unsigned = false;
  70         $scale = false;
  71         $precision = false;
  72
  73         $type = $this->_platform->getDoctrineTypeMapping($tableColumn['typename']);
  74
  75         switch (strtolower($tableColumn['typename'])) {
  76             case 'varchar':
  77                 $length = $tableColumn['length'];
  78                 $fixed = false;
  79                 break;
  80             case 'character':
  81                 $length = $tableColumn['length'];
  82                 $fixed = true;
  83                 break;
  84             case 'clob':
  85                 $length = $tableColumn['length'];
  86                 break;
  87             case 'decimal':
  88             case 'double':
  89             case 'real':
  90                 $scale = $tableColumn['scale'];
  91                 $precision = $tableColumn['length'];
  92                 break;
  93         }
  94
  95         $options = array(
  96             'length'        => $length,
  97             'unsigned'      => (bool)$unsigned,
  98             'fixed'         => (bool)$fixed,
  99             'default'       => ($tableColumn['default'] == "NULL") ? null : $tableColumn['default'],
 100             'notnull'       => (bool) ($tableColumn['nulls'] == 'N'),
 101             'scale'         => null,
 102             'precision'     => null,
 103             'platformOptions' => array(),
 104         );
 105
 106         if ($scale !== null && $precision !== null) {
 107             $options['scale'] = $scale;
 108             $options['precision'] = $precision;
 109         }
 110
 111         return new Column($tableColumn['colname'], \Doctrine\DBAL\Types\Type::getType($type), $options);
 112     }
 113
 114     protected function _getPortableTablesList($tables)
 115     {
 116         $tableNames = array();
 117         foreach ($tables as $tableRow) {
 118             $tableRow = array_change_key_case($tableRow, \CASE_LOWER);
 119             $tableNames[] = $tableRow['name'];
 120         }
 121         return $tableNames;
 122     }
 123
 124     protected function _getPortableTableIndexesList($tableIndexes, $tableName=null)
 125     {
 126         $eventManager = $this->_platform->getEventManager();
 127
 128         $indexes = array();
 129         foreach($tableIndexes as $indexKey => $data) {
 130             $data = array_change_key_case($data, \CASE_LOWER);
 131             $unique = ($data['uniquerule'] == "D") ? false : true;
 132             $primary = ($data['uniquerule'] == "P");
 133
 134             $indexName = strtolower($data['name']);
 135             if ($primary) {
 136                 $keyName = 'primary';
 137             } else {
 138                 $keyName = $indexName;
 139             }
 140
 141             $data = array(
 142                 'name' => $indexName,
 143                 'columns' => explode("+", ltrim($data['colnames'], '+')),
 144                 'unique' => $unique,
 145                 'primary' => $primary
 146             );
 147
 148             $index = null;
 149             $defaultPrevented = false;
 150
 151             if (null !== $eventManager && $eventManager->hasListeners(Events::onSchemaIndexDefinition)) {
 152                 $eventArgs = new SchemaIndexDefinitionEventArgs($data, $tableName, $this->_conn);
 153                 $eventManager->dispatchEvent(Events::onSchemaIndexDefinition, $eventArgs);
 154
 155                 $defaultPrevented = $eventArgs->isDefaultPrevented();
 156                 $index = $eventArgs->getIndex();
 157             }
 158
 159             if ( ! $defaultPrevented) {
 160                 $index = new Index($data['name'], $data['columns'], $data['unique'], $data['primary']);
 161             }
 162
 163             if ($index) {
 164                 $indexes[$indexKey] = $index;
 165             }
 166         }
 167
 168         return $indexes;
 169     }
 170
 171     protected function _getPortableTableForeignKeyDefinition($tableForeignKey)
 172     {
 173         $tableForeignKey = array_change_key_case($tableForeignKey, CASE_LOWER);
 174
 175         $tableForeignKey['deleterule'] = $this->_getPortableForeignKeyRuleDef($tableForeignKey['deleterule']);
 176         $tableForeignKey['updaterule'] = $this->_getPortableForeignKeyRuleDef($tableForeignKey['updaterule']);
 177
 178         return new ForeignKeyConstraint(
 179             array_map('trim', (array)$tableForeignKey['fkcolnames']),
 180             $tableForeignKey['reftbname'],
 181             array_map('trim', (array)$tableForeignKey['pkcolnames']),
 182             $tableForeignKey['relname'],
 183             array(
 184                 'onUpdate' => $tableForeignKey['updaterule'],
 185                 'onDelete' => $tableForeignKey['deleterule'],
 186             )
 187         );
 188     }
 189
 190     protected function _getPortableForeignKeyRuleDef($def)
 191     {
 192         if ($def == "C") {
 193             return "CASCADE";
 194         } else if ($def == "N") {
 195             return "SET NULL";
 196         }
 197         return null;
 198     }
 199
 200     protected function _getPortableViewDefinition($view)
 201     {
 202         $view = array_change_key_case($view, \CASE_LOWER);
 203         // sadly this still segfaults on PDO_IBM, see http://pecl.php.net/bugs/bug.php?id=17199
 204         //$view['text'] = (is_resource($view['text']) ? stream_get_contents($view['text']) : $view['text']);
 205         if (!is_resource($view['text'])) {
 206             $pos = strpos($view['text'], ' AS ');
 207             $sql = substr($view['text'], $pos+4);
 208         } else {
 209             $sql = '';
 210         }
 211
 212         return new View($view['name'], $sql);
 213     }
 214 }