fab2s\YaEtl\Extractors\PdoExtractor

Complexity

Total Complexity

8

Size/Duplication

Total Lines	65
Duplicated Lines	4.62 %

Coupling/Cohesion

Components	1
Dependencies	2

Importance

Changes

0

4 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	6	1
A	__destruct()	0	7	2
B	fetchRecords()	3	22	4
A	getPaginatedQuery()	0	4	1

<?php

/*
 * This file is part of YaEtl.
 *     (c) Fabrice de Stefanis / https://github.com/fab2s/YaEtl
 * This source file is licensed under the MIT license which you will
 * find in the LICENSE file or at https://opensource.org/licenses/MIT
 */

namespace fab2s\YaEtl\Extractors;

/**
 * Class PdoExtractor
 */
class PdoExtractor extends DbExtractorAbstract
{
    use PdoExtractorTrait;

    /**
     * @param \PDO        $pdo
     * @param string|null $extractQuery
     */
    public function __construct(\PDO $pdo, $extractQuery = null)
    {
        $this->configurePdo($pdo);

        parent::__construct($extractQuery);
    }

    /**
     * leave no trace
     * implement here to allow easier overidding
     */
    public function __destruct()
    {
        if ($this->driverBufferedQuery) {
            // set driver state back to where we met
            $this->pdo->setAttribute(\PDO::MYSQL_ATTR_USE_BUFFERED_QUERY, true);
        }
    }

    /**
     * @return bool
     */
    public function fetchRecords()
    {
        $extractQuery = $this->getPaginatedQuery();

        $query = $this->pdo->prepare($extractQuery);
        if (!$query->execute(!empty($this->queryBindings) ? $this->queryBindings : null)) {
            return false;
        }

        $this->extracted = new \SplDoublyLinkedList;
        $hasRecord       = false;
        while ($record = $query->fetch(\PDO::FETCH_ASSOC)) {
            $this->extracted->push($record);
            $hasRecord = true;
        }

        $query->closeCursor();
        unset($query);
        $this->extracted->rewind();

        return $hasRecord;
    }

    /**
     * This method sets offset and limit in the query
     * WARNING : if you set an offset without limit,
     * the limit will be set to  $this->maxdefaultLimit
     *
     * @return string the paginated query with current offset and limit
     */
    protected function getPaginatedQuery()
    {
        return $this->extractQuery . $this->getLimitOffsetBit();
    }
}