使用R2DBC-PostgreSQL调用返回SETOF自定义类型的PL/SQL函数列映射问题

核心问题原因 #

直接执行SELECT foo(:input)时，PostgreSQL会把返回的foo_type复合类型整体封装成单个字符串列（格式类似"(bar,1234)"），而非将其属性展开为独立字段，导致无法像普通SELECT结果那样按列取值。

解决方案1：修改SQL语句展开复合类型 #

通过两种SQL写法可以把复合类型的字段拆分为独立列，让R2DBC直接按普通列处理：

写法1：用FROM子句调用函数（推荐） #

这是PostgreSQL处理返回集合类型函数的标准方式，会自动将foo_type的所有属性展开为单独列：

SELECT * FROM foo(:input)

对应的Java代码修改：

public Flux<String> foo(String input) {
    return databaseClient
        .sql("SELECT * FROM foo(:input)")
        .bind("input", input)
        .map((row, rowMetaData) -> process(row.get("info1", String.class), row.get("info2", Long.class)))
        .all();
}

写法2：通过字段引用展开复合类型 #

如果需要保留函数调用的写法，可通过括号直接引用复合类型的字段：

-- 指定单个字段
SELECT (foo(:input)).info1, (foo(:input)).info2
-- 或通配符展开所有字段
SELECT (foo(:input)).*

这种写法适合只需要部分字段的场景，Java代码同样可以直接按字段名取值。

解决方案2：映射为Java对象（复杂场景适用） #

如果自定义类型字段较多、需要复用映射逻辑，可以创建对应Java实体类，结合Spring Data R2DBC的映射能力处理：

1. 创建Java实体类：

public class FooType {
    private String info1;
    private Long info2;

    public FooType(String info1, Long info2) {
        this.info1 = info1;
        this.info2 = info2;
    }

    // getter、setter省略
}

2. 修改Java代码映射为实体类：

public Flux<String> foo(String input) {
    return databaseClient
        .sql("SELECT * FROM foo(:input)")
        .bind("input", input)
        .map((row, rowMetaData) -> new FooType(
            row.get("info1", String.class),
            row.get("info2", Long.class)
        ))
        .map(this::process) // 假设process方法接收FooType对象
        .all();
}

注意事项 #

• 无论哪种方案，都必须确保SQL正确展开复合类型，否则仍会得到单个字符串列。
• 对于手动创建ROW或基于表查询返回的复合类型，上述方法均适用，PostgreSQL会统一处理复合类型的展开逻辑。

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