Spring JDBC "выбрать для обновления" - PullRequest
Новая
       23

Spring JDBC "выбрать для обновления"

4 голосов
/ 29 мая 2019

У меня есть следующий метод, который я использую с Spring JDBC 

public String getState() {
    String stateLink = template.queryForObject(
            "select state_url from state_scrape_queue where in_use = false ORDER BY scrape_timestamp NULLS FIRST LIMIT 1",
            (result, rowNum) -> {
                return result.getString("state_url");
            });
    return stateLink;
}

Я не могу найти пример того, как сделать для обновления с Spring JDBC.Я хочу, чтобы in_use был установлен в true, используя для обновления.

Мне нужно использовать select для обновления, так как это приложение будет использоваться многопоточным способом.Я не хочу, чтобы несколько потоков получали одну и ту же строку, и способ предотвратить это с помощью select for update

Я смог сделать это с простым JDBC, вот вопрос, который я задал, каксделать это с простым JDBC

выбрать "для обновления" с JDBC?

Кто-нибудь знает, как это будет сделано?

1 Ответ

0 голосов
/ 29 мая 2019

Это то, что я придумал, не стесняйтесь рекомендовать улучшения 

public String getState() throws SQLException {
    String state = null;

    Connection conn = DataSourceUtils.getConnection(template.getDataSource());
    try {
        conn.setAutoCommit(false);

        String[] colNames = { "id", "state_url", "in_use" };
        String query = "select " + Stream.of(colNames).collect(Collectors.joining(", "))
                + " from state_scrape_queue where in_use = false ORDER BY scrape_timestamp NULLS FIRST LIMIT 1 FOR UPDATE";
        System.out.println(query);
        try (Statement stmt = conn.createStatement(ResultSet.TYPE_SCROLL_SENSITIVE, ResultSet.CONCUR_UPDATABLE);
                ResultSet rs = stmt.executeQuery(query)) {
            while (rs.next()) {
                // Get the current values, if you need them.
                state = rs.getString(colNames[1]);

                rs.updateBoolean(colNames[2], true);
                rs.updateRow();
                conn.commit();
            }
        }
    } catch (SQLException e) {
        conn.setAutoCommit(true);
        e.printStackTrace();
    } finally {
        conn.setAutoCommit(true);
    }

    return state;
}
...