Spring WebFlux中如何在返回值为Null时抛出自定义异常？

我有一段响应式代码，希望在UserAuthenticationToken的getUserPrincipalName方法返回null时抛出自定义异常。原始代码如下：

ReactiveSecurityContextHolder.getContext()
                .map(SecurityContext::getAuthentication)
                .map(authentication ->(UserAuthenticationToken) authentication)
                .map(UserAuthenticationToken::getUserPrincipalName)
                //此处希望当getUserPrincipalName返回null时抛出异常

尝试过的无效写法 #

• 写法一：

.map(UserAuthenticationToken::getUserPrincipalName)
.switchIfEmpty(Mono.error(new MissingPrincipalException("Missing email field in the JWT token")));

• 写法二：

.map(UserAuthenticationToken::getUserPrincipalName)
.filter(principal -> !Objects.isNull(principal))
.switchIfEmpty(Mono.error(new MissingPrincipalException("Missing email field in the JWT token")))

问题原因 #

之前的写法无效是因为：map操作即使返回null，对应的Mono依然是有值的（值为null），而switchIfEmpty仅在Mono完全为空（没有发射任何元素）时才会触发，所以无法捕获null的情况。

正确解决方案 #

可以通过以下两种方式实现需求：

方案1：使用flatMap判断null

ReactiveSecurityContextHolder.getContext()
                .map(SecurityContext::getAuthentication)
                .map(authentication ->(UserAuthenticationToken) authentication)
                .map(UserAuthenticationToken::getUserPrincipalName)
                .flatMap(principal -> {
                    if (principal == null) {
                        return Mono.error(new MissingPrincipalException("Missing email field in the JWT token"));
                    }
                    return Mono.just(principal);
                });

方案2：使用handle操作符（更简洁）

ReactiveSecurityContextHolder.getContext()
                .map(SecurityContext::getAuthentication)
                .map(authentication ->(UserAuthenticationToken) authentication)
                .map(UserAuthenticationToken::getUserPrincipalName)
                .handle((principal, sink) -> {
                    if (principal == null) {
                        sink.error(new MissingPrincipalException("Missing email field in the JWT token"));
                    } else {
                        sink.next(principal);
                    }
                });

说明 #

• flatMap可以根据当前元素的值返回不同的Mono：当值为null时返回错误信号，否则返回包含该值的正常Mono。
• handle是专门用于处理元素的操作符，支持直接发射值或错误，非常适合这种需要对元素进行非空校验的场景。

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