How to wait for RestTemplate responses on shutdown?

Question

I use springc RestTemplateto send requests POSTto a web server.

When my application is disconnected (e.g. undeployed from tomcat), shutdown should be delayed until all pending responses are received (during the timeout).

RestTemplate uses HttpComponentsClientHttpRequestFactoryunder the hood.

Question: how can I say springdelay the shutdown? @PreDestroymay be one possibility, but how can I detect pending restTemplate requests?

Answer 1

I think there is no ready-made solution, as stated in https://github.com/spring-projects/spring-boot/issues/4657

Tomcat

@Component
@Scope("singleton")
public class ApplicationContextClosedListener implements ApplicationListener<ContextClosedEvent>, TomcatConnectorCustomizer {

    private static final Logger LOGGER = LoggerFactory.getLogger(ApplicationContextClosedListener.class);

    private volatile Connector connector;

    @Value("${timeout}")
    private Integer timeout;

    @Override
    public void customize(Connector connector) {
        this.connector = connector;
    }

    @Override
    public void onApplicationEvent(ContextClosedEvent event) {
        if (connector != null) {
            shutdownGracefully();
        }
    }

    private void shutdownGracefully() {
        connector.pause();
        Executor executor = connector.getProtocolHandler().getExecutor();

        if (executor instanceof ThreadPoolExecutor) {
            ThreadPoolExecutor threadPoolExecutor = (ThreadPoolExecutor) executor;

            try {
                threadPoolExecutor.shutdown();
                if (!threadPoolExecutor.awaitTermination(timeout, TimeUnit.SECONDS)) {
                    LOGGER.warn("Shutdown: Tomcat thread pool did not shut down gracefully within specified period. Proceeding with forceful shutdown");
                }
                threadPoolExecutor.shutdownNow();

                LOGGER.info("Shutdown: the executor shutdown completed");
            } catch (InterruptedException ex) {
                LOGGER.error("Shutdown: Interrupt signal received");
                threadPoolExecutor.shutdownNow();
                Thread.currentThread().interrupt();
            }
        }
    }
}

Answer 2

ExecutorService, @PreDestroy, .

@Slf4j
@Component
public class SenderService {
  private static final int TWO_SECONDS = 2;
  private RestTemplate restTemplate;
  private ExecutorService executorService;

  public SenderService() {
    this.restTemplate = new RestTemplate();
    this.executorService = Executors.newFixedThreadPool(1);
  }

  public void sendRequest() throws Exception {
    executorService.submit(() -> {
      ZonedDateTime now = ZonedDateTime.now();
      log.info("Sending request at {} ...", now);
      restTemplate.getForObject("https://httpbin.org/delay/{delay}", Void.class, TWO_SECONDS, now);
      log.info("Response received for request at {}", now);
      return null;
    }).get();
  }

  @PreDestroy
  public void destroy() throws InterruptedException {
    log.info("Shutting down sender service...");
    executorService.shutdown();
    executorService.awaitTermination(3, TimeUnit.SECONDS);
    log.info("Sender service terminated.");
  }
}

- - .

@SpringBootApplication
public class Application {

  public static void main(final String[] args) throws Exception {
    ConfigurableApplicationContext run = SpringApplication.run(Application.class, args);
    SenderService senderService = run.getBean(SenderService.class);
    while (true) {
      senderService.sendRequest();
    }
  }
}

, , delay endpoint, ExecutorService 3 , . executorService.shutdown() , , .

spring -boot tomcat, spring.