use of com.github.mgramin.sqlboot.model.resource_type.impl.jdbc.schema.table.TableJdbcResourceType in project sql-boot by sql-boot.
the class ParentTableJdbcResourceType method read.
@Override
public Stream<DbResource> read(final Uri uri) throws BootException {
try {
final TableJdbcResourceType tableJdbcResourceType = new TableJdbcResourceType(dataSource);
final Stream<DbResource> tables = tableJdbcResourceType.read(uri);
final List<DbResource> result = new ArrayList<>();
try (final Connection connection = dataSource.getConnection()) {
for (final DbResource table : tables.collect(toList())) {
final String tableSchem = table.headers().get("TABLE_SCHEM").toString();
final String tableName = table.headers().get("TABLE_NAME").toString();
final ResultSet parentTables = connection.getMetaData().getImportedKeys(null, tableSchem, tableName);
final ResultSetMetaData tableMetaData = parentTables.getMetaData();
final int columnsCount = tableMetaData.getColumnCount();
while (parentTables.next()) {
final String pkName = columnsCount >= 12 ? parentTables.getString(12) : null;
final Map<String, Object> props = new LinkedHashMap<>();
int i = 1;
for (String s : properties.keySet()) {
props.put(s, columnsCount >= i ? parentTables.getString(i++) : null);
}
result.add(new DbResourceImpl(pkName, this, new DbUri(name(), asList(tableSchem, tableName, pkName)), props));
}
}
}
return result.stream();
} catch (SQLException e) {
throw new BootException(e);
}
}
Aggregations