private function readIndexes(Table $def)
{
$res = $this->database->select()->select('t.relname as table_name, i.relname as index_name, a.attname as column_name')->select('ix.indisunique::int as is_unique,ix.indisprimary::int as is_primary')->from('pg_class t, pg_class i, pg_index ix, pg_attribute a')->where('t.oid = ix.indrelid')->where('i.oid = ix.indexrelid')->where('a.attrelid = t.oid')->where('a.attnum = ANY(ix.indkey)')->where('t.relkind = \'r\'')->where('t.relname = ?', $def->schemaName)->order('t.relname, i.relname, ix.indnatts')->query()->fetchAll();
$indexData = array();
foreach ($res as $row) {
if ($row['is_primary']) {
$row['index_name'] = 'PRIMARY';
}
$indexData[$row['index_name']]['columns'][$row['column_name']] = $row['column_name'];
$indexData[$row['index_name']]['is_unique'] = $row['is_unique'];
$indexData[$row['index_name']]['is_primary'] = $row['is_unique'];
}
$columns = $def->getColumns();
if (isset($indexData['PRIMARY'])) {
$primaryKey = array();
foreach ($indexData['PRIMARY']['columns'] as $columnName) {
$primaryKey[] = $columns->{$columnName};
}
$def->setPrimaryKey($primaryKey);
unset($indexData['PRIMARY']);
}
foreach ($indexData as $indexName => $indexInfo) {
$indexColumns = array();
foreach ($indexInfo['columns'] as $columnName) {
$indexColumns[] = $columns->{$columnName};
}
$index = new Index($indexColumns);
$index->setType($indexInfo['is_unique'] ? Index::TYPE_UNIQUE : Index::TYPE_KEY);
$index->setName($indexName);
$def->addIndex($index);
}
}
public function getTableDefinition($tableName)
{
$tableSymbol = new Symbol($tableName);
$res = $this->database->query("DESC ?", $tableSymbol);
$columns = new \stdClass();
while ($row = $res->fetchRow()) {
$type = $row['Type'];
$field = $row['Field'];
$phpType = $this->getTypeByString($type);
if ('auto_increment' === $row['Extra']) {
$phpType += Column::AUTO_ID;
}
$column = new Column($phpType);
$columns->{$field} = $column;
$column->schemaName = $field;
$notNull = $row['Null'] === 'NO';
if ($row['Default'] !== null || !$notNull) {
$column->setDefault($row['Default']);
}
$column->setFlag(Column::NOT_NULL, $notNull);
}
$definition = new Table($columns, $this->database, $tableName);
$res = $this->database->query("SHOW INDEX FROM ?", $tableSymbol);
$indexes = array();
$uniqueIndex = array();
foreach ($res as $row) {
$indexes[$row['Key_name']][$row['Seq_in_index']] = $columns->{$row['Column_name']};
$uniqueIndex[$row['Key_name']] = !$row['Non_unique'];
}
foreach ($indexes as $indexName => $indexData) {
ksort($indexData);
$index = new Index(array_values($indexData));
$index->setName($indexName);
$index->setType($uniqueIndex[$indexName] ? Index::TYPE_UNIQUE : Index::TYPE_KEY);
if ($indexName === self::_PRIMARY) {
$definition->setPrimaryKey($index->columns);
} else {
$definition->addIndex($index);
}
}
return $definition;
}
private function buildIndexes()
{
foreach ($this->indexes as $indexData) {
$type = $indexData[0];
$name = $indexData[1];
$columns = array();
foreach ($indexData[2] as $columnName) {
$columns[] = $this->columns->{$columnName};
}
$index = new Index($columns);
$index->setName($name);
$index->setType($type);
$this->table->addIndex($index);
}
}
