v.0.1
This commit is contained in:
388
vendor/illuminate/database/Query/Grammars/PostgresGrammar.php
vendored
Executable file
388
vendor/illuminate/database/Query/Grammars/PostgresGrammar.php
vendored
Executable file
@ -0,0 +1,388 @@
|
||||
<?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);
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user