Question

我有一个端点流，如示例代码块中所示。流式传输时，我通过streamHelper.getStreamSuspendCount()调用异步方法。我正在更改状态时停止此异步方法。但是，当浏览器关闭并且会话终止时，我无法访问此异步方法。更改状态时，我正在会话范围内停止异步方法。但是，当浏览器关闭并且会话终止时，我无法访问此异步方法。会话关闭后如何访问该范围？

@RequestMapping(value = "/stream/{columnId}/suspendCount", produces = MediaType.TEXT_EVENT_STREAM_VALUE)
@ResponseBody
public Flux<Integer> suspendCount(@PathVariable String columnId) {
    ColumnObject columnObject = streamHelper.findColumnObjectInListById(columnId);
    return streamHelper.getStreamSuspendCount(columnObject);
}


getStreamSuspendCount(ColumnObject columnObject) {
   ...
   //async flux
   Flux<?> newFlux = beSubscribeFlow.get(i);
   Disposable disposable = newFlux.subscribe();
   beDisposeFlow.add(disposable); // my session scope variable. if change state, i will kill disposable (dispose()).
   ...
   return Flux.fromStream(Stream.generate(() -> columnObject.getPendingObject().size())).distinctUntilChanged()
                    .doOnNext(i -> {
                        System.out.println(i);
                    }));
}

Answer 1

我认为问题的一部分是您试图获取一个Disposable，您想在会话结束时调用它。但是，这样做是您自己订阅序列。 Spring Framework还将订阅Flux返回的getStreamSuspendCount，正是该订阅需要取消，SSE客户端才能得到通知。

现在如何实现呢？您需要的是一种“阀门”，它将在接收到外部信号时取消其信号源。 takeUntilOther(Publisher<?>)就是这样做的。

因此，现在您需要一个Publisher<?>，可以将其绑定到会话生命周期（更具体地说是会话关闭事件）：takeUntilOther发出后，将立即取消其来源。

有2个选项：

会话关闭事件在类似侦听器的API中公开：使用Mono.create
您真的需要手动触发取消操作：使用MonoProcessor.create()，然后在时间到时，通过它推送任何值

以下是简化的示例，其中包含一些API来阐明：

创建

return theFluxForSSE.takeUntilOther(Mono.create(sink ->
    sessionEvent.registerListenerForClose(closeEvent -> sink.success(closeEvent))
));

MonoProcessor

MonoProcessor<String> processor = MonoProcessor.create();
beDisposeFlow.add(processor); // make it available to your session scope?
return theFluxForSSE.takeUntilOther(processor); //Spring will subscribe to this

让我们模拟一个计划任务关闭的会话：

Executors.newSingleThreadScheduledExecutor().schedule(() ->
    processor.onNext("STOP") // that's the key part: manually sending data through the processor to signal takeUntilOther
, 2, TimeUnit.SECONDS);

这是一个模拟的单元测试示例，您可以运行该示例以更好地了解发生了什么：

@Test
public void simulation() {
    Flux<Long> theFluxForSSE = Flux.interval(Duration.ofMillis(100));

    MonoProcessor<String> processor = MonoProcessor.create();
    Executors.newSingleThreadScheduledExecutor().schedule(() -> processor.onNext("STOP"), 2, TimeUnit.SECONDS);

    theFluxForSSE.takeUntilOther(processor.log())
                 .log()
                 .blockLast();
}

如何处理关闭的sse连接？

1 个答案:

创建

MonoProcessor