如何用jOOQ将一个嵌套的SQL集合类型安全地映射到一个嵌套的Java地图上

最近在Stack Overflow上有一个非常酷的问题,是关于如何用jOOQ将一个嵌套的集合映射成一个JavaMap 。在过去,我曾多次在博客上介绍过强大的MULTISET 操作符,它允许在jOOQ中嵌套集合。这一次,与其把数据嵌套到一个List<UserType> ,为什么不把它嵌套到一个Map<UserType1, UserType2>

看一下Sakila数据库,让我们看看我们如何将数据取到这个Javarecord 类型中:

record Film(
    String title,
    Map<LocalDate, BigDecimal> revenue
) {}

结果类型应该把FILM.TITLE ,以及每天每部电影赚的钱或DVD租赁店的金额包起来。我们可以使用其他的数据结构,但让我们假设这是我们想在用户界面或服务或其他方面消费的东西。

在jOOQ中,和以往一样,由于MULTISET ,我们可以在一个看起来像这样的单一查询中做到这一点:

List<Film> result =
ctx.select(
        FILM.TITLE,
        multiset(
            select(
                PAYMENT.PAYMENT_DATE.cast(LOCALDATE), 
                sum(PAYMENT.AMOUNT))
            .from(PAYMENT)
            .where(PAYMENT.rental().inventory().FILM_ID
                .eq(FILM.FILM_ID))
            .groupBy(PAYMENT.PAYMENT_DATE.cast(LOCALDATE))
            .orderBy(PAYMENT.PAYMENT_DATE.cast(LOCALDATE))
        )
        // Convert Field<Result<Record2<LocalDate, BigDecimal>>>
        // to Field<Map<LocalDate, BigDecimal>>
        .convertFrom(r -> r.collect(Records.intoMap())
   )
   .from(FILM)
   .orderBy(FILM.TITLE)

   // Convert Record2<String, Map<LocalDate, BigDecimal>>
   // to List<Film>
   .fetch(Records.mapping(Film::new))

然后我们就可以消费这个结果,比如说,像这样:

for (Film film : result) {
    System.out.println();
    System.out.println("Film %s with revenue: "
        .formatted(film.title()));

    // Inferred types are LocalDate d and BigDecimal r
    film.revenue().forEach((d, r) -> 
        System.out.println("  %s: %s".formatted(d, r))
    );
}

来产生:

Film ACADEMY DINOSAUR with revenue: 
  2005-05-27: 0.99
  2005-05-30: 1.99
  2005-06-15: 0.99
  [...]

Film ACE GOLDFINGER with revenue: 
  2005-07-07: 4.99
  2005-07-28: 9.99
  2005-08-01: 4.99
  [...]

与jOOQ一样,所有的东西都是完全类型安全的自己试试吧,改变查询中的一些列表达式,或者改变产生的recordMap 类型,看看查询是否会停止编译

这里有趣的一点是:

.convertFrom(r -> r.collect(Records.intoMap())

[Field.convertFrom()](https://www.jooq.org/javadoc/latest/org.jooq/org/jooq/Field.html#convertFrom(java.util.function.Function))方法来自jOOQ 3.15的新的临时转换API,它允许将Field<T> 列表达式临时转换为Field<U> 列表达式。在这种情况下,转换去:

  • Field<Result<Record2<LocalDate, BigDecimal>>> (多集字段类型)
  • Field<Map<LocalDate, BigDecimal>> (映射的类型)

Map<LocalDate, BigDecimal> 它是通过收集嵌套集合的所有Record2<LocalDate, BigDecimal> 记录来实现的,使用 [Records.intoMap()](https://www.jooq.org/javadoc/latest/org.jooq/org/jooq/Records.html#intoMap())收集器。该方法的签名是:

public static final <K, V, R entends Record2<K, V>> 
Collector<R, ?, Map<K, V>> intoMap() { ... }

泛型的这种特殊用法允许避免字段的键和值表达式的重复,要知道一个Record2<K, V> 的集合有一个明显的方法来收集成一个Map<K, V> (或者Map<K, List<V>> ,如果你用的是 [Records.intoGroups()](https://www.jooq.org/javadoc/latest/org.jooq/org/jooq/Records.html#intoGroups()),如果键可以重复的话)。

请注意,这两个收集器都会产生一个保留插入顺序的Map (例如:LinkedHashMap ),这样,任何MULTISET 的顺序都会被保留下来。

结论

天空是无限的,当你使用jOOQ 3.15新的嵌套集合的嵌套功能([MULTISET](https://www.jooq.org/doc/latest/manual/sql-building/column-expressions/multiset-value-constructor/)[MULTISET_AGG](https://www.jooq.org/doc/latest/manual/sql-building/column-expressions/aggregate-functions/multiset-agg-function/)) 或嵌套记录 ([ROW](https://www.jooq.org/doc/latest/manual/sql-building/column-expressions/nested-records/)).再加上特设的转换器,你可以在查询过程中把jOOQ表示法映射到任何Java表示法中,以坚持你选择的任何目标类型,包括嵌套的Map<K, V> ,有任意类型的KV

猜你喜欢

转载自juejin.im/post/7126041299928481805
今日推荐