Spring Data R2DBC单查询获取多聚合根可行方案咨询

结论 #

直接定义返回Mono<Tuple2<Book, Author>>的仓库方法不可行。

原因 #

Spring Data R2DBC 默认无法自动将联合查询的结果拆分到两个独立的实体类中：

• 它无法识别哪些字段属于Book、哪些属于Author，尤其是当两个实体存在重名字段（如id、create_time）时，字段值会被覆盖或映射错误。
• 框架没有内置逻辑将一行查询结果拆分为两个不同的实体对象并包装为Tuple2。

替代的优雅方案 #

方案1：自定义行映射器（RowMapper） #

手动实现字段到两个实体的映射，同时给SQL字段加别名避免冲突：

1. 实现自定义映射函数：

import org.springframework.data.r2dbc.convert.RowMapper;
import reactor.util.function.Tuple2;
import reactor.util.function.Tuples;

class BookAuthorRowMapper implements Function<Row, Tuple2<Book, Author>> {
    @Override
    public Tuple2<Book, Author> apply(Row row) {
        Book book = Book.builder()
                .id(row.get("book_id", Long.class))
                .title(row.get("book_title", String.class))
                .authorId(row.get("book_author_id", Long.class))
                // 补充其他Book字段
                .build();

        Author author = Author.builder()
                .id(row.get("author_id", Long.class))
                .name(row.get("author_name", String.class))
                // 补充其他Author字段
                .build();

        return Tuples.of(book, author);
    }
}

2. 在仓库接口中使用该映射器：

@Repository
public interface BookRepository extends ReactiveCrudRepository<Book, Long> {

    @Query(value = """
           SELECT b.id AS book_id, b.title AS book_title, b.author_id AS book_author_id,
                  a.id AS author_id, a.name AS author_name
           FROM book b
           JOIN author a ON b.author_id = a.id
           WHERE b.id = :bookId
           """,
           rowMapper = BookAuthorRowMapper.class)
    Mono<Tuple2<Book, Author>> findTupleById(long bookId);
}

方案2：使用嵌套DTO（推荐） #

定义包含Book和Author的嵌套DTO，Spring Data R2DBC支持通过字段前缀自动映射嵌套对象：

1. 定义嵌套DTO：

record BookWithAuthorDto(Book book, Author author) {}

2. 仓库方法中使用带前缀的SQL查询：

@Repository
public interface BookRepository extends ReactiveCrudRepository<Book, Long> {

    @Query("""
           SELECT b.id AS book_id, b.title AS book_title, b.author_id AS book_author_id,
                  a.id AS author_id, a.name AS author_name
           FROM book b
           JOIN author a ON b.author_id = a.id
           WHERE b.id = :bookId
           """)
    Mono<BookWithAuthorDto> findNestedDtoById(long bookId);
}

框架会自动将前缀为book_的字段映射到Book对象，author_前缀的字段映射到Author对象，无需手动编写映射逻辑，结构清晰且维护成本低。

内容的提问来源于stack exchange，提问作者banterCZ