Java application graceful shutdown

AppShutdown.java

ESRunner runner = new ESRunner();

// Gracefully shutdown
final Thread mainThread = Thread.currentThread();

Runtime.getRuntime().addShutdownHook(new Thread(() -> {
    try {
        runner.shutdown();  // Clean up work
        mainThread.join(200);  // wait main thread to stop
    } catch (InterruptedException e) {
        Thread.currentThread().interrupt();
    }
}));

runner.run();



// In ESRunner:

private final AtomicBoolean closed = new AtomicBoolean(false);

public void run() {
  try {

      while (!closed.get()) {
          // Doing stuff
      }
  } catch (WakeupException e) {
      // Ignore exception if closing
      if (!closed.get()) throw e;
  } finally {
      consumer.close();
      client.close();
  }
}

// Shutdown hook which can be called from a separate thread
public void shutdown() {
  log.info("Shutting down es runner");
  closed.set(true);
  consumer.wakeup();
}