Spring WebFlux控制器中高效获取并利用原始请求体对象

// 原始控制器方法
@PostMapping("url")
public Mono<MyResponse> getMyResponse(@RequestBody Mono<MyRequest> myRequestMono) {
    return urlService.getUrl(myRequestMono)
            .doOnNext(url -> {
                // 在这里如何访问 myRequestMono 中包含的 MyRequest 对象？
                System.out.println("Generated URL: Successfully ");
            })
            .map(dto -> MyResponse.builder().url(dto).build())
            .doOnError(e -> System.out.println("Error " + e));
}

// 服务层方法
public Mono<String> getUrl(Mono<MyRequest> myRequestMono) {
    return myRequestMono.map(myRequest -> {
        // 在这里可以访问 MyRequest 对象
        callSomething(myRequest);
        return "something";
    });
}

import org.springframework.web.bind.annotation.*;
import reactor.core.publisher.Mono;

// 假设 MyRequest 和 MyResponse 是定义好的数据传输对象
class MyRequest {
    private String requestId;
    private String data;
    // 构造函数、Getter、Setter、toString等
    public String getRequestId() { return requestId; }
    public void setRequestId(String requestId) { this.requestId = requestId; }
    public String getData() { return data; }
    public void setData(String data) { this.data = data; }
    @Override
    public String toString() {
        return "MyRequest{" + "requestId='" + requestId + '\'' + ", data='" + data + '\'' + '}';
    }
}

class MyResponse {
    private String url;
    // 构造函数、Getter、Setter等
    public MyResponse(String url) { this.url = url; }
    public String getUrl() { return url; }
    public void setUrl(String url) { this.url = url; }

    public static MyResponseBuilder builder() {
        return new MyResponseBuilder();
    }

    public static class MyResponseBuilder {
        private String url;
        public MyResponseBuilder url(String url) {
            this.url = url;
            return this;
        }
        public MyResponse build() {
            return new MyResponse(url);
        }
    }
}

// 假设 UrlService 是一个注入的服务
class UrlService {
    public Mono<String> getUrl(Mono<MyRequest> myRequestMono) {
        return myRequestMono.map(myRequest -> {
            System.out.println("Service processing request: " + myRequest.getRequestId());
            // 模拟一些业务逻辑
            return "https://example.com/generated/" + myRequest.getData();
        });
    }
}

@RestController
@RequestMapping("/api")
public class MyController {

    private final UrlService urlService;

    public MyController(UrlService urlService) {
        this.urlService = urlService;
    }

    @PostMapping("url")
    public Mono<MyResponse> getMyResponse(@RequestBody MyRequest myRequest) { // 关键改变：直接接收 MyRequest
        System.out.println("Received request in controller: " + myRequest.getRequestId());

        // 将 MyRequest 包装成 Mono.just() 以便传入期望 Mono<MyRequest> 的服务层方法
        return urlService.getUrl(Mono.just(myRequest))
                .doOnNext(url -> {
                    // 现在可以直接访问 myRequest 对象了
                    System.out.println("Generated URL: Successfully for request ID: " + myRequest.getRequestId() + ", URL: " + url);
                })
                .map(url -> MyResponse.builder().url(url).build())
                .doOnError(e -> {
                    // 在错误处理中也可以访问 myRequest
                    System.out.println("Error occurred for request ID: " + myRequest.getRequestId() + ", Error: " + e.getMessage());
                });
    }
}