173 lines
		
	
	
		
			5.3 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
			
		
		
	
	
			173 lines
		
	
	
		
			5.3 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
<?php
 | 
						|
/**
 | 
						|
 * @link http://www.yiiframework.com/
 | 
						|
 * @copyright Copyright (c) 2008 Yii Software LLC
 | 
						|
 * @license http://www.yiiframework.com/license/
 | 
						|
 */
 | 
						|
 | 
						|
namespace yii\db\conditions;
 | 
						|
 | 
						|
use yii\db\ExpressionBuilderInterface;
 | 
						|
use yii\db\ExpressionBuilderTrait;
 | 
						|
use yii\db\ExpressionInterface;
 | 
						|
use yii\db\Query;
 | 
						|
 | 
						|
/**
 | 
						|
 * Class InConditionBuilder builds objects of [[InCondition]]
 | 
						|
 *
 | 
						|
 * @author Dmytro Naumenko <d.naumenko.a@gmail.com>
 | 
						|
 * @since 2.0.14
 | 
						|
 */
 | 
						|
class InConditionBuilder implements ExpressionBuilderInterface
 | 
						|
{
 | 
						|
    use ExpressionBuilderTrait;
 | 
						|
 | 
						|
 | 
						|
    /**
 | 
						|
     * Method builds the raw SQL from the $expression that will not be additionally
 | 
						|
     * escaped or quoted.
 | 
						|
     *
 | 
						|
     * @param ExpressionInterface|InCondition $expression the expression to be built.
 | 
						|
     * @param array $params the binding parameters.
 | 
						|
     * @return string the raw SQL that will not be additionally escaped or quoted.
 | 
						|
     */
 | 
						|
    public function build(ExpressionInterface $expression, array &$params = [])
 | 
						|
    {
 | 
						|
        $operator = $expression->getOperator();
 | 
						|
        $column = $expression->getColumn();
 | 
						|
        $values = $expression->getValues();
 | 
						|
 | 
						|
        if ($column === []) {
 | 
						|
            // no columns to test against
 | 
						|
            return $operator === 'IN' ? '0=1' : '';
 | 
						|
        }
 | 
						|
 | 
						|
        if ($values instanceof Query) {
 | 
						|
            return $this->buildSubqueryInCondition($operator, $column, $values, $params);
 | 
						|
        }
 | 
						|
 | 
						|
        if (!is_array($values) && !$values instanceof \Traversable) {
 | 
						|
            // ensure values is an array
 | 
						|
            $values = (array) $values;
 | 
						|
        }
 | 
						|
        if ($column instanceof \Traversable || ((is_array($column) || $column instanceof \Countable) && count($column) > 1)) {
 | 
						|
            return $this->buildCompositeInCondition($operator, $column, $values, $params);
 | 
						|
        }
 | 
						|
 | 
						|
        if (is_array($column)) {
 | 
						|
            $column = reset($column);
 | 
						|
        }
 | 
						|
 | 
						|
        $sqlValues = $this->buildValues($expression, $values, $params);
 | 
						|
        if (empty($sqlValues)) {
 | 
						|
            return $operator === 'IN' ? '0=1' : '';
 | 
						|
        }
 | 
						|
 | 
						|
        if (strpos($column, '(') === false) {
 | 
						|
            $column = $this->queryBuilder->db->quoteColumnName($column);
 | 
						|
        }
 | 
						|
        if (count($sqlValues) > 1) {
 | 
						|
            return "$column $operator (" . implode(', ', $sqlValues) . ')';
 | 
						|
        }
 | 
						|
 | 
						|
        $operator = $operator === 'IN' ? '=' : '<>';
 | 
						|
 | 
						|
        return $column . $operator . reset($sqlValues);
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Builds $values to be used in [[InCondition]]
 | 
						|
     *
 | 
						|
     * @param ConditionInterface|InCondition $condition
 | 
						|
     * @param array $values
 | 
						|
     * @param array $params the binding parameters
 | 
						|
     * @return array of prepared for SQL placeholders
 | 
						|
     */
 | 
						|
    protected function buildValues(ConditionInterface $condition, $values, &$params)
 | 
						|
    {
 | 
						|
        $sqlValues = [];
 | 
						|
        $column = $condition->getColumn();
 | 
						|
 | 
						|
        foreach ($values as $i => $value) {
 | 
						|
            if (is_array($value) || $value instanceof \ArrayAccess) {
 | 
						|
                $value = isset($value[$column]) ? $value[$column] : null;
 | 
						|
            }
 | 
						|
            if ($value === null) {
 | 
						|
                $sqlValues[$i] = 'NULL';
 | 
						|
            } elseif ($value instanceof ExpressionInterface) {
 | 
						|
                $sqlValues[$i] = $this->queryBuilder->buildExpression($value, $params);
 | 
						|
            } else {
 | 
						|
                $sqlValues[$i] = $this->queryBuilder->bindParam($value, $params);
 | 
						|
            }
 | 
						|
        }
 | 
						|
 | 
						|
        return $sqlValues;
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Builds SQL for IN condition.
 | 
						|
     *
 | 
						|
     * @param string $operator
 | 
						|
     * @param array|string $columns
 | 
						|
     * @param Query $values
 | 
						|
     * @param array $params
 | 
						|
     * @return string SQL
 | 
						|
     */
 | 
						|
    protected function buildSubqueryInCondition($operator, $columns, $values, &$params)
 | 
						|
    {
 | 
						|
        $sql = $this->queryBuilder->buildExpression($values, $params);
 | 
						|
 | 
						|
        if (is_array($columns)) {
 | 
						|
            foreach ($columns as $i => $col) {
 | 
						|
                if (strpos($col, '(') === false) {
 | 
						|
                    $columns[$i] = $this->queryBuilder->db->quoteColumnName($col);
 | 
						|
                }
 | 
						|
            }
 | 
						|
 | 
						|
            return '(' . implode(', ', $columns) . ") $operator $sql";
 | 
						|
        }
 | 
						|
 | 
						|
        if (strpos($columns, '(') === false) {
 | 
						|
            $columns = $this->queryBuilder->db->quoteColumnName($columns);
 | 
						|
        }
 | 
						|
 | 
						|
        return "$columns $operator $sql";
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Builds SQL for IN condition.
 | 
						|
     *
 | 
						|
     * @param string $operator
 | 
						|
     * @param array|\Traversable $columns
 | 
						|
     * @param array $values
 | 
						|
     * @param array $params
 | 
						|
     * @return string SQL
 | 
						|
     */
 | 
						|
    protected function buildCompositeInCondition($operator, $columns, $values, &$params)
 | 
						|
    {
 | 
						|
        $vss = [];
 | 
						|
        foreach ($values as $value) {
 | 
						|
            $vs = [];
 | 
						|
            foreach ($columns as $column) {
 | 
						|
                if (isset($value[$column])) {
 | 
						|
                    $vs[] = $this->queryBuilder->bindParam($value[$column], $params);
 | 
						|
                } else {
 | 
						|
                    $vs[] = 'NULL';
 | 
						|
                }
 | 
						|
            }
 | 
						|
            $vss[] = '(' . implode(', ', $vs) . ')';
 | 
						|
        }
 | 
						|
 | 
						|
        if (empty($vss)) {
 | 
						|
            return $operator === 'IN' ? '0=1' : '';
 | 
						|
        }
 | 
						|
 | 
						|
        $sqlColumns = [];
 | 
						|
        foreach ($columns as $i => $column) {
 | 
						|
            $sqlColumns[] = strpos($column, '(') === false ? $this->queryBuilder->db->quoteColumnName($column) : $column;
 | 
						|
        }
 | 
						|
 | 
						|
        return '(' . implode(', ', $sqlColumns) . ") $operator (" . implode(', ', $vss) . ')';
 | 
						|
    }
 | 
						|
}
 |