Daniel/hotel
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update parser_sql

Browse Source
This commit is contained in:
shreesh arora
2021-05-03 01:06:30 +05:30
parent b7ef2391cc
commit d91def3d84
172 changed files with 8743 additions and 2571 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
tools/parser_sql/processors/SetProcessor.php
View File

@@ -30,33 +30,34 @@
* DAMAGE.
*/
require_once(dirname(__FILE__) . '/AbstractProcessor.php');
require_once(dirname(__FILE__) . '/ExpressionListProcessor.php');
require_once(dirname(__FILE__) . '/../utils/ExpressionType.php');
namespace PHPSQLParser\processors;
use PHPSQLParser\utils\ExpressionType;
/**
*
*
* This class processes the SET statements.
*
*
* @author arothe
*
*
*/
class SetProcessor extends AbstractProcessor {
private $expressionListProcessor;
public function __construct() {
$this->expressionListProcessor = new ExpressionListProcessor();
protected function processExpressionList($tokens) {
$processor = new ExpressionListProcessor($this->options);
return $processor->process($tokens);
}
/**
* A SET list is simply a list of key = value expressions separated by comma (,).
* This function produces a list of the key/value expressions.
*/
protected function getAssignment($base_expr) {
$assignment = $this->expressionListProcessor->process($this->splitSQLIntoTokens($base_expr));
protected function processAssignment($base_expr) {
$assignment = $this->processExpressionList($this->splitSQLIntoTokens($base_expr));
// TODO: if the left side of the assignment is a reserved keyword, it should be changed to colref
return array('expr_type' => ExpressionType::EXPRESSION, 'base_expr' => trim($base_expr),
'sub_tree' => $assignment);
'sub_tree' => (empty($assignment) ? false : $assignment));
}
public function process($tokens, $isUpdate = false) {
@@ -66,13 +67,15 @@ class SetProcessor extends AbstractProcessor {
$varType = false;
foreach ($tokens as $token) {
$upper = strtoupper(trim($token));
$trim = trim($token);
$upper = strtoupper($trim);
switch ($upper) {
case 'LOCAL':
case 'SESSION':
case 'GLOBAL':
if (!$isUpdate) {
$result[] = array('expr_type' => ExpressionType::RESERVED, 'base_expr' => $trim);
$varType = $this->getVariableType("@@" . $upper . ".");
$baseExpr = "";
continue 2;
@@ -80,7 +83,7 @@ class SetProcessor extends AbstractProcessor {
break;
case ',':
$assignment = $this->getAssignment($baseExpr);
$assignment = $this->processAssignment($baseExpr);
if (!$isUpdate && $varType !== false) {
$assignment['sub_tree'][0]['expr_type'] = $varType;
}
@@ -95,7 +98,7 @@ class SetProcessor extends AbstractProcessor {
}
if (trim($baseExpr) !== "") {
$assignment = $this->getAssignment($baseExpr);
$assignment = $this->processAssignment($baseExpr);
if (!$isUpdate && $varType !== false) {
$assignment['sub_tree'][0]['expr_type'] = $varType;
}