forked from yiisoft/yii2
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Fixes yiisoft#15357: Added multi statement support for `yii\db\sqlite…
…\Command`
- Loading branch information
1 parent
eac6a5d
commit 315855f
Showing
4 changed files
with
197 additions
and
7 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,116 @@ | ||
| <?php | ||
| /** | ||
| * @link http://www.yiiframework.com/ | ||
| * @copyright Copyright (c) 2008 Yii Software LLC | ||
| * @license http://www.yiiframework.com/license/ | ||
| */ | ||
|
|
||
| namespace yii\db\sqlite; | ||
|
|
||
| use yii\db\SqlToken; | ||
| use yii\helpers\StringHelper; | ||
|
|
||
| /** | ||
| * Command represents an SQLite's SQL statement to be executed against a database. | ||
| * | ||
| * {@inheritdoc} | ||
| * | ||
| * @author Sergey Makinen <[email protected]> | ||
| * @since 2.0.14 | ||
| */ | ||
| class Command extends \yii\db\Command | ||
| { | ||
| /** | ||
| * @inheritdoc | ||
| */ | ||
| public function execute() | ||
| { | ||
| $sql = $this->getSql(); | ||
| $params = $this->params; | ||
| $statements = $this->splitStatements($sql, $params); | ||
| if ($statements === false) { | ||
| return parent::execute(); | ||
| } | ||
|
|
||
| $result = null; | ||
| foreach ($statements as $statement) { | ||
| list($statementSql, $statementParams) = $statement; | ||
| $this->setSql($statementSql)->bindValues($statementParams); | ||
| $result = parent::execute(); | ||
| } | ||
| $this->setSql($sql)->bindValues($params); | ||
| return $result; | ||
| } | ||
|
|
||
| /** | ||
| * @inheritdoc | ||
| */ | ||
| protected function queryInternal($method, $fetchMode = null) | ||
| { | ||
| $sql = $this->getSql(); | ||
| $params = $this->params; | ||
| $statements = $this->splitStatements($sql, $params); | ||
| if ($statements === false) { | ||
| return parent::queryInternal($method, $fetchMode); | ||
| } | ||
|
|
||
| list($lastStatementSql, $lastStatementParams) = array_pop($statements); | ||
| foreach ($statements as $statement) { | ||
| list($statementSql, $statementParams) = $statement; | ||
| $this->setSql($statementSql)->bindValues($statementParams); | ||
| parent::execute(); | ||
| } | ||
| $this->setSql($lastStatementSql)->bindValues($lastStatementParams); | ||
| $result = parent::queryInternal($method, $fetchMode); | ||
| $this->setSql($sql)->bindValues($params); | ||
| return $result; | ||
| } | ||
|
|
||
| /** | ||
| * Splits the specified SQL code into individual SQL statements and returns them | ||
| * or `false` if there's a single statement. | ||
| * @param string $sql | ||
| * @param array $params | ||
| * @return string[]|false | ||
| */ | ||
| private function splitStatements($sql, $params) | ||
| { | ||
| $semicolonIndex = strpos($sql, ';'); | ||
| if ($semicolonIndex === false || $semicolonIndex === StringHelper::byteLength($sql) - 1) { | ||
| return false; | ||
| } | ||
|
|
||
| $tokenizer = new SqlTokenizer($sql); | ||
| $codeToken = $tokenizer->tokenize(); | ||
| if (count($codeToken->getChildren()) === 1) { | ||
| return false; | ||
| } | ||
|
|
||
| $statements = []; | ||
| foreach ($codeToken->getChildren() as $statement) { | ||
| $statements[] = [$statement->getSql(), $this->extractUsedParams($statement, $params)]; | ||
| } | ||
| return $statements; | ||
| } | ||
|
|
||
| /** | ||
| * Returns named bindings used in the specified statement token. | ||
| * @param SqlToken $statement | ||
| * @param array $params | ||
| * @return array | ||
| */ | ||
| private function extractUsedParams(SqlToken $statement, $params) | ||
| { | ||
| preg_match_all('/(?P<placeholder>[:][a-zA-Z0-9_]+)/', $statement->getSql(), $matches, PREG_SET_ORDER); | ||
| $result = []; | ||
| foreach ($matches as $match) { | ||
| $phName = ltrim($match['placeholder'], ':'); | ||
| if (isset($params[$phName])) { | ||
| $result[$phName] = $params[$phName]; | ||
| } elseif (isset($params[':' . $phName])) { | ||
| $result[':' . $phName] = $params[':' . $phName]; | ||
| } | ||
| } | ||
| return $result; | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters