@SuppressWarnings("unchecked") public BT lookupMatch(PT record) { int hashCode = hash(probeComparator.hash(record)); int index = hashIndex(hashCode, data.length); pairComparator.setReference(record); HashEntry entry = data[index]; while (entry != null) { if (entryHashCode(entry) == hashCode && pairComparator.equalToReference((BT) entry.getValue())) { return (BT) entry.getValue(); } entry = entryNext(entry); } return null; }
/** * Calls the <code>JoinFunction#join()</code> method for all two key-value pairs that share the * same key and come from different inputs. The output of the <code>join()</code> method is * forwarded. * * <p>This method first zig-zags between the two sorted inputs in order to find a common key, and * then calls the join stub with the cross product of the values. * * @throws Exception Forwards all exceptions from the user code and the I/O system. * @see * org.apache.flink.runtime.operators.util.JoinTaskIterator#callWithNextKey(org.apache.flink.api.common.functions.FlatJoinFunction, * org.apache.flink.util.Collector) */ @Override public boolean callWithNextKey( final FlatJoinFunction<T1, T2, O> joinFunction, final Collector<O> collector) throws Exception { if (!this.iterator1.nextKey() || !this.iterator2.nextKey()) { // consume all remaining keys (hack to prevent remaining inputs during iterations, lets get // rid of this soon) while (this.iterator1.nextKey()) ; while (this.iterator2.nextKey()) ; return false; } final TypePairComparator<T1, T2> comparator = this.pairComparator; comparator.setReference(this.iterator1.getCurrent()); T2 current2 = this.iterator2.getCurrent(); // zig zag while (true) { // determine the relation between the (possibly composite) keys final int comp = comparator.compareToReference(current2); if (comp == 0) { break; } if (comp < 0) { if (!this.iterator2.nextKey()) { return false; } current2 = this.iterator2.getCurrent(); } else { if (!this.iterator1.nextKey()) { return false; } comparator.setReference(this.iterator1.getCurrent()); } } // here, we have a common key! call the join function with the cross product of the // values final Iterator<T1> values1 = this.iterator1.getValues(); final Iterator<T2> values2 = this.iterator2.getValues(); crossMatchingGroup(values1, values2, joinFunction, collector); return true; }