java - Webflux 如何从流内部设置 cookie

Question

我想知道谁可以从 ServerResponse 正文中设置 cookie。
我找到了几个解决方案，但这些解决方案是从这样的外部 body 功能中保存 cookie。

ServerResponse.BodyBuilder response = ServerResponse.ok();
response.cookie(ResponseCookie.from("mycookie", "myVal").build());
return response.body(Mono.just("Welcome"), String.class);

这可以节省 cookie，如果我们没有动态 cookie 值，这个解决方案就很好。
现在我想写来自 db 的 cookie 值。
像这样。

return response.body(
                Mono.just(101)
                        .flatMap(i ->
                            userRepository
                                    .findById(i)
                                    .map(user -> {
                                        Long time = user.getLastLoginTime();
                                        // set time as cookie,
                                        // I would like to save cookie from here
                                        return user.getEmail();
                                    })
                        )
                , String.class);

那么如何从 map 函数内部保存 cookie 并且由于我正在返回流，webflux 将自动订阅它，直到我的流不被执行。

Answer 1

使用 ServerWebExchange 定义 Controller 的方法签名参数如下:

public Mono<String> cookieSetter(ServerWebExchange exchange)

...然后您可以使用 exchange.getResponse().addCookie() react 链中的任何地方。 (我会将其作为专用的副作用来做，而不是将其捆绑在单个 map 调用中。)所以您可以执行以下操作:

return response.body(
        Mono.just(101)
                .flatMap(i ->
                        userRepository
                                .findById(i)
                                .doOnNext(user -> {
                                    exchange.getResponse().addCookie(ResponseCookie.from("time", user.getLastLoginTime())
                                            .httpOnly(true)
                                            .build());
                                })
                                .map(User::getEmail)
                )
        , String.class);