MAPREDUCE-4771. KeyFieldBasedPartitioner not partitioning properly when configured (jlowe via bobby)
git-svn-id: https://svn.apache.org/repos/asf/hadoop/common/trunk@1405975 13f79535-47bb-0310-9956-ffa450edef68
This commit is contained in:
parent
b1aa62a848
commit
0eadfcfcf5
|
@ -631,6 +631,9 @@ Release 0.23.5 - UNRELEASED
|
|||
|
||||
MAPREDUCE-4763 repair test TestUmbilicalProtocolWithJobToken (Ivan A.
|
||||
Veselovsky via bobby)
|
||||
|
||||
MAPREDUCE-4771. KeyFieldBasedPartitioner not partitioning properly when
|
||||
configured (jlowe via bobby)
|
||||
|
||||
Release 0.23.4 - UNRELEASED
|
||||
|
||||
|
|
|
@ -63,6 +63,7 @@ public class KeyFieldBasedPartitioner<K2, V2> extends Partitioner<K2, V2>
|
|||
|
||||
public void setConf(Configuration conf) {
|
||||
this.conf = conf;
|
||||
keyFieldHelper = new KeyFieldHelper();
|
||||
String keyFieldSeparator =
|
||||
conf.get(MRJobConfig.MAP_OUTPUT_KEY_FIELD_SEPERATOR, "\t");
|
||||
keyFieldHelper.setKeyFieldSeparator(keyFieldSeparator);
|
||||
|
|
|
@ -17,17 +17,18 @@
|
|||
*/
|
||||
package org.apache.hadoop.mapred.lib;
|
||||
|
||||
import static org.junit.Assert.assertEquals;
|
||||
|
||||
import org.apache.hadoop.io.Text;
|
||||
import org.apache.hadoop.mapred.JobConf;
|
||||
import org.apache.hadoop.mapred.lib.KeyFieldBasedPartitioner;
|
||||
import org.junit.Test;
|
||||
|
||||
import junit.framework.TestCase;
|
||||
|
||||
public class TestKeyFieldBasedPartitioner extends TestCase {
|
||||
public class TestKeyFieldBasedPartitioner {
|
||||
|
||||
/**
|
||||
* Test is key-field-based partitioned works with empty key.
|
||||
*/
|
||||
@Test
|
||||
public void testEmptyKey() throws Exception {
|
||||
KeyFieldBasedPartitioner<Text, Text> kfbp =
|
||||
new KeyFieldBasedPartitioner<Text, Text>();
|
||||
|
@ -37,4 +38,18 @@ public class TestKeyFieldBasedPartitioner extends TestCase {
|
|||
assertEquals("Empty key should map to 0th partition",
|
||||
0, kfbp.getPartition(new Text(), new Text(), 10));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testMultiConfigure() {
|
||||
KeyFieldBasedPartitioner<Text, Text> kfbp =
|
||||
new KeyFieldBasedPartitioner<Text, Text>();
|
||||
JobConf conf = new JobConf();
|
||||
conf.set(KeyFieldBasedPartitioner.PARTITIONER_OPTIONS, "-k1,1");
|
||||
kfbp.setConf(conf);
|
||||
Text key = new Text("foo\tbar");
|
||||
Text val = new Text("val");
|
||||
int partNum = kfbp.getPartition(key, val, 4096);
|
||||
kfbp.configure(conf);
|
||||
assertEquals(partNum, kfbp.getPartition(key,val, 4096));
|
||||
}
|
||||
}
|
Loading…
Reference in New Issue