diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/Statements/ReplaceStatement.php | 160 |
1 files changed, 142 insertions, 18 deletions
diff --git a/src/Statements/ReplaceStatement.php b/src/Statements/ReplaceStatement.php index 10c5e63..df00272 100644 --- a/src/Statements/ReplaceStatement.php +++ b/src/Statements/ReplaceStatement.php @@ -8,10 +8,15 @@ */ namespace SqlParser\Statements; +use SqlParser\Parser; +use SqlParser\Token; +use SqlParser\TokensList; use SqlParser\Statement; +use SqlParser\Statements\SelectStatement; +use SqlParser\Components\Array2d; use SqlParser\Components\IntoKeyword; +use SqlParser\Components\OptionsArray; use SqlParser\Components\SetOperation; -use SqlParser\Components\Array2d; /** * `REPLACE` statement. @@ -26,6 +31,14 @@ use SqlParser\Components\Array2d; * [INTO] tbl_name * SET col_name={expr | DEFAULT}, ... * + * or + * + * REPLACE [LOW_PRIORITY | DELAYED] + * [INTO] tbl_name + * [PARTITION (partition_name,...)] + * [(col_name,...)] + * SELECT ... + * * @category Statements * @package SqlParser * @subpackage Statements @@ -46,22 +59,6 @@ class ReplaceStatement extends Statement ); /** - * The clauses of this statement, in order. - * - * @see Statement::$CLAUSES - * - * @var array - */ - public static $CLAUSES = array( - 'REPLACE' => array('REPLACE', 2), - // Used for options. - '_OPTIONS' => array('_OPTIONS', 1), - 'INTO' => array('FROM', 3), - 'VALUES' => array('VALUES', 1), - 'SET' => array('PARTITION', 3), - ); - - /** * Tables used as target for this statement. * * @var IntoKeyword @@ -76,9 +73,136 @@ class ReplaceStatement extends Statement public $values; /** - * The replaced values. + * If SET clause is present + * holds the SetOperation * * @var SetOperation[] */ public $set; + + /** + * If SELECT clause is present + * holds the SelectStatement + * + * @var SelectStatement + */ + public $select; + + /** + * @return string + */ + public function build() + { + $ret = 'REPLACE ' . $this->options + . ' INTO ' . $this->into; + + if ($this->values != NULL && count($this->values) > 0) { + $ret .= ' VALUES ' . Array2d::build($this->values); + } elseif ($this->set != NULL && count($this->set) > 0) { + $ret .= ' SET ' . SetOperation::build($this->set); + } elseif ($this->select != NULL && count($this->select) > 0) { + $ret .= ' ' . $this->select->build(); + } + + return $ret; + } + + + /** + * @param Parser $parser The instance that requests parsing. + * @param TokensList $list The list of tokens to be parsed. + * + * @return void + */ + public function parse(Parser $parser, TokensList $list) + { + ++$list->idx; // Skipping `REPLACE`. + + // parse any options if provided + $this->options = OptionsArray::parse( + $parser, + $list, + static::$OPTIONS + ); + ++$list->idx; + + $token = $list->tokens[$list->idx]; + + /** + * The state of the parser. + * + * Below are the states of the parser. + * + * 0 ---------------------------------[ INTO ]----------------------------------> 1 + * + * 1 -------------------------[ VALUES/VALUE/SET/SELECT ]-----------------------> 2 + * + * @var int $state + */ + $state = 0; + + for (; $list->idx < $list->count; ++$list->idx) { + /** + * Token parsed at this moment. + * + * @var Token $token + */ + $token = $list->tokens[$list->idx]; + + // End of statement. + if ($token->type === Token::TYPE_DELIMITER) { + break; + } + + // Skipping whitespaces and comments. + if (($token->type === Token::TYPE_WHITESPACE) || ($token->type === Token::TYPE_COMMENT)) { + continue; + } + + if ($state === 0) { + if ($token->type === Token::TYPE_KEYWORD + && $token->value !== 'INTO' + ) { + $parser->error(__('Unexpected keyword.'), $token); + break; + } else { + ++$list->idx; + $this->into = IntoKeyword::parse($parser, $list); + } + + $state = 1; + } elseif ($state === 1) { + if ($token->type === Token::TYPE_KEYWORD) { + if ($token->value === 'VALUE' + || $token->value === 'VALUES' + ) { + ++$list->idx; // skip VALUES + + $this->values = Array2d::parse($parser, $list); + } elseif ($token->value === 'SET') { + ++$list->idx; // skip SET + + $this->set = SetOperation::parse($parser, $list); + } elseif ($token->value === 'SELECT') { + $this->select = new SelectStatement($parser, $list); + } else { + $parser->error( + __('Unexpected keyword.'), + $token + ); + break; + } + $state = 2; + } else { + $parser->error( + __('Unexpected token.'), + $token + ); + break; + } + } + } + + --$list->idx; + } } |