Search in sources :

Example 1 with BuildAndPushMapper

use of voldemort.store.readonly.mr.BuildAndPushMapper in project voldemort by voldemort.

the class HadoopStoreWriterTest method init.

private void init() {
    tmpOutPutDirectory = TestUtils.createTempDir();
    // Setup before each test method
    conf = new JobConf();
    conf.setInt(AbstractStoreBuilderConfigurable.NUM_CHUNKS, numChunks);
    conf.set("final.output.dir", tmpOutPutDirectory.getAbsolutePath());
    conf.set("mapred.output.dir", tmpOutPutDirectory.getAbsolutePath());
    conf.set("mapred.task.id", "1234");
    /**
         * We don't have to test different types of checksums. That's covered in
         * {@link voldemort.store.readonly.checksum.CheckSumTests}.
         */
    conf.set(VoldemortBuildAndPushJob.CHECKSUM_TYPE, CheckSum.CheckSumType.NONE.name());
    // generate a list of storeDefinitions.
    List<StoreDefinition> storeDefList = Lists.newArrayList(ServerTestUtils.getStoreDef("test", // Replication Factor 1, since we are testing a one node "cluster"
    1, // preferred/required reads/writes all at 1
    1, // preferred/required reads/writes all at 1
    1, // preferred/required reads/writes all at 1
    1, // preferred/required reads/writes all at 1
    1, RoutingStrategyType.CONSISTENT_STRATEGY.toString()));
    String storesXML = new StoreDefinitionsMapper().writeStoreList(storeDefList);
    conf.set("stores.xml", storesXML);
    String clusterXML = new ClusterMapper().writeCluster(ServerTestUtils.getLocalCluster(1));
    conf.set("cluster.xml", clusterXML);
    // We leverage the real mapper used in the BnP job to generate data with the proper format
    mapper = new BuildAndPushMapper();
    mapper.configure(conf);
    testCollector = new TestCollector();
    testCollectorWrapper = new TestCollectorWrapper();
    testCollectorWrapper.setCollector(testCollector);
}
Also used : StoreDefinition(voldemort.store.StoreDefinition) StoreDefinitionsMapper(voldemort.xml.StoreDefinitionsMapper) ClusterMapper(voldemort.xml.ClusterMapper) BuildAndPushMapper(voldemort.store.readonly.mr.BuildAndPushMapper) JobConf(org.apache.hadoop.mapred.JobConf)

Aggregations

JobConf (org.apache.hadoop.mapred.JobConf)1 StoreDefinition (voldemort.store.StoreDefinition)1 BuildAndPushMapper (voldemort.store.readonly.mr.BuildAndPushMapper)1 ClusterMapper (voldemort.xml.ClusterMapper)1 StoreDefinitionsMapper (voldemort.xml.StoreDefinitionsMapper)1