使用 R2DB2 在 select 中动态选择列

Question

我一直在尝试传递我想要选择的列，但开箱即用它似乎是不可能的。我尝试过类似

@Query("SELECT :columns FROM USERS u WHERE u.LOCALE = :locale AND u.id IN (:ids)")
Flux<Users> retrieveExportData(@Param("columns") String columns,
                                       @Param("locale") String locale,
                                       @Param("ids") String[] ids);

与

private final R2dbcEntityTemplate template;

我尝试创建自己的查询，但这不起作用，因为它必须是 Criteria 类型，而这只会造成不值得的复杂性。

如果我能添加这样的列就好了

    criteriaList.add(
        Criteria.where("LOCALE").is(locale)
    );
    Criteria criteria = Criteria.from(criteriaList);

然后像这样执行

    Flux<Users> users = this.template.select(User.class)
        .matching(Query.query(criteria))
        .all();

或者只是像我的第一个示例一样调用存储库。

有没有人能够成功地做到这一点？

----- 更新 1 -----

我试过这样做：

导入 org.springframework.r2dbc.core.DatabaseClient;

DatabaseClient databaseClient = DatabaseClient.create(connectionFactory);

String sql = "SELECT " + columns + "  FROM USERS u WHERE u.LOCALE=" + locale + " AND u.id IN (" + ids + ")";

return databaseClient.sql(sql)
    .fetch()
    .all().cast(User.class);

但是从 Spring 4.3.6.RELEASE 开始，LinkedCaseInsensitiveMap 不再扩展 LinkedHashMap 和 HashMap，只实现了 Map 接口。

这会导致

Cannot cast org.springframework.util.LinkedCaseInsensitiveMap to User at java.base/java.lang.Class.cast

错误。

然后我尝试了答案中建议的 jooq 方法，但它只会产生语法错误。示例

private final DSLContext ctx = DSL.using(connectionFactory);
private final Users users = ctx.newRecord(Users.USERS); <-- USERS not found

@Query("SELECT :columns FROM USERS u WHERE u.LOCALE = :locale AND u.id IN (:ids)")
public Flux<Users> retrieveExportData(
    List<Field<?>> columns,
    String locale,
    String[] ids
) {
    return Flux.from(ctx
        .select(columns)
        .from("USERS")
        .where(users.LOCALE.eq(locale)) <--- LOCALE not found
        .and(users.ID.in(ids)) <--- ID not found
    ).map(r -> r.into(Users.class)); <---- into not found
}

图书馆看起来很有希望。我会努力让它发挥作用。

Answer 1

除了实际的绑定值之外，您不能用这样的语法元素替换绑定参数 (:columns)。对于这种类型的动态 SQL，您将不得不求助于某种查询构建机制。

也许看看jOOQ, which has R2DBC support？您的实现将如下所示：

@Query("SELECT :columns FROM USERS u WHERE u.LOCALE = :locale AND u.id IN (:ids)")
public Flux<Users> retrieveExportData(
    List<Field<?>> columns,
    String locale,
    String[] ids
) {
    return Flux.from(ctx
        .select(columns)
        .from(USERS)
        .where(USERS.LOCALE.eq(locale))
        .and(USERS.ID.in(ids))
    ).map(r -> r.into(Users.class));
}

免责声明：我为 jOOQ 背后的公司工作。

Answer 2

你可以这样写动态SQL。

import org.springframework.r2dbc.core.DatabaseClient;

DatabaseClient databaseClient = DatabaseClient.create(connectionFactory);

String sql = "SELECT " + columns + "  FROM USERS u WHERE u.LOCALE=" + locale + " AND u.id IN (" + ids + ")";

return databaseClient.sql(sql)
    .fetch()
    .all().cast(User.class);

你将需要这个依赖

implementation "org.springframework.boot:spring-boot-starter-data-r2dbc:2.6.2"