use of org.jaxdb.www.ddlx_0_5.xLygluGCXAA.$Column in project jaxdb by jaxdb.
the class Decompiler method createDDL.
public static Schema createDDL(final Connection connection) throws SQLException {
final DBVendor vendor = DBVendor.valueOf(connection.getMetaData());
final Decompiler decompiler = Decompiler.getDecompiler(vendor);
final DatabaseMetaData metaData = connection.getMetaData();
try (final ResultSet tableRows = metaData.getTables(null, null, null, new String[] { "TABLE" })) {
final Schema schema = new Schema();
final Map<String, List<$CheckReference>> tableNameToChecks = decompiler.getCheckConstraints(connection);
final Map<String, List<$Table.Constraints.Unique>> tableNameToUniques = decompiler.getUniqueConstraints(connection);
final Map<String, $Table.Indexes> tableNameToIndexes = decompiler.getIndexes(connection);
final Map<String, Map<String, $ForeignKeyUnary>> tableNameToForeignKeys = decompiler.getForeignKeys(connection);
final Map<String, $Column> columnNameToColumn = new HashMap<>();
final Map<Integer, $Column> columnNumberToColumn = new TreeMap<>();
final Map<String, TreeMap<Short, String>> indexNameToIndex = new HashMap<>();
final Map<String, String> indexNameToType = new HashMap<>();
final Map<String, Boolean> indexNameToUnique = new HashMap<>();
while (tableRows.next()) {
final String tableName = tableRows.getString(3);
final $Table table = new Schema.Table();
table.setName$(new $Named.Name$(tableName.toLowerCase()));
schema.addTable(table);
try (final ResultSet columnRows = metaData.getColumns(null, null, tableName, null)) {
while (columnRows.next()) {
final String columnName = columnRows.getString("COLUMN_NAME").toLowerCase();
final String typeName = columnRows.getString("TYPE_NAME");
final int columnSize = columnRows.getInt("COLUMN_SIZE");
final String _default = columnRows.getString("COLUMN_DEF");
final int index = columnRows.getInt("ORDINAL_POSITION");
final String nullable = columnRows.getString("IS_NULLABLE");
final String autoIncrement = columnRows.getString("IS_AUTOINCREMENT");
final int decimalDigits = columnRows.getInt("DECIMAL_DIGITS");
final $Column column = decompiler.makeColumn(columnName.toLowerCase(), typeName, columnSize, decimalDigits, _default, nullable.length() == 0 ? null : "YES".equals(nullable), autoIncrement.length() == 0 ? null : "YES".equals(autoIncrement));
columnNameToColumn.put(columnName, column);
columnNumberToColumn.put(index, column);
}
columnNumberToColumn.values().forEach(table::addColumn);
try (final ResultSet primaryKeyRows = metaData.getPrimaryKeys(null, null, tableName)) {
while (primaryKeyRows.next()) {
final String columnName = primaryKeyRows.getString("COLUMN_NAME").toLowerCase();
if (table.getConstraints() == null)
table.setConstraints(new $Table.Constraints());
if (table.getConstraints().getPrimaryKey() == null)
table.getConstraints().setPrimaryKey(new $Table.Constraints.PrimaryKey());
final $Table.Constraints.PrimaryKey.Column column = new $Table.Constraints.PrimaryKey.Column();
column.setName$(new $Table.Constraints.PrimaryKey.Column.Name$(columnName));
table.getConstraints().getPrimaryKey().addColumn(column);
}
}
final List<$Table.Constraints.Unique> uniques = tableNameToUniques == null ? null : tableNameToUniques.get(tableName);
if (uniques != null && uniques.size() > 0) {
if (table.getConstraints() == null)
table.setConstraints(new $Table.Constraints());
for (final $Table.Constraints.Unique unique : uniques) table.getConstraints().addUnique(unique);
}
try (final ResultSet indexRows = metaData.getIndexInfo(null, null, tableName, false, true)) {
while (indexRows.next()) {
final String columnName = indexRows.getString("COLUMN_NAME").toLowerCase();
if (columnName == null)
continue;
final String indexName = indexRows.getString("INDEX_NAME").toLowerCase();
TreeMap<Short, String> indexes = indexNameToIndex.get(indexName);
if (indexes == null)
indexNameToIndex.put(indexName, indexes = new TreeMap<>());
final short ordinalPosition = indexRows.getShort("ORDINAL_POSITION");
indexes.put(ordinalPosition, columnName);
final String type = getType(indexRows.getShort("TYPE"));
final String currentType = indexNameToType.get(indexName);
if (currentType == null)
indexNameToType.put(indexName, type);
else if (!type.equals(currentType))
throw new IllegalStateException("Expected " + type + " = " + currentType);
final boolean unique = !indexRows.getBoolean("NON_UNIQUE");
final Boolean currentUnique = indexNameToUnique.get(indexName);
if (currentUnique == null)
indexNameToUnique.put(indexName, unique);
else if (unique != currentUnique)
throw new IllegalStateException("Expected " + unique + " = " + currentType);
}
}
final $Table.Indexes indexes = tableNameToIndexes == null ? null : tableNameToIndexes.get(tableName);
if (indexes != null)
table.setIndexes(indexes);
final List<$CheckReference> checks = tableNameToChecks == null ? null : tableNameToChecks.get(tableName);
if (checks != null)
for (final $CheckReference check : checks) addCheck(columnNameToColumn.get(check.getColumn$().text()), check);
final Map<String, $ForeignKeyUnary> foreignKeys = tableNameToForeignKeys == null ? null : tableNameToForeignKeys.get(tableName);
if (foreignKeys != null)
for (final Map.Entry<String, $ForeignKeyUnary> entry : foreignKeys.entrySet()) columnNameToColumn.get(entry.getKey().toLowerCase()).setForeignKey(entry.getValue());
}
columnNameToColumn.clear();
columnNumberToColumn.clear();
indexNameToIndex.clear();
indexNameToType.clear();
}
return schema;
}
}
use of org.jaxdb.www.ddlx_0_5.xLygluGCXAA.$Column in project jaxdb by jaxdb.
the class DerbyDecompiler method getUniqueConstraints.
@Override
@SuppressWarnings("null")
Map<String, List<$Table.Constraints.Unique>> getUniqueConstraints(final Connection connection) throws SQLException {
final Map<String, List<String>> tableNameToColumns = getTables(connection);
final PreparedStatement statement = connection.prepareStatement(constraintsSql);
final ResultSet rows = statement.executeQuery();
final Map<String, List<$Table.Constraints.Unique>> tableNameToUniques = new HashMap<>();
String lastTable = null;
List<$Table.Constraints.Unique> uniques = null;
while (rows.next()) {
final String tableName = rows.getString(2);
if (!tableName.equals(lastTable)) {
lastTable = tableName;
tableNameToUniques.put(tableName, uniques = new ArrayList<>());
}
final List<String> columns = tableNameToColumns.get(tableName);
final String descriptor = rows.getString(3);
final int close = descriptor.lastIndexOf(')');
final int open = descriptor.lastIndexOf('(', close - 1);
final String[] colRefs = descriptor.substring(open + 1, close).split(",");
final $Table.Constraints.Unique unique = new $Table.Constraints.Unique();
uniques.add(unique);
for (int i = 0; i < colRefs.length; ++i) {
colRefs[i] = columns.get(Integer.valueOf(colRefs[i].trim()) - 1);
final $Table.Constraints.Unique.Column column = new $Table.Constraints.Unique.Column();
column.setName$(new $Table.Constraints.Unique.Column.Name$(colRefs[i].toLowerCase()));
unique.addColumn(column);
}
}
return tableNameToUniques;
}
use of org.jaxdb.www.ddlx_0_5.xLygluGCXAA.$Column in project jaxdb by jaxdb.
the class DerbyDecompiler method getForeignKeys.
@Override
@SuppressWarnings("null")
Map<String, Map<String, $ForeignKeyUnary>> getForeignKeys(final Connection connection) throws SQLException {
final Map<String, List<String>> tableNameToColumns = getTables(connection);
final PreparedStatement statement = connection.prepareStatement(foreignKeySql);
final ResultSet rows = statement.executeQuery();
final Map<String, Map<String, $ForeignKeyUnary>> tableNameToForeignKeys = new HashMap<>();
String lastTable = null;
Map<String, $ForeignKeyUnary> columnNameToForeignKey = null;
while (rows.next()) {
final String tableName = rows.getString(2);
final List<String> columnNames = tableNameToColumns.get(tableName);
if (!tableName.equals(lastTable)) {
lastTable = tableName;
tableNameToForeignKeys.put(tableName, columnNameToForeignKey = new HashMap<>());
}
final String primaryTable = rows.getString(6);
final String primaryDescriptor = rows.getString(7);
final String primaryColumn = tableNameToColumns.get(primaryTable).get(Integer.valueOf(primaryDescriptor.substring(primaryDescriptor.lastIndexOf('(') + 1, primaryDescriptor.lastIndexOf(')'))) - 1);
final $ForeignKeyUnary foreignKey = new $Column.ForeignKey();
foreignKey.setReferences$(new References$(primaryTable.toLowerCase()));
foreignKey.setColumn$(new Column$(primaryColumn.toLowerCase()));
final String deleteRule = rows.getString(4);
final $ChangeRule.Enum onDelete = deleteRule == null ? null : "S".equals(deleteRule) ? $ChangeRule.RESTRICT : "C".equals(deleteRule) ? $ChangeRule.CASCADE : "U".equals(deleteRule) ? $ChangeRule.SET_20NULL : null;
if (onDelete != null)
foreignKey.setOnDelete$(new OnDelete$(onDelete));
final String updateRule = rows.getString(5);
final $ChangeRule.Enum onUpdate = updateRule == null ? null : "S".equals(updateRule) ? $ChangeRule.RESTRICT : null;
if (onUpdate != null)
foreignKey.setOnUpdate$(new OnUpdate$(onUpdate));
final String foreignDescriptor = rows.getString(3);
final String foreignColumn = columnNames.get(Integer.valueOf(foreignDescriptor.substring(foreignDescriptor.lastIndexOf('(') + 1, foreignDescriptor.lastIndexOf(')'))) - 1);
columnNameToForeignKey.put(foreignColumn, foreignKey);
}
return tableNameToForeignKeys;
}
use of org.jaxdb.www.ddlx_0_5.xLygluGCXAA.$Column in project jaxdb by jaxdb.
the class DerbyDecompiler method getIndexes.
@Override
@SuppressWarnings("null")
Map<String, $Table.Indexes> getIndexes(final Connection connection) throws SQLException {
final Map<String, List<String>> tableNameToColumns = getTables(connection);
final PreparedStatement statement = connection.prepareStatement(indexSql);
final ResultSet rows = statement.executeQuery();
final Map<String, $Table.Indexes> tableNameToIndexes = new HashMap<>();
String lastTable = null;
$Table.Indexes indexes = null;
while (rows.next()) {
final String tableName = rows.getString(2);
final List<String> columnNames = tableNameToColumns.get(tableName);
if (!tableName.equals(lastTable)) {
lastTable = tableName;
tableNameToIndexes.put(tableName, indexes = new $Table.Indexes());
}
final String descriptor = rows.getString(3);
final boolean unique = descriptor.startsWith("UNIQUE");
final $IndexType.Enum type = descriptor.startsWith("HASH") ? $Index.Type$.HASH : $Index.Type$.BTREE;
final $Table.Indexes.Index index = new $Table.Indexes.Index();
indexes.addIndex(index);
if (!$Index.Type$.BTREE.equals(type))
index.setType$(new $Index.Type$(type));
if (unique)
index.setUnique$(new $Index.Unique$(unique));
final String[] columnNumbers = descriptor.substring(descriptor.lastIndexOf('(') + 1, descriptor.lastIndexOf(')')).split(",");
for (final String columnNumber : columnNumbers) {
final String columnName = columnNames.get(Integer.valueOf(columnNumber.trim()) - 1);
final $Table.Indexes.Index.Column column = new $Table.Indexes.Index.Column();
column.setName$(new $Table.Indexes.Index.Column.Name$(columnName.toLowerCase()));
index.addColumn(column);
}
}
return tableNameToIndexes;
}
use of org.jaxdb.www.ddlx_0_5.xLygluGCXAA.$Column in project jaxdb by jaxdb.
the class Generator method registerColumns.
private static void registerColumns(final $Table table, final Set<? super String> tableNames, final Map<? super String, ColumnRef> columnNameToColumn) throws GeneratorExecutionException {
final String tableName = table.getName$().text();
final List<String> violations = new ArrayList<>();
String nameViolation = checkNameViolation(tableName);
if (nameViolation != null)
violations.add(nameViolation);
if (tableNames.contains(tableName))
throw new GeneratorExecutionException("Circular table dependency detected: " + tableName);
tableNames.add(tableName);
final List<$Column> columns = table.getColumn();
if (columns != null) {
for (int c = 0, len = columns.size(); c < len; ++c) {
final $Column column = columns.get(c);
final String columnName = column.getName$().text();
nameViolation = checkNameViolation(columnName);
if (nameViolation != null)
violations.add(nameViolation);
final ColumnRef existing = columnNameToColumn.get(columnName);
if (existing != null)
throw new GeneratorExecutionException("Duplicate column definition: " + tableName + "." + columnName);
columnNameToColumn.put(columnName, new ColumnRef(column, c));
}
}
if (violations.size() > 0)
violations.forEach(logger::warn);
}
Aggregations