Java 使用JDBC参数化IN子句的最佳方法是什么？

Z时代
2024-01-10
分类：问答

说我有一个查询表格

SELECT * FROM MYTABLE WHERE MYCOL in (?)

我想参数化in。

有没有一种简单的方法可以在Java中使用JDBC进行操作，而该方法可以在多个数据库上运行而无需修改SQL本身？

我发现的最接近的问题与C＃有关，我想知道Java / JDBC是否有所不同。

回答：

在JDBC中，确实没有直接的方法可以做到这一点。某些 JDBC驱动程序似乎支持PreparedStatement#setArray()该IN子句。我只是不确定那是哪个。

你可以仅使用带有String#join()和的辅助方法，并Collections#nCopies()为IN子句生成占位符，使用另一个辅助方法来使用设置循环中的所有值PreparedStatement#setObject()。

public static String preparePlaceHolders(int length) {
    return String.join(",", Collections.nCopies(length, "?"));
}
public static void setValues(PreparedStatement preparedStatement, Object... values) throws SQLException {
    for (int i = 0; i < values.length; i++) {
        preparedStatement.setObject(i + 1, values[i]);
    }
}

使用方法如下：

private static final String SQL_FIND = "SELECT id, name, value FROM entity WHERE id IN (%s)";
public List<Entity> find(Set<Long> ids) throws SQLException {
    List<Entity> entities = new ArrayList<Entity>();
    String sql = String.format(SQL_FIND, preparePlaceHolders(ids.size()));
    try (
        Connection connection = dataSource.getConnection();
        PreparedStatement statement = connection.prepareStatement(sql);
    ) {
        setValues(statement, ids.toArray());
        try (ResultSet resultSet = statement.executeQuery()) {
            while (resultSet.next()) {
                entities.add(map(resultSet));
            }
        }
    }
    return entities;
}
private static Entity map(ResultSet resultSet) throws SQLException {
    Enitity entity = new Entity();
    entity.setId(resultSet.getLong("id"));
    entity.setName(resultSet.getString("name"));
    entity.setValue(resultSet.getInt("value"));
    return entity;
}