PostgresDbCleanerExtension loadTablesToClean

PostgresDbCleanerExtension.kt

@Throws(SQLException::class)
private fun loadTablesToClean(connection: Connection): List<TableData> {
    val databaseMetaData = connection.metaData
    val resultSet = databaseMetaData.getTables(
        connection.catalog, null, null, arrayOf("TABLE"))

    val tablesToClean = mutableListOf<TableData>()
    while (resultSet.next()) {
        val table = TableData(
            schema = resultSet.getString("TABLE_SCHEM"),
            name = resultSet.getString("TABLE_NAME")
        )

        if (!TABLES_TO_IGNORE.contains(table)) {
            tablesToClean.add(table)
        }
    }

    return tablesToClean
}