Spring Cloud Gateway聚合多微服务Mono/Flux结果的实现方法

问题说明 #

基于Spring Cloud Gateway开发API网关时，需要完成跨微服务的结果聚合：

• 微服务A返回Mono<ObjectA>结果，ObjectA内包含关联ObjectB的ID集合listIdObjectB
• 微服务B支持传入ID列表，返回Flux<ObjectB>结果
• 需要在网关层将两个服务的返回结果组装为单个Mono<MyDto>类型的统一响应

现有代码修正 #

你提供的ServiceBClient存在构造函数命名笔误，会导致Spring依赖注入失败，先修正该问题，同时补充空参数判断避免无效调用：

@Service
public class ServiceBClient {

    private final WebClient webClient;

    // 原代码构造函数误写为ServiceAClient，修正为与类名一致
    public ServiceBClient(WebClient.Builder builder) {
        this.webClient = builder.baseUrl("lb://microservice-B/B/").build();
    }

    public Flux<ObjectB> getListObjectB(List<Long> ids){
        // ID列表为空时直接返回空Flux，避免无效HTTP请求
        if (ids == null || ids.isEmpty()) {
            return Flux.empty();
        }
        return webClient
                .get()
                .uri("{ids}" , ids)
                .retrieve()
                .bodyToFlux(ObjectB.class);
    }
}

核心聚合逻辑实现 #

聚合逻辑基于Reactor的flatMap操作符实现：先调用微服务A获取基础数据，再从结果中提取关联ID调用微服务B，最后将两个结果组装为目标DTO，代码如下：

@Service
@AllArgsConstructor
public class CombinedService {

    private final ServiceAClient serviceAClient;
    private final ServiceBClient serviceBClient;

    public Mono<MyDto> getDetails(String id){
        return serviceAClient.getObjectA(id)
                // 可根据业务需求调整空结果处理逻辑，这里示例为抛出业务异常
                .switchIfEmpty(Mono.error(new IllegalArgumentException("对应资源不存在，id：" + id)))
                .flatMap(objectA -> serviceBClient.getListObjectB(objectA.getListIdObjectB())
                        // 将Flux<ObjectB>收集为List类型
                        .collectList()
                        // 组装最终MyDto对象
                        .map(bList -> {
                            MyDto dto = new MyDto();
                            dto.setId(objectA.getId());
                            dto.setName(objectA.getName());
                            dto.setDescription(objectA.getDescription());
                            dto.setStart(objectA.getStart());
                            dto.setEnd(objectA.getEnd());
                            dto.setListObjectB(bList);
                            return dto;
                        })
                );
    }
}

逻辑说明 #

• 由于微服务B的查询参数完全依赖微服务A的返回结果，调用链路为串行执行，符合业务逻辑要求
• collectList()会自动等待Flux中所有元素发射完成后，将所有结果收集为List，适配MyDto中listObjectB的字段类型
• 空参数判断、空结果处理逻辑可根据自身业务需求调整，比如微服务A返回空时直接返回空的MyDto而非抛出异常

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