Yiisoft\Db\Mssql\Condition\InConditionBuilder

Complexity

Total Complexity

12

Size/Duplication

Total Lines	59
Duplicated Lines	0 %

Test Coverage

Coverage

90%

Importance

Changes	1
Bugs	0	Features	0

2 Methods

Rating	Name	Duplication	Size	Complexity
A	buildSubqueryInCondition()	0	7	2
B	buildCompositeInCondition()	0	26	10

<?php

declare(strict_types=1);

namespace Yiisoft\Db\Mssql\Condition;

use Traversable;
use Yiisoft\Db\Exception\Exception;
use Yiisoft\Db\Exception\InvalidArgumentException;
use Yiisoft\Db\Exception\InvalidConfigException;
use Yiisoft\Db\Exception\NotSupportedException;
use Yiisoft\Db\Mssql\Connection;
use Yiisoft\Db\Query\Conditions\InConditionBuilder as AbstractInConditionBuilder;
use Yiisoft\Db\Query\Query;

use function implode;
use function is_array;
use function strpos;

final class InConditionBuilder extends AbstractInConditionBuilder
{
    /**
     * Builds SQL for IN condition.
     *
     * @param string $operator
     * @param array|string $columns
     * @param Query $values
     * @param array $params
     *
     * @throws Exception|InvalidArgumentException|InvalidConfigException|NotSupportedException
     *
     * @return string SQL
     */
    protected function buildSubqueryInCondition(string $operator, $columns, Query $values, array &$params = []): string
    {
        if (is_array($columns)) {
            throw new NotSupportedException(__METHOD__ . ' is not supported by MSSQL.');
        }

        return parent::buildSubqueryInCondition($operator, $columns, $values, $params);
    }

    /**
     * Builds SQL for IN condition.
     *
     * @param string|null $operator
     * @param array|Traversable $columns
     * @param array|iterable $values
     * @param array $params
     *
     * @return string SQL
     */
    protected function buildCompositeInCondition(?string $operator, $columns, $values, array &$params = []): string
    {
        /** @var Connection $db */
        $db = $this->queryBuilder->getDb();

        $quotedColumns = [];
        foreach ($columns as $i => $column) {
            $quotedColumns[$i] = strpos($column, '(') === false
                ? $db->quoteColumnName($column) : $column;
        }

        $vss = [];
        foreach ($values as $value) {
            $vs = [];
            foreach ($columns as $i => $column) {
                if (isset($value[$column])) {
                    $phName = $this->queryBuilder->bindParam($value[$column], $params);
                    $vs[] = $quotedColumns[$i] . ($operator === 'IN' ? ' = ' : ' != ') . $phName;
                } else {
                    $vs[] = $quotedColumns[$i] . ($operator === 'IN' ? ' IS' : ' IS NOT') . ' NULL';
                }
            }
            $vss[] = '(' . implode($operator === 'IN' ? ' AND ' : ' OR ', $vs) . ')';
        }

        return '(' . implode($operator === 'IN' ? ' OR ' : ' AND ', $vss) . ')';
    }
}