389 lines
		
	
	
		
			11 KiB
		
	
	
	
		
			PHP
		
	
	
		
			Executable File
		
	
	
	
	
			
		
		
	
	
			389 lines
		
	
	
		
			11 KiB
		
	
	
	
		
			PHP
		
	
	
		
			Executable File
		
	
	
	
	
<?php
 | 
						|
 | 
						|
namespace Illuminate\Database\Query\Grammars;
 | 
						|
 | 
						|
use Illuminate\Database\Query\Builder;
 | 
						|
use Illuminate\Support\Arr;
 | 
						|
use Illuminate\Support\Str;
 | 
						|
 | 
						|
class PostgresGrammar extends Grammar
 | 
						|
{
 | 
						|
    /**
 | 
						|
     * All of the available clause operators.
 | 
						|
     *
 | 
						|
     * @var array
 | 
						|
     */
 | 
						|
    protected $operators = [
 | 
						|
        '=', '<', '>', '<=', '>=', '<>', '!=',
 | 
						|
        'like', 'not like', 'between', 'ilike', 'not ilike',
 | 
						|
        '~', '&', '|', '#', '<<', '>>', '<<=', '>>=',
 | 
						|
        '&&', '@>', '<@', '?', '?|', '?&', '||', '-', '-', '#-',
 | 
						|
        'is distinct from', 'is not distinct from',
 | 
						|
    ];
 | 
						|
 | 
						|
    /**
 | 
						|
     * {@inheritdoc}
 | 
						|
     *
 | 
						|
     * @param  \Illuminate\Database\Query\Builder  $query
 | 
						|
     * @param  array  $where
 | 
						|
     * @return string
 | 
						|
     */
 | 
						|
    protected function whereBasic(Builder $query, $where)
 | 
						|
    {
 | 
						|
        if (Str::contains(strtolower($where['operator']), 'like')) {
 | 
						|
            return sprintf(
 | 
						|
                '%s::text %s %s',
 | 
						|
                $this->wrap($where['column']),
 | 
						|
                $where['operator'],
 | 
						|
                $this->parameter($where['value'])
 | 
						|
            );
 | 
						|
        }
 | 
						|
 | 
						|
        return parent::whereBasic($query, $where);
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Compile a "where date" clause.
 | 
						|
     *
 | 
						|
     * @param  \Illuminate\Database\Query\Builder  $query
 | 
						|
     * @param  array  $where
 | 
						|
     * @return string
 | 
						|
     */
 | 
						|
    protected function whereDate(Builder $query, $where)
 | 
						|
    {
 | 
						|
        $value = $this->parameter($where['value']);
 | 
						|
 | 
						|
        return $this->wrap($where['column']).'::date '.$where['operator'].' '.$value;
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Compile a "where time" clause.
 | 
						|
     *
 | 
						|
     * @param  \Illuminate\Database\Query\Builder  $query
 | 
						|
     * @param  array  $where
 | 
						|
     * @return string
 | 
						|
     */
 | 
						|
    protected function whereTime(Builder $query, $where)
 | 
						|
    {
 | 
						|
        $value = $this->parameter($where['value']);
 | 
						|
 | 
						|
        return $this->wrap($where['column']).'::time '.$where['operator'].' '.$value;
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Compile a date based where clause.
 | 
						|
     *
 | 
						|
     * @param  string  $type
 | 
						|
     * @param  \Illuminate\Database\Query\Builder  $query
 | 
						|
     * @param  array  $where
 | 
						|
     * @return string
 | 
						|
     */
 | 
						|
    protected function dateBasedWhere($type, Builder $query, $where)
 | 
						|
    {
 | 
						|
        $value = $this->parameter($where['value']);
 | 
						|
 | 
						|
        return 'extract('.$type.' from '.$this->wrap($where['column']).') '.$where['operator'].' '.$value;
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Compile the "select *" portion of the query.
 | 
						|
     *
 | 
						|
     * @param  \Illuminate\Database\Query\Builder  $query
 | 
						|
     * @param  array  $columns
 | 
						|
     * @return string|null
 | 
						|
     */
 | 
						|
    protected function compileColumns(Builder $query, $columns)
 | 
						|
    {
 | 
						|
        // If the query is actually performing an aggregating select, we will let that
 | 
						|
        // compiler handle the building of the select clauses, as it will need some
 | 
						|
        // more syntax that is best handled by that function to keep things neat.
 | 
						|
        if (! is_null($query->aggregate)) {
 | 
						|
            return;
 | 
						|
        }
 | 
						|
 | 
						|
        if (is_array($query->distinct)) {
 | 
						|
            $select = 'select distinct on ('.$this->columnize($query->distinct).') ';
 | 
						|
        } elseif ($query->distinct) {
 | 
						|
            $select = 'select distinct ';
 | 
						|
        } else {
 | 
						|
            $select = 'select ';
 | 
						|
        }
 | 
						|
 | 
						|
        return $select.$this->columnize($columns);
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Compile a "JSON contains" statement into SQL.
 | 
						|
     *
 | 
						|
     * @param  string  $column
 | 
						|
     * @param  string  $value
 | 
						|
     * @return string
 | 
						|
     */
 | 
						|
    protected function compileJsonContains($column, $value)
 | 
						|
    {
 | 
						|
        $column = str_replace('->>', '->', $this->wrap($column));
 | 
						|
 | 
						|
        return '('.$column.')::jsonb @> '.$value;
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Compile a "JSON length" statement into SQL.
 | 
						|
     *
 | 
						|
     * @param  string  $column
 | 
						|
     * @param  string  $operator
 | 
						|
     * @param  string  $value
 | 
						|
     * @return string
 | 
						|
     */
 | 
						|
    protected function compileJsonLength($column, $operator, $value)
 | 
						|
    {
 | 
						|
        $column = str_replace('->>', '->', $this->wrap($column));
 | 
						|
 | 
						|
        return 'json_array_length(('.$column.')::json) '.$operator.' '.$value;
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Compile the lock into SQL.
 | 
						|
     *
 | 
						|
     * @param  \Illuminate\Database\Query\Builder  $query
 | 
						|
     * @param  bool|string  $value
 | 
						|
     * @return string
 | 
						|
     */
 | 
						|
    protected function compileLock(Builder $query, $value)
 | 
						|
    {
 | 
						|
        if (! is_string($value)) {
 | 
						|
            return $value ? 'for update' : 'for share';
 | 
						|
        }
 | 
						|
 | 
						|
        return $value;
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Compile an insert ignore statement into SQL.
 | 
						|
     *
 | 
						|
     * @param  \Illuminate\Database\Query\Builder  $query
 | 
						|
     * @param  array  $values
 | 
						|
     * @return string
 | 
						|
     */
 | 
						|
    public function compileInsertOrIgnore(Builder $query, array $values)
 | 
						|
    {
 | 
						|
        return $this->compileInsert($query, $values).' on conflict do nothing';
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Compile an insert and get ID statement into SQL.
 | 
						|
     *
 | 
						|
     * @param  \Illuminate\Database\Query\Builder  $query
 | 
						|
     * @param  array  $values
 | 
						|
     * @param  string  $sequence
 | 
						|
     * @return string
 | 
						|
     */
 | 
						|
    public function compileInsertGetId(Builder $query, $values, $sequence)
 | 
						|
    {
 | 
						|
        return $this->compileInsert($query, $values).' returning '.$this->wrap($sequence ?: 'id');
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Compile an update statement into SQL.
 | 
						|
     *
 | 
						|
     * @param  \Illuminate\Database\Query\Builder  $query
 | 
						|
     * @param  array  $values
 | 
						|
     * @return string
 | 
						|
     */
 | 
						|
    public function compileUpdate(Builder $query, array $values)
 | 
						|
    {
 | 
						|
        if (isset($query->joins) || isset($query->limit)) {
 | 
						|
            return $this->compileUpdateWithJoinsOrLimit($query, $values);
 | 
						|
        }
 | 
						|
 | 
						|
        return parent::compileUpdate($query, $values);
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Compile the columns for an update statement.
 | 
						|
     *
 | 
						|
     * @param  \Illuminate\Database\Query\Builder  $query
 | 
						|
     * @param  array  $values
 | 
						|
     * @return string
 | 
						|
     */
 | 
						|
    protected function compileUpdateColumns(Builder $query, array $values)
 | 
						|
    {
 | 
						|
        return collect($values)->map(function ($value, $key) {
 | 
						|
            $column = last(explode('.', $key));
 | 
						|
 | 
						|
            if ($this->isJsonSelector($key)) {
 | 
						|
                return $this->compileJsonUpdateColumn($column, $value);
 | 
						|
            }
 | 
						|
 | 
						|
            return $this->wrap($column).' = '.$this->parameter($value);
 | 
						|
        })->implode(', ');
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Prepares a JSON column being updated using the JSONB_SET function.
 | 
						|
     *
 | 
						|
     * @param  string  $key
 | 
						|
     * @param  mixed  $value
 | 
						|
     * @return string
 | 
						|
     */
 | 
						|
    protected function compileJsonUpdateColumn($key, $value)
 | 
						|
    {
 | 
						|
        $segments = explode('->', $key);
 | 
						|
 | 
						|
        $field = $this->wrap(array_shift($segments));
 | 
						|
 | 
						|
        $path = '\'{"'.implode('","', $segments).'"}\'';
 | 
						|
 | 
						|
        return "{$field} = jsonb_set({$field}::jsonb, {$path}, {$this->parameter($value)})";
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Compile an update statement with joins or limit into SQL.
 | 
						|
     *
 | 
						|
     * @param  \Illuminate\Database\Query\Builder  $query
 | 
						|
     * @param  array  $values
 | 
						|
     * @return string
 | 
						|
     */
 | 
						|
    protected function compileUpdateWithJoinsOrLimit(Builder $query, array $values)
 | 
						|
    {
 | 
						|
        $table = $this->wrapTable($query->from);
 | 
						|
 | 
						|
        $columns = $this->compileUpdateColumns($query, $values);
 | 
						|
 | 
						|
        $alias = last(preg_split('/\s+as\s+/i', $query->from));
 | 
						|
 | 
						|
        $selectSql = $this->compileSelect($query->select($alias.'.ctid'));
 | 
						|
 | 
						|
        return "update {$table} set {$columns} where {$this->wrap('ctid')} in ({$selectSql})";
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Prepare the bindings for an update statement.
 | 
						|
     *
 | 
						|
     * @param  array  $bindings
 | 
						|
     * @param  array  $values
 | 
						|
     * @return array
 | 
						|
     */
 | 
						|
    public function prepareBindingsForUpdate(array $bindings, array $values)
 | 
						|
    {
 | 
						|
        $values = collect($values)->map(function ($value, $column) {
 | 
						|
            return is_array($value) || ($this->isJsonSelector($column) && ! $this->isExpression($value))
 | 
						|
                ? json_encode($value)
 | 
						|
                : $value;
 | 
						|
        })->all();
 | 
						|
 | 
						|
        $cleanBindings = Arr::except($bindings, 'select');
 | 
						|
 | 
						|
        return array_values(
 | 
						|
            array_merge($values, Arr::flatten($cleanBindings))
 | 
						|
        );
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Compile a delete statement into SQL.
 | 
						|
     *
 | 
						|
     * @param  \Illuminate\Database\Query\Builder  $query
 | 
						|
     * @return string
 | 
						|
     */
 | 
						|
    public function compileDelete(Builder $query)
 | 
						|
    {
 | 
						|
        if (isset($query->joins) || isset($query->limit)) {
 | 
						|
            return $this->compileDeleteWithJoinsOrLimit($query);
 | 
						|
        }
 | 
						|
 | 
						|
        return parent::compileDelete($query);
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Compile a delete statement with joins or limit into SQL.
 | 
						|
     *
 | 
						|
     * @param  \Illuminate\Database\Query\Builder  $query
 | 
						|
     * @return string
 | 
						|
     */
 | 
						|
    protected function compileDeleteWithJoinsOrLimit(Builder $query)
 | 
						|
    {
 | 
						|
        $table = $this->wrapTable($query->from);
 | 
						|
 | 
						|
        $alias = last(preg_split('/\s+as\s+/i', $query->from));
 | 
						|
 | 
						|
        $selectSql = $this->compileSelect($query->select($alias.'.ctid'));
 | 
						|
 | 
						|
        return "delete from {$table} where {$this->wrap('ctid')} in ({$selectSql})";
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Compile a truncate table statement into SQL.
 | 
						|
     *
 | 
						|
     * @param  \Illuminate\Database\Query\Builder  $query
 | 
						|
     * @return array
 | 
						|
     */
 | 
						|
    public function compileTruncate(Builder $query)
 | 
						|
    {
 | 
						|
        return ['truncate '.$this->wrapTable($query->from).' restart identity cascade' => []];
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Wrap the given JSON selector.
 | 
						|
     *
 | 
						|
     * @param  string  $value
 | 
						|
     * @return string
 | 
						|
     */
 | 
						|
    protected function wrapJsonSelector($value)
 | 
						|
    {
 | 
						|
        $path = explode('->', $value);
 | 
						|
 | 
						|
        $field = $this->wrapSegments(explode('.', array_shift($path)));
 | 
						|
 | 
						|
        $wrappedPath = $this->wrapJsonPathAttributes($path);
 | 
						|
 | 
						|
        $attribute = array_pop($wrappedPath);
 | 
						|
 | 
						|
        if (! empty($wrappedPath)) {
 | 
						|
            return $field.'->'.implode('->', $wrappedPath).'->>'.$attribute;
 | 
						|
        }
 | 
						|
 | 
						|
        return $field.'->>'.$attribute;
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     *Wrap the given JSON selector for boolean values.
 | 
						|
     *
 | 
						|
     * @param  string  $value
 | 
						|
     * @return string
 | 
						|
     */
 | 
						|
    protected function wrapJsonBooleanSelector($value)
 | 
						|
    {
 | 
						|
        $selector = str_replace(
 | 
						|
            '->>', '->',
 | 
						|
            $this->wrapJsonSelector($value)
 | 
						|
        );
 | 
						|
 | 
						|
        return '('.$selector.')::jsonb';
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Wrap the given JSON boolean value.
 | 
						|
     *
 | 
						|
     * @param  string  $value
 | 
						|
     * @return string
 | 
						|
     */
 | 
						|
    protected function wrapJsonBooleanValue($value)
 | 
						|
    {
 | 
						|
        return "'".$value."'::jsonb";
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Wrap the attributes of the give JSON path.
 | 
						|
     *
 | 
						|
     * @param  array  $path
 | 
						|
     * @return array
 | 
						|
     */
 | 
						|
    protected function wrapJsonPathAttributes($path)
 | 
						|
    {
 | 
						|
        return array_map(function ($attribute) {
 | 
						|
            return filter_var($attribute, FILTER_VALIDATE_INT) !== false
 | 
						|
                        ? $attribute
 | 
						|
                        : "'$attribute'";
 | 
						|
        }, $path);
 | 
						|
    }
 | 
						|
}
 |