-
Notifications
You must be signed in to change notification settings - Fork 3.4k
/
UpdateQuery.php
137 lines (124 loc) · 4.01 KB
/
UpdateQuery.php
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
<?php
declare(strict_types=1);
/**
* CakePHP(tm) : Rapid Development Framework (https://cakephp.org)
* Copyright (c) Cake Software Foundation, Inc. (https://cakefoundation.org)
*
* Licensed under The MIT License
* For full copyright and license information, please see the LICENSE.txt
* Redistributions of files must retain the above copyright notice.
*
* @copyright Copyright (c) Cake Software Foundation, Inc. (https://cakefoundation.org)
* @link https://cakephp.org CakePHP(tm) Project
* @since 4.5.0
* @license https://opensource.org/licenses/mit-license.php MIT License
*/
namespace Cake\Database\Query;
use Cake\Database\Expression\QueryExpression;
use Cake\Database\ExpressionInterface;
use Cake\Database\Query;
use Closure;
/**
* This class is used to generate UPDATE queries for the relational database.
*/
class UpdateQuery extends Query
{
/**
* Type of this query.
*
* @var string
*/
protected string $_type = self::TYPE_UPDATE;
/**
* List of SQL parts that will be used to build this query.
*
* @var array<string, mixed>
*/
protected array $_parts = [
'comment' => null,
'with' => [],
'update' => [],
'modifier' => [],
'join' => [],
'set' => [],
'where' => null,
'order' => null,
'limit' => null,
'epilog' => null,
];
/**
* Create an update query.
*
* Can be combined with set() and where() methods to create update queries.
*
* @param \Cake\Database\ExpressionInterface|string $table The table you want to update.
* @return $this
*/
public function update(ExpressionInterface|string $table)
{
$this->_dirty();
$this->_parts['update'][0] = $table;
return $this;
}
/**
* Set one or many fields to update.
*
* ### Examples
*
* Passing a string:
*
* ```
* $query->update('articles')->set('title', 'The Title');
* ```
*
* Passing an array:
*
* ```
* $query->update('articles')->set(['title' => 'The Title'], ['title' => 'string']);
* ```
*
* Passing a callback:
*
* ```
* $query->update('articles')->set(function (ExpressionInterface $exp) {
* return $exp->eq('title', 'The title', 'string');
* });
* ```
*
* @param \Cake\Database\Expression\QueryExpression|\Closure|array|string $key The column name or array of keys
* + values to set. This can also be a QueryExpression containing a SQL fragment.
* It can also be a Closure, that is required to return an expression object.
* @param mixed $value The value to update $key to. Can be null if $key is an
* array or QueryExpression. When $key is an array, this parameter will be
* used as $types instead.
* @param array<string, string>|string $types The column types to treat data as.
* @return $this
*/
public function set(QueryExpression|Closure|array|string $key, mixed $value = null, array|string $types = [])
{
if (empty($this->_parts['set'])) {
$this->_parts['set'] = $this->newExpr()->setConjunction(',');
}
if ($key instanceof Closure) {
$exp = $this->newExpr()->setConjunction(',');
/** @var \Cake\Database\Expression\QueryExpression $setExpr */
$setExpr = $this->_parts['set'];
$setExpr->add($key($exp));
return $this;
}
if (is_array($key) || $key instanceof ExpressionInterface) {
$types = (array)$value;
/** @var \Cake\Database\Expression\QueryExpression $setExpr */
$setExpr = $this->_parts['set'];
$setExpr->add($key, $types);
return $this;
}
if (!is_string($types)) {
$types = null;
}
/** @var \Cake\Database\Expression\QueryExpression $setExpr */
$setExpr = $this->_parts['set'];
$setExpr->eq($key, $value, $types);
return $this;
}
}