如何在客户端连接关闭时终止后端API后续服务调用？

核心思路 #

通过检测当前HTTP请求的连接状态，在每一步外部API调用后判断客户端是否已断开连接，若已断开则立即终止后续代码执行，避免无效的资源消耗。

具体实现步骤 #

1. 传递请求对象到Service层：在Controller中注入HttpServletRequest，将其作为参数传递给Service方法，用于后续连接状态检测。
2. 封装连接检查工具方法：在Service中实现复用的检查逻辑，统一判断客户端连接是否关闭。
3. 关键节点插入检查：在每个外部API调用完成后执行连接检查，若检测到连接关闭则中断线程并抛出异常，终止后续流程。
4. 配置事务回滚：确保流程中断时事务能正确回滚，避免数据不一致问题。

修改后的代码示例 #

Controller类

@RestController
@RequestMapping(path = "/employee")
public class EmployeeController {

    @Autowired
    private EmployeeService employeeService;

    @GetMapping(path = "/detail/{employeeId}")
    public ResponseEntity<EmployeeDetailResponse> employeeDetail(@PathVariable("employeeId") Long employeeId, HttpServletRequest request)  {
        return new ResponseEntity<>(employeeService.getEmployeeDetail(employeeId, request), HttpStatus.OK);
    }
}

Service类

@Service
public class EmployeeServiceImpl implements EmployeeService {

    @Autowired
    private EmployeeRepository employeeRepository;
    
    @Autowired
    private IdentityService identityService;
 
    @Autowired
    private HRService hrService;
    
    @Autowired
    private AccountService accountService;

    @Transactional(rollbackFor = RuntimeException.class)
    public EmployeeDetailResponse getEmployeeDetail(Long employeeId, HttpServletRequest request) {
        Employee employee = employeeRepository.findById(employeeId);

        // 调用identityService获取地址和电话信息
        IdentityDetail identityDetail = identityService.getIdentity(employeeId);

        // 检查客户端连接状态
        checkClientConnection(request);

        // 调用hrService获取雇佣信息
        EmplyementDetail employementDetail = hrService.getEmployementDetail(employeeId);

        // 再次检查连接状态
        checkClientConnection(request);

        // 调用accountService获取薪资信息
        AccountDetail accountDetail = accountService.getAccountDetail(employeeId);

        return mergeResponses(employee, identityDetail, employementDetail, accountDetail);
    }

    /**
     * 检查客户端连接是否已关闭，若关闭则中断线程并抛出异常
     */
    private void checkClientConnection(HttpServletRequest request) {
        if (request.isClosed()) {
            Thread.currentThread().interrupt();
            throw new IllegalStateException("客户端连接已关闭，终止后续操作");
        }
    }
}

注意事项 #

• Servlet版本要求：request.isClosed()方法需要Servlet 4.0及以上版本支持（对应Tomcat 9+）。若使用更低版本Servlet，可尝试通过request.getAsyncContext().hasCompleted()间接判断，但准确性稍差。
• 下游服务的中断处理：若下游服务包含阻塞IO操作，需确保其能响应线程中断信号（捕获InterruptedException并终止执行），否则中断可能无法生效。
• 事务回滚配置：@Transactional注解需指定rollbackFor = RuntimeException.class，确保抛出异常时事务能正确回滚。

内容的提问来源于stack exchange，提问作者MishraJi