edgardmessias\db\ibm\db2\Connection

Complexity

Total Complexity

5

Size/Duplication

Total Lines	75
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	1

Test Coverage

Coverage

91.67%

Importance

Changes

0

1 Method

Rating	Name	Duplication	Size	Complexity
A	initConnection()	0	18	5

<?php

/**
 * @link http://www.yiiframework.com/
 * @copyright Copyright (c) 2008 Yii Software LLC
 * @license http://www.yiiframework.com/license/
 */

namespace edgardmessias\db\ibm\db2;

use PDO;

/**
 * @author Edgard Messias <edgardmessias@gmail.com>
 * @since 1.0
 */
class Connection extends \yii\db\Connection
{

    /**
     * @inheritdoc
     */
    public $commandClass = 'edgardmessias\db\ibm\db2\Command';

    /**
     * @var bool|null set to true if working on iSeries
     */

    public $isISeries = null;


    /**
     * @var string need to be set if isISeries is set to true
     */

    public $defaultSchema;


    /**
     * @var array PDO attributes (name => value) that should be set when calling [[open()]]
     * to establish a DB connection. Please refer to the
     * [PHP manual](http://www.php.net/manual/en/function.PDO-setAttribute.php) for
     * details about available attributes.
     */
    public $attributes = [
        PDO::ATTR_CASE => PDO::CASE_NATURAL,
        PDO::ATTR_STRINGIFY_FETCHES => true,
    ];
    
    /**
     * @var array mapping between PDO driver names and [[Schema]] classes.
     * The keys of the array are PDO driver names while the values the corresponding
     * schema class name or configuration. Please refer to [[Yii::createObject()]] for
     * details on how to specify a configuration.
     *
     * This property is mainly used by [[getSchema()]] when fetching the database schema information.
     * You normally do not need to set this property unless you want to use your own
     * [[Schema]] class to support DBMS that is not supported by Yii.
     */
    public $schemaMap = [
        'ibm'   => 'edgardmessias\db\ibm\db2\Schema', // IBM DB2
        'odbc'   => 'edgardmessias\db\ibm\db2\Schema', // IBM DB2 ODBC
    ];
    
    /**
     * Initializes the DB connection.
     * This method is invoked right after the DB connection is established.
     * The default implementation turns on `PDO::ATTR_EMULATE_PREPARES`
     * if [[emulatePrepare]] is true, and sets the database [[charset]] if it is not empty.
     * It then triggers an [[EVENT_AFTER_OPEN]] event.
     */
    protected function initConnection()
    {
        parent::initConnection();

        if($this->isISeries === null){
            try {
                $stmt = $this->pdo->query('SELECT * FROM QSYS2.SYSTABLES FETCH FIRST 1 ROW ONLY');
                $this->isISeries = boolval($stmt);
            } catch (\Exception $ex) {
                $this->isISeries = false;
            }
        }

        if($this->defaultSchema && !$this->isISeries){
            $this->pdo->exec('SET CURRENT SCHEMA ' . $this->pdo->quote($this->defaultSchema));
        }
        
    }

}