From 9f0e97ff2542a5125fbe7a62c9e88c097e9f0bb7 Mon Sep 17 00:00:00 2001 From: p0larbeer Date: Mon, 2 Dec 2013 23:31:44 -0800 Subject: [PATCH] Create Schema.php --- framework/yii/db/oci/Schema.php | 325 ++++++++++++++++++++++++++++++++++++++++ 1 file changed, 325 insertions(+) create mode 100644 framework/yii/db/oci/Schema.php diff --git a/framework/yii/db/oci/Schema.php b/framework/yii/db/oci/Schema.php new file mode 100644 index 0000000..2e247d9 --- /dev/null +++ b/framework/yii/db/oci/Schema.php @@ -0,0 +1,325 @@ +db); + } + + public function quoteTableName($name) { + if (strpos($name, '.') === false) { + return $this->quoteSimpleTableName($name); + } + $parts = explode('.', $name); + foreach ($parts as $i => $part) { + $parts[$i] = $this->quoteSimpleTableName($part); + } + return implode('.', $parts); + } + + public function quoteSimpleTableName($name) { + return '"' . $name . '"'; + //return $name; + } + + public function quoteColumnName($name) { + if (($pos = strrpos($name, '.')) !== false) { + $prefix = $this->quoteTableName(substr($name, 0, $pos)) . '.'; + $name = substr($name, $pos + 1); + } else { + $prefix = ''; + } + return $prefix . ($name === '*' ? $name : $this->quoteSimpleColumnName($name)); + } + + public function quoteSimpleColumnName($name) { + return '"' . $name . '"'; + //return $name; + } + + public function loadTableSchema($name) { + $table = new TableSchema(); + $this->resolveTableNames($table, $name); + // $this->findPrimaryKeys($table); + if ($this->findColumns($table)) { + //$this->findForeignKeys($table); + $this->findConstraints($table); + return $table; + } + } + + /** + * Resolves the table name and schema name (if any). + * + * @param TableSchema $table the table metadata object + * @param string $name the table name + */ + protected function resolveTableNames($table, $name) { + $parts = explode('.', str_replace('"', '', $name)); + if (isset($parts[1])) { + $table->schemaName = $parts[0]; + $table->name = $parts[1]; + } else { + $table->name = $parts[0]; + } + + if ($table->schemaName === null) { + $table->schemaName = $this->getDefaultSchema(); + } + } + + /** + * + * @return string default schema. + */ + public function getDefaultSchema() { + if (!strlen($this->_defaultSchema)) { + $this->setDefaultSchema(strtoupper($this->db->username)); + } + + return $this->_defaultSchema; + } + + /** + * + * @param string $schema default schema. + */ + public function setDefaultSchema($schema) { + $this->_defaultSchema = $schema; + } + + public function getLastInsertID($sequenceName = '') { + if ($this->db->isActive) { + $sql = "select {$sequenceName}.currval from dual"; + return $this->db->createCommand($sql)->queryScalar(); + } else { + throw new InvalidCallException('DB Connection is not active.'); + } + } + + /** + * Collects the table column metadata. + * @param COciTableSchema $table the table metadata + * @return boolean whether the table exists in the database + */ + protected function findColumns($table) { + $schemaName = $table->schemaName; + $tableName = $table->name; + + $sql = << 0 then ',' || a.data_scale else '' end + || ')' + when data_type = 'DATE' then '' + when data_type = 'NUMBER' then '' + else '(' || to_char(a.data_length) || ')' + end as data_type, + a.nullable, a.data_default, + ( SELECT D.constraint_type + FROM ALL_CONS_COLUMNS C + inner join ALL_constraints D on D.OWNER = C.OWNER and D.constraint_name = C.constraint_name + WHERE C.OWNER = B.OWNER + and C.table_name = B.object_name + and C.column_name = A.column_name + and D.constraint_type = 'P') as Key, + com.comments as column_comment +FROM ALL_TAB_COLUMNS A +inner join ALL_OBJECTS B ON b.owner = a.owner and ltrim(B.OBJECT_NAME) = ltrim(A.TABLE_NAME) +LEFT JOIN user_col_comments com ON (A.table_name = com.table_name AND A.column_name = com.column_name) +WHERE + a.owner = '{$schemaName}' + and (b.object_type = 'TABLE' or b.object_type = 'VIEW') + and b.object_name = '{$tableName}' +ORDER by a.column_id +EOD; + + + try { + $columns = $this->db->createCommand($sql)->queryAll(); + } catch (\Exception $e) { + return false; + } + + foreach ($columns as $column) { + $c = $this->createColumn($column); + + $table->columns[$c->name] = $c; + if ($c->isPrimaryKey) { + if ($table->primaryKey === null) { + $table->primaryKey = $c->name; + } elseif (is_string($table->primaryKey)) { + $table->primaryKey = array($table->primaryKey, $c->name); + } else { + $table->primaryKey[] = $c->name; + } + $sql = "select sequence_name +from user_tables tabs +join user_triggers trigs + on trigs.table_name = tabs.table_name +join user_dependencies deps + on deps.name = trigs.trigger_name +join user_sequences seqs + on seqs.sequence_name = deps.referenced_name +where tabs.table_name = '{$tableName}' + and trigs.triggering_event = 'INSERT'"; + try { + $seqName = $this->db->createCommand($sql)->queryScalar(); + } catch (\Exception $e) { + return false; + } + $table->sequenceName = $seqName; + $c->autoIncrement = true; + } + } + return true; + } + + protected function createColumn($column) { + $c = new ColumnSchema; + $c->name = $column['COLUMN_NAME']; + $c->allowNull = $column['NULLABLE'] === 'Y'; + $c->isPrimaryKey = strpos($column['KEY'], 'P') !== false; + $c->extract($column['DATA_TYPE'], $column['DATA_DEFAULT']); + $c->comment = $column['COLUMN_COMMENT'] === null ? '' : $column['COLUMN_COMMENT']; + + return $c; + } + + protected function findConstraints($table) { + $sql = << 'P' + order by d.constraint_name, c.position +EOD; + $command = $this->db->createCommand($sql); + foreach ($command->queryAll() as $row) { + if ($row['CONSTRAINT_TYPE'] === 'R') { // foreign key + $name = $row["COLUMN_NAME"]; + $table->foreignKeys[$name] = array($row["TABLE_REF"], $row["COLUMN_REF"]); + /* + if (isset($table->columns[$name])) { + $table->columns[$name]->isForeignKey = true; + } + * + */ + } + } + } + + /** + * Returns all table names in the database. + * @param string $schema the schema of the tables. Defaults to empty string, meaning the current or default 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 = '') { + if ($schema === '') { + $sql = <<db->createCommand($sql); + } else { + $sql = <<db > createCommand($sql); + $command->bindParam(':schema', $schema); + } + + $rows = $command->queryAll(); + $names = array(); + foreach ($rows as $row) { + if ($schema === $this->getDefaultSchema() || $schema === '') { + $names[] = $row['TABLE_NAME']; + } else { + $names[] = $row['TABLE_SCHEMA'] . '.' . $row['TABLE_NAME']; + } + } + return $names; + } + + /** + * Builds a SQL statement for renaming a DB table. + * @param string $table the table to be renamed. The name will be properly quoted by the method. + * @param string $newName the new table name. The name will be properly quoted by the method. + * @return string the SQL statement for renaming a DB table. + * @since 1.1.6 + */ + public function renameTable($table, $newName) { + return 'ALTER TABLE ' . $this->quoteTableName($table) . ' RENAME TO ' . $this->quoteTableName($newName); + } + + /** + * 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. + * @param string $type the new column type. The {@link getColumnType} method will be invoked to convert abstract column type (if any) + * into the physical one. Anything that is not recognized as abstract type will be kept in the generated SQL. + * For example, 'string' will be turned into 'varchar(255)', while 'string not null' will become 'varchar(255) not null'. + * @return string the SQL statement for changing the definition of a column. + * @since 1.1.6 + */ + public function alterColumn($table, $column, $type) { + $type = $this->getColumnType($type); + $sql = 'ALTER TABLE ' . $this->quoteTableName($table) . ' MODIFY ' + . $this->quoteColumnName($column) . ' ' + . $this->getColumnType($type); + return $sql; + } + + /** + * Builds a SQL statement for dropping an index. + * @param string $name the name of the index to be dropped. The name will be properly quoted by the method. + * @param string $table the table whose index is to be dropped. The name will be properly quoted by the method. + * @return string the SQL statement for dropping an index. + * @since 1.1.6 + */ + public function dropIndex($name, $table) { + return 'DROP INDEX ' . $this->quoteTableName($name); + } + + /** + * Resets 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 CDbTableSchema $table the table schema 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. + * @since 1.1.13 + */ + public function resetSequence($table, $value = 1) { + $seq = $table->name . "_SEQ"; + if ($table->sequenceName !== null) { + $this->db->createCommand("DROP SEQUENCE " . $seq)->execute(); + + $createSequenceSql = <<< SQL +create sequence $seq +start with $value +increment by 1 +nomaxvalue +nocache +SQL; + $this->db->createCommand($createSequenceSql)->execute(); + } + } + +}