public int await() throws Exception { latch.await(); if (exception != null) { exception.addSuppressed(new DrillRuntimeException("Exception in executor threadpool")); throw exception; } return count.get(); }