Apache
/
druid
mirror of https://github.com/apache/druid.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #2119 from navis/fix-sporadic-testfail2 

Fix sporadic fail of DruidCoordinatorTest#testCoordinatorRun
This commit is contained in:
Himanshu 2015-12-21 23:28:32 -06:00
parent e38b7554e4 7b24d1f355
commit 7b5fd76058
2 changed files with 14 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
server/src/main/java/io/druid/server/coordinator/DruidCoordinator.java
View File

@ -247,6 +247,16 @@ public class DruidCoordinator
return retVal;
}
CountingMap<String> getLoadPendingDatasources() {
final CountingMap<String> retVal = new CountingMap<>();
for (LoadQueuePeon peon : loadManagementPeons.values()) {
for (DataSegment segment : peon.getSegmentsToLoad()) {
retVal.add(segment.getDataSource(), 1);
}
}
return retVal;
}
public Map<String, Double> getLoadStatus()
{
Map<String, Double> loadStatus = Maps.newHashMap();

4
server/src/test/java/io/druid/server/coordinator/DruidCoordinatorTest.java
View File

@ -338,6 +338,10 @@ public class DruidCoordinatorTest extends CuratorTestBase
Assert.assertEquals(1, segmentAvailability.size());
Assert.assertEquals(0l, segmentAvailability.get(dataSource));
while (coordinator.getLoadPendingDatasources().get(dataSource).get() > 0) {
Thread.sleep(50);
}
Map<String, CountingMap<String>> replicationStatus = coordinator.getReplicationStatus();
Assert.assertNotNull(replicationStatus);
Assert.assertEquals(1, replicationStatus.entrySet().size());