Spring Boot应用程序阻止StreamingResponseBody

时间:2020-10-07 15:39:56

标签: java spring-boot streamingresponsebody

我有一个Spring Boot应用程序,在其中创建了一个POST方法,该方法以流方式将数据发送给调用方。下面的代码:

@RequestMapping(value = "/mapmatchstreaming", method = RequestMethod.POST)
public ResponseEntity<StreamingResponseBody> handleRequest(@RequestParam(value = "data", required = true) String data, @RequestParam(value = "mnr", required = true) Boolean mnr) {
    logger.info("/mapmatchstreaming endpoint");

    try {
        Semaphore semaphore = new Semaphore(1);
        ObjectMapper mapper = new ObjectMapper();

        StreamingResponseBody responseBody = new StreamingResponseBody() {
            @Override
            public void writeTo (OutputStream outputStream) throws IOException {
                // For each map
                DataReader dataReader = new DataReader(data, "2020.06.011");

                for(String mapRoot: dataReader.getMapsFolders()) {
                    dataReader = new DataReader(data, "2020.06.011");
                    DistributedMapMatcherStreaming distributedMapMatcher = new DistributedMapMatcherStreaming(dataReader.getTraces(), mapRoot, dataReader.getBoundingBox());
                    distributedMapMatcher.mapMatchBatch(new DistributedMapMatcherResult() {
                        @Override
                        public void onCorrectlyMapMatched(MapMatchedTrajectory mapMatchedTrajectory) {
                            try {
                                semaphore.acquire();
                                outputStream.write(mapper.writeValueAsString(mapMatchedTrajectory).getBytes());
                                outputStream.flush();
                            }
                            catch (Exception e) {
                                e.printStackTrace();
                                logger.error(String.format("Writing to output stream error: %s", e.getMessage()));
                            } finally{
                                semaphore.release();
                            }
                        }
                    });
                }
            }
        };

        return new ResponseEntity<StreamingResponseBody>(responseBody, HttpStatus.OK);
    }
    catch (Exception e) {
        logger.error(String.format("Map-matching result ERROR: %s", ExceptionUtils.getStackTrace(e)));
        return new ResponseEntity<StreamingResponseBody>(HttpStatus.BAD_REQUEST);
    }
}

它工作得很好,但是问题是,即使我设置了server.tomcat.threads.max=1,如果有多次调用到达此方法,则所有这些都将并行运行。在非流式版本中,每个下一个呼叫都等待当前的呼叫完成。

在Spring中是否有可能阻止流式通话?谢谢。

编辑:我暂时通过只使用1个许可的全局信号量解决了问题,但是我认为这不是理想的解决方案。

0 个答案:

没有答案