InConditionBuilder.php 5.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172
  1. <?php
  2. /**
  3. * @link http://www.yiiframework.com/
  4. * @copyright Copyright (c) 2008 Yii Software LLC
  5. * @license http://www.yiiframework.com/license/
  6. */
  7. namespace yii\db\conditions;
  8. use yii\db\ExpressionBuilderInterface;
  9. use yii\db\ExpressionBuilderTrait;
  10. use yii\db\ExpressionInterface;
  11. use yii\db\Query;
  12. /**
  13. * Class InConditionBuilder builds objects of [[InCondition]]
  14. *
  15. * @author Dmytro Naumenko <d.naumenko.a@gmail.com>
  16. * @since 2.0.14
  17. */
  18. class InConditionBuilder implements ExpressionBuilderInterface
  19. {
  20. use ExpressionBuilderTrait;
  21. /**
  22. * Method builds the raw SQL from the $expression that will not be additionally
  23. * escaped or quoted.
  24. *
  25. * @param ExpressionInterface|InCondition $expression the expression to be built.
  26. * @param array $params the binding parameters.
  27. * @return string the raw SQL that will not be additionally escaped or quoted.
  28. */
  29. public function build(ExpressionInterface $expression, array &$params = [])
  30. {
  31. $operator = $expression->getOperator();
  32. $column = $expression->getColumn();
  33. $values = $expression->getValues();
  34. if ($column === []) {
  35. // no columns to test against
  36. return $operator === 'IN' ? '0=1' : '';
  37. }
  38. if ($values instanceof Query) {
  39. return $this->buildSubqueryInCondition($operator, $column, $values, $params);
  40. }
  41. if (!is_array($values) && !$values instanceof \Traversable) {
  42. // ensure values is an array
  43. $values = (array) $values;
  44. }
  45. if ($column instanceof \Traversable || ((is_array($column) || $column instanceof \Countable) && count($column) > 1)) {
  46. return $this->buildCompositeInCondition($operator, $column, $values, $params);
  47. }
  48. if (is_array($column)) {
  49. $column = reset($column);
  50. }
  51. $sqlValues = $this->buildValues($expression, $values, $params);
  52. if (empty($sqlValues)) {
  53. return $operator === 'IN' ? '0=1' : '';
  54. }
  55. if (strpos($column, '(') === false) {
  56. $column = $this->queryBuilder->db->quoteColumnName($column);
  57. }
  58. if (count($sqlValues) > 1) {
  59. return "$column $operator (" . implode(', ', $sqlValues) . ')';
  60. }
  61. $operator = $operator === 'IN' ? '=' : '<>';
  62. return $column . $operator . reset($sqlValues);
  63. }
  64. /**
  65. * Builds $values to be used in [[InCondition]]
  66. *
  67. * @param ConditionInterface|InCondition $condition
  68. * @param array $values
  69. * @param array $params the binding parameters
  70. * @return array of prepared for SQL placeholders
  71. */
  72. protected function buildValues(ConditionInterface $condition, $values, &$params)
  73. {
  74. $sqlValues = [];
  75. $column = $condition->getColumn();
  76. foreach ($values as $i => $value) {
  77. if (is_array($value) || $value instanceof \ArrayAccess) {
  78. $value = isset($value[$column]) ? $value[$column] : null;
  79. }
  80. if ($value === null) {
  81. $sqlValues[$i] = 'NULL';
  82. } elseif ($value instanceof ExpressionInterface) {
  83. $sqlValues[$i] = $this->queryBuilder->buildExpression($value, $params);
  84. } else {
  85. $sqlValues[$i] = $this->queryBuilder->bindParam($value, $params);
  86. }
  87. }
  88. return $sqlValues;
  89. }
  90. /**
  91. * Builds SQL for IN condition.
  92. *
  93. * @param string $operator
  94. * @param array|string $columns
  95. * @param Query $values
  96. * @param array $params
  97. * @return string SQL
  98. */
  99. protected function buildSubqueryInCondition($operator, $columns, $values, &$params)
  100. {
  101. $sql = $this->queryBuilder->buildExpression($values, $params);
  102. if (is_array($columns)) {
  103. foreach ($columns as $i => $col) {
  104. if (strpos($col, '(') === false) {
  105. $columns[$i] = $this->queryBuilder->db->quoteColumnName($col);
  106. }
  107. }
  108. return '(' . implode(', ', $columns) . ") $operator $sql";
  109. }
  110. if (strpos($columns, '(') === false) {
  111. $columns = $this->queryBuilder->db->quoteColumnName($columns);
  112. }
  113. return "$columns $operator $sql";
  114. }
  115. /**
  116. * Builds SQL for IN condition.
  117. *
  118. * @param string $operator
  119. * @param array|\Traversable $columns
  120. * @param array $values
  121. * @param array $params
  122. * @return string SQL
  123. */
  124. protected function buildCompositeInCondition($operator, $columns, $values, &$params)
  125. {
  126. $vss = [];
  127. foreach ($values as $value) {
  128. $vs = [];
  129. foreach ($columns as $column) {
  130. if (isset($value[$column])) {
  131. $vs[] = $this->queryBuilder->bindParam($value[$column], $params);
  132. } else {
  133. $vs[] = 'NULL';
  134. }
  135. }
  136. $vss[] = '(' . implode(', ', $vs) . ')';
  137. }
  138. if (empty($vss)) {
  139. return $operator === 'IN' ? '0=1' : '';
  140. }
  141. $sqlColumns = [];
  142. foreach ($columns as $i => $column) {
  143. $sqlColumns[] = strpos($column, '(') === false ? $this->queryBuilder->db->quoteColumnName($column) : $column;
  144. }
  145. return '(' . implode(', ', $sqlColumns) . ") $operator (" . implode(', ', $vss) . ')';
  146. }
  147. }