You are viewing a plain text version of this content. The canonical link for it is here.
Posted to common-commits@hadoop.apache.org by sh...@apache.org on 2009/04/02 22:08:21 UTC
svn commit: r761407 - in /hadoop/core/trunk: ./
src/core/org/apache/hadoop/io/ src/mapred/org/apache/hadoop/mapred/lib/
src/mapred/org/apache/hadoop/mapreduce/lib/partition/
src/test/org/apache/hadoop/mapreduce/lib/partition/
Author: shv
Date: Thu Apr 2 20:08:21 2009
New Revision: 761407
URL: http://svn.apache.org/viewvc?rev=761407&view=rev
Log:
HADOOP-5528. Add a configurable hash partitioner operating on ranges of BinaryComparable keys. Contributed by Klaas Bosteels.
Added:
hadoop/core/trunk/src/mapred/org/apache/hadoop/mapred/lib/BinaryPartitioner.java
hadoop/core/trunk/src/mapred/org/apache/hadoop/mapreduce/lib/partition/BinaryPartitioner.java
hadoop/core/trunk/src/test/org/apache/hadoop/mapreduce/lib/partition/TestBinaryPartitioner.java
Modified:
hadoop/core/trunk/CHANGES.txt
hadoop/core/trunk/src/core/org/apache/hadoop/io/WritableComparator.java
Modified: hadoop/core/trunk/CHANGES.txt
URL: http://svn.apache.org/viewvc/hadoop/core/trunk/CHANGES.txt?rev=761407&r1=761406&r2=761407&view=diff
==============================================================================
--- hadoop/core/trunk/CHANGES.txt (original)
+++ hadoop/core/trunk/CHANGES.txt Thu Apr 2 20:08:21 2009
@@ -68,6 +68,9 @@
HADOOP-5363. Add support for proxying connections to multiple clusters with
different versions to hdfsproxy. (Zhiyong Zhang via cdouglas)
+ HADOOP-5528. Add a configurable hash partitioner operating on ranges of
+ BinaryComparable keys. (Klaas Bosteels via shv)
+
IMPROVEMENTS
HADOOP-4565. Added CombineFileInputFormat to use data locality information
Modified: hadoop/core/trunk/src/core/org/apache/hadoop/io/WritableComparator.java
URL: http://svn.apache.org/viewvc/hadoop/core/trunk/src/core/org/apache/hadoop/io/WritableComparator.java?rev=761407&r1=761406&r2=761407&view=diff
==============================================================================
--- hadoop/core/trunk/src/core/org/apache/hadoop/io/WritableComparator.java (original)
+++ hadoop/core/trunk/src/core/org/apache/hadoop/io/WritableComparator.java Thu Apr 2 20:08:21 2009
@@ -135,12 +135,17 @@
}
/** Compute hash for binary data. */
- public static int hashBytes(byte[] bytes, int length) {
+ public static int hashBytes(byte[] bytes, int offset, int length) {
int hash = 1;
- for (int i = 0; i < length; i++)
+ for (int i = offset; i < offset + length; i++)
hash = (31 * hash) + (int)bytes[i];
return hash;
}
+
+ /** Compute hash for binary data. */
+ public static int hashBytes(byte[] bytes, int length) {
+ return hashBytes(bytes, 0, length);
+ }
/** Parse an unsigned short from a byte array. */
public static int readUnsignedShort(byte[] bytes, int start) {
Added: hadoop/core/trunk/src/mapred/org/apache/hadoop/mapred/lib/BinaryPartitioner.java
URL: http://svn.apache.org/viewvc/hadoop/core/trunk/src/mapred/org/apache/hadoop/mapred/lib/BinaryPartitioner.java?rev=761407&view=auto
==============================================================================
--- hadoop/core/trunk/src/mapred/org/apache/hadoop/mapred/lib/BinaryPartitioner.java (added)
+++ hadoop/core/trunk/src/mapred/org/apache/hadoop/mapred/lib/BinaryPartitioner.java Thu Apr 2 20:08:21 2009
@@ -0,0 +1,43 @@
+/**
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements. See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership. The ASF 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 org.apache.hadoop.mapred.lib;
+
+import org.apache.hadoop.io.BinaryComparable;
+import org.apache.hadoop.mapred.JobConf;
+import org.apache.hadoop.mapred.Partitioner;
+
+/**
+ * Partition {@link BinaryComparable} keys using a configurable part of
+ * the bytes array returned by {@link BinaryComparable#getBytes()}.
+ *
+ * @see org.apache.hadoop.mapreduce.lib.partition.BinaryPartitioner
+ * @deprecated Use
+ * {@link org.apache.hadoop.mapreduce.lib.partition.BinaryPartitioner}
+ * instead.
+ */
+@Deprecated
+public class BinaryPartitioner<V>
+ extends org.apache.hadoop.mapreduce.lib.partition.BinaryPartitioner<V>
+ implements Partitioner<BinaryComparable, V> {
+
+ public void configure(JobConf job) {
+ super.setConf(job);
+ }
+
+}
Added: hadoop/core/trunk/src/mapred/org/apache/hadoop/mapreduce/lib/partition/BinaryPartitioner.java
URL: http://svn.apache.org/viewvc/hadoop/core/trunk/src/mapred/org/apache/hadoop/mapreduce/lib/partition/BinaryPartitioner.java?rev=761407&view=auto
==============================================================================
--- hadoop/core/trunk/src/mapred/org/apache/hadoop/mapreduce/lib/partition/BinaryPartitioner.java (added)
+++ hadoop/core/trunk/src/mapred/org/apache/hadoop/mapreduce/lib/partition/BinaryPartitioner.java Thu Apr 2 20:08:21 2009
@@ -0,0 +1,138 @@
+/**
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements. See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership. The ASF 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 org.apache.hadoop.mapreduce.lib.partition;
+
+import org.apache.hadoop.conf.Configurable;
+import org.apache.hadoop.conf.Configuration;
+import org.apache.hadoop.io.BinaryComparable;
+import org.apache.hadoop.io.WritableComparator;
+import org.apache.hadoop.mapreduce.Partitioner;
+
+/**
+ * <p>Partition {@link BinaryComparable} keys using a configurable part of
+ * the bytes array returned by {@link BinaryComparable#getBytes()}.</p>
+ *
+ * <p>The subarray to be used for the partitioning can be defined by means
+ * of the following properties:
+ * <ul>
+ * <li>
+ * <i>mapred.binary.partitioner.left.offset</i>:
+ * left offset in array (0 by default)
+ * </li>
+ * <li>
+ * <i>mapred.binary.partitioner.right.offset</i>:
+ * right offset in array (-1 by default)
+ * </li>
+ * </ul>
+ * Like in Python, both negative and positive offsets are allowed, but
+ * the meaning is slightly different. In case of an array of length 5,
+ * for instance, the possible offsets are:
+ * <pre><code>
+ * +---+---+---+---+---+
+ * | B | B | B | B | B |
+ * +---+---+---+---+---+
+ * 0 1 2 3 4
+ * -5 -4 -3 -2 -1
+ * </code></pre>
+ * The first row of numbers gives the position of the offsets 0...5 in
+ * the array; the second row gives the corresponding negative offsets.
+ * Contrary to Python, the specified subarray has byte <code>i</code>
+ * and <code>j</code> as first and last element, repectively, when
+ * <code>i</code> and <code>j</code> are the left and right offset.
+ *
+ * <p>For Hadoop programs written in Java, it is advisable to use one of
+ * the following static convenience methods for setting the offsets:
+ * <ul>
+ * <li>{@link #setOffsets}</li>
+ * <li>{@link #setLeftOffset}</li>
+ * <li>{@link #setRightOffset}</li>
+ * </ul></p>
+ */
+public class BinaryPartitioner<V> extends Partitioner<BinaryComparable, V>
+ implements Configurable {
+
+ private static final String LEFT_OFFSET_PROPERTY_NAME =
+ "mapred.binary.partitioner.left.offset";
+ private static final String RIGHT_OFFSET_PROPERTY_NAME =
+ "mapred.binary.partitioner.right.offset";
+
+ /**
+ * Set the subarray to be used for partitioning to
+ * <code>bytes[left:(right+1)]</code> in Python syntax.
+ *
+ * @param conf configuration object
+ * @param left left Python-style offset
+ * @param right right Python-style offset
+ */
+ public static void setOffsets(Configuration conf, int left, int right) {
+ conf.setInt(LEFT_OFFSET_PROPERTY_NAME, left);
+ conf.setInt(RIGHT_OFFSET_PROPERTY_NAME, right);
+ }
+
+ /**
+ * Set the subarray to be used for partitioning to
+ * <code>bytes[offset:]</code> in Python syntax.
+ *
+ * @param conf configuration object
+ * @param offset left Python-style offset
+ */
+ public static void setLeftOffset(Configuration conf, int offset) {
+ conf.setInt(LEFT_OFFSET_PROPERTY_NAME, offset);
+ }
+
+ /**
+ * Set the subarray to be used for partitioning to
+ * <code>bytes[:(offset+1)]</code> in Python syntax.
+ *
+ * @param conf configuration object
+ * @param offset right Python-style offset
+ */
+ public static void setRightOffset(Configuration conf, int offset) {
+ conf.setInt(RIGHT_OFFSET_PROPERTY_NAME, offset);
+ }
+
+
+ private Configuration conf;
+ private int leftOffset, rightOffset;
+
+ public void setConf(Configuration conf) {
+ this.conf = conf;
+ leftOffset = conf.getInt(LEFT_OFFSET_PROPERTY_NAME, 0);
+ rightOffset = conf.getInt(RIGHT_OFFSET_PROPERTY_NAME, -1);
+ }
+
+ public Configuration getConf() {
+ return conf;
+ }
+
+ /**
+ * Use (the specified slice of the array returned by)
+ * {@link BinaryComparable#getBytes()} to partition.
+ */
+ @Override
+ public int getPartition(BinaryComparable key, V value, int numPartitions) {
+ int length = key.getLength();
+ int leftIndex = (leftOffset + length) % length;
+ int rightIndex = (rightOffset + length) % length;
+ int hash = WritableComparator.hashBytes(key.getBytes(),
+ leftIndex, rightIndex - leftIndex + 1);
+ return (hash & Integer.MAX_VALUE) % numPartitions;
+ }
+
+}
Added: hadoop/core/trunk/src/test/org/apache/hadoop/mapreduce/lib/partition/TestBinaryPartitioner.java
URL: http://svn.apache.org/viewvc/hadoop/core/trunk/src/test/org/apache/hadoop/mapreduce/lib/partition/TestBinaryPartitioner.java?rev=761407&view=auto
==============================================================================
--- hadoop/core/trunk/src/test/org/apache/hadoop/mapreduce/lib/partition/TestBinaryPartitioner.java (added)
+++ hadoop/core/trunk/src/test/org/apache/hadoop/mapreduce/lib/partition/TestBinaryPartitioner.java Thu Apr 2 20:08:21 2009
@@ -0,0 +1,103 @@
+/**
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements. See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership. The ASF 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 org.apache.hadoop.mapreduce.lib.partition;
+
+import org.apache.hadoop.conf.Configuration;
+import org.apache.hadoop.io.BinaryComparable;
+import org.apache.hadoop.io.BytesWritable;
+import org.apache.hadoop.util.ReflectionUtils;
+
+import junit.framework.TestCase;
+
+public class TestBinaryPartitioner extends TestCase {
+
+ public void testDefaultOffsets() {
+ Configuration conf = new Configuration();
+ BinaryPartitioner<?> partitioner =
+ ReflectionUtils.newInstance(BinaryPartitioner.class, conf);
+
+ BinaryComparable key1 = new BytesWritable(new byte[] { 1, 2, 3, 4, 5 });
+ BinaryComparable key2 = new BytesWritable(new byte[] { 1, 2, 3, 4, 5 });
+ int partition1 = partitioner.getPartition(key1, null, 10);
+ int partition2 = partitioner.getPartition(key2, null, 10);
+ assertEquals(partition1, partition2);
+
+ key1 = new BytesWritable(new byte[] { 1, 2, 3, 4, 5 });
+ key2 = new BytesWritable(new byte[] { 6, 2, 3, 4, 5 });
+ partition1 = partitioner.getPartition(key1, null, 10);
+ partition2 = partitioner.getPartition(key2, null, 10);
+ assertTrue(partition1 != partition2);
+
+ key1 = new BytesWritable(new byte[] { 1, 2, 3, 4, 5 });
+ key2 = new BytesWritable(new byte[] { 1, 2, 3, 4, 6 });
+ partition1 = partitioner.getPartition(key1, null, 10);
+ partition2 = partitioner.getPartition(key2, null, 10);
+ assertTrue(partition1 != partition2);
+ }
+
+ public void testCustomOffsets() {
+ Configuration conf = new Configuration();
+ BinaryComparable key1 = new BytesWritable(new byte[] { 1, 2, 3, 4, 5 });
+ BinaryComparable key2 = new BytesWritable(new byte[] { 6, 2, 3, 7, 8 });
+
+ BinaryPartitioner.setOffsets(conf, 1, -3);
+ BinaryPartitioner<?> partitioner =
+ ReflectionUtils.newInstance(BinaryPartitioner.class, conf);
+ int partition1 = partitioner.getPartition(key1, null, 10);
+ int partition2 = partitioner.getPartition(key2, null, 10);
+ assertEquals(partition1, partition2);
+
+ BinaryPartitioner.setOffsets(conf, 1, 2);
+ partitioner = ReflectionUtils.newInstance(BinaryPartitioner.class, conf);
+ partition1 = partitioner.getPartition(key1, null, 10);
+ partition2 = partitioner.getPartition(key2, null, 10);
+ assertEquals(partition1, partition2);
+
+ BinaryPartitioner.setOffsets(conf, -4, -3);
+ partitioner = ReflectionUtils.newInstance(BinaryPartitioner.class, conf);
+ partition1 = partitioner.getPartition(key1, null, 10);
+ partition2 = partitioner.getPartition(key2, null, 10);
+ assertEquals(partition1, partition2);
+ }
+
+ public void testLowerBound() {
+ Configuration conf = new Configuration();
+ BinaryPartitioner.setLeftOffset(conf, 0);
+ BinaryPartitioner<?> partitioner =
+ ReflectionUtils.newInstance(BinaryPartitioner.class, conf);
+ BinaryComparable key1 = new BytesWritable(new byte[] { 1, 2, 3, 4, 5 });
+ BinaryComparable key2 = new BytesWritable(new byte[] { 6, 2, 3, 4, 5 });
+ int partition1 = partitioner.getPartition(key1, null, 10);
+ int partition2 = partitioner.getPartition(key2, null, 10);
+ assertTrue(partition1 != partition2);
+ }
+
+ public void testUpperBound() {
+ Configuration conf = new Configuration();
+ BinaryPartitioner.setRightOffset(conf, 4);
+ BinaryPartitioner<?> partitioner =
+ ReflectionUtils.newInstance(BinaryPartitioner.class, conf);
+ BinaryComparable key1 = new BytesWritable(new byte[] { 1, 2, 3, 4, 5 });
+ BinaryComparable key2 = new BytesWritable(new byte[] { 1, 2, 3, 4, 6 });
+ int partition1 = partitioner.getPartition(key1, null, 10);
+ int partition2 = partitioner.getPartition(key2, null, 10);
+ assertTrue(partition1 != partition2);
+ }
+
+}