Browse Source

fixed bug with forgotten param, fixed behavior for one table integrity

tags/2.0.0-beta
Mark 11 years ago
parent
commit
65b543f68f
  1. 5
      framework/yii/db/Command.php
  2. 8
      framework/yii/db/pgsql/QueryBuilder.php
  3. 2
      framework/yii/db/sqlite/Schema.php

5
framework/yii/db/Command.php

@ -743,12 +743,13 @@ class Command extends \yii\base\Component
* @param boolean $check whether to turn on or off the integrity check.
* @param string $schema the schema name of the tables. Defaults to empty string, meaning the current
* or default schema.
* @param string $table the table name.
* @return Command the command object itself
* @throws NotSupportedException if this is not supported by the underlying DBMS
*/
public function checkIntegrity($check = true, $schema = '')
public function checkIntegrity($check = true, $schema = '', $table = '')
{
$sql = $this->db->getQueryBuilder()->checkIntegrity($check, $schema);
$sql = $this->db->getQueryBuilder()->checkIntegrity($check, $schema, $table);
return $this->setSql($sql);
}
}

8
framework/yii/db/pgsql/QueryBuilder.php

@ -99,14 +99,14 @@ class QueryBuilder extends \yii\db\QueryBuilder
/**
* Builds a SQL statement for enabling or disabling integrity check.
* @param boolean $check whether to turn on or off the integrity check.
* @param string $table the table name. Meaningless for MySQL.
* @param string $schema the schema of the tables. Meaningless for MySQL.
* @param string $schema the schema of the tables.
* @param string $table the table name.
* @return string the SQL statement for checking integrity
*/
public function checkIntegrity($check = true, $schema = '', $table = '')
{
$enable=$check ? 'ENABLE' : 'DISABLE';
$tableNames=$this->db->schema->findTableNames($schema);
$enable = $check ? 'ENABLE' : 'DISABLE';
$tableNames = $table ? [$table] : $this->db->schema->findTableNames($schema);
$command = '';
foreach($tableNames as $tableName)

2
framework/yii/db/sqlite/Schema.php

@ -87,7 +87,7 @@ class Schema extends \yii\db\Schema
* If not empty, the returned table names will be prefixed with the schema name.
* @return array all table names in the database.
*/
protected function findTableNames($schema = '')
public function findTableNames($schema = '')
{
$sql = "SELECT DISTINCT tbl_name FROM sqlite_master WHERE tbl_name<>'sqlite_sequence'";
return $this->db->createCommand($sql)->queryColumn();

Loading…
Cancel
Save