org.apache.flink.types.NormalizableKey Java Examples

@Override
public void putNormalizedKey(Record record, MemorySegment target, int offset, int numBytes) {
	int i = 0;
	try {
		for (; i < this.numLeadingNormalizableKeys & numBytes > 0; i++)
		{
			int len = this.normalizedKeyLengths[i]; 
			len = numBytes >= len ? len : numBytes;
			((NormalizableKey<?>) record.getField(this.keyFields[i], this.transientKeyHolders[i])).copyNormalizedKey(target, offset, len);
			numBytes -= len;
			offset += len;
		}
	}
	catch (NullPointerException npex) {
		throw new NullKeyFieldException(this.keyFields[i]);
	}
}
 

@Override
public void putNormalizedKey(Record record, MemorySegment target, int offset, int numBytes) {
	int i = 0;
	try {
		for (; i < this.numLeadingNormalizableKeys & numBytes > 0; i++)
		{
			int len = this.normalizedKeyLengths[i]; 
			len = numBytes >= len ? len : numBytes;
			((NormalizableKey<?>) record.getField(this.keyFields[i], this.transientKeyHolders[i])).copyNormalizedKey(target, offset, len);
			numBytes -= len;
			offset += len;
		}
	}
	catch (NullPointerException npex) {
		throw new NullKeyFieldException(this.keyFields[i]);
	}
}
 

@Override
public int getNormalizeKeyLen() {
	ensureReferenceInstantiated();

	NormalizableKey<?> key = (NormalizableKey<?>) reference;
	return key.getMaxNormalizedKeyLen();
}
 

@Override
public int getNormalizeKeyLen() {
	ensureReferenceInstantiated();

	NormalizableKey<?> key = (NormalizableKey<?>) reference;
	return key.getMaxNormalizedKeyLen();
}
 

@Override
public int getNormalizeKeyLen() {
	ensureReferenceInstantiated();

	NormalizableKey<?> key = (NormalizableKey<?>) reference;
	return key.getMaxNormalizedKeyLen();
}
 

@Override
public int getNormalizeKeyLen() {
	if (this.reference == null) {
		this.reference = InstantiationUtil.instantiate(this.type);
	}

	NormalizableKey<?> key = (NormalizableKey<?>) this.reference;
	return key.getMaxNormalizedKeyLen();
}
 

@Override
public int getNormalizeKeyLen() {
	if (this.reference == null) {
		this.reference = InstantiationUtil.instantiate(this.type);
	}

	NormalizableKey<?> key = (NormalizableKey<?>) this.reference;
	return key.getMaxNormalizedKeyLen();
}
 

@Override
public int getNormalizeKeyLen() {
	if (reference == null) {
		reference = InstantiationUtil.instantiate(type, Value.class);
	}
	
	NormalizableKey<?> key = (NormalizableKey<?>) reference;
	return key.getMaxNormalizedKeyLen();
}
 

@Override
public int getNormalizeKeyLen() {
	if (reference == null) {
		reference = InstantiationUtil.instantiate(type, Value.class);
	}
	
	NormalizableKey<?> key = (NormalizableKey<?>) reference;
	return key.getMaxNormalizedKeyLen();
}
 

@Override
public boolean supportsNormalizedKey() {
	return NormalizableKey.class.isAssignableFrom(type);
}
 

@Override
public boolean supportsNormalizedKey() {
	return NormalizableKey.class.isAssignableFrom(NullValueArray.class);
}
 

/**
 * Creates a new comparator that compares Pact Records by the subset of fields as described
 * by the given key positions and types.
 * 
 * @param keyFields The positions of the key fields.
 * @param keyTypes The types (classes) of the key fields.
 * @param sortDirection The direction for sorting. A value of <i>true</i> indicates ascending for an attribute,
 *                  a value of <i>false</i> indicated descending. If the parameter is <i>null</i>, then
 *                  all order comparisons will assume ascending order on all fields.
 */
public RecordComparator(int[] keyFields, Class<? extends Value>[] keyTypes, boolean[] sortDirection) {
	this.keyFields = keyFields;
	
	// instantiate fields to extract keys into
	this.keyHolders = new Value[keyTypes.length];
	this.transientKeyHolders = new Value[keyTypes.length];
	for (int i = 0; i < keyTypes.length; i++) {
		if (keyTypes[i] == null) {
			throw new NullPointerException("Key type " + i + " is null.");
		}
		this.keyHolders[i] = InstantiationUtil.instantiate(keyTypes[i], Value.class);
		this.transientKeyHolders[i] = InstantiationUtil.instantiate(keyTypes[i], Value.class);
	}
	
	// set up auxiliary fields for normalized key support
	this.normalizedKeyLengths = new int[keyFields.length];
	int nKeys = 0;
	int nKeyLen = 0;
	boolean inverted = false;
	for (int i = 0; i < this.keyHolders.length; i++) {
		Value k = this.keyHolders[i];
		if (k instanceof NormalizableKey) {
			if (sortDirection != null) {
				if (sortDirection[i] && inverted) {
					break;
				} else if (i == 0 && !sortDirection[0]) {
					inverted = true;
				}
			}
			nKeys++;
			final int len = ((NormalizableKey) k).getMaxNormalizedKeyLen();
			if (len < 0) {
				throw new RuntimeException("Data type " + k.getClass().getName() + 
					" specifies an invalid length for the normalized key: " + len);
			}
			this.normalizedKeyLengths[i] = len;
			nKeyLen += this.normalizedKeyLengths[i];
			if (nKeyLen < 0) {
				nKeyLen = Integer.MAX_VALUE;
				break;
			}
		} else {
			break;
		}
	}
	this.numLeadingNormalizableKeys = nKeys;
	this.normalizableKeyPrefixLen = nKeyLen;
	
	this.temp1 = new Record();
	this.temp2 = new Record();
	
	if (sortDirection != null) {
		this.ascending = sortDirection;
	} else {
		this.ascending = new boolean[keyFields.length];
		for (int i = 0; i < this.ascending.length; i++) {
			this.ascending[i] = true;
		}
	}
}
 

/**
 * Creates a new comparator that compares Pact Records by the subset of fields as described
 * by the given key positions and types.
 * 
 * @param keyFields The positions of the key fields.
 * @param keyTypes The types (classes) of the key fields.
 * @param sortDirection The direction for sorting. A value of <i>true</i> indicates ascending for an attribute,
 *                  a value of <i>false</i> indicated descending. If the parameter is <i>null</i>, then
 *                  all order comparisons will assume ascending order on all fields.
 */
public RecordComparator(int[] keyFields, Class<? extends Value>[] keyTypes, boolean[] sortDirection) {
	this.keyFields = keyFields;
	
	// instantiate fields to extract keys into
	this.keyHolders = new Value[keyTypes.length];
	this.transientKeyHolders = new Value[keyTypes.length];
	for (int i = 0; i < keyTypes.length; i++) {
		if (keyTypes[i] == null) {
			throw new NullPointerException("Key type " + i + " is null.");
		}
		this.keyHolders[i] = InstantiationUtil.instantiate(keyTypes[i], Value.class);
		this.transientKeyHolders[i] = InstantiationUtil.instantiate(keyTypes[i], Value.class);
	}
	
	// set up auxiliary fields for normalized key support
	this.normalizedKeyLengths = new int[keyFields.length];
	int nKeys = 0;
	int nKeyLen = 0;
	boolean inverted = false;
	for (int i = 0; i < this.keyHolders.length; i++) {
		Value k = this.keyHolders[i];
		if (k instanceof NormalizableKey) {
			if (sortDirection != null) {
				if (sortDirection[i] && inverted) {
					break;
				} else if (i == 0 && !sortDirection[0]) {
					inverted = true;
				}
			}
			nKeys++;
			final int len = ((NormalizableKey) k).getMaxNormalizedKeyLen();
			if (len < 0) {
				throw new RuntimeException("Data type " + k.getClass().getName() + 
					" specifies an invalid length for the normalized key: " + len);
			}
			this.normalizedKeyLengths[i] = len;
			nKeyLen += this.normalizedKeyLengths[i];
			if (nKeyLen < 0) {
				nKeyLen = Integer.MAX_VALUE;
				break;
			}
		} else {
			break;
		}
	}
	this.numLeadingNormalizableKeys = nKeys;
	this.normalizableKeyPrefixLen = nKeyLen;
	
	this.temp1 = new Record();
	this.temp2 = new Record();
	
	if (sortDirection != null) {
		this.ascending = sortDirection;
	} else {
		this.ascending = new boolean[keyFields.length];
		for (int i = 0; i < this.ascending.length; i++) {
			this.ascending[i] = true;
		}
	}
}
 

@Override
public boolean supportsNormalizedKey() {
	return NormalizableKey.class.isAssignableFrom(StringValueArray.class);
}
 

@Override
public boolean supportsNormalizedKey() {
	return NormalizableKey.class.isAssignableFrom(FloatValueArray.class);
}
 

@Override
public boolean supportsNormalizedKey() {
	return NormalizableKey.class.isAssignableFrom(IntValue.class);
}
 

@Override
public boolean supportsNormalizedKey() {
	return NormalizableKey.class.isAssignableFrom(ByteValue.class);
}
 

@Override
public boolean supportsNormalizedKey() {
	return NormalizableKey.class.isAssignableFrom(CharValue.class);
}
 

@Override
public boolean supportsNormalizedKey() {
	return NormalizableKey.class.isAssignableFrom(ByteValue.class);
}
 

@Override
public boolean supportsNormalizedKey() {
	return NormalizableKey.class.isAssignableFrom(NullValue.class);
}
 

@Override
public void putNormalizedKey(FloatValue record, MemorySegment target, int offset, int numBytes) {
	NormalizableKey<?> key = (NormalizableKey<?>) record;
	key.copyNormalizedKey(target, offset, numBytes);
}
 

@Override
public int getNormalizeKeyLen() {
	NormalizableKey<?> key = (NormalizableKey<?>) reference;
	return key.getMaxNormalizedKeyLen();
}
 

@Override
public boolean supportsNormalizedKey() {
	return NormalizableKey.class.isAssignableFrom(FloatValue.class);
}
 

@Override
public boolean supportsNormalizedKey() {
	return NormalizableKey.class.isAssignableFrom(NullValue.class);
}
 

@Override
public boolean supportsNormalizedKey() {
	return NormalizableKey.class.isAssignableFrom(BooleanValue.class);
}
 

@Override
public void putNormalizedKey(T record, MemorySegment target, int offset, int numBytes) {
	NormalizableKey<?> key = (NormalizableKey<?>) record;
	key.copyNormalizedKey(target, offset, numBytes);
}
 

@Override
public int getNormalizeKeyLen() {
	NormalizableKey<?> key = (NormalizableKey<?>) reference;
	return key.getMaxNormalizedKeyLen();
}
 

@Override
public boolean supportsNormalizedKey() {
	return NormalizableKey.class.isAssignableFrom(type);
}
 

@Override
public boolean supportsNormalizedKey() {
	return NormalizableKey.class.isAssignableFrom(CharValueArray.class);
}
 

@Override
public boolean supportsNormalizedKey() {
	return NormalizableKey.class.isAssignableFrom(type);
}