WbmQueryManager\Services\QueryManagerDb

Complexity

Total Complexity

18

Size/Duplication

Total Lines	122
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	0

Importance

Changes

0

8 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	5	1
A	getConnection()	0	12	3
A	query()	0	8	2
A	getRowCount()	0	7	2
A	getColumnCount()	0	7	2
A	fetchAll()	0	13	4
A	nextResult()	0	7	2
A	close()	0	9	2

<?php
/**
 * Query Manager
 * Copyright (c) Webmatch GmbH
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 */

namespace WbmQueryManager\Services;

use Shopware\Components\DependencyInjection\Container;

if (!class_exists('Zend_Db_Adapter_Mysqli')) {
    require_once __DIR__ . '/../Library/Zend/Db/Adapter/Mysqli.php';
    require_once __DIR__ . '/../Library/Zend/Db/Adapter/Mysqli/Exception.php';
    require_once __DIR__ . '/../Library/Zend/Db/Statement/Mysqli.php';
    require_once __DIR__ . '/../Library/Zend/Db/Statement/Mysqli/Exception.php';
}

/**
 * Class QueryManagerDb
 * @package WbmQueryManager\Services
 */
class QueryManagerDb implements QueryManagerDbInterface {

    /**
     * @var Container
     */
    private $container;

    /**
     * @var \Enlight_Components_Db_Adapter_Pdo_Mysql
     */
    private $db;

    /**
     * QueryManagerDb constructor.
     * @param Container $container
     * @param \Enlight_Components_Db_Adapter_Pdo_Mysql $db
     */
    public function __construct(Container $container, \Enlight_Components_Db_Adapter_Pdo_Mysql $db)
    {
        $this->container = $container;
        $this->db = $db;
    }

    /**
     * @param $query
     * @return \Enlight_Components_Db_Adapter_Pdo_Mysql|\mysqli
     */
    private function getConnection($query)
    {
        if(function_exists('mysqli_connect')) {
            $config = $this->container->getParameter('shopware.db');
            $mysqli = new \Zend_Db_Adapter_Mysqli($config);
            $connection = $mysqli->getConnection();
            if($connection->multi_query($query)){
                return $connection;
            }
        }
        return $this->db;
    }

    /**
     * @param $query
     * @return \mysqli|\Zend_Db_Statement_Pdo
     */
    public function query($query)
    {
        $connection = $this->getConnection($query);
        if(get_class($connection) == 'mysqli'){
            return $connection;
        }
        return $connection->query($query);
    }

    /**
     * @param \mysqli|\Zend_Db_Statement_Pdo $query
     * @return int
     */
    public function getRowCount($query)
    {
        if(get_class($query) == 'mysqli'){
            return $query->affected_rows;
        }
        return $query->rowCount();
    }

    /**
     * @param \mysqli|\Zend_Db_Statement_Pdo $query
     * @return int
     */
    public function getColumnCount($query)
    {
        if(get_class($query) == 'mysqli'){
            return $query->field_count;
        }
        return $query->columnCount();
    }

    /**
     * @param \mysqli|\Zend_Db_Statement_Pdo $query
     * @return array
     */
    public function fetchAll($query)
    {
        if(get_class($query) == 'mysqli'){
            $results = array();
            if($result = $query->store_result()){
                while($row = $result->fetch_assoc()){
                    $results[] = $row;
                }
            }
            return $results;
        }
        return $query->fetchAll();
    }

    /**
     * @param \mysqli|\Zend_Db_Statement_Pdo $query
     * @return bool
     */
    public function nextResult($query)
    {
        if(get_class($query) == 'mysqli'){
            return $query->next_result();
        }
        return false;
    }

    /**
     * @param \mysqli|\Zend_Db_Statement_Pdo $query
     * @return bool
     */
    public function close($query)
    {
        if(get_class($query) == 'mysqli'){
            return $query->close();
        } else {
            $query->closeCursor();
        }
        return true;
    }

}