我们如何停止后端进程？

Question

在我的 Spring Boot 应用程序中，我有以下 REST 服务

@GetMapping("/testThread")
public Map<String, Object> testThread(HttpServletRequest request){

    for (int i=0;i<100000;i++){
        System.out.println(i);
    }
    return null;
}

在这个网络服务中，我从 POSTMAN 工具调用，并且在这个网络服务循环中运行了 100 000 次。

我的要求是在从邮递员调用网络服务之后，我可以使用 Cancel Request 按钮取消该请求，然后在后端 循环也应该停止。

到目前为止，循环正在完成执行。

我们有什么选择吗？

Answer 1

你有两个原型选项，第一个是请求线程以某种方式取消请求。当您在返回响应之前进行工作时，理论上您可以在每次迭代时将一些数据发送到管道中，例如，一个空格字符，然后刷新输出写入器。如果调用者关闭了管道，您将收到一个 IOException 并终止循环。不过这很hacky。

另一种方法是为工作线程返回某种标识符，以便可以发送取消命令来标识应该取消哪个线程。为此，您需要将工作线程推入新线程并立即返回包含工作标识符的响应。这意味着即使工作线程仍在工作，初始客户端调用也会立即得到响应。

请注意，以下示例仅适用于单个服务器或粘性负载平衡环境。当然，您可以将状态图扩展到数据库表以实现更复杂的架构。

您可能还想添加更多端点以获取当前状态或结果。

private enum State {
   Running,
   Cancel
}

private final Map<String, State> state = Collections.synchronizedMap(new HashMap<>());
private final ExecutorService executor = Executors.newCachedThreadPool();

@GetMapping("/cancelThread/{id}")
public Map<String, Object> testThread(@PathVariable String id) {
    state.computeIfPresent(id, (k, v) -> State.Cancel);
}

@GetMapping("/testThread")
public Map<String, Object> testThread(HttpServletRequest request) {
    String id = java.util.UUID.randomUUID().toString()
    state.put(id, State.Running);

    executor.submit(() -> doWork(id));

    Map<String, Object> response = new HashMap<>();
    response.put("transactionId", id);
    return response;
}

private void doWork(String id) {
    executor.submit(() -); 

    try {
        for (int i = 0; i < 100000; i++) {
          if (state.get(id) == State.Cancel) {
              break;
          }

          System.out.println(i);
        }
    } finally {
        state.remove(id);
    }
}