150 lines
4.3 KiB
PHP
150 lines
4.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\pgsql;
|
|
|
|
use yii\db\ArrayExpression;
|
|
use yii\db\ExpressionBuilderInterface;
|
|
use yii\db\ExpressionBuilderTrait;
|
|
use yii\db\ExpressionInterface;
|
|
use yii\db\JsonExpression;
|
|
use yii\db\Query;
|
|
|
|
/**
|
|
* Class ArrayExpressionBuilder builds [[ArrayExpression]] for PostgreSQL DBMS.
|
|
*
|
|
* @author Dmytro Naumenko <d.naumenko.a@gmail.com>
|
|
* @since 2.0.14
|
|
*/
|
|
class ArrayExpressionBuilder implements ExpressionBuilderInterface
|
|
{
|
|
use ExpressionBuilderTrait;
|
|
|
|
|
|
/**
|
|
* {@inheritdoc}
|
|
* @param ArrayExpression|ExpressionInterface $expression the expression to be built
|
|
*/
|
|
public function build(ExpressionInterface $expression, array &$params = [])
|
|
{
|
|
$value = $expression->getValue();
|
|
if ($value === null) {
|
|
return 'NULL';
|
|
}
|
|
|
|
if ($value instanceof Query) {
|
|
list ($sql, $params) = $this->queryBuilder->build($value, $params);
|
|
return $this->buildSubqueryArray($sql, $expression);
|
|
}
|
|
|
|
$placeholders = $this->buildPlaceholders($expression, $params);
|
|
|
|
return 'ARRAY[' . implode(', ', $placeholders) . ']' . $this->getTypehint($expression);
|
|
}
|
|
|
|
/**
|
|
* Builds placeholders array out of $expression values
|
|
* @param ExpressionInterface|ArrayExpression $expression
|
|
* @param array $params the binding parameters.
|
|
* @return array
|
|
*/
|
|
protected function buildPlaceholders(ExpressionInterface $expression, &$params)
|
|
{
|
|
$value = $expression->getValue();
|
|
|
|
$placeholders = [];
|
|
if ($value === null || !is_array($value) && !$value instanceof \Traversable) {
|
|
return $placeholders;
|
|
}
|
|
|
|
if ($expression->getDimension() > 1) {
|
|
foreach ($value as $item) {
|
|
$placeholders[] = $this->build($this->unnestArrayExpression($expression, $item), $params);
|
|
}
|
|
return $placeholders;
|
|
}
|
|
|
|
foreach ($value as $item) {
|
|
if ($item instanceof Query) {
|
|
list ($sql, $params) = $this->queryBuilder->build($item, $params);
|
|
$placeholders[] = $this->buildSubqueryArray($sql, $expression);
|
|
continue;
|
|
}
|
|
|
|
$item = $this->typecastValue($expression, $item);
|
|
if ($item instanceof ExpressionInterface) {
|
|
$placeholders[] = $this->queryBuilder->buildExpression($item, $params);
|
|
continue;
|
|
}
|
|
|
|
$placeholders[] = $this->queryBuilder->bindParam($item, $params);
|
|
}
|
|
|
|
return $placeholders;
|
|
}
|
|
|
|
/**
|
|
* @param ArrayExpression $expression
|
|
* @param mixed $value
|
|
* @return ArrayExpression
|
|
*/
|
|
private function unnestArrayExpression(ArrayExpression $expression, $value)
|
|
{
|
|
$expressionClass = get_class($expression);
|
|
|
|
return new $expressionClass($value, $expression->getType(), $expression->getDimension()-1);
|
|
}
|
|
|
|
/**
|
|
* @param ArrayExpression $expression
|
|
* @return string the typecast expression based on [[type]].
|
|
*/
|
|
protected function getTypehint(ArrayExpression $expression)
|
|
{
|
|
if ($expression->getType() === null) {
|
|
return '';
|
|
}
|
|
|
|
$result = '::' . $expression->getType();
|
|
$result .= str_repeat('[]', $expression->getDimension());
|
|
|
|
return $result;
|
|
}
|
|
|
|
/**
|
|
* Build an array expression from a subquery SQL.
|
|
*
|
|
* @param string $sql the subquery SQL.
|
|
* @param ArrayExpression $expression
|
|
* @return string the subquery array expression.
|
|
*/
|
|
protected function buildSubqueryArray($sql, ArrayExpression $expression)
|
|
{
|
|
return 'ARRAY(' . $sql . ')' . $this->getTypehint($expression);
|
|
}
|
|
|
|
/**
|
|
* Casts $value to use in $expression
|
|
*
|
|
* @param ArrayExpression $expression
|
|
* @param mixed $value
|
|
* @return JsonExpression
|
|
*/
|
|
protected function typecastValue(ArrayExpression $expression, $value)
|
|
{
|
|
if ($value instanceof ExpressionInterface) {
|
|
return $value;
|
|
}
|
|
|
|
if (in_array($expression->getType(), [Schema::TYPE_JSON, Schema::TYPE_JSONB], true)) {
|
|
return new JsonExpression($value);
|
|
}
|
|
|
|
return $value;
|
|
}
|
|
}
|