Fix resource leaking in Liquibase dialect

Author: alex268

liquibase-dialect/src/main/java/tech/ydb/liquibase/change/InsertDataChangeYdb.java

@@ -6,6 +6,7 @@
 import java.util.Iterator;
 import java.util.List;
 import java.util.Map;
+
 import liquibase.change.ChangeMetaData;
 import liquibase.change.ColumnConfig;
 import liquibase.change.DatabaseChange;
@@ -34,14 +35,13 @@ public SqlStatement[] generateStatements(Database database) {
 
         try {
             JdbcConnection jdbcConnection = (JdbcConnection) database.getConnection();
-            ResultSet resultSet = jdbcConnection.getMetaData()
-                    .getColumns(null, null, getTableName(), null);
-
-            while (resultSet.next()) {
-                columnToLiquibaseDataType.put(
-                        resultSet.getString("COLUMN_NAME").toLowerCase(),
-                        DataTypeFactory.getInstance()
-                                .fromDescription(resultSet.getString("TYPE_NAME"), database));
+            try (ResultSet rs = jdbcConnection.getMetaData().getColumns(null, null, getTableName(), null)) {
+                while (rs.next()) {
+                    columnToLiquibaseDataType.put(
+                            rs.getString("COLUMN_NAME").toLowerCase(),
+                            DataTypeFactory.getInstance().fromDescription(rs.getString("TYPE_NAME"), database)
+                    );
+                }
             }
 
             StringBuilder yqlInsert = new StringBuilder()

liquibase-dialect/src/main/java/tech/ydb/liquibase/change/LoadDataChangeYdb.java

@@ -1,6 +1,5 @@
 package tech.ydb.liquibase.change;
 
-import com.opencsv.exceptions.CsvMalformedLineException;
 import java.io.IOException;
 import java.sql.ResultSet;
 import java.util.ArrayList;
@@ -9,6 +8,8 @@
 import java.util.LinkedHashMap;
 import java.util.List;
 import java.util.Map;
+
+import com.opencsv.exceptions.CsvMalformedLineException;
 import liquibase.Scope;
 import liquibase.change.ChangeMetaData;
 import liquibase.change.DatabaseChange;
@@ -23,6 +24,7 @@
 import liquibase.statement.core.InsertStatement;
 import liquibase.statement.core.RawSqlStatement;
 import liquibase.util.csv.CSVReader;
+
 import tech.ydb.liquibase.database.YdbDatabase;
 
 /**
@@ -60,17 +62,16 @@ public SqlStatement[] generateStatements(Database database) {
             }
 
             JdbcConnection jdbcConnection = (JdbcConnection) database.getConnection();
-            ResultSet resultSet = jdbcConnection.getMetaData()
-                    .getColumns(null, null, tableName, null);
-
-            while (resultSet.next()) {
-                columnToLiquibaseDataType.put(
-                        resultSet.getString("COLUMN_NAME").toLowerCase(),
-                        DataTypeFactory.getInstance()
-                                .fromDescription(resultSet.getString("TYPE_NAME"), database));
-            }
+            try (ResultSet resultSet = jdbcConnection.getMetaData().getColumns(null, null, tableName, null)) {
+                while (resultSet.next()) {
+                    columnToLiquibaseDataType.put(
+                            resultSet.getString("COLUMN_NAME").toLowerCase(),
+                            DataTypeFactory.getInstance()
+                                    .fromDescription(resultSet.getString("TYPE_NAME"), database));
+                }
 
-            return getSqlStatements(database, reader, headers);
+                return getSqlStatements(database, reader, headers);
+            }
         } catch (CsvMalformedLineException e) {
             throw new RuntimeException("Error parsing " + getRelativeTo() + " on line " + e.getLineNumber() + ": " + e.getMessage());
         } catch (UnexpectedLiquibaseException ule) {