Yiisoft\Db\Mssql\BatchQueryResult

Complexity

Total Complexity

6

Size/Duplication

Total Lines	31
Duplicated Lines	0 %

Test Coverage

Coverage

100%

Importance

Changes	2
Bugs	0	Features	0

1 Method

Rating	Name	Duplication	Size	Complexity
A	getRows()	0	19	6

<?php

declare(strict_types=1);

namespace Yiisoft\Db\Mssql;

use PDOException;
use Yiisoft\Db\Query\BatchQueryResult as BaseBatchQueryResult;

class BatchQueryResult extends BaseBatchQueryResult
{
    /**
     * @var int MSSQL error code for exception that is thrown when last batch is size less than specified batch size
     *
     * {@see https://github.com/yiisoft/yii2/issues/10023}
     */
    private int $mssqlNoMoreRowsErrorCode = -13;

    /**
     * Reads and collects rows for batch.
     */
    protected function getRows(): array
    {
        $rows = [];
        $count = 0;

        try {
            do {
                $this->dataReader?->next();
                /** @psalm-var array|bool $row */
                $row = $this->dataReader?->current();
            } while ($row && ($rows[] = $row) && ++$count < $this->batchSize);
        } catch (PDOException $e) {
            /** @psalm-var array{0:string, 1:int|null, 2:string} $errorInfo */
            $errorInfo = $e->errorInfo;
            $errorInfo[1] ??= null;
            $rows = $errorInfo[1] !== $this->mssqlNoMoreRowsErrorCode ? throw $e : [];
        }

        return $rows;
    }
}