Examples with KeyDescription org.apache.hadoop.mapreduce.lib.partition.KeyFieldHelper.KeyDescription used on opensource projects

Example 1 with KeyDescription

use of org.apache.hadoop.mapreduce.lib.partition.KeyFieldHelper.KeyDescription in project hadoop by apache.

the class KeyFieldBasedComparator method compare.

public int compare(byte[] b1, int s1, int l1, byte[] b2, int s2, int l2) {
    int n1 = WritableUtils.decodeVIntSize(b1[s1]);
    int n2 = WritableUtils.decodeVIntSize(b2[s2]);
    List<KeyDescription> allKeySpecs = keyFieldHelper.keySpecs();
    if (allKeySpecs.size() == 0) {
        return compareBytes(b1, s1 + n1, l1 - n1, b2, s2 + n2, l2 - n2);
    }
    int[] lengthIndicesFirst = keyFieldHelper.getWordLengths(b1, s1 + n1, s1 + l1);
    int[] lengthIndicesSecond = keyFieldHelper.getWordLengths(b2, s2 + n2, s2 + l2);
    for (KeyDescription keySpec : allKeySpecs) {
        int startCharFirst = keyFieldHelper.getStartOffset(b1, s1 + n1, s1 + l1, lengthIndicesFirst, keySpec);
        int endCharFirst = keyFieldHelper.getEndOffset(b1, s1 + n1, s1 + l1, lengthIndicesFirst, keySpec);
        int startCharSecond = keyFieldHelper.getStartOffset(b2, s2 + n2, s2 + l2, lengthIndicesSecond, keySpec);
        int endCharSecond = keyFieldHelper.getEndOffset(b2, s2 + n2, s2 + l2, lengthIndicesSecond, keySpec);
        int result;
        if ((result = compareByteSequence(b1, startCharFirst, endCharFirst, b2, startCharSecond, endCharSecond, keySpec)) != 0) {
            return result;
        }
    }
    return 0;
}

Example 2 with KeyDescription

use of org.apache.hadoop.mapreduce.lib.partition.KeyFieldHelper.KeyDescription in project hadoop by apache.

the class KeyFieldBasedPartitioner method getPartition.

public int getPartition(K2 key, V2 value, int numReduceTasks) {
    byte[] keyBytes;
    List<KeyDescription> allKeySpecs = keyFieldHelper.keySpecs();
    if (allKeySpecs.size() == 0) {
        return getPartition(key.toString().hashCode(), numReduceTasks);
    }
    try {
        keyBytes = key.toString().getBytes("UTF-8");
    } catch (UnsupportedEncodingException e) {
        throw new RuntimeException("The current system does not " + "support UTF-8 encoding!", e);
    }
    // return 0 if the key is empty
    if (keyBytes.length == 0) {
        return 0;
    }
    int[] lengthIndicesFirst = keyFieldHelper.getWordLengths(keyBytes, 0, keyBytes.length);
    int currentHash = 0;
    for (KeyDescription keySpec : allKeySpecs) {
        int startChar = keyFieldHelper.getStartOffset(keyBytes, 0, keyBytes.length, lengthIndicesFirst, keySpec);
        // no key found! continue
        if (startChar < 0) {
            continue;
        }
        int endChar = keyFieldHelper.getEndOffset(keyBytes, 0, keyBytes.length, lengthIndicesFirst, keySpec);
        currentHash = hashCode(keyBytes, startChar, endChar, currentHash);
    }
    return getPartition(currentHash, numReduceTasks);
}