背景
最近有需求實現類似于 QueryBuilder 的謂詞語句阿纤,就去翻看了它的源碼。先看兩個例子
例子1
$qb = $em->createQueryBuilder();
$qb->select('*')->from('User', 'u')->where('u.id = 1');
echo $qb->getDQL();
例子2
$qb = $em->createQueryBuilder();
$qb->select('*')
->from('User', 'u')
->where('u.id = 1')
->andWhere('u.score >= 90')
->orWhere('u.score <= 100');
echo $qb->getDQL();
是不是有點懵逼夷陋,希望看完我的文章欠拾,希望你有所收獲。
接下來會以例子2講解骗绕,分別解釋 where藐窄、andWhere、orWhere 方法酬土,圖文并茂荆忍,一步步教你理解上述PHP代碼轉換為sql語句的原理。
代碼不難,建議大家配合源碼食用刹枉。[相關源碼在此]https://github.com/doctrine/orm/blob/master/lib/Doctrine/ORM/QueryBuilder.php(https://github.com/doctrine/orm/blob/master/lib/Doctrine/ORM/QueryBuilder.php
)
添加謂詞語句
先看 QueryBuilder::where
方法叽唱,生成了一個Andx
謂詞對象 $predicates
,
public function where($predicates)
{
if ( ! (func_num_args() == 1 && $predicates instanceof Expr\Composite)) {
$predicates = new Expr\Andx(func_get_args());
}
return $this->add('where', $predicates);
}
執(zhí)行 var_export($predicates)
后查看成員變量如下嘶卧,separator
是 where 子句條件之間的連接符尔觉,有 AND 和 OR。parts
是由條件組成的數組芥吟。
// var_export($predicates)的輸出
Doctrine\ORM\Query\Expr\Andx::__set_state(array(
'separator' => ' AND ',
'allowedClasses' =>
array (
0 => 'Doctrine\\ORM\\Query\\Expr\\Comparison',
1 => 'Doctrine\\ORM\\Query\\Expr\\Func',
2 => 'Doctrine\\ORM\\Query\\Expr\\Orx',
3 => 'Doctrine\\ORM\\Query\\Expr\\Andx',
),
'preSeparator' => '(',
'postSeparator' => ')',
'parts' =>
array (
0 => 'u.id = 1',
),
))
接下來執(zhí)行的return $this->add('where', $predicates);
函數很簡單侦铜,把謂詞對象作為QueryBuilder::_dqlParts
中的key為where的value吉懊,打印QueryBuilder對象如下
// 此函數有刪減蕉鸳,完整請看官方源碼
// QueryBuilder的add方法
public function add($dqlPartName, $dqlPart, $append = false)
{
$isMultiple = is_array($this->_dqlParts[$dqlPartName])
&& !($dqlPartName == 'join' && !$append);
// $isMultiple 這里的值為 false
$this->_dqlParts[$dqlPartName] = ($isMultiple) ? [$dqlPart] : $dqlPart;
$this->_state = self::STATE_DIRTY;
return $this;
}
// _dqlParts 結構
array (
...
'where' =>
Doctrine\ORM\Query\Expr\Andx::__set_state(array(
'separator' => ' AND ',
'allowedClasses' =>
array (
0 => 'Doctrine\\ORM\\Query\\Expr\\Comparison',
1 => 'Doctrine\\ORM\\Query\\Expr\\Func',
2 => 'Doctrine\\ORM\\Query\\Expr\\Orx',
3 => 'Doctrine\\ORM\\Query\\Expr\\Andx',
),
'preSeparator' => '(',
'postSeparator' => ')',
'parts' =>
array (
0 => 'u.id = 1',
),
))
...
接下來具體看 QueryBuilder::andWhere
方法严嗜,
getDQLPart取出的是剛才設置的Andx對象返帕,接著執(zhí)行Andx的addMultiple方法羡疗,最終調用的是Andx::add方法串远,這個方法最終是把'u.score >= 90'加入到Andx::parts數組中
// QueryBuilder
public function andWhere()
{
$args = func_get_args();
$where = $this->getDQLPart('where');
if ($where instanceof Expr\Andx) {
$where->addMultiple($args);
} else {
array_unshift($args, $where);
$where = new Expr\Andx($args);
}
return $this->add('where', $where);
}
// Andx
public function add($arg)
{
if ( $arg !== null && (!$arg instanceof self || $arg->count() > 0) ) {
// If we decide to keep Expr\Base instances, we can use this check
if ( ! is_string($arg)) {
$class = get_class($arg);
if ( ! in_array($class, $this->allowedClasses)) {
throw new \InvalidArgumentException("Expression of type '$class' not allowed in this context.");
}
}
$this->parts[] = $arg;
}
return $this;
}
所以此時Andx::parts數組中有了兩個元素
Doctrine\ORM\Query\Expr\Andx::__set_state(array(
'separator' => ' AND ',
'allowedClasses' =>
array (
0 => 'Doctrine\\ORM\\Query\\Expr\\Comparison',
1 => 'Doctrine\\ORM\\Query\\Expr\\Func',
2 => 'Doctrine\\ORM\\Query\\Expr\\Orx',
3 => 'Doctrine\\ORM\\Query\\Expr\\Andx',
),
'preSeparator' => '(',
'postSeparator' => ')',
'parts' =>
array (
0 => 'u.id = 1',
1 => 'u.score >= 90',
),
))
繼續(xù)看QueryBuilder::orWhere
方法汹族,取出的 $where
是剛剛andWhere
執(zhí)行后設置的 Andx 對象锚国,執(zhí)行 array_unshift($args, $where)
語句后蒙袍,形成的 $args
由一個 Andx 對象和一個字符串 'u.score <= 100'
組成俊卤。
public function orWhere()
{
$args = func_get_args();
$where = $this->getDQLPart('where');
if ($where instanceof Expr\Orx) {
$where->addMultiple($args);
} else {
array_unshift($args, $where);
$where = new Expr\Orx($args);
}
return $this->add('where', $where);
}
將包含 Andx 對象和字符串 u.score <= 100
的 $args
數組作為 Orx 對象的構造方法,Orx 構造函數的內部實現是 addMultiple
方法害幅,最終調用 Orx::add
方法將 $args
中的元素全部都加到 Orx對象 的 $parts
對象中消恍,最終 Orx 對象的 parts
內容的示意圖的階段 3 所示
我整理了一下添加邏輯如下所示
解析謂詞語句
謂詞對象轉換成謂詞語句其實就是一句話,
$queryPart = $this->getDQLPart($queryPartName);
echo $queryPart;
不要覺得奇怪以现,對象也可以當作字符串用狠怨,引用PHP手冊上的原話
__toString() 方法用于一個類被當成字符串時應怎樣回應。
謂詞對象的__toString
的實現在Doctrine\ORM\Query\Expr\Composite
邑遏,一起來看看
public function __toString()
{
if ($this->count() === 1) {
return (string) $this->parts[0];
}
$components = [];
foreach ($this->parts as $part) {
$components[] = $this->processQueryPart($part);
}
return implode($this->separator, $components);
}
private function processQueryPart($part)
{
$queryPart = (string) $part;
if (is_object($part) && $part instanceof self && $part->count() > 1) {
return $this->preSeparator . $queryPart . $this->postSeparator;
}
if (stripos($queryPart, ' OR ') !== false || stripos($queryPart, ' AND ') !== false) {
return $this->preSeparator . $queryPart . $this->postSeparator;
}
return $queryPart;
}
帶入之前組合好的 Orx 對象佣赖,一起來分析下。Orx 對象parts屬性的兩個元素分別會被帶入processQueryPart執(zhí)行记盒。
Andx 你先來憎蛤,走到$queryPart = (string) $part
,我們希望$part
被當作字符串處理,繼續(xù)回到__toString
纪吮,這里是個遞歸蹂午。
Andx對象parts屬性的兩個字符串元素繼續(xù)帶入processQueryPart執(zhí)行。這兩個字符串經過處理會作為Andx對象的$components
的元素彬碱,最終經過implode($this->separator, $components)
返回字符串 u.id=1 and u.score >= 90
豆胸,此時的值會被返回到 $queryPart
。
接下來執(zhí)行到的是 return $this->preSeparator . $queryPart . $this->postSeparator;
到 Orx 對象的 $components
的數組中去巷疼。
// $part 是 Andx 對象
// Andx 對象 經過字符串化后成了 u.id=1 and u.score >= 90晚胡,賦值給 $queryPart
$queryPart = (string) $part;
// 因為Andx對象有兩個條件灵奖,所以左右兩邊會被加上括號,最終返回 (u.id = 1 AND u.score >= 90)
if (is_object($part) && $part instanceof self && $part->count() > 1) {
return $this->preSeparator . $queryPart . $this->postSeparator;
}
Orx 對象parts屬性的第一個元素已經處理完畢估盘,接下來是第二個元素u.score <= 100
瓷患,字符串就很簡單了,直接返回到 Orx 對象的 $components
中去遣妥!
現在來看看Orx 對象的 $components
中間有啥了
array (
0 => '(u.id = 1 AND u.score >= 90)',
1 => 'u.score <= 100',
)
再用implode切割成字符串 結果就是出來了(u.id = 1 AND u.score >= 90) OR u.score <= 100
,解析完畢擅编!看不懂的同學看我整理的流程圖。
總結
具體細節(jié)大家可以使用 phpStorm + xdebug 單步調試研究箫踩。