mirror of https://github.com/apache/druid.git
Wait for any pending realtime task to complete before disabling datasource (#3757)
Noticed this in our internal testing, sometimes realtime index tasks in kafkaIndexing service can get stuck waiting for handoff if datasource is disabled before there task completion. This is a workaround to ensure integration tests do not hit this case until https://github.com/druid-io/druid/issues/1729 is fixed.
This commit is contained in:
parent
06d0ef9c6c
commit
361af4c94f
|
@ -69,6 +69,9 @@ public abstract class AbstractIndexerTest
|
|||
|
||||
protected void unloadAndKillData(final String dataSource, String start, String end) throws Exception
|
||||
{
|
||||
// Wait for any existing index tasks to complete before disabling the datasource otherwise
|
||||
// realtime tasks can get stuck waiting for handoff. https://github.com/druid-io/druid/issues/1729
|
||||
waitForAllTasksToComplete();
|
||||
Interval interval = new Interval(start + "/" + end);
|
||||
coordinator.unloadSegmentsForDataSource(dataSource, interval);
|
||||
RetryUtil.retryUntilFalse(
|
||||
|
@ -82,6 +85,10 @@ public abstract class AbstractIndexerTest
|
|||
}, "Segment Unloading"
|
||||
);
|
||||
coordinator.deleteSegmentsDataSource(dataSource, interval);
|
||||
waitForAllTasksToComplete();
|
||||
}
|
||||
|
||||
protected void waitForAllTasksToComplete(){
|
||||
RetryUtil.retryUntilTrue(
|
||||
new Callable<Boolean>()
|
||||
{
|
||||
|
|
Loading…
Reference in New Issue