spring-boot - Spring WebFlux - 如何从请求中获取数据？

Question

尝试将我的 Spring Boot 应用程序迁移到 WebFlux，我开始转换 API 层，同时保持存储库完好无损(即数据库访问是同步和阻塞的)。我遇到了如何从 Mono/Flux 类型获取数据并将它们转发到存储库的问题。

考虑以下内容

@POST
@Path("/register")
public String register( String body ) throws Exception
{
    ObjectMapper objectMapper = json();

    User user = objectMapper.readValue( body, User.class );

    int random = getRandomNumber( 111111, 999999 );

    String uuid = null;

    //first, check if user already did registration from that phone
    UserDbRecord userDbRecord = UserDAO.getInstance().getUserByPhone( user.phone );

    if( userDbRecord != null )
    {
        logger.info( "register. User already exist with phone: " + user.phone + ", id: " + userDbRecord.getId() );

        uuid = userDbRecord.getToken();
    }
    else
    {
        uuid = UUID.randomUUID().toString();
    }

    SMS.send( user.phone, random );

    Auth auth = new Auth();
    auth.token = uuid;

    return objectMapper.writeValueAsString( auth );
}

因此尝试执行以下操作:

public Mono<ServerResponse> register( ServerRequest request )
{
    Mono<User> user = request.bodyToMono( User.class );

    Mono<UserDbRecord> userDbRecord = user.flatMap( u -> Mono.just( userRepository.findByPhone( u.phone ) ) );

    int random = getRandomNumber( 111111, 999999 );

    String uuid = null;

    //first, check if user already did registration from that phone

    //now what???
    if( userDbRecord != null )
    {
        logger.info( "register. User already exist with phone: " + userDbRecord.getPhone() + ", id: " + userDbRecord.getId() );

        uuid = userDbRecord.getToken();
    }
    else
    {
        uuid = UUID.randomUUID().toString();
    }

    SMS.send( user.phone, random );

    Auth auth = new Auth();
    auth.token = uuid;

    return ok().contentType( APPLICATION_JSON ).syncBody( auth );
}

检查 userDbRecord Mono 是否为空以及如果不为空则从中提取电话属性的最佳方法是什么？

Answer 1

重新思考数据处理方式

在使用 RxJava 或 Project Reactor 的响应式编程中，从头到尾继续您的流程非常重要。

在您的情况下，您必须将命令式验证/检查替换为响应式(Reactive)验证/检查:

public Mono<ServerResponse> register( ServerRequest request )
{
    return request
        .bodyToMono( User.class )
        // make sure you use Reactive DataBase Access in order to 
        // get the all benefits of Non-Blocking I/O with Project Reactor
        // if you use JPA - consider Moving to R2DBC r2dbc.io
        .flatMap( user -> // <1>
            Mono.just( userRepository.findByPhone( user.phone ) ) // <2>
                .map(userDbRecord -> {
                    logger.info( "register. User already exist with phone: " + userDbRecord.getPhone() + ", id: " + userDbRecord.getId() );
                    return userDbRecord.getToken();
                })
                .switchIfEmpty(Mono.fromSupplier(() -> UUID.randomUUID().toString())) <3>
                .flatMap(uuid -> {
                    SMS.send( user.phone, random ); <4>
                    Auth auth = new Auth();
                    auth.token = uuid;
                    return ok().contentType( APPLICATION_JSON ).syncBody( auth );
                })
        );
}

以上示例展示了如何将命令式 Controller 的方法重写为响应式(Reactive)方法。我在下面对它们进行了一些评论和描述:

1. 这里我使用 flatMap 来保持对创建的闭包中的 User 实体的访问。
2. 确保您从头到尾都使用非阻塞、响应式 I/O -> 忽略此规则可能会导致否定 WebFlux 的所有优势。如果您使用 JPA，请考虑迁移到 R2DBC 和 Spring Data R2DBC，它们为您提供了一种响应式、非阻塞的 JPA 替代品
3. 频繁生成 UUID 会导致线程阻塞 -> https://stackoverflow.com/a/14533384/4891253
4. 确保这是非阻塞的