Как правильно использовать try-with-resources в соединении с БД? - PullRequest
Новая
       30

Как правильно использовать try-with-resources в соединении с БД?

0 голосов
/ 08 января 2020

Я оглянулся, но не могу найти ответ на свой вопрос.

Вот контекст : мне нужно подключиться к базе данных в моем Java запрограммируйте и выполните запрос SQL, который я не могу контролировать и не знаю заранее. Для этого я использую приведенный ниже код. 

public Collection<HashMap<String, String>> runQuery(String request, int maxRows) {

    List<HashMap<String, String>> resultList = new ArrayList<>();

    DataSource datasource = null;

    try {

        Context initContext = new InitialContext();
        datasource = (DataSource) initContext.lookup("java:jboss/datasources/xxxxDS"); 

    } catch (NamingException ex) {

        // throw something.
    }

    try (Connection conn = datasource.getConnection();
         Statement statement = conn.createStatement();
         ResultSet rs = statement.executeQuery(request); ) {

        while (rs.next()) 
        {
            HashMap<String, String> map = new HashMap<>();

            for (int i = 1; i <= rs.getMetaData().getColumnCount(); i++) {
                map.put(rs.getMetaData().getColumnName(i).toUpperCase(), rs.getString(i));
            }

            resultList.add(map);             
        }

    } catch (SQLException ex) {

        // throw something.
    }

    return resultList;       
}

Проблема, с которой я сталкиваюсь: Как видите, есть еще один параметр maxRows, который я не использую. Мне нужно указать это для statement, но я не могу сделать это в try-with-resources.

Я хотел бы избежать увеличения когнитивной сложности этого метода, вложив еще один try-with-resources в первый, чтобы указать максимальное количество строк (как в этом примере кода). 

try (Connection conn = datasource.getConnection();
 Statement statement = conn.createStatement(); ) {

    statement.setMaxRows(maxRows);

    try (ResultSet rs = statement.executeQuery(request); ) {

        while (rs.next()) 
        {
            HashMap<String, String> map = new HashMap<>();

            for (int i = 1; i <= rs.getMetaData().getColumnCount(); i++) {
                map.put(rs.getMetaData().getColumnName(i).toUpperCase(), rs.getString(i));
            }

            resultList.add(map);             
        }
    }

} catch (SQLException ex) {

    // throw something.
}

Есть ли способ сделать это только с одним try-with-resources?

Ответы [ 2 ]

3 голосов
/ 08 января 2020

Если вам подходит go для дополнительного метода, то это может быть возможно только с одним try-ресурсом

Вместо Statement statement = conn.createStatement();

Statement statement = createStatement(conn, maxRows);

Внутри этого нового метода создайте объект Statement, установите maxRows и верните объект obj.

1 голос
/ 08 января 2020

Вы можете добавить вспомогательный метод, чтобы сделать это в одном блоке try-with-resources: 

    private static <T, E extends Exception> T configured(T resource, ThrowingConsumer<? super T, E> configuration) throws E {
        configuration.accept(resource);
        return resource;
    }

    private interface ThrowingConsumer<T, E extends Exception> {

        void accept(T value) throws E;
    }

И использовать его так: 

        try (Connection conn = null
              ; Statement statement = configured(conn.createStatement(), stmt -> stmt.setMaxRows(maxRows))
              ; ResultSet rs = statement.executeQuery(request)) {

        }
...