@Override public void unannounceSegments(Iterable<DataSegment> segments) { for (DataSegment segment : segments) { unannounceSegment(segment); } }
@Test public void testSkipLoadSpec() throws Exception { skipLoadSpec = true; Iterator<DataSegment> segIter = testSegments.iterator(); DataSegment firstSegment = segIter.next(); segmentAnnouncer.announceSegment(firstSegment); List<String> zNodes = cf.getChildren().forPath(testSegmentsPath); for (String zNode : zNodes) { DataSegment announcedSegment = Iterables.getOnlyElement(segmentReader.read(joiner.join(testSegmentsPath, zNode))); Assert.assertEquals(announcedSegment, firstSegment); Assert.assertNull(announcedSegment.getLoadSpec()); } segmentAnnouncer.unannounceSegment(firstSegment); Assert.assertTrue(cf.getChildren().forPath(testSegmentsPath).isEmpty()); }
@Test public void testSkipDimensions() throws Exception { skipDimensionsAndMetrics = true; Iterator<DataSegment> segIter = testSegments.iterator(); DataSegment firstSegment = segIter.next(); segmentAnnouncer.announceSegment(firstSegment); List<String> zNodes = cf.getChildren().forPath(testSegmentsPath); for (String zNode : zNodes) { DataSegment announcedSegment = Iterables.getOnlyElement(segmentReader.read(joiner.join(testSegmentsPath, zNode))); Assert.assertEquals(announcedSegment, firstSegment); Assert.assertTrue(announcedSegment.getDimensions().isEmpty()); Assert.assertTrue(announcedSegment.getMetrics().isEmpty()); } segmentAnnouncer.unannounceSegment(firstSegment); Assert.assertTrue(cf.getChildren().forPath(testSegmentsPath).isEmpty()); }
@Test public void testRun() throws Exception { segmentAnnouncer.announceSegments(testSegments); waitForSync(batchServerInventoryView, testSegments); DruidServer server = Iterables.get(batchServerInventoryView.getInventory(), 0); Set<DataSegment> segments = Sets.newHashSet(server.getSegments()); Assert.assertEquals(testSegments, segments); DataSegment segment1 = makeSegment(101); DataSegment segment2 = makeSegment(102); segmentAnnouncer.announceSegment(segment1); segmentAnnouncer.announceSegment(segment2); testSegments.add(segment1); testSegments.add(segment2); waitForSync(batchServerInventoryView, testSegments); Assert.assertEquals(testSegments, Sets.newHashSet(server.getSegments())); segmentAnnouncer.unannounceSegment(segment1); segmentAnnouncer.unannounceSegment(segment2); testSegments.remove(segment1); testSegments.remove(segment2); waitForSync(batchServerInventoryView, testSegments); Assert.assertEquals(testSegments, Sets.newHashSet(server.getSegments())); }
Assert.assertEquals(2, snapshot.getCounter().getCounter()); segmentAnnouncer.unannounceSegment(firstSegment); segmentAnnouncer.unannounceSegment(secondSegment);
segmentAnnouncer.unannounceSegment(segment); testSegments.remove(segment);
testSegments.add(oldSegment); segmentAnnouncer.unannounceSegment(oldSegment); testSegments.remove(oldSegment); segmentAnnouncer.unannounceSegment(segment2); testSegments.remove(segment2);
@Override public void unannounceSegments(Iterable<DataSegment> segments) { for (DataSegment segment : segments) { unannounceSegment(segment); } }