001/* 002 * Copyright (c) 2007-2017 Xplenty, Inc. All Rights Reserved. 003 * 004 * Project and contact information: http://www.cascading.org/ 005 * 006 * This file is part of the Cascading project. 007 * 008 * Licensed under the Apache License, Version 2.0 (the "License"); 009 * you may not use this file except in compliance with the License. 010 * You may obtain a copy of the License at 011 * 012 * http://www.apache.org/licenses/LICENSE-2.0 013 * 014 * Unless required by applicable law or agreed to in writing, software 015 * distributed under the License is distributed on an "AS IS" BASIS, 016 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 017 * See the License for the specific language governing permissions and 018 * limitations under the License. 019 */ 020 021package cascading.tuple.hadoop.util; 022 023import java.io.IOException; 024import java.util.Comparator; 025 026import cascading.CascadingException; 027import cascading.tuple.io.TuplePair; 028import org.apache.hadoop.conf.Configuration; 029 030/** Class TuplePairComparator is an implementation of {@link org.apache.hadoop.io.RawComparator}. */ 031public class GroupingSortingComparator extends DeserializerComparator<TuplePair> 032 { 033 Class[] sortTypes; 034 Comparator[] sortComparators; 035 036 @Override 037 public void setConf( Configuration conf ) 038 { 039 super.setConf( conf ); 040 041 if( conf == null ) 042 return; 043 044 sortTypes = tupleSerialization.getSortTypes(); 045 046 sortComparators = deserializeComparatorsFor( "cascading.sort.comparator" ); 047 sortComparators = delegatingComparatorsFor( sortTypes, sortComparators ); 048 } 049 050 public int compare( byte[] b1, int s1, int l1, byte[] b2, int s2, int l2 ) 051 { 052 try 053 { 054 lhsBuffer.reset( b1, s1, l1 ); 055 rhsBuffer.reset( b2, s2, l2 ); 056 057 int c = compareTuples( keyTypes, groupComparators ); 058 059 if( c != 0 ) 060 return c; 061 062 return compareTuples( sortTypes, sortComparators ); 063 } 064 catch( IOException exception ) 065 { 066 throw new CascadingException( exception ); 067 } 068 finally 069 { 070 lhsBuffer.clear(); 071 rhsBuffer.clear(); 072 } 073 } 074 075 public int compare( TuplePair lhs, TuplePair rhs ) 076 { 077 int c = compareTuples( groupComparators, lhs.getLhs(), rhs.getLhs() ); 078 079 if( c != 0 ) 080 return c; 081 082 return compareTuples( sortComparators, lhs.getRhs(), rhs.getRhs() ); 083 } 084 }