Записать sql ResultSet в CSV-файл - PullRequest
Новая
       4

Записать sql ResultSet в CSV-файл

1 голос
/ 19 марта 2020

У меня есть функция, которая подключается к базе данных, выполняет запрос sql и пытается записать результат в файл. 

public class DBReader {

    public ResultSet executeSQL(SqlContainer sc) {
        String url = "url";
        String user = "user";
        String password = "pw";
        Connection con;
        Statement stmt;
        ResultSet rs;

        try {
            System.out.println("=== Started SQL ===");
            Class.forName("com.ibm.db2.jcc.DB2Driver");
            con = DriverManager.getConnection(url, user, password);
            con.setAutoCommit(false);
            stmt = con.createStatement();

            rs = stmt.executeQuery(sc.getSQL());


            CSVWriter writer = new CSVWriter(new FileWriter("result.csv"));
            writer.writeAll(rs, true);

            stmt.close();
            con.commit();
            con.close();

            System.out.println("    Finished SQL");
            return rs;
        } catch (SQLException | ClassNotFoundException | IOException e) {
            e.printStackTrace();
        }
        return null;
    }

}

Тем не менее, когда я выполняю вышеуказанную функцию, файл, который выходит, опорожнить. Я использовал следующий код, чтобы проверить, пустой ли набор результатов, но если я выполню: 

while (rs.next()) {
                String value = rs.getString(4);
                System.out.println("Something = " + value);
            }

, он вернет некоторые значения.

Я бы предпочел любую помощь или кивок справа направление, заранее спасибо.

1 Ответ

3 голосов
/ 19 марта 2020

Вам также следует закрыть writer:

writer.close();

Или, что еще лучше, вы можете использовать оператор try-with-resource, который автоматически закрывает его: 

        try(CSVWriter writer = new CSVWriter(new FileWriter("result.csv"));) {
            System.out.println("=== Started SQL ===");
            Class.forName("com.ibm.db2.jcc.DB2Driver");
            con = DriverManager.getConnection(url, user, password);
            con.setAutoCommit(false);
            stmt = con.createStatement();

            rs = stmt.executeQuery(sc.getSQL());

            writer.writeAll(rs, true);

            stmt.close();
            con.commit();
            con.close();

            System.out.println("    Finished SQL");
            return rs;
        } catch (SQLException | ClassNotFoundException | IOException e) {
            e.printStackTrace();
        }
...