From b46aff12ae5684f4c73b094844d06c1305ffb1f3 Mon Sep 17 00:00:00 2001 From: Bingkun Guo Date: Thu, 7 May 2015 12:35:12 -0700 Subject: [PATCH] Unit test for IndexGeneratorJob --- .../druid/indexer/IndexGeneratorJobTest.java | 346 ++++++++++++++++++ 1 file changed, 346 insertions(+) create mode 100644 indexing-hadoop/src/test/java/io/druid/indexer/IndexGeneratorJobTest.java diff --git a/indexing-hadoop/src/test/java/io/druid/indexer/IndexGeneratorJobTest.java b/indexing-hadoop/src/test/java/io/druid/indexer/IndexGeneratorJobTest.java new file mode 100644 index 00000000000..bfaac4a465b --- /dev/null +++ b/indexing-hadoop/src/test/java/io/druid/indexer/IndexGeneratorJobTest.java @@ -0,0 +1,346 @@ +/* + * Licensed to Metamarkets Group Inc. (Metamarkets) under one + * or more contributor license agreements. See the NOTICE file + * distributed with this work for additional information + * regarding copyright ownership. Metamarkets licenses this file + * to you under the Apache License, Version 2.0 (the + * "License"); you may not use this file except in compliance + * with the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, + * software distributed under the License is distributed on an + * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY + * KIND, either express or implied. See the License for the + * specific language governing permissions and limitations + * under the License. + */ + +package io.druid.indexer; + +import com.fasterxml.jackson.databind.InjectableValues; +import com.fasterxml.jackson.databind.ObjectMapper; +import com.fasterxml.jackson.databind.jsontype.NamedType; +import com.google.common.collect.ImmutableList; +import com.google.common.collect.ImmutableMap; +import com.google.common.collect.Lists; +import com.google.common.collect.Maps; +import com.metamx.common.Granularity; +import io.druid.data.input.impl.CSVParseSpec; +import io.druid.data.input.impl.DimensionsSpec; +import io.druid.data.input.impl.StringInputRowParser; +import io.druid.data.input.impl.TimestampSpec; +import io.druid.granularity.QueryGranularity; +import io.druid.jackson.DefaultObjectMapper; +import io.druid.query.aggregation.AggregatorFactory; +import io.druid.query.aggregation.LongSumAggregatorFactory; +import io.druid.segment.indexing.DataSchema; +import io.druid.segment.indexing.granularity.UniformGranularitySpec; +import io.druid.timeline.DataSegment; +import io.druid.timeline.partition.HashBasedNumberedShardSpec; +import io.druid.timeline.partition.ShardSpec; +import io.druid.timeline.partition.SingleDimensionShardSpec; +import org.apache.commons.io.FileUtils; +import org.joda.time.DateTime; +import org.joda.time.DateTimeComparator; +import org.joda.time.Interval; +import org.junit.Assert; +import org.junit.Before; +import org.junit.Rule; +import org.junit.Test; +import org.junit.rules.TemporaryFolder; +import org.junit.runner.RunWith; +import org.junit.runners.Parameterized; + +import java.io.File; +import java.io.IOException; +import java.util.Arrays; +import java.util.Collection; +import java.util.List; +import java.util.Map; + +@RunWith(Parameterized.class) +public class IndexGeneratorJobTest +{ + + @Parameterized.Parameters(name = "partitionType={0}, interval={1}, shardInfoForEachSegment={2}, data={3}") + public static Collection constructFeed() + { + return Arrays.asList( + new Object[][]{ + { + "single", + "2014-10-22T00:00:00Z/P2D", + new String[][][]{ + { + { null, "c.example.com" }, + { "c.example.com", "e.example.com" }, + { "e.example.com", "g.example.com" }, + { "g.example.com", "i.example.com" }, + { "i.example.com", null } + }, + { + { null, "c.example.com" }, + { "c.example.com", "e.example.com" }, + { "e.example.com", "g.example.com" }, + { "g.example.com", "i.example.com" }, + { "i.example.com", null } + } + }, + ImmutableList.of( + "2014102200,a.example.com,100", + "2014102200,b.exmaple.com,50", + "2014102200,c.example.com,200", + "2014102200,d.example.com,250", + "2014102200,e.example.com,123", + "2014102200,f.example.com,567", + "2014102200,g.example.com,11", + "2014102200,h.example.com,251", + "2014102200,i.example.com,963", + "2014102200,j.example.com,333", + "2014102300,a.example.com,100", + "2014102300,b.exmaple.com,50", + "2014102300,c.example.com,200", + "2014102300,d.example.com,250", + "2014102300,e.example.com,123", + "2014102300,f.example.com,567", + "2014102300,g.example.com,11", + "2014102300,h.example.com,251", + "2014102300,i.example.com,963", + "2014102300,j.example.com,333" + ) + }, + { + "hashed", + "2014-10-22T00:00:00Z/P1D", + new Integer[][][]{ + { + { 0, 4 }, + { 1, 4 }, + { 2, 4 }, + { 3, 4 } + } + }, + ImmutableList.of( + "2014102200,a.example.com,100", + "2014102201,b.exmaple.com,50", + "2014102202,c.example.com,200", + "2014102203,d.example.com,250", + "2014102204,e.example.com,123", + "2014102205,f.example.com,567", + "2014102206,g.example.com,11", + "2014102207,h.example.com,251", + "2014102208,i.example.com,963", + "2014102209,j.example.com,333", + "2014102210,k.example.com,253", + "2014102211,l.example.com,321", + "2014102212,m.example.com,3125", + "2014102213,n.example.com,234", + "2014102214,o.example.com,325", + "2014102215,p.example.com,3533", + "2014102216,q.example.com,587" + ) + } + } + ); + } + + public final @Rule TemporaryFolder temporaryFolder = new TemporaryFolder(); + + private ObjectMapper mapper; + private HadoopDruidIndexerConfig config; + private File dataFile; + private File tmpDir; + private Interval interval; + private String partitionType; + private Object[][][] shardInfoForEachSegment; + private List data; + + public IndexGeneratorJobTest( + String partitionType, + String interval, + Object[][][] shardInfoForEachSegment, + List data + ) throws IOException + { + this.partitionType = partitionType; + this.shardInfoForEachSegment = shardInfoForEachSegment; + this.interval = new Interval(interval); + this.data = data; + } + + @Before + public void setUp() throws Exception + { + mapper = new DefaultObjectMapper(); + mapper.registerSubtypes(new NamedType(HashBasedNumberedShardSpec.class, "hashed")); + mapper.registerSubtypes(new NamedType(SingleDimensionShardSpec.class, "single")); + InjectableValues inject = new InjectableValues.Std().addValue(ObjectMapper.class, mapper); + mapper.setInjectableValues(inject); + + dataFile = temporaryFolder.newFile(); + tmpDir = temporaryFolder.newFolder(); + + FileUtils.writeLines(dataFile, data); + + config = new HadoopDruidIndexerConfig( + new HadoopIngestionSpec( + new DataSchema( + "website", + new StringInputRowParser( + new CSVParseSpec( + new TimestampSpec("timestamp", "yyyyMMddHH", null), + new DimensionsSpec(ImmutableList.of("host"), null, null), + null, + ImmutableList.of("timestamp", "host", "visited_num") + ) + ), + new AggregatorFactory[]{new LongSumAggregatorFactory("visited_num", "visited_num")}, + new UniformGranularitySpec( + Granularity.DAY, QueryGranularity.NONE, ImmutableList.of(this.interval) + ) + ), + new HadoopIOConfig( + ImmutableMap.of( + "paths", + dataFile.getCanonicalPath(), + "type", + "static" + ), + null, + tmpDir.getCanonicalPath() + ), + new HadoopTuningConfig( + tmpDir.getCanonicalPath(), + null, + null, + null, + null, + null, + false, + false, + false, + false, + null, + false, + false, + false, + null, + null + ) + ) + ); + + config.setShardSpecs(loadShardSpecs(partitionType, shardInfoForEachSegment)); + config = HadoopDruidIndexerConfig.fromSpec(config.getSchema()); + } + + private List constructShardSpecFromShardInfo(String partitionType, Object[][] shardInfoForEachShard) + { + List specs = Lists.newArrayList(); + if (partitionType.equals("hashed")) { + for (Integer[] shardInfo : (Integer[][]) shardInfoForEachShard) { + specs.add(new HashBasedNumberedShardSpec(shardInfo[0], shardInfo[1], HadoopDruidIndexerConfig.jsonMapper)); + } + } else if (partitionType.equals("single")) { + int partitionNum = 0; + for (String[] shardInfo : (String[][]) shardInfoForEachShard) { + specs.add(new SingleDimensionShardSpec("host", shardInfo[0], shardInfo[1], partitionNum++)); + } + } else { + throw new RuntimeException(String.format("Invalid partition type:[%s]", partitionType)); + } + + return specs; + } + + private Map> loadShardSpecs( + String partitionType, + Object[][][] shardInfoForEachShard + ) + { + Map> shardSpecs = Maps.newTreeMap(DateTimeComparator.getInstance()); + int shardCount = 0; + int segmentNum = 0; + for (Interval segmentGranularity : config.getSegmentGranularIntervals().get()) { + List specs = constructShardSpecFromShardInfo(partitionType, shardInfoForEachShard[segmentNum++]); + List actualSpecs = Lists.newArrayListWithExpectedSize(specs.size()); + for (int i = 0; i < specs.size(); ++i) { + actualSpecs.add(new HadoopyShardSpec(specs.get(i), shardCount++)); + } + + shardSpecs.put(segmentGranularity.getStart(), actualSpecs); + } + + return shardSpecs; + } + + @Test + public void testIndexGeneratorJob() throws IOException + { + verifyJob(new IndexGeneratorJob(config)); + } + + @Test + public void testLegacyIndexGeneratorJob() throws IOException + { + verifyJob(new LegacyIndexGeneratorJob(config)); + } + + private void verifyJob(IndexGeneratorJob job) throws IOException + { + JobHelper.runJobs(ImmutableList.of(job), config); + + int segmentNum = 0; + for (DateTime currTime = interval.getStart(); currTime.isBefore(interval.getEnd()); currTime = currTime.plusDays(1)) { + Object[][] shardInfo = shardInfoForEachSegment[segmentNum++]; + File segmentOutputFolder = new File( + String.format( + "%s/%s/%s_%s/%s", + config.getSchema().getIOConfig().getSegmentOutputPath(), + config.getSchema().getDataSchema().getDataSource(), + currTime.toString(), + currTime.plusDays(1).toString(), + config.getSchema().getTuningConfig().getVersion() + ) + ); + Assert.assertTrue(segmentOutputFolder.exists()); + Assert.assertEquals(shardInfo.length, segmentOutputFolder.list().length); + + for (int partitionNum = 0; partitionNum < shardInfo.length; ++partitionNum) { + File individualSegmentFolder = new File(segmentOutputFolder, Integer.toString(partitionNum)); + Assert.assertTrue(individualSegmentFolder.exists()); + + File descriptor = new File(individualSegmentFolder, "descriptor.json"); + File indexZip = new File(individualSegmentFolder, "index.zip"); + Assert.assertTrue(descriptor.exists()); + Assert.assertTrue(indexZip.exists()); + + DataSegment dataSegment = mapper.readValue(descriptor, DataSegment.class); + Assert.assertEquals("website", dataSegment.getDataSource()); + Assert.assertEquals(config.getSchema().getTuningConfig().getVersion(), dataSegment.getVersion()); + Assert.assertEquals(new Interval(currTime, currTime.plusDays(1)), dataSegment.getInterval()); + Assert.assertEquals("local", dataSegment.getLoadSpec().get("type")); + Assert.assertEquals(indexZip.getCanonicalPath(), dataSegment.getLoadSpec().get("path")); + Assert.assertEquals("host", dataSegment.getDimensions().get(0)); + Assert.assertEquals("visited_num", dataSegment.getMetrics().get(0)); + Assert.assertEquals(Integer.valueOf(9), dataSegment.getBinaryVersion()); + if (partitionType.equals("hashed")) { + Integer[] hashShardInfo = (Integer[]) shardInfo[partitionNum]; + HashBasedNumberedShardSpec spec = (HashBasedNumberedShardSpec) dataSegment.getShardSpec(); + Assert.assertEquals((int) hashShardInfo[0], spec.getPartitionNum()); + Assert.assertEquals((int) hashShardInfo[1], spec.getPartitions()); + } else if (partitionType.equals("single")) { + String[] singleDimensionShardInfo = (String[]) shardInfo[partitionNum]; + SingleDimensionShardSpec spec = (SingleDimensionShardSpec) dataSegment.getShardSpec(); + Assert.assertEquals(singleDimensionShardInfo[0], spec.getStart()); + Assert.assertEquals(singleDimensionShardInfo[1], spec.getEnd()); + } else { + throw new RuntimeException(String.format("Invalid partition type:[%s]", partitionType)); + } + } + } + } + +}