如何返回一个Mono，其中响应实体可以是两种不同的类型

Question

我是新来的Spring Web通量/Reactor核心，并试图执行以下功能：

>

如果返回用户，则返回“用户”类型的响应实体。如果为空，则返回“字符串”类型的响应实体

以下代码给出了一个类型错误，因为. defaultIf空（）期望类型为user的响应实体。您能建议正确的运算符/方法来实现此功能吗？

@PostMapping("api/user/login/webApp")
public Mono<ResponseEntity> login(@RequestBody Credentials credentials, ServerWebExchange serverWebExchange) {
     return userService.loginWebApp(credentials, serverWebExchange)
             .map(user -> ResponseEntity.status(HttpStatus.OK).body(user))
             .defaultIfEmpty(ResponseEntity.status(HttpStatus.UNAUTHORIZED).body("Invalid username or password"));
}

Answer 1

我将执行以下操作：

>

abstract class Response {
}

为每种响应（如UserResponse、ErrorResponse、NotFoundResponse等）创建单独的类，并从基本的响应类扩展它们

class UserResponse extends Response {
    private String login;
    private String password;

    public UserResponse(String login, String password) {
        this.login = login;
        this.password = password;
    }

    @JsonGetter("login")
    public String getLogin() {
        return login;
    }

    @JsonSetter("login")
    public void setLogin(String login) {
        this.login = login;
    }

    @JsonGetter("password")
    public String getPassword() {
        return password;
    }

    @JsonSetter("password")
    public void setPassword(String password) {
        this.password = password;
    }
}

class ErrorResponse extends Response {
    private String errorMessage;

    public ErrorResponse(String errorMessage) {
        this.errorMessage = errorMessage;
    }

    @JsonGetter("error_message")
    public String getErrorMessage() {
        return errorMessage;
    }

    @JsonSetter("error_message")
    public void setErrorMessage(String errorMessage) {
        this.errorMessage = errorMessage;
    }
}

显式设置返回值的类型Mono

@GetMapping("/test/{login}")
public Mono<ResponseEntity<Response>> test(@PathVariable(value = "login") String login) {
    return loginWebApp(login)
            .map(userResponse -> ResponseEntity.status(HttpStatus.OK).body(userResponse))
            .defaultIfEmpty(ResponseEntity.status(HttpStatus.UNAUTHORIZED).body(new ErrorResponse("bad login")));
}

现在让我们尝试一下错误的登录：

良好的登录：

完整代码可在此处找到

Answer 2

您可以使用强制转换操作符来向下转换泛型，我相信WebFlux仍然能够封送用户和字符串：

@PostMapping("api/user/login/webApp")
public Mono<ResponseEntity> login(@RequestBody Credentials credentials, ServerWebExchange serverWebExchange) {
     return userService.loginWebApp(credentials, serverWebExchange)
             .map(user -> ResponseEntity.status(HttpStatus.OK).body(user))
             .cast(ResponseEntity.class)
             .defaultIfEmpty(ResponseEntity.status(HttpStatus.UNAUTHORIZED).body("Invalid username or password"));
}