fab2s/YaEtl

<?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\Laravel\Extractors;

use fab2s\NodalFlow\NodalFlowException;
use fab2s\YaEtl\Extractors\PaginatedQueryInterface;
use fab2s\YaEtl\Extractors\PdoUniqueKeyExtractor;
use fab2s\YaEtl\Extractors\UniqueKeyExtractorAbstract;
use fab2s\YaEtl\YaEtlException;
use Illuminate\Database\Query\Builder;

/**
 * Class UniqueKeyExtractor
 */
class UniqueKeyExtractor extends PdoUniqueKeyExtractor implements PaginatedQueryInterface
{
    use DelayedExtractQueryTrait;

    /**
     * Generic extraction from tables with unique (composite) key
     *
     * @param ?Builder     $extractQuery
     * @param array|string $uniqueKey    can be either a unique key name as
     *                                   string ('id' by default, will be ordered asc) or an associative array :
     *                                   [
     *                                   'uniqueKeyName' => 'order', // eg 'asc' or 'desc'
     *                                   ]
     *                                   or, for a unique composite key :
     *                                   [
     *                                   'compositeKey1' => 'asc',
     *                                   'compositeKey2' => 'desc',
     *                                   // ...
     *                                   ]
     *
     * @throws YaEtlException
     * @throws NodalFlowException
     */
    public function __construct(?Builder $extractQuery = null, $uniqueKey = 'id')
    {
        if ($extractQuery !== null) {
            $this->setExtractQuery($extractQuery);
        }

        // delay configuring pdo to flow start
        UniqueKeyExtractorAbstract::__construct(null, $uniqueKey);
    }

    /**
     * This method sets offset and limit in the query
     *
     * @return string the paginated query with current offset and limit
     */
    public function getPaginatedQuery(): string
    {
        if ($this->joinFrom) {
            $extractQuery = $this->extractQuery
                ->whereIn($this->uniqueKeyName, $this->uniqueKeyValues);
        } else {
            $extractQuery = $this->extractQuery
                ->offset($this->offset)
                ->limit($this->batchSize);
        }

        $this->queryBindings = $extractQuery->getRawBindings();
        $this->queryBindings = $this->queryBindings['where'];

        return $extractQuery->toSql();
    }
}