|
|
|
@ -8,6 +8,8 @@
|
|
|
|
|
|
|
|
|
|
namespace yii\db\pgsql; |
|
|
|
|
|
|
|
|
|
use yii\base\InvalidParamException; |
|
|
|
|
|
|
|
|
|
/** |
|
|
|
|
* QueryBuilder is the query builder for PostgreSQL databases. |
|
|
|
|
* |
|
|
|
@ -62,6 +64,65 @@ class QueryBuilder extends \yii\db\QueryBuilder
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
/** |
|
|
|
|
* Creates a SQL statement for resetting the sequence value of a table's primary key. |
|
|
|
|
* The sequence will be reset such that the primary key of the next new row inserted |
|
|
|
|
* will have the specified value or 1. |
|
|
|
|
* @param string $tableName the name of the table whose primary key sequence will be reset |
|
|
|
|
* @param mixed $value the value for the primary key of the next new row inserted. If this is not set, |
|
|
|
|
* the next new row's primary key will have a value 1. |
|
|
|
|
* @return string the SQL statement for resetting sequence |
|
|
|
|
* @throws InvalidParamException if the table does not exist or there is no sequence associated with the table. |
|
|
|
|
*/ |
|
|
|
|
public function resetSequence($tableName, $value = null) |
|
|
|
|
{ |
|
|
|
|
$table = $this->db->getTableSchema($tableName); |
|
|
|
|
if ($table !== null && $table->sequenceName !== null) { |
|
|
|
|
$sequence = '"' . $table->sequenceName . '"'; |
|
|
|
|
|
|
|
|
|
if (strpos($sequence, '.') !== false) { |
|
|
|
|
$sequence = str_replace('.', '"."', $sequence); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
$tableName = $this->db->quoteTableName($tableName); |
|
|
|
|
if ($value === null) { |
|
|
|
|
$key = reset($table->primaryKey); |
|
|
|
|
$value = "(SELECT COALESCE(MAX(\"{$key}\"),0) FROM {$tableName})+1"; |
|
|
|
|
} else { |
|
|
|
|
$value = (int)$value; |
|
|
|
|
} |
|
|
|
|
return "SELECT SETVAL('$sequence',$value,false)"; |
|
|
|
|
} elseif ($table === null) { |
|
|
|
|
throw new InvalidParamException("Table not found: $tableName"); |
|
|
|
|
} else { |
|
|
|
|
throw new InvalidParamException("There is not sequence associated with table '$tableName'."); |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
/** |
|
|
|
|
* Builds a SQL statement for enabling or disabling integrity check. |
|
|
|
|
* @param boolean $check whether to turn on or off the integrity check. |
|
|
|
|
* @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'; |
|
|
|
|
$schema = $schema ? $schema : $this->db->schema->defaultSchema; |
|
|
|
|
$tableNames = $table ? [$table] : $this->db->schema->findTableNames($schema); |
|
|
|
|
$command = ''; |
|
|
|
|
|
|
|
|
|
foreach ($tableNames as $tableName) { |
|
|
|
|
$tableName = '"' . $schema . '"."' . $tableName . '"'; |
|
|
|
|
$command .= "ALTER TABLE $tableName $enable TRIGGER ALL; "; |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
#enable to have ability to alter several tables |
|
|
|
|
$this->db->pdo->setAttribute(\PDO::ATTR_EMULATE_PREPARES, true); |
|
|
|
|
return $command; |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
/** |
|
|
|
|
* Builds a SQL statement for changing the definition of a column. |
|
|
|
|
* @param string $table the table whose column is to be changed. The table name will be properly quoted by the method. |
|
|
|
|
* @param string $column the name of the column to be changed. The name will be properly quoted by the method. |
|
|
|
|