如何在Spring WebFlux中使用@PreAuthorize？

Question

在Spring Web中，我使用@PreAuthorize with SpEl来检查当前用户的权限。类似这样的东西：

@Component
public class CustomSecurity {
    public boolean checkPermission(){
        Authentication authentication = SecurityContextHolder.getContext()
                .getAuthentication();
        CurrentAccount currentAccount = (CurrentAccount)authentication.getPrincipal();
        return currentAccount.getUsername().equals("admin");
    }
}

在RestController中：

@PreAuthorize("@customSecurity.checkPermission()")
@GetMapping("/")
public Object getWidgetValues() {
    return "Hello admin";
}

现在我尝试使用WebFlux。写入了reactiveCheckPermission。

public boolean checkPermission2() {
    return ReactiveSecurityContextHolder.getContext()
            .map(SecurityContext::getAuthentication)
            .map(Authentication::getPrincipal)
            .map(o -> (CurrentAccount) o)
            .map(currentAccount -> currentAccount.getUsername().equals("admin"))
            .block();
}

但是它抛出IllegalStateException（"block（）/block First（）/block Final（）是阻塞的，线程并行不支持

将boolean更改为Mono，但@PreAuthroze只需要boolean，而不需要Mono。

如何在WebFlux中使用@PreAuthorize？

Answer 1

如果您想更手动地执行此操作，您可以：

@GetMapping("/resources")
public Flux<Object> getResources() {
 return ReactiveSecurityContextHolder.getContext()
        .map(SecurityContext::getAuthentication)
        .map(Authentication::getPrincipal)
        .map(o -> (CurrentAccount) o)
        .filter(currentAccount -> 
            currentAccount.getUsername().equals("admin"))
        .switchIfEmpty(Mono.error(new ResponseStatusException(FORBIDDEN)))
        .then(<here further processing>);
}

Answer 2

我找到了一个解决方案。

@PreAuthorize("@customSecurity.checkPermission(#account)")
@GetMapping("/")
public Object getWidgetValues(@AuthenticationPrincipal(expression = "account") Account account) {
    return "Hello admin";
}

其中，ReactiveUserDetailsService中使用的CurrentAccount

public class CurrentAccount extends User {
private Account account;

public CurrentAccount(Account account) {
    super(account.getLogin(), account.getPassword(), true, true,
            true, !account.isLocked(),
            AuthorityUtils.createAuthorityList(account.getRole().name()));
    this.account = account;
}